【修改】 优化 共用 zookeeper client

loit-build-component/loit-keygen-leaf-zk/src/main/java/com/loit/component/keygen/leaf/LeafSegmentKeyGenerator.java

@@ -43,6 +43,11 @@ public final class LeafSegmentKeyGenerator implements LeafKeyGenerator {
 
     private long step;
 
+    /**
+     * 是否第一次加载
+     */
+    private Boolean isInit = true;
+
     @Getter
     @Setter
     private Properties properties = new Properties();
@@ -60,17 +65,36 @@ public final class LeafSegmentKeyGenerator implements LeafKeyGenerator {
     @Override
     public synchronized Comparable<?> generateKey() {
         String leafKey = getLeafKey();
-        if (null == leafRegistryCenter) {
+        if (isInit) {
             initLeafSegmentKeyGenerator(leafKey);
         } else {
             increaseIdWhenLeafKeyStoredInCenter(leafKey);
         }
+
+        isInit = false;
         return id;
     }
 
-    private void initLeafSegmentKeyGenerator(final String leafKey) {
+    /**
+     * 构建zookeeper注册中心
+     */
+    public RegistryCenter buidLeafRegistryCenter() {
         RegistryCenterConfiguration leafConfiguration = getRegistryCenterConfiguration();
         leafRegistryCenter = new RegistryCenterServiceLoader().load(leafConfiguration);
+
+        return leafRegistryCenter;
+    }
+
+    /**
+     * 设置zookeeper注册中心(共用注册中心)
+     */
+    public void setLeafRegistryCenter(RegistryCenter registryCenter) {
+        this.leafRegistryCenter = registryCenter;
+    }
+
+    private void initLeafSegmentKeyGenerator(final String leafKey) {
+        buidLeafRegistryCenter();
+
         step = getStep();
         id = initializeId(leafKey);
         initializeLeafKeyInCenter(leafKey, id, step);

loit-build-component/loit-keygen-leaf-zk/src/main/java/com/loit/component/keygen/leaf/LeafSegmentKeyGeneratorFactory.java

 package com.loit.component.keygen.leaf;
 
+import com.loit.component.keygen.leaf.core.reg.RegistryCenter;
 import org.apache.commons.lang3.StringUtils;
 
 import java.util.LinkedHashMap;
@@ -15,6 +16,8 @@ public class LeafSegmentKeyGeneratorFactory {
 
     private Properties globleProperties = new Properties();
 
+    private RegistryCenter registryCenter;
+
     /**
      * 根据 leafKey 标识不同的keyGenerator
      */
@@ -36,6 +39,12 @@ public class LeafSegmentKeyGeneratorFactory {
             properties.setProperty("registry.center.type", globleProperties.getProperty("registry.center.type", LeafSegmentKeyGenerator.DEFAULT_REGISTRY_CENTER));
             leafSegmentKeyGenerator.setProperties(properties);
 
+            if (registryCenter == null) {
+                registryCenter = leafSegmentKeyGenerator.buidLeafRegistryCenter();
+            } else {
+                leafSegmentKeyGenerator.setLeafRegistryCenter(registryCenter);
+            }
+
             keyGeneratorMap.put(leafKey, leafSegmentKeyGenerator);
         }
 

loit-build-component/loit-keygen-leaf-zk/src/test/java/com/loit/component/keygen/leaf/LeafSegmentKeyGeneratorFactoryTest.java

@@ -22,9 +22,11 @@ public class LeafSegmentKeyGeneratorFactoryTest {
         properties.setProperty("leaf.segment.step", "100");
         leafSegmentKeyGeneratorFactory.setGlobleProperties(properties);
 
-        for (int i = 0; i < 12; i++) {
-            Comparable<?> t_order_item = leafSegmentKeyGeneratorFactory.getKeyGenerator("t_order_item").generateKey();
-            System.out.println(t_order_item);
+        for (int i = 0; i < 20; i++) {
+            for (int j = 0; j < 50; j++) {
+                Comparable<?> t_order_item = leafSegmentKeyGeneratorFactory.getKeyGenerator("t_order_item" + i).generateKey();
+                System.out.println("t_order_item" + i + ": " + t_order_item);
+            }
         }
 
     }