@intercepts拦截更新语句实现create_time,update_time自动注入

开箱即用

@Intercepts({
        @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
@Component
public class SaveAndUpdateInteceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        Object parameter = args[1];
        if (parameter instanceof Map) {
            ((Map<?, ?>) parameter).values().forEach(PojoUtil::setCommonField);
        } else {
            PojoUtil.setCommonField(parameter);
        }
        return invocation.proceed();
    }
}

public class PojoUtil {
    static Logger logger = LoggerFactory.getLogger(PojoUtil.class);

    public static <T> void setCommonField(T t, String userId) {
        Object id = SpringReflectionUtils.getField("id", t);

        if (Objects.isNull(id) || StringUtils.isEmpty(id.toString())) {
            SpringReflectionUtils.setField("createUserId", t, userId);
            SpringReflectionUtils.setField("createTime", t, new Date());
            try {
                SpringReflectionUtils.setField("rev", t, 1);
            } catch (Exception e) {
                //ignore
            }
        }
        SpringReflectionUtils.setField("updateTime", t, new Date());
        SpringReflectionUtils.setField("updateUserId", t, userId);
        SpringReflectionUtils.setField("deleteFlag", t, "0");
    }

    public static <T> void setCommonField(T t) {
        setCommonField(t, "1");
    }
}

public class SpringReflectionUtils {
    private static final Logger logger = LoggerFactory.getLogger(SpringReflectionUtils.class);

    /**
     * Set the field represented by the supplied {@linkplain Field field object} on
     * the specified {@linkplain Object target object} to the specified {@code value}.
     * <p>In accordance with {@link Field#set(Object, Object)} semantics, the new value
     * is automatically unwrapped if the underlying field has a primitive type.
     * <p>This method does not support setting {@code static final} fields.
     * <p>Thrown exceptions are handled via a call to {@link #handleReflectionException(Exception)}.
     *
     * @param fieldName the field to set
     * @param target    the target object on which to set the field
     *                  (or {@code null} for a static field)
     * @param value     the value to set (may be {@code null})
     */![在这里插入图片描述](https://img-blog.csdnimg.cn/350856429ff04344b5744b463bec27b2.jpeg#pic_center)

    public static void setField(String fieldName, Object target, Object value) {
        Field field = ReflectionUtils.findField(target.getClass(), fieldName);
        Assert.notNull(target,"对象不能为空");
        if (field == null) {
            logger.warn("未找到类[{}]的[{}]字段", target.getClass(), fieldName);
            return;
        }
        ReflectionUtils.makeAccessible(field);
        ReflectionUtils.setField(field, target, value);
    }


    /**
     * Get the field represented by the supplied {@link Field field object} on the
     * specified {@link Object target object}. In accordance with {@link Field#get(Object)}
     * semantics, the returned value is automatically wrapped if the underlying field
     * has a primitive type.
     * <p>Thrown exceptions are handled via a call to {@link #handleReflectionException(Exception)}.
     * @param field the field to get
     * @param target the target object from which to get the field
     * (or {@code null} for a static field)
     * @return the field's current value
     */
    @Nullable
    public static Object getField(String fieldName, Object target) {
        Field field = ReflectionUtils.findField(target.getClass(), fieldName);
        if (field == null){return null;}
        ReflectionUtils.makeAccessible(field);
        return ReflectionUtils.getField(field, target);
    }
}