neo4j export: escape \ (#263)

fixes #261
joernio · Oct 17, 2024 · 78b5dc0 · 78b5dc0
1 parent ab93c28
commit 78b5dc0
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/formats/src/main/scala/flatgraph/formats/neo4jcsv/Neo4jCsvExporter.scala b/formats/src/main/scala/flatgraph/formats/neo4jcsv/Neo4jCsvExporter.scala
@@ -2,15 +2,16 @@ package flatgraph.formats.neo4jcsv
 
 import com.github.tototoshi.csv.*
 import flatgraph.formats.{ExportResult, Exporter, writeFile}
-import flatgraph.{Edge, GNode, Graph, Schema}
+import flatgraph.{Edge, GNode, Schema}
 
 import java.nio.file.Path
+import java.util.regex.Pattern
 import scala.collection.mutable
 import scala.jdk.CollectionConverters.*
-import scala.jdk.OptionConverters.RichOptional
 import scala.util.Using
 
 object Neo4jCsvExporter extends Exporter {
+  val BackslashRegex = Pattern.compile("""\\""") // matches `\`
 
   override def defaultFileExtension = "csv"
 
@@ -74,7 +75,7 @@ object Neo4jCsvExporter extends Exporter {
         }
 
         val specialColumns       = Seq(node.id.toString, node.label)
-        val propertyValueColumns = columnDefinitions.propertyValues(node.propertyOption)
+        val propertyValueColumns = columnDefinitions.propertyValues(node.propertyOption).map(escapeSpecialCharacters)
         writer.writeRow(specialColumns ++ propertyValueColumns)
         nodeCount += 1
       }
@@ -162,6 +163,11 @@ object Neo4jCsvExporter extends Exporter {
     }
   }
 
+  private def escapeSpecialCharacters(propertyValue: String): String = {
+    // replace `\` with `\\` - otherwise neo4j import complains
+    BackslashRegex.matcher(propertyValue).replaceAll("""\\\\""")
+  }
+
   private case class EdgeFilesContext(
     label: String,
     headerFile: Path,