add leaf starter ..

Author: xlorne

pom.xml

@@ -39,6 +39,7 @@
 		<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>
+		<perf4j.version>0.9.16</perf4j.version>
 
 	</properties>
 
@@ -74,6 +75,12 @@
 				<version>${org.reflections.version}</version>
 			</dependency>
 
+			<dependency>
+				<groupId>org.perf4j</groupId>
+				<artifactId>perf4j</artifactId>
+				<version>${perf4j.version}</version>
+			</dependency>
+
 			<dependency>
 				<groupId>com.alibaba</groupId>
 				<artifactId>fastjson</artifactId>

springboot-starter-leaf/pom.xml

@@ -20,6 +20,12 @@
             <groupId>org.reflections</groupId>
             <artifactId>reflections</artifactId>
         </dependency>
+
+        <dependency>
+            <groupId>org.perf4j</groupId>
+            <artifactId>perf4j</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>org.springframework.data</groupId>
             <artifactId>spring-data-commons</artifactId>

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/AutoConfiguration.java

@@ -1,5 +1,9 @@
 package com.codingapi.springboot.leaf;
 
+import com.codingapi.springboot.leaf.exception.InitException;
+import com.codingapi.springboot.leaf.segment.dao.IDAllocDao;
+import com.codingapi.springboot.leaf.segment.dao.impl.IDAllocDaoImpl;
+import com.codingapi.springboot.leaf.service.SegmentService;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
@@ -10,8 +14,18 @@
 @Configuration
 public class AutoConfiguration {
 
+    @Bean
+    public IDAllocDao allocDao(){
+        return new IDAllocDaoImpl();
+    }
+
+    @Bean
+    public SegmentService segmentService(IDAllocDao allocDao) throws InitException {
+        return new SegmentService(allocDao);
+    }
+
     @Bean(initMethod = "init")
-    public LeafClient leafClient(){
-        return new LeafClient();
+    public LeafClient leafClient(SegmentService segmentService){
+        return new LeafClient(segmentService);
     }
 }

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/IDGen.java

@@ -0,0 +1,12 @@
+package com.codingapi.springboot.leaf;
+
+
+import com.codingapi.springboot.leaf.common.Result;
+
+public interface IDGen {
+
+    Result get(String key);
+
+    boolean init();
+
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafClient.java

@@ -1,17 +1,38 @@
 package com.codingapi.springboot.leaf;
 
+import com.codingapi.springboot.leaf.common.Result;
+import com.codingapi.springboot.leaf.common.Status;
+import com.codingapi.springboot.leaf.exception.InitException;
+import com.codingapi.springboot.leaf.exception.LeafServerException;
+import com.codingapi.springboot.leaf.exception.NoKeyException;
+import com.codingapi.springboot.leaf.segment.model.LeafAlloc;
+import com.codingapi.springboot.leaf.service.SegmentService;
 import lombok.extern.slf4j.Slf4j;
 
+import java.util.Date;
+
 /**
  * @author lorne
  * @since 1.0.0
  */
 @Slf4j
 public class LeafClient {
 
+    private SegmentService segmentService;
+
+    public LeafClient(SegmentService segmentService) {
+        this.segmentService = segmentService;
+    }
+
     public long segmentGetId(String key){
-        //todo leaf
-        return 1;
+        Result result =  segmentService.getId(key);
+        if (key == null || key.isEmpty()) {
+            throw new NoKeyException();
+        }
+        if (result.getStatus().equals(Status.EXCEPTION)) {
+            throw new LeafServerException(result.toString());
+        }
+        return result.getId();
     }
 
 
@@ -22,8 +43,14 @@ public long segmentGetId(String key){
      * @param maxId 开始的最大Id 1
      * @return  执行状态
      */
-    public boolean segmentPush(String key, int step, int maxId){
+    public boolean segmentPush(String key, int step, int maxId) throws InitException {
         //todo leaf
+        LeafAlloc alloc = new LeafAlloc();
+        alloc.setKey(key);
+        alloc.setStep(step);
+        alloc.setMaxId(maxId);
+        alloc.setUpdateTime(new Date());
+        segmentService.save(alloc);
         return true;
     }
 

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/LeafUtils.java

@@ -1,6 +1,7 @@
 package com.codingapi.springboot.leaf;
 
 import lombok.Setter;
+import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 
 import java.util.Set;
@@ -50,7 +51,8 @@ private long segmentGetId(Class<?> clazz){
     }
 
 
-     boolean push(String key,int step,int maxId){
+     @SneakyThrows
+     boolean push(String key, int step, int maxId){
         if(leafClient==null){
             throw new RuntimeException("springboot init finish then to running.");
         }

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/common/CheckVO.java

@@ -0,0 +1,16 @@
+package com.codingapi.springboot.leaf.common;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+
+@Setter
+@Getter
+@AllArgsConstructor
+public class CheckVO {
+
+    private long timestamp;
+    private int workID;
+
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/common/PropertyFactory.java

@@ -0,0 +1,23 @@
+package com.codingapi.springboot.leaf.common;
+
+import lombok.extern.slf4j.Slf4j;
+
+import java.io.IOException;
+import java.util.Properties;
+
+@Slf4j
+public class PropertyFactory {
+
+    private static final Properties prop = new Properties();
+    static {
+        try {
+            prop.load(PropertyFactory.class.getClassLoader().getResourceAsStream("leaf.properties"));
+        } catch (IOException e) {
+            log.warn("Load Properties Ex", e);
+        }
+    }
+
+    public static Properties getProperties() {
+        return prop;
+    }
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/common/Result.java

@@ -0,0 +1,15 @@
+package com.codingapi.springboot.leaf.common;
+
+import lombok.*;
+
+@Setter
+@Getter
+@ToString
+@AllArgsConstructor
+@NoArgsConstructor
+public class Result {
+
+    private long id;
+    private Status status;
+
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/common/Status.java

@@ -0,0 +1,6 @@
+package com.codingapi.springboot.leaf.common;
+
+public enum  Status {
+    SUCCESS,
+    EXCEPTION
+}