MyBatis主键拦截器，简单代码实现

最新推荐文章于 2024-08-05 00:09:55 发布

blazingfire

最新推荐文章于 2024-08-05 00:09:55 发布

阅读量511

点赞数

分类专栏： java笔记

本文链接：https://blog.csdn.net/blazingfire/article/details/103797257

版权

本文介绍了如何在MyBatis中自定义主键拦截器，通过简单代码实现自动赋值主键，提高数据库操作效率。

摘要由CSDN通过智能技术生成

主要代码如下：

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;

@Component
@Intercepts(@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class}))
public class PrimaryKeyInterceptor implements Interceptor {

    static Logger log = LoggerFactory.getLogger(PrimaryKeyInterceptor.class);

    private static ConcurrentHashMap<String, AtomicLong> seeds = new ConcurrentHashMap<>();

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        Connection connection = (Connection) args[0];
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        BoundSql boundSql = statementHandler.getBoundSql();

        //MetaObject是Mybatis提供的一个用于方便、优雅访问访问对象的反射工具
        //同时它支持对JavaBean、Collection、Map三种类型对象的操作。
        MetaObject metaObject = Syste