mybatis插件-自動更新updateTime

@Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})})
public class AutoSetUpdateTime implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0];
        Object parameterObject = args[1];

        BoundSql boundSql = ms.getBoundSql(parameterObject);

        String sql = boundSql.getSql();
        MySqlStatementParser mySqlStatementParser = new MySqlStatementParser(sql);
        SQLStatement statement = mySqlStatementParser.parseStatement();

        MySqlUpdateStatement mySqlUpdateStatement = (MySqlUpdateStatement)statement;
        List<SQLUpdateSetItem> list =  mySqlUpdateStatement.getItems();

        boolean containModifyTime = false;
        for (SQLUpdateSetItem setItem : list){
            String colnumName = setItem.getColumn().toString();
            if("gmt_modified".equals(colnumName)){
                containModifyTime = true;
            }
        }

        if(!containModifyTime) {
            if (null != parameterObject) {
                String[] sqls = sql.split(";");
                StringBuilder newSql = new StringBuilder();
                for (int i=0;i<sqls.length;i++){
                    if(i>0){
                        newSql.append(";");
                    }
                    int index = sqls[i].indexOf(" where ");
                    String subSql = sqls[i].substring(0,index);
                    newSql.append(subSql).append(",gmt_modified=").append(System.currentTimeMillis());
                    String whereSql = sqls[i].substring(index);
                    newSql.append(whereSql);
                }

                BoundSql bs = new BoundSql(ms.getConfiguration(),newSql.toString(),boundSql.getParameterMappings(),parameterObject);

                MappedStatement newMs = copyFromMappedStatement(ms, new LimitInterceptorBysqlParser.BoundSqlSqlSource(bs));
                for (ParameterMapping mapping : boundSql.getParameterMappings()) {
                    String prop = mapping.getProperty();
                    if (boundSql.hasAdditionalParameter(prop)) {
                        bs.setAdditionalParameter(prop, boundSql.getAdditionalParameter(prop));
                    }
                }
                args[0] = newMs;
            }
        }
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target,this);
    }

    @Override
    public void setProperties(Properties properties) {

    }

    private MappedStatement copyFromMappedStatement(MappedStatement ms, SqlSource newSqlSource) {
        MappedStatement.Builder builder = new MappedStatement.Builder(ms.getConfiguration(), ms.getId(), newSqlSource, ms.getSqlCommandType());
        builder.resource(ms.getResource());
        builder.fetchSize(ms.getFetchSize());
        builder.statementType(ms.getStatementType());
        builder.keyGenerator(ms.getKeyGenerator());
        if (ms.getKeyProperties() != null && ms.getKeyProperties().length > 0) {
            builder.keyProperty(ms.getKeyProperties()[0]);
        }
        builder.timeout(ms.getTimeout());
        builder.parameterMap(ms.getParameterMap());
        builder.resultMaps(ms.getResultMaps());
        builder.resultSetType(ms.getResultSetType());
        builder.cache(ms.getCache());
        builder.flushCacheRequired(ms.isFlushCacheRequired());
        builder.useCache(ms.isUseCache());
        return builder.build();
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章