feat: make save to projects

archguard · Apr 2, 2022 · 97d1ff5 · 97d1ff5
1 parent ddc569c
commit 97d1ff5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/code_repository/src/main/kotlin/org/archguard/scanner/common/ClassRepository.kt b/code_repository/src/main/kotlin/org/archguard/scanner/common/ClassRepository.kt
@@ -402,6 +402,7 @@ class ClassRepository(systemId: String, language: String, workspace: String) {
             }
             if (field.TypeValue.contains("\n")) {
                 println("contains \n will give up: --------------\n-------------")
+                println(field.TypeValue)
                 name = ""
             }
 

diff --git a/code_repository/src/main/kotlin/org/archguard/scanner/common/DatamapRepository.kt b/code_repository/src/main/kotlin/org/archguard/scanner/common/DatamapRepository.kt
@@ -34,7 +34,11 @@ class DatamapRepository(systemId: String, language: String, workspace: String) {
         values["function_name"] = relation.functionName
 
         values["tables"] = relation.tables.joinToString(",")
-        values["sqls"] = ""
+
+        // `,` is a symbol in sql
+        values["sqls"] = relation.sqls.joinToString("\n::archguard::\n") {
+            it.replace("\'", "\'\'")
+        }
 
         batch.add("data_code_database_relation", values)
         return id

diff --git a/scan_sourcecode/src/main/kotlin/org/archguard/scanner/sourcecode/Runner.kt b/scan_sourcecode/src/main/kotlin/org/archguard/scanner/sourcecode/Runner.kt
@@ -123,9 +123,7 @@ class Runner : CliktCommand(help = "scan git to sql") {
                 val apiAnalyser = MysqlAnalyser()
                 val records = dataStructs.flatMap { data ->
                     apiAnalyser.analysisByNode(data, "")
-                }
-                    .toList()
-
+                }.toList()
 
                 val datamapRepository = DatamapRepository(systemId, language, path)
                 datamapRepository.saveRelations(records)