Minor refactoring

SANSA-Stack · Mar 6, 2020 · 01b5268 · 01b5268
1 parent e355f46
commit 01b5268
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/sansa-owl-spark/src/main/scala/net/sansa_stack/owl/spark/writers/RDFJsonWriter.scala b/sansa-owl-spark/src/main/scala/net/sansa_stack/owl/spark/writers/RDFJsonWriter.scala
@@ -9,18 +9,11 @@ import org.eclipse.rdf4j.rio.RDFFormat
 import org.eclipse.rdf4j.rio.rdfjson.RDFJSONWriter
 import org.semanticweb.owlapi.apibinding.OWLManager
 import org.semanticweb.owlapi.formats.RDFJsonDocumentFormat
-import org.semanticweb.owlapi.model.{OWLAxiom, OWLOntology}
-import org.semanticweb.owlapi.rio.RioRenderer
+import org.semanticweb.owlapi.model.OWLAxiom
 
 import net.sansa_stack.owl.spark.rdd.OWLAxiomsRDD
 
 
-protected class SANSARioRenderer(ont: OWLOntology, rdfJsonWriter: RDFJSONWriter, docFormat: RDFJsonDocumentFormat)
-  extends RioRenderer(ont, rdfJsonWriter, docFormat) {
-
-  override def renderOntologyHeader(): Unit = None
-}
-
 object RDFJsonWriter extends OWLWriterBase {
   override def save(filePath: String, owlAxioms: OWLAxiomsRDD): Unit = {
     owlAxioms.mapPartitionsWithIndex((idx: Int, partition: Iterator[OWLAxiom]) => if (partition.hasNext) {

diff --git a/sansa-owl-spark/src/main/scala/net/sansa_stack/owl/spark/writers/SANSARioRenderer.scala b/sansa-owl-spark/src/main/scala/net/sansa_stack/owl/spark/writers/SANSARioRenderer.scala
@@ -0,0 +1,12 @@
+package net.sansa_stack.owl.spark.writers
+
+import org.eclipse.rdf4j.rio.helpers.AbstractRDFWriter
+import org.semanticweb.owlapi.formats.{RDFJsonLDDocumentFormat, RioRDFNonPrefixDocumentFormat}
+import org.semanticweb.owlapi.model.OWLOntology
+import org.semanticweb.owlapi.rio.RioRenderer
+
+class SANSARioRenderer(ont: OWLOntology, rdfJsonWriter: AbstractRDFWriter, docFormat: RioRDFNonPrefixDocumentFormat)
+  extends RioRenderer(ont, rdfJsonWriter, docFormat) {
+
+  override def renderOntologyHeader(): Unit = None
+}