mybatis plus中怎么自定義主鍵生成器IKeyGenerator

本篇文章為大家展示了mybatis plus中怎么自定義主鍵生成器IKeyGenerator，內容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。

1. 繼承IKeyGenerator

@Slf4j
@Component
public class MybatisKeyGenerator implements IKeyGenerator {
 
	@Value("${server.worker-id}")
	private Integer workerId;
 
	@Value("${server.data-center-id}")
	private Integer dataCenterId;
 
	@Override
	public String executeSql(String incrementerName) {
		log.info("mybatis plus keyGenerator: " + incrementerName + "(" + workerId + "," + dataCenterId + ")");
		long uid = new SnowflakeIdWorker(workerId, dataCenterId).nextId();
		return "select " + uid + " from dual";
	}
}

2. 定義entity, 需要有@KeySequence注解，和IdType.INPUT或者IdType.ID_WORKER

@Data
@TableName("test1")
@KeySequence("mybatisKeyGenerator")
public class Test {
	@TableId(type = IdType.INPUT)
	private Long id;
	private String name;
}

 3.測試:

@Autowired
private TestMapper testMapper;
@Override
public void run(String... args) throws Exception {
	Test test = new Test();
	test.setName("CCC"+ LocalDateTime.now());
	testMapper.insert(test);
	System.out.println(">>>> test:"+ test);
}

源碼查看：

1. com.baomidou.mybatisplus.autoconfigure.MybatisPlusAutoConfiguration

// 從spring容器中獲取IKeyGenerator
//注入主鍵生成器
if (this.applicationContext.getBeanNamesForType(IKeyGenerator.class, false,
	false).length > 0) {
	IKeyGenerator keyGenerator = this.applicationContext.getBean(IKeyGenerator.class);
	globalConfig.getDbConfig().setKeyGenerator(keyGenerator);
}

2. 在掃描實體時候，會掃描KeySequence注解

/* 開啟了自定義 KEY 生成器 */
if (null != dbConfig.getKeyGenerator()) {
   tableInfo.setKeySequence(clazz.getAnnotation(KeySequence.class));
}

3. 插入數據時：com.baomidou.mybatisplus.core.injector.methods.Insert

// 表包含主鍵處理邏輯,如果不包含主鍵當普通字段處理
if (StringUtils.isNotEmpty(tableInfo.getKeyProperty())) {
	if (tableInfo.getIdType() == IdType.AUTO) {
		/** 自增主鍵 */
		keyGenerator = new Jdbc3KeyGenerator();
		keyProperty = tableInfo.getKeyProperty();
		keyColumn = tableInfo.getKeyColumn();
	} else {
 
        // 如果有注解，則獲取自定義的主鍵生成器
		if (null != tableInfo.getKeySequence()) {
            // 這里獲取的就是注入的生成器，執行生成器接口executeSql，傳入值為注解里面的value
			keyGenerator = TableInfoHelper.genKeyGenerator(tableInfo, builderAssistant, sqlMethod.getMethod(), languageDriver);
			keyProperty = tableInfo.getKeyProperty();
			keyColumn = tableInfo.getKeyColumn();
		}
	}
}

4. 

    /**
     * <p>
     * 自定義 KEY 生成器
     * </p>
     */
    public static KeyGenerator genKeyGenerator(TableInfo tableInfo, MapperBuilderAssistant builderAssistant,
                                               String baseStatementId, LanguageDriver languageDriver) {
        IKeyGenerator keyGenerator = GlobalConfigUtils.getKeyGenerator(builderAssistant.getConfiguration());
        if (null == keyGenerator) {
            throw new IllegalArgumentException("not configure IKeyGenerator implementation class.");
        }
        String id = baseStatementId + SelectKeyGenerator.SELECT_KEY_SUFFIX;
        Class<?> resultTypeClass = tableInfo.getKeySequence().clazz();
        StatementType statementType = StatementType.PREPARED;
        String keyProperty = tableInfo.getKeyProperty();
        String keyColumn = tableInfo.getKeyColumn();
 
        // 執行語句keyGenerator.executeSql(tableInfo.getKeySequence().value())
        SqlSource sqlSource = languageDriver.createSqlSource(builderAssistant.getConfiguration(),
            keyGenerator.executeSql(tableInfo.getKeySequence().value()), null);
        builderAssistant.addMappedStatement(id, sqlSource, statementType, SqlCommandType.SELECT, null, null, null,
            null, null, resultTypeClass, null, false, false, false,
            new NoKeyGenerator(), keyProperty, keyColumn, null, languageDriver, null);
        id = builderAssistant.applyCurrentNamespace(id, false);
        MappedStatement keyStatement = builderAssistant.getConfiguration().getMappedStatement(id, false);
        SelectKeyGenerator selectKeyGenerator = new SelectKeyGenerator(keyStatement, true);
        builderAssistant.getConfiguration().addKeyGenerator(id, selectKeyGenerator);
        return selectKeyGenerator;
    }

上述內容就是mybatis plus中怎么自定義主鍵生成器IKeyGenerator，你們學到知識或技能了嗎？如果還想學到更多技能或者豐富自己的知識儲備，歡迎關注億速云行業資訊頻道。