执行器线程数量配置

cgb-springcloud · Jul 11, 2023 · c48db42 · c48db42
1 parent a689eed
commit c48db42
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 10 deletions.
diff --git a/...thub/linyuzai/connection/loadbalance/core/message/retry/AbstractMessageRetryStrategy.java b/...thub/linyuzai/connection/loadbalance/core/message/retry/AbstractMessageRetryStrategy.java
@@ -43,11 +43,13 @@ public class RetryRunnable implements Runnable {
 
         @Override
         public void run() {
+            int times = getTimes();
+            String currentTimes = current + "/" + times;
+            concept.getLogger().info("Start retry and current is " + currentTimes);
             retryable.accept(e -> {
-                int times = getTimes();
+                String retryErrorMessage = "Retry failed " + currentTimes;
                 int newCurrent = current + 1;
                 int period = getPeriod(newCurrent);
-                String retryErrorMessage = "Retry failed " + current + "/" + times;
                 MessageRetryException retryException =
                         new MessageRetryException(retryErrorMessage, e, error);
                 if (current < times) {

diff --git a/.../java/com/github/linyuzai/connection/loadbalance/netty/NettyLoadBalanceConfiguration.java b/.../java/com/github/linyuzai/connection/loadbalance/netty/NettyLoadBalanceConfiguration.java
@@ -165,9 +165,11 @@ public MessageIdempotentVerifierFactory nettyMessageIdempotentVerifierFactory()
     }
 
     @Bean
-    public ScheduledExecutorFactory nettyScheduledExecutorFactory() {
-        return new ScheduledExecutorFactoryImpl()
-                .addScopes(NettyScoped.NAME);
+    public ScheduledExecutorFactory nettyScheduledExecutorFactory(NettyLoadBalanceProperties properties) {
+        ScheduledExecutorFactoryImpl factory = new ScheduledExecutorFactoryImpl();
+        factory.setSize(properties.getExecutor().getSize());
+        factory.addScopes(NettyScoped.NAME);
+        return factory;
     }
 
     @Bean

diff --git a/...ain/java/com/github/linyuzai/connection/loadbalance/netty/NettyLoadBalanceProperties.java b/...ain/java/com/github/linyuzai/connection/loadbalance/netty/NettyLoadBalanceProperties.java
@@ -20,6 +20,8 @@ public class NettyLoadBalanceProperties {
      */
     private LoadBalanceProperties loadBalance = new LoadBalanceProperties();
 
+    private ExecutorProperties executor = new ExecutorProperties();
+
     @Data
     public static class ServerProperties {
 
@@ -67,6 +69,12 @@ public enum Subscriber {
         }
     }
 
+    @Data
+    public static class ExecutorProperties {
+
+        private int size = 1;
+    }
+
     @Data
     public static class MessageProperties {
 

diff --git a/...m/github/linyuzai/connection/loadbalance/websocket/EnableWebSocketLoadBalanceConcept.java b/...m/github/linyuzai/connection/loadbalance/websocket/EnableWebSocketLoadBalanceConcept.java
@@ -15,7 +15,6 @@
 @Inherited
 @EnableConnectionLoadBalanceConfiguration
 @EnableConfigurationProperties(WebSocketLoadBalanceProperties.class)
-@Import({WebSocketLoadBalanceConfiguration.class,
-        WebSocketLoadBalanceImportSelector.class})
+@Import({WebSocketLoadBalanceImportSelector.class, WebSocketLoadBalanceConfiguration.class})
 public @interface EnableWebSocketLoadBalanceConcept {
 }
diff --git a/...m/github/linyuzai/connection/loadbalance/websocket/WebSocketLoadBalanceConfiguration.java b/...m/github/linyuzai/connection/loadbalance/websocket/WebSocketLoadBalanceConfiguration.java
@@ -150,9 +150,12 @@ public MessageIdempotentVerifierFactory wsMessageIdempotentVerifierFactory() {
     }
 
     @Bean
-    public ScheduledExecutorFactory wsScheduledExecutorFactory() {
-        return new ScheduledExecutorFactoryImpl()
-                .addScopes(WebSocketScoped.NAME);
+    public ScheduledExecutorFactory wsScheduledExecutorFactory(WebSocketLoadBalanceProperties properties) {
+        ScheduledExecutorFactoryImpl factory = new ScheduledExecutorFactoryImpl();
+        factory.setSize(properties.getExecutor().getSize());
+        factory.addScopes(WebSocketScoped.NAME);
+        return factory;
+
     }
 
     @Bean

diff --git a/.../com/github/linyuzai/connection/loadbalance/websocket/WebSocketLoadBalanceProperties.java b/.../com/github/linyuzai/connection/loadbalance/websocket/WebSocketLoadBalanceProperties.java
@@ -26,6 +26,8 @@ public class WebSocketLoadBalanceProperties {
      */
     private LoadBalanceProperties loadBalance = new LoadBalanceProperties();
 
+    private ExecutorProperties executor = new ExecutorProperties();
+
     @Data
     public static class ServerProperties {
 
@@ -166,6 +168,12 @@ public static class MonitorProperties {
         }
     }
 
+    @Data
+    public static class ExecutorProperties {
+
+        private int size = 1;
+    }
+
     @Data
     public static class MessageProperties {