make accept backlog tunable

facebookarchive · Feb 22, 2014 · 6fd662a · 6fd662a
1 parent a03b01c
commit 6fd662a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/nifty-load-tester/src/main/java/com/facebook/nifty/perf/LoadTesterNettyConfigProvider.java b/nifty-load-tester/src/main/java/com/facebook/nifty/perf/LoadTesterNettyConfigProvider.java
@@ -20,13 +20,18 @@
 import com.google.inject.Provider;
 
 public class LoadTesterNettyConfigProvider implements Provider<NettyServerConfig> {
-    public LoadTesterNettyConfigProvider() {
+    private final NiftyLoadTester.LoadTesterConfig config;
+
+    @Inject
+    public LoadTesterNettyConfigProvider(NiftyLoadTester.LoadTesterConfig config)
+    {
+        this.config = config;
     }
 
     @Override
     public NettyServerConfig get() {
         NettyServerConfigBuilder configBuilder = new NettyServerConfigBuilder();
-        configBuilder.getServerSocketChannelConfig().setBacklog(1024);
+        configBuilder.getServerSocketChannelConfig().setBacklog(config.getAcceptBacklog());
         return configBuilder.build();
     }
 }
diff --git a/nifty-load-tester/src/main/java/com/facebook/nifty/perf/NiftyLoadTester.java b/nifty-load-tester/src/main/java/com/facebook/nifty/perf/NiftyLoadTester.java
@@ -116,6 +116,7 @@ public static class LoadTesterConfig
         private boolean useTaskQueue = false;
         private int numTaskThreads = 8;
         private int queuedResponseLimit = 500;
+        private int acceptBacklog = 1024;
 
         public int getServerPort()
         {
@@ -171,5 +172,16 @@ public void setQueuedResponseLimit(int queuedResponseLimit)
         {
             this.queuedResponseLimit = queuedResponseLimit;
         }
+
+        public int getAcceptBacklog()
+        {
+            return acceptBacklog;
+        }
+
+        @Config("acceptBacklog")
+        public void setAcceptBacklog(int acceptBacklog)
+        {
+            this.acceptBacklog = acceptBacklog;
+        }
     }
 }