add leaf starter ..

Author: xlorne

springboot-starter-leaf/pom.xml

@@ -21,6 +21,11 @@
             <artifactId>reflections</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>commons-dbutils</groupId>
+            <artifactId>commons-dbutils</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>org.perf4j</groupId>
             <artifactId>perf4j</artifactId>

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/h2/DataHelper.java

@@ -0,0 +1,56 @@
+package com.codingapi.springboot.leaf.h2;
+
+import org.apache.commons.dbutils.QueryRunner;
+import org.apache.commons.dbutils.ResultSetHandler;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.SQLException;
+
+public class DataHelper {
+
+    private DataSource dataSource;
+
+    private QueryRunner queryRunner;
+
+    public DataHelper(DataSource dataSource) {
+        this.dataSource = dataSource;
+        this.queryRunner = new QueryRunner();
+    }
+
+    public void execute(String sql,Object...params) throws DbException {
+        try {
+            Connection connection = dataSource.getConnection();
+            queryRunner.execute(connection,sql,params);
+            connection.close();
+        } catch (SQLException e) {
+            throw new DbException(e);
+        }
+    }
+
+    public void update(String sql,Object...params) throws DbException {
+        try {
+            Connection connection = dataSource.getConnection();
+            connection.setAutoCommit(false);
+            queryRunner.update(connection,sql,params);
+            connection.commit();
+            connection.close();
+        } catch (SQLException e) {
+            throw new DbException(e);
+        }
+    }
+
+    public <T> void query(String sql, ResultSetHandler<T> handler, Object...params)throws DbException{
+        try {
+            Connection connection = dataSource.getConnection();
+            queryRunner.query(connection,sql,handler,params);
+            connection.close();
+        } catch (SQLException e) {
+            throw new DbException(e);
+        }
+    }
+
+
+
+
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/h2/DbException.java

@@ -0,0 +1,24 @@
+package com.codingapi.springboot.leaf.h2;
+
+public class DbException extends RuntimeException{
+
+    public DbException() {
+    }
+
+    public DbException(String message) {
+        super(message);
+    }
+
+    public DbException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public DbException(Throwable cause) {
+        super(cause);
+    }
+
+    public DbException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+
+}

springboot-starter-leaf/src/main/java/com/codingapi/springboot/leaf/segment/dao/impl/IDAllocDaoImpl.java

@@ -1,5 +1,6 @@
 package com.codingapi.springboot.leaf.segment.dao.impl;
 
+import com.codingapi.springboot.leaf.h2.DataHelper;
 import com.codingapi.springboot.leaf.segment.dao.IDAllocDao;
 import com.codingapi.springboot.leaf.segment.model.LeafAlloc;
 import lombok.AllArgsConstructor;
@@ -11,34 +12,50 @@
 @AllArgsConstructor
 public class IDAllocDaoImpl implements IDAllocDao {
 
+    private DataHelper dataHelper;
+
     @Override
     public List<LeafAlloc> getAllLeafAllocs() {
         //todo leaf
+        String sql = "select * from leaf_alloc";
         return new ArrayList<>();
     }
 
     @Override
     public LeafAlloc updateMaxIdAndGetLeafAlloc(String tag) {
         //todo leaf
+        //UPDATE leaf_alloc SET maxId = maxId + step WHERE key = :key
+
+        //select * from leaf_alloc where key = :key
         LeafAlloc leafAlloc = new LeafAlloc();
         return leafAlloc;
     }
 
     @Override
     public LeafAlloc updateMaxIdByCustomStepAndGetLeafAlloc(LeafAlloc leafAlloc) {
+        //UPDATE leaf_alloc SET maxId = maxId + :step WHERE key = :key
+
+        //select * from leaf_alloc where key = :key
         //todo leaf
         return leafAlloc;
     }
 
     @Override
     public List<String> getAllTags() {
+        String sql = "select tag from leaf_alloc";
         //todo leaf
         return Arrays.asList("tags");
     }
 
     @Override
     public boolean save(LeafAlloc leafInfo) {
+        String sql = "insert into leaf_alloc (max_id, step, update_time, tag) values (?, ?, ?, ?)";
         //todo leaf
         return true;
     }
+
+    public void init(){
+        String sql = "create table leaf_alloc (tag varchar(128) not null, max_id bigint, step integer, update_time timestamp, primary key (tag))";
+
+    }
 }