Added initializeCohort

sksamuel · Apr 29, 2024 · 36df7b2 · 36df7b2
1 parent 2e4fa90
commit 36df7b2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/cohort-vertx/src/main/kotlin/com/sksamuel/cohort/vertx/HealthVerticle.kt b/cohort-vertx/src/main/kotlin/com/sksamuel/cohort/vertx/HealthVerticle.kt
@@ -15,13 +15,14 @@ import io.vertx.ext.web.Router
 import org.slf4j.LoggerFactory
 import java.time.ZoneOffset
 
-fun Router.cohort(
-   configure: CohortConfiguration.() -> Unit = {},
-) {
+fun initializeCohort(configure: CohortConfiguration.() -> Unit = {}): CohortConfiguration {
+   return CohortConfiguration().also(configure)
+}
+
+fun Router.cohort(cohort: CohortConfiguration) {
 
    val router = this
    val logger = LoggerFactory.getLogger(Router::class.java)
-   val cohort = CohortConfiguration().also(configure)
 
    if (cohort.heapDump) {
       router.get("${cohort.endpointPrefix}/heapdump")