Task: Add query for account with completable future and want to test with flux and fetch accounts with subscriptions

dgs-graphql/src/main/kotlin/com/cubetiqs/graphql/demo/config/ManagementConfig.kt

@@ -1,8 +1,10 @@
 package com.cubetiqs.graphql.demo.config
 
 import org.springframework.context.annotation.Configuration
+import org.springframework.scheduling.annotation.EnableAsync
 import org.springframework.transaction.annotation.EnableTransactionManagement
 
 @Configuration
 @EnableTransactionManagement
+@EnableAsync
 class ManagementConfig

dgs-graphql/src/main/kotlin/com/cubetiqs/graphql/demo/repository/AccountRepository.kt

@@ -2,7 +2,12 @@ package com.cubetiqs.graphql.demo.repository
 
 import com.cubetiqs.graphql.demo.domain.account.Account
 import org.springframework.data.jpa.repository.JpaRepository
+import org.springframework.scheduling.annotation.Async
 import org.springframework.stereotype.Repository
+import java.util.concurrent.CompletableFuture
 
 @Repository
-interface AccountRepository : JpaRepository<Account, Long>
+interface AccountRepository : JpaRepository<Account, Long> {
+    @Async
+    fun readAllBy(): CompletableFuture<Collection<Account>>
+}

dgs-graphql/src/main/kotlin/com/cubetiqs/graphql/demo/resolver/subscription/HelloSubscriptionResolver.kt

@@ -1,16 +1,34 @@
 package com.cubetiqs.graphql.demo.resolver.subscription
 
 import com.cubetiqs.graphql.demo.context.GSubscription
+import com.cubetiqs.graphql.demo.domain.account.Account
+import com.cubetiqs.graphql.demo.repository.AccountRepository
 import com.netflix.graphql.dgs.DgsSubscription
 import graphql.schema.DataFetchingEnvironment
+import kotlinx.coroutines.reactor.awaitSingle
 import org.reactivestreams.Publisher
+import org.springframework.beans.factory.annotation.Autowired
 import reactor.core.publisher.Flux
+import reactor.core.publisher.Mono
+import reactor.kotlin.core.publisher.toFlux
+import reactor.kotlin.core.publisher.toMono
 import java.time.Duration
 
 @GSubscription
 class HelloSubscriptionResolver {
+    @Autowired
+    private lateinit var accountRepository: AccountRepository
+
     @DgsSubscription(field = "hello")
     fun hello(env: DataFetchingEnvironment): Publisher<Int> {
         return Flux.range(1, 10).delayElements(Duration.ofSeconds(1))
     }
+
+    @DgsSubscription(field = "fetchAccounts")
+    fun fetchAccounts(env: DataFetchingEnvironment): Publisher<Collection<Account>> {
+        return Flux.generate<Collection<Account>?> { sink ->
+            sink.next(accountRepository.findAll())
+        }.delayElements(Duration.ofSeconds(1))
+        // return Mono.fromFuture(accounts).toFlux().delayElements(Duration.ofSeconds(1))
+    }
 }

dgs-graphql/src/main/resources/schema/schema.graphqls

@@ -12,6 +12,8 @@ type Query {
 
 type Subscription {
     hello: Int
+
+    fetchAccounts: [Account]
 }
 
 type Mutation {