add leaf starter

xlorne · xlorne · commit 7aa33db00f65 · 2022-07-29T00:13:59.000+08:00
diff --git a/pom.xml b/pom.xml
@@ -128,6 +128,12 @@
 				<version>${codingapi.framework.version}</version>
 			</dependency>
 
+			<dependency>
+				<groupId>com.codingapi.springboot</groupId>
+				<artifactId>springboot-starter-leaf</artifactId>
+				<version>${codingapi.framework.version}</version>
+			</dependency>
+
 			<dependency>
 				<groupId>commons-dbutils</groupId>
 				<artifactId>commons-dbutils</artifactId>
diff --git a/springboot-example/pom.xml b/springboot-example/pom.xml
@@ -35,6 +35,11 @@
 			<artifactId>springboot-starter-security-jwt</artifactId>
 		</dependency>
 
+		<dependency>
+			<groupId>com.codingapi.springboot</groupId>
+			<artifactId>springboot-starter-leaf</artifactId>
+		</dependency>
+
 
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
diff --git a/springboot-example/src/main/java/com/codingapi/springboot/example/ExampleApplication.java b/springboot-example/src/main/java/com/codingapi/springboot/example/ExampleApplication.java
@@ -1,5 +1,6 @@
 package com.codingapi.springboot.example;
 
+import com.codingapi.springboot.leaf.EnableLeaf;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.context.annotation.Bean;
@@ -9,6 +10,7 @@
 import java.util.Locale;
 
 @SpringBootApplication
