add leaf starter

xlorne · xlorne · commit aeebb0472209 · 2022-07-26T11:44:52.000+08:00
diff --git a/pom.xml b/pom.xml
@@ -2,6 +2,9 @@
 <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 	<modelVersion>4.0.0</modelVersion>
 	<packaging>pom</packaging>
+    <modules>
+        <module>springboot-starter-leaf</module>
+    </modules>
     <parent>
 		<groupId>org.springframework.boot</groupId>
 		<artifactId>spring-boot-starter-parent</artifactId>
@@ -35,6 +38,7 @@
 		<jsonwebtoken.jjwt.version>0.11.5</jsonwebtoken.jjwt.version>
 		<commons-io.version>2.11.0</commons-io.version>
 		<commons-dbutils.version>1.7</commons-dbutils.version>
+		<org.reflections.version>0.10.2</org.reflections.version>
 
 	</properties>
 
@@ -63,6 +67,13 @@
 
 	<dependencyManagement>
 		<dependencies>
+
+			<dependency>
+				<groupId>org.reflections</groupId>
+				<artifactId>reflections</artifactId>
+				<version>${org.reflections.version}</version>
+			</dependency>
+
 			<dependency>
 				<groupId>com.alibaba</groupId>
 				<artifactId>fastjson</artifactId>
@@ -106,6 +117,11 @@
 				<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>
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>
 			<artifactId>spring-boot-starter-web</artifactId>
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,16 +20,16 @@
 @NoArgsConstructor
 @ToString
 @Entity(name = "t_demo")
-public class Demo  {
+public class Demo implements LeafIdGenerate {
 
     @Id
-    @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Integer id;
 
     private String name;
 
     public Demo(String name) {
         this.name = name;
+        this.id = generateIntId();
     }
 
     public void changeName(String name){
diff --git a/springboot-example/src/main/resources/application.properties b/springboot-example/src/main/resources/application.properties
@@ -1,4 +1,6 @@
 
+server.port=8085
+
 spring.datasource.driver-class-name=org.h2.Driver
 spring.datasource.url=jdbc:h2:file:./demo.db
 
diff --git a/springboot-starter-leaf/pom.xml b/springboot-starter-leaf/pom.xml
@@ -0,0 +1,29 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>springboot-parent</artifactId>
+        <groupId>com.codingapi.springboot</groupId>
+        <version>1.1.2</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>springboot-starter-leaf</artifactId>
+
+    <properties>
+        <java.version>1.8</java.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.reflections</groupId>
+            <artifactId>reflections</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.data</groupId>
+            <artifactId>spring-data-commons</artifactId>
+        </dependency>
+    </dependencies>
+
+</project>
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,17 @@
+package com.codingapi.springboot.leaf;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * @author lorne
+ * @since 1.0.0
+ */
+@Configuration
+public class AutoConfiguration {
+
+    @Bean(initMethod = "init")
+    public LeafClient leafClient(){
+        return new LeafClient();
+    }
+}
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,48 @@
+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.info("classes:{}",classes);
+
+        LeafUtils.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,15 @@
+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/LeafClient.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafClient.java
@@ -0,0 +1,34 @@
+package com.codingapi.springboot.leaf;
+
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * @author lorne
+ * @since 1.0.0
+ */
+@Slf4j
+public class LeafClient {
+
+    public long segmentGetId(String key){
+        //todo leaf
+        return 1;
+    }
+
+
+    /**
+     * 数据库模式添加 数据
+     * @param key   key 关键字
+     * @param step  每次获取数据长度 2000
+     * @param maxId 开始的最大Id 1
+     * @return  执行状态
+     */
+    public boolean segmentPush(String key, int step, int maxId){
+        //todo leaf
+        return true;
+    }
+
+    public void init(){
+        LeafUtils.getInstance().setLeafClient(this);
+    }
+
+}
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 LeafUtils.getInstance().generateId(getClass());
+    }
+
+    default int generateIntId(){
+        return (int)LeafUtils.getInstance().generateId(getClass());
+    }
+
+    default String generateStringId(){
+        return String.valueOf(LeafUtils.getInstance().generateId(getClass()));
+    }
+
+}
diff --git a/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafUtils.java b/springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafUtils.java
@@ -0,0 +1,75 @@
+package com.codingapi.springboot.leaf;
+
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.Set;
+
+/**
+ * @author lorne
+ * @since 1.0.0
+ */
+@Slf4j
+public class LeafUtils {
+
+    private LeafClient leafClient;
+    @Setter
+    private Set<Class<? extends LeafIdGenerate>> classes;
+
+    private LeafUtils(){
+    }
+
+    private static LeafUtils instance;
+
+    public static LeafUtils getInstance() {
+        if(instance==null){
+            synchronized (LeafUtils.class){
+                if(instance==null){
+                    instance = new LeafUtils();
+                }
+            }
+        }
+        return instance;
+    }
+
+    protected void setLeafClient(LeafClient leafClient){
+        this.leafClient = leafClient;
+        this.initClass();
+    }
+
+
+    public long generateId(Class<?> clazz){
+        return segmentGetId(clazz);
+    }
+
+    private long segmentGetId(Class<?> clazz){
+        if(leafClient==null){
+            throw new RuntimeException("springboot init finish then to running.");
+        }
+        return leafClient.segmentGetId(clazz.getName());
+    }
+
+
+     boolean push(String key,int step,int maxId){
+        if(leafClient==null){
+            throw new RuntimeException("springboot init finish then to running.");
+        }
+        return leafClient.segmentPush(key,step,maxId);
+    }
+
+
+    private void initClass(){
+        if(classes!=null&&classes.size()>0) {
+            for (Class<?> clazz : classes) {
+                try {
+                    LeafUtils.getInstance().push(clazz.getName(), 2000, 1);
+                } catch (Exception e) {
+                    e.printStackTrace();
+                }
+            }
+        }
+
+        log.info("leaf init finish.");
+    }
+
+}
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
