发布自定义分布式主键ID

Author: chengxy-nds

springboot-sharding-jdbc/pom.xml

@@ -51,6 +51,13 @@
 			<scope>runtime</scope>
 		</dependency>
 
+		<!-- 集成数据库密码加密工具jasypt  jdk8版本-->
+		<dependency>
+			<groupId>com.github.ulisesbocchio</groupId>
+			<artifactId>jasypt-spring-boot-starter</artifactId>
+			<version>1.14</version>
+		</dependency>
+
 	</dependencies>
 
 	<build>

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/contoller/TestController.java

@@ -42,7 +42,7 @@ public class TestController {
     @ResponseBody
     public String insertOrder() {
 
-        for (int i = 0; i < 1; i++) {
+        for (int i = 0; i < 3; i++) {
 //            HintManager.clear();
 //            HintManager hintManager = HintManager.getInstance();
 //            hintManager.setDatabaseShardingValue(1);

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/key/IncrementShardingKeyGenerator.java

@@ -0,0 +1,42 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.xiaofu.sharding.key;
+
+import lombok.Getter;
+import lombok.Setter;
+import org.apache.shardingsphere.spi.keygen.ShardingKeyGenerator;
+
+import java.util.Properties;
+import java.util.concurrent.atomic.AtomicInteger;
+
+public final class IncrementShardingKeyGenerator implements ShardingKeyGenerator {
+    
+    @Getter
+    private final String type = "INCREMENT";
+    
+    private final AtomicInteger count = new AtomicInteger();
+    
+    @Getter
+    @Setter
+    private Properties properties = new Properties();
+    
+    @Override
+    public Comparable<?> generateKey() {
+        return count.incrementAndGet();
+    }
+}

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/key/MyKeyGenerator.java renamed to springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/key/MyShardingKeyGenerator.java

@@ -4,23 +4,32 @@
 import org.springframework.stereotype.Component;
 
 import java.util.Properties;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
- * @Author: xinzhifu
- * @Description:
+ * @Author: xiaofu
+ * @Description: 自定义主键生成器
  */
 @Component
-public class MyKeyGenerator implements ShardingKeyGenerator {
+public class MyShardingKeyGenerator implements ShardingKeyGenerator {
 
 
+    private final AtomicInteger count = new AtomicInteger();
+
+    /**
+     * 自定义的生成方案类型
+     */
     @Override
-    public Comparable<?> generateKey() {
-        return 1;
+    public String getType() {
+        return "XXX";
     }
 
+    /**
+     * 核心方法-生成主键ID
+     */
     @Override
-    public String getType() {
-        return "xiaofukey";
+    public Comparable<?> generateKey() {
+        return count.incrementAndGet();
     }
 
     @Override

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/model/TOrder.java

@@ -1,14 +1,13 @@
 package com.xiaofu.sharding.model;
 
-import com.baomidou.mybatisplus.annotation.TableId;
 import lombok.Data;
 
 import java.math.BigDecimal;
 
 @Data
 public class TOrder {
 
-    @TableId(value = "order_id")
+    //@TableId(value = "order_id")
     private Long orderId;
 
     private Long userId;

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/model/TOrderDto.java

@@ -1,14 +1,13 @@
 package com.xiaofu.sharding.model;
 
-import com.baomidou.mybatisplus.annotation.TableId;
 import lombok.Data;
 
 import java.math.BigDecimal;
 
 @Data
 public class TOrderDto {
 
-    @TableId(value = "order_id")
+    //@TableId(value = "order_id")
     private Long orderId;
 
     private String orderNo;

springboot-sharding-jdbc/src/main/java/com/xiaofu/sharding/utils/JasyptUtil.java

@@ -0,0 +1,54 @@
+package com.xiaofu.sharding.utils;
+
+import lombok.extern.slf4j.Slf4j;
+import org.jasypt.util.text.BasicTextEncryptor;
+
+/**
+ * @Author: xinzhifu
+ * @Description:
+ */
+@Slf4j
+public class JasyptUtil {
+
+    /**
+     * 加密方法
+     *
+     * @param salt         盐值
+     * @param targetString 待加密字符串
+     * @return 密文
+     */
+    public static String encrypt(String salt, String targetString) {
+        BasicTextEncryptor encryptor = new BasicTextEncryptor();
+        encryptor.setPassword(salt);
+        return encryptor.encrypt(targetString);
+    }
+
+    /**
+     * 解密方法
+     *
+     * @param salt         盐值
+     * @param targetString 待解密字符串
+     * @return 明文
+     */
+    public static String decrypt(String salt, String targetString) {
+        BasicTextEncryptor encryptor = new BasicTextEncryptor();
+        encryptor.setPassword(salt);
+        return encryptor.decrypt(targetString);
+    }
+
+    public static void main(String[] args) {
+
+        String salt = "order_database";
+        String username = "root";
+        String password = "xinzhifu521";
+        // 进行加密操作
+        String usernameEncryptStr = encrypt(salt, username);
+        String passwordEncryptStr = encrypt(salt, password);
+        // 进行解密操作
+        String usernameDecryptStr = decrypt(salt, usernameEncryptStr);
+        String passwordDecryptStr = decrypt(salt, passwordEncryptStr);
+        // 输出明文和密文
+        log.info("usernameEncrypt :{}", usernameEncryptStr);
+        log.info("passwordEncryptStr :{}", passwordEncryptStr);
+    }
+}

springboot-sharding-jdbc/src/main/resources/META-INF/services/org.apache.shardingsphere.spi.keygen.ShardingKeyGenerator

@@ -1 +1 @@
-com.xiaofu.sharding.key.MyKeyGenerator
+com.xiaofu.sharding.key.MyShardingKeyGenerator

springboot-sharding-jdbc/src/main/resources/application.properties

@@ -19,20 +19,21 @@ spring.shardingsphere.datasource.ds-1.password=xinzhifu521
 spring.shardingsphere.sharding.tables.t_order.actual-data-nodes=ds-$->{0..1}.t_order_$->{0..2}
 ### 分库策略
 # 分库分片健
-spring.shardingsphere.sharding.tables.t_order.database-strategy.inline.sharding-column=order_id
+spring.shardingsphere.sharding.tables.t_order.database-strategy.inline.sharding-column=user_id
 # 分库分片算法
-spring.shardingsphere.sharding.tables.t_order.database-strategy.inline.algorithm-expression=ds-$->{order_id % 2}
+spring.shardingsphere.sharding.tables.t_order.database-strategy.inline.algorithm-expression=ds-$->{user_id % 2}
 # 分表策略
 # 分表分片健
-spring.shardingsphere.sharding.tables.t_order.table-strategy.inline.sharding-column=order_id
+spring.shardingsphere.sharding.tables.t_order.table-strategy.inline.sharding-column=user_id
 # 分表算法
-spring.shardingsphere.sharding.tables.t_order.table-strategy.inline.algorithm-expression=t_order_$->{order_id % 3}
+spring.shardingsphere.sharding.tables.t_order.table-strategy.inline.algorithm-expression=t_order_$->{user_id % 3}
 # 自增主键字段
 spring.shardingsphere.sharding.tables.t_order.key-generator.column=order_id
 # 自增主键ID 生成方案
-spring.shardingsphere.sharding.tables.t_order.key-generator.type=SNOWFLAKE
+spring.shardingsphere.sharding.tables.t_order.key-generator.type=XXX
 # 工作机器唯一 id
-spring.shardingsphere.sharding.tables.t_order.key-generator.worker.id=6
+spring.shardingsphere.sharding.tables.t_order.key-generator.props.worker.id=0000
+
 
 # 配置分片表 t_order_item
 spring.shardingsphere.sharding.tables.t_order_item.actual-data-nodes=ds-$->{0..1}.t_order_item_$->{0..2}
@@ -61,3 +62,4 @@ mybatis-plus.map-underscore-to-camel-case=true
 # 开启热部署（更改文件后，自动重启）
 spring.devtools.restart.enabled=true
 
+