v1.2.1 可手动指定某个应用的负载均衡策略

Author: codingmiao

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>org.wowtools.springcloudext</groupId>
     <artifactId>ngineureka</artifactId>
-    <version>1.2-SNAPSHOT</version>
+    <version>1.2.1</version>
 
     <parent>
         <groupId>org.springframework.boot</groupId>

src/main/java/org/wowtools/springcloudext/ngineureka/service/NginxService.java

@@ -8,6 +8,8 @@
 
 import java.io.*;
 import java.util.HashMap;
+import java.util.Properties;
+import java.util.Set;
 
 /**
  * 操作nginx的服务
@@ -49,6 +51,51 @@ public class NginxService {
             }
         }
 
+    }
+
+    /**
+     * 负载均衡策略
+     */
+    private final HashMap<String, String> loadBalancingStrategys;
+
+    {
+        String path = Constant.rootPath + "loadBalancingStrategy.properties";
+        if (new File(path).exists()) {
+            Properties p = new Properties();
+            try {
+                p.load(ResourcesReader.readStream(path));
+            } catch (IOException e) {
+                throw new RuntimeException("loadBalancingStrategy.properties config err", e);
+            }
+            Set<String> appNames = p.stringPropertyNames();
+            loadBalancingStrategys = new HashMap<>(appNames.size());
+            for (String appName : appNames) {
+                String value = p.getProperty(appName);
+                String strategy;
+                switch (value) {
+                    case "least_conn":
+                        strategy = "\tleast_conn;\n";
+                        break;
+                    case "ip_hash":
+                        strategy = "\tip_hash;\n";
+                        break;
+                    case "polling":
+                        strategy = "";
+                        break;
+                    case "url_hash":
+                        strategy = "\thash $request_uri;\n";
+                        break;
+                    case "fair":
+                        strategy = "\tfair;\n";
+                        break;
+                    default:
+                        strategy = "\tleast_conn;\n";
+                }
+                loadBalancingStrategys.put(appName, strategy);
+            }
+        } else {
+            loadBalancingStrategys = new HashMap<>(0);
+        }
 
     }
 
@@ -77,9 +124,14 @@ private void writeCfg(Record record) {
             for (ServiceRecord service : services) {
                 String[] urls = service.getServiceUrls();
                 String appName = service.getName();
+                //upstream
                 sbUpstream.append("upstream upstream-").append(appName).append("{\n");
-                sbUpstream.append("\tleast_conn;\n");
-
+                String loadBalancingStrategy = loadBalancingStrategys.get(appName);
+                if (null == loadBalancingStrategy) {
+                    loadBalancingStrategy = "\tleast_conn;\n";
+                }
+                sbUpstream.append(loadBalancingStrategy);
+                //location
                 sbServer.append("location ^~ /").append(appName).append("/ {\n");
                 sbServer.append("\tproxy_pass http://upstream-").append(appName).append(";\n");
                 String param = locationParams.get(appName);

src/main/resources/loadBalancingStrategy.properties

@@ -0,0 +1,8 @@
+#负载均衡策略配置，支持以下策略
+#least_conn 默认，最小连接数
+#ip_hash 通过客户端请求ip进行hash，再通过hash值选择后端server
+#polling 轮询
+#url_hash 通过请求url进行hash，再通过hash值选择后端server（需安装对应模块）
+#fair 按后端服务器的响应时间来分配请求，响应时间短的优先分配（需安装对应模块）
+#示例 myapp=url_hash 表示名称为myapp的这个应用使用url_hash策略
+ngineureka=polling