逍遥阁

mybatis插件开发

关键词: mybatis

mybatis插件开发

```
package dry.test.mybatisplug;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
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.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.springframework.stereotype.Component;

import java.util.List;

@Component
@Slf4j
@Intercepts(
        {
                @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
                @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}),
        }
)
public class QueryInterceptor implements Interceptor {

@Override
    public Object intercept(Invocation invocation) throws Throwable {
        Executor executor = (Executor) invocation.getTarget();
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0];
        Object parameter = args[1];
        RowBounds rowBounds = (RowBounds) args[2];
        ResultHandler resultHandler = (ResultHandler) args[3];
        BoundSql boundSql;
        CacheKey cacheKey;
        if (args.length == 4) {
            boundSql = ms.getBoundSql(parameter);
            cacheKey = executor.createCacheKey(ms, parameter, rowBounds, boundSql);
        } else {
            cacheKey = (CacheKey) args[4];
            boundSql = (BoundSql) args[5];
        }
        log.info("[self]sql={}", ms.getSqlSource().getBoundSql(parameter).getSql());
        List<Object> oldList = executor.query(ms, parameter, rowBounds, resultHandler, cacheKey, boundSql);
        if (oldList.isEmpty()) {
            return oldList;
        }
        Class<?> resultType = ms.getResultMaps().get(0).getType();
        String oldJsonString = JSONObject.toJSONString(oldList);
        JSONArray oldArray = JSONObject.parseArray(oldJsonString);
        JSONArray newArray = new JSONArray();
        for (int i = 0; i < oldArray.size(); i++) {
            JSONObject item = oldArray.getJSONObject(i);
            item.put("statusText", "This is test");
            newArray.add(item);
        }
        String newJsonString = JSONObject.toJSONString(newArray);
        List<?> newList = JSONObject.parseArray(newJsonString, resultType);
        log.info("[self]newJsonString={}", newJsonString);
        return newList;
    }

}
```

admin
2021-08-14 17:03:12
mysql

文章分类

文章标签

文章归档

友情链接

mybatis插件开发