diff options
Diffstat (limited to 'messagebus')
-rw-r--r-- | messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetwork.java | 3 | ||||
-rwxr-xr-x | messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetworkParams.java | 11 |
2 files changed, 13 insertions, 1 deletions
diff --git a/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetwork.java b/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetwork.java index b0724ad6029..1c41f87d1ee 100644 --- a/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetwork.java +++ b/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetwork.java @@ -87,7 +87,8 @@ public class RPCNetwork implements Network, MethodHandler { public RPCNetwork(RPCNetworkParams params, SlobrokConfigSubscriber slobrokConfig) { this.slobroksConfig = slobrokConfig; identity = params.getIdentity(); - orb = new Supervisor(new Transport("mbus-rpc-" + identity.getServicePrefix(), params.getNumNetworkThreads(), shouldEnableTcpNodelay(params.getOptimization()))); + orb = new Supervisor(new Transport("mbus-rpc-" + identity.getServicePrefix(), params.getNumNetworkThreads(), + shouldEnableTcpNodelay(params.getOptimization()), params.getTransportEventsBeforeWakeup())); orb.setMaxInputBufferSize(params.getMaxInputBufferSize()); orb.setMaxOutputBufferSize(params.getMaxOutputBufferSize()); targetPool = new RPCTargetPool(params.getConnectionExpireSecs(), params.getNumTargetsPerSpec()); diff --git a/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetworkParams.java b/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetworkParams.java index e77cddd8b06..db22363785d 100755 --- a/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetworkParams.java +++ b/messagebus/src/main/java/com/yahoo/messagebus/network/rpc/RPCNetworkParams.java @@ -21,6 +21,8 @@ public class RPCNetworkParams { private double connectionExpireSecs = 30; private int numTargetsPerSpec = 1; private int numNetworkThreads = 2; + + private int transportEventsBeforeWakeup = 1; public enum Optimization {LATENCY, THROUGHPUT} Optimization optimization = Optimization.LATENCY; @@ -216,4 +218,13 @@ public class RPCNetworkParams { this.maxOutputBufferSize = maxOutputBufferSize; return this; } + + public int getTransportEventsBeforeWakeup() { + return transportEventsBeforeWakeup; + } + + public RPCNetworkParams setTransportEventsBeforeWakeup(int transportEventsBeforeWakeup) { + this.transportEventsBeforeWakeup = transportEventsBeforeWakeup; + return this; + } } |