Parcourir la source

Update 添加 ID 生成工具类,支持多种 ID 生成方式

Yue il y a 1 semaine
Parent
commit
ecfc44dee1

+ 129 - 0
SERVER/VberAdminPlusV3/vber-common/vber-common-mybatis/src/main/java/com/vber/common/mybatis/utils/IdGeneratorUtil.java

@@ -0,0 +1,129 @@
+package com.vber.common.mybatis.utils;
+
+import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
+import com.baomidou.mybatisplus.core.toolkit.IdWorker;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+import com.vber.common.core.utils.SpringUtils;
+
+/**
+ * ID 生成工具类
+ *
+ * @author Iwb
+ */
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+public final class IdGeneratorUtil {
+
+    private static final IdentifierGenerator GENERATOR = SpringUtils.getBean(IdentifierGenerator.class);
+
+    /**
+     * 生成字符串类型主键 ID
+     * <p>
+     * 调用 {@link IdentifierGenerator#nextId(Object)},返回 String 格式 ID。
+     * </p>
+     *
+     * @return 字符串格式主键 ID
+     */
+    public static String nextId() {
+        return GENERATOR.nextId(null).toString();
+    }
+
+    /**
+     * 生成 Long 类型主键 ID
+     * <p>
+     * 自动将生成的数字型主键转换为 Long 类型
+     * </p>
+     *
+     * @return Long 类型主键 ID
+     */
+    public static Long nextLongId() {
+        return GENERATOR.nextId(null).longValue();
+    }
+
+    /**
+     * 生成 Number 类型主键 ID
+     * <p>
+     * 推荐在需要保留原始 Number 类型时使用
+     * </p>
+     *
+     * @return Number 类型主键 ID
+     */
+    public static Number nextNumberId() {
+        return GENERATOR.nextId(null);
+    }
+
+    /**
+     * 根据实体生成数字型主键 ID
+     * <p>
+     * 若自定义的 {@link IdentifierGenerator} 根据实体内容生成 ID,则可以使用本方法
+     * </p>
+     *
+     * @param entity 实体对象
+     * @return Number 类型主键 ID
+     */
+    public static Number nextId(Object entity) {
+        return GENERATOR.nextId(entity);
+    }
+
+    /**
+     * 根据实体生成字符串主键 ID
+     * <p>
+     * 与 {@link #nextId(Object)} 类似,但返回 String 类型
+     * </p>
+     *
+     * @param entity 实体对象
+     * @return 字符串格式主键 ID
+     */
+    public static String nextStringId(Object entity) {
+        return GENERATOR.nextId(entity).toString();
+    }
+
+    /**
+     * 生成 32 位 UUID
+     * <p>
+     * 底层使用 {@link IdWorker#get32UUID()}
+     * </p>
+     *
+     * @return 32 位 UUID 字符串
+     */
+    public static String nextUUID() {
+        return IdWorker.get32UUID();
+    }
+
+    /**
+     * 根据实体生成 32 位 UUID
+     * <p>
+     * 默认 {@link IdentifierGenerator#nextUUID(Object)} 实现忽略实体,但保留该方法便于扩展。
+     * </p>
+     *
+     * @param entity 实体对象
+     * @return 32 位 UUID 字符串
+     */
+    public static String nextUUID(Object entity) {
+        return GENERATOR.nextUUID(entity);
+    }
+
+    /**
+     * 生成带指定前缀的字符串主键 ID
+     * <p>
+     * 示例:prefix = "ORD",生成结果形如:{@code ORD20251211000123}
+     * </p>
+     *
+     * @param prefix 自定义前缀
+     * @return 带前缀的字符串主键 ID
+     */
+    public static String nextIdWithPrefix(String prefix) {
+        return prefix + nextId();
+    }
+
+    /**
+     * 生成带前缀的 UUID
+     *
+     * @param prefix 前缀
+     * @return prefix + UUID
+     */
+    public static String nextUUIDWithPrefix(String prefix) {
+        return prefix + nextUUID();
+    }
+
+}

+ 4 - 5
SERVER/VberAdminPlusV3/vber-modules/vber-generator/src/main/java/com/vber/generator/service/GenTableServiceImpl.java

@@ -8,7 +8,6 @@ import com.baomidou.dynamic.datasource.annotation.DS;
 import com.baomidou.dynamic.datasource.annotation.DSTransactional;
 import com.baomidou.dynamic.datasource.annotation.DSTransactional;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.vber.common.core.constant.Constants;
 import com.vber.common.core.constant.Constants;
@@ -19,6 +18,7 @@ import com.vber.common.core.utils.file.FileUtils;
 import com.vber.common.json.utils.JsonUtils;
 import com.vber.common.json.utils.JsonUtils;
 import com.vber.common.mybatis.core.page.PageQuery;
 import com.vber.common.mybatis.core.page.PageQuery;
 import com.vber.common.mybatis.core.page.TableDataInfo;
 import com.vber.common.mybatis.core.page.TableDataInfo;
+import com.vber.common.mybatis.utils.IdGeneratorUtil;
 import com.vber.generator.config.GenConfig;
 import com.vber.generator.config.GenConfig;
 import com.vber.generator.constant.GenConstants;
 import com.vber.generator.constant.GenConstants;
 import com.vber.generator.domain.GenMenu;
 import com.vber.generator.domain.GenMenu;
@@ -60,11 +60,10 @@ import java.util.zip.ZipOutputStream;
 @Service
 @Service
 public class GenTableServiceImpl implements IGenTableService {
 public class GenTableServiceImpl implements IGenTableService {
 
 
-    private static final String[] TABLE_IGNORE = new String[]{"sj_", "act_", "flw_", "gen_"};
+    private static final String[] TABLE_IGNORE = new String[] { "sj_", "act_", "flw_", "gen_" };
     private final GenTableMapper baseMapper;
     private final GenTableMapper baseMapper;
     private final GenTableColumnMapper genTableColumnMapper;
     private final GenTableColumnMapper genTableColumnMapper;
     private final GenMenuMapper genMenuMapper;
     private final GenMenuMapper genMenuMapper;
-    private final IdentifierGenerator identifierGenerator;
 
 
     /**
     /**
      * 获取代码生成地址
      * 获取代码生成地址
@@ -523,7 +522,7 @@ public class GenTableServiceImpl implements IGenTableService {
     }
     }
 
 
     private GenMenu buildMenuBtn(Long mId, String btnName, String perms, String icon, String btnClass,
     private GenMenu buildMenuBtn(Long mId, String btnName, String perms, String icon, String btnClass,
-                                 String btnScript) {
+            String btnScript) {
         GenMenu menu = new GenMenu();
         GenMenu menu = new GenMenu();
         menu.setParentId(mId);
         menu.setParentId(mId);
         menu.setMenuName(btnName);
         menu.setMenuName(btnName);
@@ -643,7 +642,7 @@ public class GenTableServiceImpl implements IGenTableService {
     private @NotNull List<String> getTemplates(GenTable table) {
     private @NotNull List<String> getTemplates(GenTable table) {
         List<Long> menuIds = new ArrayList<>();
         List<Long> menuIds = new ArrayList<>();
         for (int i = 0; i < 6; i++) {
         for (int i = 0; i < 6; i++) {
-            menuIds.add(identifierGenerator.nextId(null).longValue());
+            menuIds.add(IdGeneratorUtil.nextLongId());
         }
         }
         table.setMenuIds(menuIds);
         table.setMenuIds(menuIds);
         // 设置主键列信息
         // 设置主键列信息