Utils.java

package won.protocol.message;

import org.apache.jena.query.Dataset;
import org.apache.jena.query.DatasetFactory;
import org.apache.jena.query.Query;
import org.apache.jena.query.QueryFactory;
import org.apache.jena.rdf.model.Model;
import org.apache.jena.rdf.model.ModelFactory;
import org.apache.jena.riot.Lang;
import org.apache.jena.riot.RDFDataMgr;
import org.apache.jena.riot.RDFFormat;
import won.protocol.util.RdfUtils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

/**
 * User: ypanchenko
 * Date: 05.08.2014
 */
public class Utils
{

  public static Dataset createTestDataset(String resourceName) throws IOException {

    InputStream is = Utils.class.getResourceAsStream(resourceName);
    Dataset dataset = DatasetFactory.createGeneral();
    RDFDataMgr.read(dataset, is, RDFFormat.TRIG.getLang());
    is.close();
    return dataset;

  }

  public static Model createTestModel(String resourceName) throws IOException {

    InputStream is = Utils.class.getResourceAsStream(resourceName);
    Model model = ModelFactory.createDefaultModel();
    RDFDataMgr.read(model, is, RDFFormat.TURTLE.getLang());
    //model1.read(new InputStreamReader(is1), RESOURCE_URI, FileUtils.langTurtle);
    is.close();
    return model;

  }

  public static Query createTestQuery(String resourceName) throws IOException {
    String queryString = loadQueryFromResource(resourceName);
    return QueryFactory.create(queryString);
  }

  private static String loadQueryFromResource(final String path) {
    BufferedReader reader = null;
    StringBuilder sb = new StringBuilder();
    String line;
    try {

      reader = new BufferedReader(new InputStreamReader(Utils.class.getResourceAsStream(path),
                                                        "UTF-8"));
      while ((line = reader.readLine()) != null) {
        sb.append(line);
        sb.append("\n");
      }

    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (reader != null) {
        try {
          reader.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }
    return sb.toString();
  }

  public static void print(Model model) {
    System.out.println(RdfUtils.writeModelToString(model, Lang.TURTLE));
  }

  public static void print(Dataset dataset) {
    System.out.println(RdfUtils.writeDatasetToString(dataset, Lang.TRIG));
  }
}