+@EnableLeaf
 public class ExampleApplication {
 
 	public static void main(String[] args) {
diff --git a/springboot-example/src/main/java/com/codingapi/springboot/example/domain/Demo.java b/springboot-example/src/main/java/com/codingapi/springboot/example/domain/Demo.java
@@ -2,14 +2,13 @@
 
 import com.codingapi.springboot.example.event.DemoNameChangeEvent;
 import com.codingapi.springboot.framework.event.EventPusher;
+import com.codingapi.springboot.leaf.LeafIdGenerate;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 import lombok.Setter;
 import lombok.ToString;
 
 import javax.persistence.Entity;
-import javax.persistence.GeneratedValue;
-import javax.persistence.GenerationType;
 import javax.persistence.Id;
 
 /**
@@ -21,15 +20,15 @@
 @NoArgsConstructor
 @ToString
 @Entity(name = "t_demo")
-public class Demo  {
+public class Demo  implements LeafIdGenerate {
 
     @Id
-    @GeneratedValue(strategy = GenerationType.AUTO)
     private Integer id;
 
     private String name;
 
     public Demo(String name) {
+        this.id = generateIntId();
         this.name = name;
     }
 
diff --git a/springboot-starter-leaf/pom.xml b/springboot-starter-leaf/pom.xml
@@ -17,6 +17,12 @@
 
 
     <dependencies>
+
+        <dependency>
+            <groupId>org.reflections</groupId>
+            <artifactId>reflections</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>org.perf4j</groupId>
             <artifactId>perf4j</artifactId>
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/AutoConfiguration.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/AutoConfiguration.java
@@ -0,0 +1,31 @@
+package com.codingapi.springboot.leaf;
+
+import com.sankuai.inf.leaf.IDGen;
+import com.sankuai.inf.leaf.segment.SegmentIDGenImpl;
+import com.sankuai.inf.leaf.segment.dao.IDAllocDao;
+import com.sankuai.inf.leaf.segment.dao.impl.IDAllocDaoImpl;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class AutoConfiguration {
+
+    @Bean
+    @ConditionalOnMissingBean
+    public IDAllocDao allocDao(){
+        return new IDAllocDaoImpl("jdbc:h2:mem:leaf;DB_CLOSE_DELAY=-1");
+    }
+
+
+    @Bean
+    public IDGen idGen(IDAllocDao allocDao){
+        return new SegmentIDGenImpl(allocDao);
+    }
+
+    @Bean(initMethod = "init")
+    public Leaf leafClient(IDGen idGen, IDAllocDao allocDao){
+        return new Leaf(idGen,allocDao);
+    }
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/AutoConfigurationImportSelector.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/AutoConfigurationImportSelector.java
@@ -0,0 +1,47 @@
+package com.codingapi.springboot.leaf;
+
+import lombok.extern.slf4j.Slf4j;
+import org.reflections.Reflections;
+import org.reflections.scanners.Scanners;
+import org.reflections.util.ConfigurationBuilder;
+import org.springframework.beans.factory.support.BeanDefinitionRegistry;
+import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
+import org.springframework.core.type.AnnotationMetadata;
+
+import java.util.*;
+
+@Slf4j
+public class AutoConfigurationImportSelector implements ImportBeanDefinitionRegistrar {
+
+    private Set<Class<? extends LeafIdGenerate>> classes;
+
+    @Override
+    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
+        String className = importingClassMetadata.getClassName();
+        List<String> packageNames = new ArrayList<>();
+        try {
+            packageNames.add(Class.forName(className).getPackage().getName());
+
+            Map<String,Object> annotations =  importingClassMetadata.getAnnotationAttributes(EnableLeaf.class.getName());
+            assert annotations != null;
+            String[] packages = (String[])annotations.get("scanBasePackages");
+            packageNames.addAll(Arrays.asList(packages));
+
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+
+        Reflections reflections =  new Reflections(new ConfigurationBuilder()
+                .forPackages(packageNames.toArray(new String[]{}))
+                .addScanners(Scanners.TypesAnnotated,Scanners.SubTypes));
+
+        this.classes = reflections.getSubTypesOf(LeafIdGenerate.class);
+        log.debug("classes:{}",classes);
+
+        LeafContext.getInstance().setClasses(classes);
+
+    }
+
+
+}
+
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/EnableLeaf.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/EnableLeaf.java
@@ -0,0 +1,14 @@
+package com.codingapi.springboot.leaf;
+import org.springframework.context.annotation.Import;
+
+import java.lang.annotation.*;
+
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+@Import(AutoConfigurationImportSelector.class)
+public @interface EnableLeaf {
+
+    String[] scanBasePackages() default {};
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/Leaf.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/Leaf.java
@@ -0,0 +1,65 @@
+package com.codingapi.springboot.leaf;
+
+
+import com.codingapi.springboot.leaf.exception.LeafServerException;
+import com.codingapi.springboot.leaf.exception.NoKeyException;
+import com.sankuai.inf.leaf.IDGen;
+import com.sankuai.inf.leaf.common.Result;
+import com.sankuai.inf.leaf.common.Status;
+import com.sankuai.inf.leaf.segment.dao.IDAllocDao;
+import com.sankuai.inf.leaf.segment.model.LeafAlloc;
+import lombok.AllArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.Date;
+
+/**
+ * @author lorne
+ * @since 1.0.0
+ */
+@Slf4j
+@AllArgsConstructor
+class Leaf {
+
+    private final IDGen idGen;
+    private final IDAllocDao idAllocDao;
+
+
+    long segmentGetId(String key){
+        Result result =  idGen.get(key);
+        if (key == null || key.isEmpty()) {
+            throw new NoKeyException();
+        }
+        if (result.getStatus().equals(Status.EXCEPTION)) {
+            throw new LeafServerException(result.toString());
+        }
+        return result.getId();
+    }
+
+
+    /**
+     * 数据库模式添加 数据
+     * @param key   key 关键字
+     * @param step  每次获取数据长度 2000
+     * @param maxId 开始的最大Id 1
+     * @return  执行状态
+     */
+    boolean segmentPush(String key, int step, int maxId)  {
+        LeafAlloc alloc = new LeafAlloc();
+        alloc.setKey(key);
+        alloc.setStep(step);
+        alloc.setMaxId(maxId);
+        alloc.setUpdateTime(new Date());
+        return idAllocDao.save(alloc);
+    }
+
+    void initIdGen(){
+        idGen.init();
+    }
+
+     void init(){
+        LeafContext.getInstance().setLeaf(this);
+    }
+
+}
+
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafContext.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafContext.java
@@ -0,0 +1,70 @@
+package com.codingapi.springboot.leaf;
+
+import com.codingapi.springboot.leaf.exception.LeafServerException;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.Set;
+
+/**
+ * @author lorne
+ * @since 1.0.0
+ */
+@Slf4j
+class LeafContext {
+
+    private Leaf leaf;
+    @Setter
+    private Set<Class<? extends LeafIdGenerate>> classes;
+
+    private LeafContext(){
+    }
+
+    private static LeafContext instance;
+
+    public static LeafContext getInstance() {
+        if(instance==null){
+            synchronized (LeafContext.class){
+                if(instance==null){
+                    instance = new LeafContext();
+                }
+            }
+        }
+        return instance;
+    }
+
+    protected void setLeaf(Leaf leaf){
+        this.leaf = leaf;
+        this.initClass();
+    }
+
+
+    long generateId(Class<?> clazz){
+        return segmentGetId(clazz);
+    }
+
+    private long segmentGetId(Class<?> clazz){
+        return leaf.segmentGetId(clazz.getName());
+    }
+
+
+    boolean push(String key, int step, int maxId){
+        return leaf.segmentPush(key,step,maxId);
+    }
+
+
+    private void initClass(){
+        if(classes!=null&&classes.size()>0) {
+            for (Class<?> clazz : classes) {
+                try {
+                    LeafContext.getInstance().push(clazz.getName(), 2000, 1);
+                } catch (Exception e) {
+                    throw new LeafServerException(e);
+                }
+            }
+        }
+        leaf.initIdGen();
+        log.info("leaf init finish.");
+    }
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafIdGenerate.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafIdGenerate.java
@@ -0,0 +1,18 @@
+package com.codingapi.springboot.leaf;
+
+public interface LeafIdGenerate {
+
+
+    default long generateLongId(){
+        return LeafContext.getInstance().generateId(getClass());
+    }
+
+    default int generateIntId(){
+        return (int) LeafContext.getInstance().generateId(getClass());
+    }
+
+    default String generateStringId(){
+        return String.valueOf(LeafContext.getInstance().generateId(getClass()));
+    }
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/exception/LeafServerException.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/exception/LeafServerException.java
@@ -0,0 +1,23 @@
+package com.codingapi.springboot.leaf.exception;
+
+public class LeafServerException extends RuntimeException{
+
+    public LeafServerException() {
+    }
+
+    public LeafServerException(String message) {
+        super(message);
+    }
+
+    public LeafServerException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public LeafServerException(Throwable cause) {
+        super(cause);
+    }
+
+    public LeafServerException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/exception/NoKeyException.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/exception/NoKeyException.java
@@ -0,0 +1,6 @@
+package com.codingapi.springboot.leaf.exception;
+
+public class NoKeyException extends RuntimeException{
+
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/sankuai/inf/leaf/segment/SegmentIDGenImpl.java b/springboot-starter-leaf/src/main/java/com/sankuai/inf/leaf/segment/SegmentIDGenImpl.java
@@ -42,6 +42,10 @@ public class SegmentIDGenImpl implements IDGen {
     private Map<String, SegmentBuffer> cache = new ConcurrentHashMap<String, SegmentBuffer>();
     private IDAllocDao dao;
 
+    public SegmentIDGenImpl(IDAllocDao dao) {
+        this.dao = dao;
+    }
+
     public static class UpdateThreadFactory implements ThreadFactory {
 
         private static int threadInitNumber = 0;
diff --git a/springboot-starter-leaf/src/main/resources/META-INF/spring.factories b/springboot-starter-leaf/src/main/resources/META-INF/spring.factories
@@ -0,0 +1,2 @@
+org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
+com.codingapi.springboot.leaf.AutoConfiguration
diff --git a/springboot-starter-leaf/src/test/java/com/sankuai/inf/leaf/IDGenTest.java b/springboot-starter-leaf/src/test/java/com/sankuai/inf/leaf/IDGenTest.java
@@ -27,11 +27,10 @@ public void beforeInitIdGen() {
         alloc.setMaxId(0);
         alloc.setStep(100);
         alloc.setKey("leaf-segment-test");
+        allocDao.save(alloc);
 
         //init id gen
-        allocDao.save(alloc);
-        idGen = new SegmentIDGenImpl();
-        ((SegmentIDGenImpl) idGen).setDao(allocDao);
+        idGen = new SegmentIDGenImpl(allocDao);
         idGen.init();
     }
 
