From 2c6b5f5890c29f4ecf0d4c7788c70e35ebe7da12 Mon Sep 17 00:00:00 2001
From: Max <coding@maax.gr>
Date: Sun, 17 Jul 2022 15:34:15 +0200
Subject: [PATCH] extract logic to FeatureInstance class

---
 .../kotlin/com/apurebase/kgraphql/KtorFeature.kt    | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/kgraphql-ktor/src/main/kotlin/com/apurebase/kgraphql/KtorFeature.kt b/kgraphql-ktor/src/main/kotlin/com/apurebase/kgraphql/KtorFeature.kt
index bc7d5d08..471a9e2f 100644
--- a/kgraphql-ktor/src/main/kotlin/com/apurebase/kgraphql/KtorFeature.kt
+++ b/kgraphql-ktor/src/main/kotlin/com/apurebase/kgraphql/KtorFeature.kt
@@ -46,6 +46,17 @@ class GraphQL(val schema: Schema) {
     companion object Feature: ApplicationFeature<Application, Configuration, GraphQL> {
         override val key = AttributeKey<GraphQL>("KGraphQL")
 
+        private val rootFeature = FeatureInstance("KGraphQL")
+
+        override fun install(pipeline: Application, configure: Configuration.() -> Unit): GraphQL {
+            return rootFeature.install(pipeline, configure)
+        }
+    }
+
+    class FeatureInstance(featureKey: String = "KGraphQL"): ApplicationFeature<Application, Configuration, GraphQL> {
+
+        override val key = AttributeKey<GraphQL>(featureKey)
+
         override fun install(pipeline: Application, configure: Configuration.() -> Unit): GraphQL {
             val config = Configuration().apply(configure)
             val schema = KGraphQL.schema {
@@ -121,6 +132,8 @@ class GraphQL(val schema: Schema) {
                 }
             })
         }.toString()
+
     }
 
+
 }