lodsve.mybatis.plugins.repository.BaseRepositoryInterceptor Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2018  Sun.Hao
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */

package lodsve.mybatis.plugins.repository;

import lodsve.mybatis.repository.BaseRepository;
import lodsve.mybatis.repository.bean.*;
import lodsve.mybatis.repository.helper.EntityHelper;
import lodsve.mybatis.repository.helper.MapperHelper;
import lodsve.mybatis.repository.provider.BaseMapperProvider;
import lodsve.mybatis.repository.provider.ExternalProvider;
import org.apache.ibatis.binding.MapperMethod.ParamMap;
import org.apache.ibatis.builder.annotation.ProviderSqlSource;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.springframework.util.Assert;

import java.sql.SQLSyntaxErrorException;
import java.time.LocalDateTime;
import java.util.Properties;
import java.util.ServiceLoader;

/**
 * 通用Mapper拦截器.
 *
 * @author sunhao(sunhao.java @ gmail.com)
 * @version V1.0, 15/7/13 下午4:31
 */
@Intercepts({
        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
        @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
public class BaseRepositoryInterceptor implements Interceptor {
    private final static String LOGIC_DELETE_WITH_MODIFIED_BY_MAPPED_STATEMENT_ID = "logicDeleteByIdWithModifiedBy";

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] objects = invocation.getArgs();
        MappedStatement ms = (MappedStatement) objects[0];
        // 参数
        ParamMap parameter = ((objects[1] instanceof ParamMap) ? (ParamMap) objects[1] : null);
        String msId = ms.getId();
        //不需要拦截的方法直接返回
        if (MapperHelper.isMapperMethod(msId)) {
            // 第一次经过处理后，就不会是ProviderSqlSource了，一开始高并发时可能会执行多次，但不影响。以后就不会在执行了
            if (ms.getSqlSource() instanceof ProviderSqlSource) {
                MapperHelper.resetSqlSource(ms, parameter);
            }
        }

        // 要排除logicDeleteWithModifiedBy这个方法，因为这个方法每次都会修改参数，所以得每次都从这边走
        if (msId.contains(LOGIC_DELETE_WITH_MODIFIED_BY_MAPPED_STATEMENT_ID)) {
            handleParams(ms, parameter);
        }

        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        MapperHelper.registerMapper(BaseRepository.class);

        ServiceLoader serviceLoader = ServiceLoader.load(ExternalProvider.class);
        serviceLoader.iterator().forEachRemaining(obj -> obj.provider().stream().filter(Class::isInterface).forEach(MapperHelper::registerMapper));

        if (target instanceof Executor) {
            return Plugin.wrap(target, this);
        } else {
            return target;
        }
    }

    @Override
    public void setProperties(Properties properties) {

    }

    private void handleParams(MappedStatement ms, ParamMap parameter) throws SQLSyntaxErrorException {
        Assert.notNull(parameter, "parameter must be non-null!");

        BaseMapperProvider mapperProvider = MapperHelper.getMapperProvider(ms.getId());

        Class entityClass = mapperProvider.getSelectReturnType(ms);
        EntityTable table = EntityHelper.getEntityTable(entityClass);
        DeleteColumn deleteColumn = table.getDeleteColumn();
        if (null == deleteColumn) {
            throw new SQLSyntaxErrorException("不支持逻辑删除！没有@LogicDelete注解");
        }

        IdColumn idColumn = table.getIdColumn();
        LastModifiedByColumn modifiedByColumn = table.getModifiedByColumn();
        LastModifiedDateColumn modifiedDateColumn = table.getModifiedDateColumn();
        DisabledDateColumn disabledDateColumn = table.getDisabledDateColumn();

        handleParamMap(parameter, idColumn, modifiedByColumn, modifiedDateColumn, disabledDateColumn);
    }


    @SuppressWarnings("unchecked")
    private void handleParamMap(ParamMap parameter, IdColumn idColumn, LastModifiedByColumn modifiedByColumn, LastModifiedDateColumn modifiedDateColumn, DisabledDateColumn disabledDateColumn) {
        // 修改参数
        ParamMap