From ecd130198ca59d02bd6f35214854ea73138bdb93 Mon Sep 17 00:00:00 2001
From: Fabrice Lecomte <fabrice.lecomte.be@gmail.com>
Date: Tue, 30 Jul 2019 23:01:09 +0200
Subject: [PATCH] fix: Requester.addFunction()

---
 .../kotlin/fr/postgresjson/connexion/Requester.kt   | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

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
     }