use extension function instead of resolverKF

pgutkowski · Mar 22, 2019 · e8dfd02 · e8dfd02
1 parent cc27bdd
commit e8dfd02
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/jmh/kotlin/com/github/pgutkowski/kgraphql/BenchmarkSchema.kt b/src/jmh/kotlin/com/github/pgutkowski/kgraphql/BenchmarkSchema.kt
@@ -39,7 +39,7 @@ object BenchmarkSchema {
             resolver(threeResolver)
         }
         query("threeKF"){
-            resolverKF(HasOneResolver::oneResolver)
+            HasOneResolver::oneResolver.toResolver()
         }
     }
 }
diff --git a/src/main/kotlin/com/github/pgutkowski/kgraphql/schema/dsl/QueryOrMutationDSL.kt b/src/main/kotlin/com/github/pgutkowski/kgraphql/schema/dsl/QueryOrMutationDSL.kt
@@ -26,7 +26,7 @@ class QueryOrMutationDSL(
         return ResolverDSL(this)
     }
 
-    fun <T>resolverKF(function: KFunction<T>) = resolver(FunctionWrapper.on(function))
+    fun <T>KFunction<T>.toResolver() = resolver(FunctionWrapper.on(this))
 
     fun <T>resolver(function: () -> T) = resolver(FunctionWrapper.on(function))
 

diff --git a/src/test/kotlin/com/github/pgutkowski/kgraphql/schema/SchemaBuilderTest.kt b/src/test/kotlin/com/github/pgutkowski/kgraphql/schema/SchemaBuilderTest.kt
@@ -181,11 +181,11 @@ class SchemaBuilderTest {
 
         val tested = defaultSchema {
             query("mainActor") {
-                resolverKF(actorService::getMainActor)
+                actorService::getMainActor.toResolver()
             }
 
             query("actorById") {
-                resolverKF(actorService::getActor)
+                actorService::getActor.toResolver()
             }
 
             type<Actor> {