use vertx periodic await

tmio · Jul 7, 2023 · 1f1d1ed · 1f1d1ed
1 parent a8083fb
commit 1f1d1ed
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/dns-discovery/src/main/kotlin/org/apache/tuweni/discovery/DNSDaemon.kt b/dns-discovery/src/main/kotlin/org/apache/tuweni/discovery/DNSDaemon.kt
@@ -17,7 +17,7 @@
 package org.apache.tuweni.discovery
 
 import io.vertx.core.Vertx
-import kotlinx.coroutines.runBlocking
+import io.vertx.kotlin.coroutines.setPeriodicAwait
 import org.apache.tuweni.devp2p.EthereumNodeRecord
 import org.slf4j.LoggerFactory
 
@@ -60,7 +60,7 @@ class DNSDaemon @JvmOverloads constructor(
   fun start() {
     logger.trace("Starting DNSDaemon for $enrLink")
     val task = DNSTimerTask(vertx, seq, enrLink, this::updateRecords)
-    this.periodicHandle = vertx.setPeriodic(period, task::run)
+    vertx.setPeriodicAwait(period, task::run)
   }
 
   /**
@@ -84,10 +84,8 @@ internal class DNSTimerTask(
     val logger = LoggerFactory.getLogger(DNSTimerTask::class.java)
   }
 
-  fun run(@Suppress("UNUSED_PARAMETER") timerId: Long) {
+  suspend fun run(@Suppress("UNUSED_PARAMETER") timerId: Long) {
     logger.debug("Refreshing DNS records with $enrLink")
-    runBlocking {
-      records(dnsResolver.collectAll(enrLink))
-    }
+    records(dnsResolver.collectAll(enrLink))
   }
 }