Use a typesafe Duration instead of integer milliseondst status.

2 files changed, 5 insertions, 2 deletions

diff --git a/messagebus/src/main/java/com/yahoo/messagebus/MessageBus.java b/messagebus/src/main/java/com/yahoo/messagebus/MessageBus.java
index c47fe88dc71..c3279d994c9 100644
--- a/messagebus/src/main/java/com/yahoo/messagebus/MessageBus.java
+++ b/messagebus/src/main/java/com/yahoo/messagebus/MessageBus.java
@@ -2,6 +2,8 @@
 package com.yahoo.messagebus;
 
 import com.yahoo.concurrent.SystemTimer;
+
+import java.time.Duration;
 import java.util.logging.Level;
 import com.yahoo.messagebus.network.Network;
 import com.yahoo.messagebus.network.NetworkMultiplexer;
@@ -88,7 +90,7 @@ public class MessageBus implements ConfigHandler, NetworkOwner, MessageHandler,
     }
 
     private void sendBlockedMessages() {
-        int timeout = SystemTimer.adjustTimeoutByDetectedHz(10);
+        long timeout = SystemTimer.adjustTimeoutByDetectedHz(Duration.ofMillis(10)).toMillis();
         while (! destroyed.get()) {
             for (SendBlockedMessages sender : blockedSenders.keySet()) {
                 if (!sender.trySend()) {
diff --git a/messagebus/src/main/java/com/yahoo/messagebus/Messenger.java b/messagebus/src/main/java/com/yahoo/messagebus/Messenger.java
index 4f36cb3b328..871f53396b4 100755
--- a/messagebus/src/main/java/com/yahoo/messagebus/Messenger.java
+++ b/messagebus/src/main/java/com/yahoo/messagebus/Messenger.java
@@ -3,6 +3,7 @@ package com.yahoo.messagebus;
 
 import com.yahoo.concurrent.SystemTimer;
 
+import java.time.Duration;
 import java.util.logging.Level;
 
 import java.util.ArrayDeque;
@@ -149,7 +150,7 @@ public class Messenger implements Runnable {
 
     @Override
     public void run() {
-        int timeoutMS = SystemTimer.adjustTimeoutByDetectedHz(100);
+        long timeoutMS = SystemTimer.adjustTimeoutByDetectedHz(Duration.ofMillis(100)).toMillis();
         while (true) {
             Task task = null;
             synchronized (this) {