前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >mybatis-plus实现对创建时间和更新时间的自动填充

mybatis-plus实现对创建时间和更新时间的自动填充

作者头像
一缕82年的清风
发布2021-12-06 10:41:53
2K0
发布2021-12-06 10:41:53
举报
文章被收录于专栏:lsqingfeng

我们在项目的开发当中,基本上没张表里都有创建时间和更新时间,而且我们每次在新增或修改数据的时候,也都要把这两个时间更新成当前时间,当然我们也可以在数据库层面设置更新时更新,否则就只能在代码中出现很多重复的如下代码:

xxx.setCreateTime(new Date());

xxx.setUpdateTime(new Date());

而mybatis-plus给我们提供一种方式,可以自动帮我们更新这两个字段,在写业务逻辑的时候就不用去关注类似上面这种重复的代码,一劳永逸,但是要注意的是,必须字段名称一致,就是每张表的创建时间都叫create_time ,更新时间叫update_time:好了,话不多说。给出代码:

1. 添加一个配置类:

代码语言:javascript
复制
import com.baomidou.mybatisplus.core.config.GlobalConfig;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


@Configuration
public class MybatisPlusConf {
    /**
     * 自动填充功能
     */
    @Bean
    public GlobalConfig globalConfig() {
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setMetaObjectHandler(new MetaHandler());
        return globalConfig;
    }

}

2. 添加一个Handler

代码语言:javascript
复制
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.util.Date;


@Component
public class MetaHandler implements MetaObjectHandler {

    /**
     * 新增数据执行
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        boolean hasSetter = metaObject.hasSetter("createTime");
        if (hasSetter) {
            this.setFieldValByName("createTime", new Date(), metaObject);
            this.setFieldValByName("updateTime", new Date(), metaObject);
        }
    }

    /**
     * 更新数据执行
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        Object val = getFieldValByName("updateTime", metaObject);
        if (val == null) {
            this.setFieldValByName("updateTime", new Date(), metaObject);
        }
    }

}

这里要注意:如果你的实体中日期是Date() 类型,上面 就用new Date(), 如果是LocalDateTime类型,就把new Date() 替换为 LocalDateTIme.now();

当然我们也可以使用上篇文章中提到的Mybatis拦截器,拦截instert 和 update方法: 配置如下;

代码语言:javascript
复制
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.sql.Timestamp;
import java.time.LocalDateTime;

import org.apache.ibatis.binding.MapperMethod.ParamMap;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Signature;
import org.springframework.stereotype.Component;


@Component
@Intercepts({ @Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }) })
public class CreateTimeSetInterceptor implements Interceptor {

	private static final String CREATE_TIME_SETTER = "setCreateTime";
	private static final String UPDATE_TIME_SETTER = "setUpdateTime";

	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
		Object parameter = invocation.getArgs()[1];

		if (ms.getSqlCommandType() == SqlCommandType.INSERT) {
			setTimeIfNeccesary(parameter, CREATE_TIME_SETTER);
			setTimeIfNeccesary(parameter, UPDATE_TIME_SETTER);
		} else if (ms.getSqlCommandType() == SqlCommandType.UPDATE) {
			setTimeIfNeccesary(parameter, UPDATE_TIME_SETTER);
		}
		return invocation.getMethod().invoke(invocation.getTarget(), invocation.getArgs());
	}

	private void setTimeIfNeccesary(Object param, String methodName) {
		Class<?> cls = param.getClass();

		if (cls == ParamMap.class) {
			@SuppressWarnings("unchecked")
			ParamMap<Object> map = (ParamMap<Object>) param;
			map.entrySet().forEach(entry -> {
				if (!entry.getKey().equals("et")) {
					setIfSetterExist(entry.getValue(), methodName);
				}
			});
		} else {
			setIfSetterExist(param, methodName);
		}
	}

	private void setIfSetterExist(Object param, String methodName) {
		Class<?> cls = param.getClass();
		try {
			Method m = null;
			try {
				m = cls.getDeclaredMethod(methodName, new Class[] { LocalDateTime.class });
				if (m != null) {
					m.setAccessible(true);
					m.invoke(param, LocalDateTime.now());
				}
			} catch (NoSuchMethodException e1) {
				m = cls.getDeclaredMethod(methodName, new Class[] { Timestamp.class });
				if (m != null) {
					m.setAccessible(true);
					m.invoke(param, new Timestamp(System.currentTimeMillis()));
				}
			}
		} catch (NoSuchMethodException | SecurityException | IllegalAccessException | IllegalArgumentException
				| InvocationTargetException e) {
			e.printStackTrace();
		}
	}

}

好了关于今天的内容就介绍到这里了。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/01/27 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档