fix: Requester.addFunction()

2019-07-30 23:01:09 +02:00
parent e819e1d027
commit ecd130198c
1 changed files with 4 additions and 9 deletions
--- a/src/main/kotlin/fr/postgresjson/connexion/Requester.kt
+++ b/src/main/kotlin/fr/postgresjson/connexion/Requester.kt
@@ -43,15 +43,10 @@ class Requester(
    }

    fun addFunction(functionsDirectory: File): Requester {
-        functionsDirectory.walk().filter {
-            it.isDirectory
-        }.forEach { directory ->
-            directory.walk().filter {
-                it.isFile
-            }.forEach { file ->
-                val fileContent = file.readText()
-                addFunction(fileContent)
-            }
+        functionsDirectory.walk()
+            .filter { it.isFile && it.extension == "sql" }
+            .forEach {
+                addFunction(it.readText())
            }
        return this
    }