diff --git a/src/main/kotlin/fr/postgresjson/connexion/Requester.kt b/src/main/kotlin/fr/postgresjson/connexion/Requester.kt index 2d34c1c..06367d4 100644 --- a/src/main/kotlin/fr/postgresjson/connexion/Requester.kt +++ b/src/main/kotlin/fr/postgresjson/connexion/Requester.kt @@ -43,16 +43,11 @@ 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 }