MongoPlus逻辑删除时remove报错

MongoPlus逻辑删除报错

报错信息：

Caused by: java.lang.NoSuchMethodException: com.anwen.mongo.execute.instance.SessionExecute.executeUpdate(org.bson.conversions.Bson,org.bson.conversions.Bson,com.mongodb.client.MongoCollection)

解决方案

自定义逻辑删除替换器

1. 创建替换器类CustomLogicRemoveReplacer

import com.anwen.mongo.cache.global.CollectionLogicDeleteCache;
import com.anwen.mongo.enums.ExecuteMethodEnum;
import com.anwen.mongo.enums.SpecialConditionEnum;
import com.anwen.mongo.execute.Execute;
import com.anwen.mongo.logic.LogicDeleteHandler;
import com.anwen.mongo.model.LogicDeleteResult;
import com.anwen.mongo.model.MutablePair;
import com.anwen.mongo.replacer.Replacer;
import com.anwen.mongo.support.BoolFunction;
import com.mongodb.BasicDBObject;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.UpdateResult;
import org.bson.Document;
import org.bson.conversions.Bson;

import java.lang.reflect.Method;
import java.util.Collections;
import java.util.Objects;

/**
 * 自定义逻辑删除替换器
 * @author anwen
 */
public class CustomLogicRemoveReplacer implements Replacer {

    @Override
    @SuppressWarnings("unchecked")
    public Object invoke(Object proxy, Object target, Method method, Object[] args) throws Throwable {

        // 是否忽略
        if (CollectionLogicDeleteCache.getLogicIgnore()) {
            return method.invoke(target, args);
        }
        // 转换为MongoCollection类
        MongoCollection<Document> collection = (MongoCollection<Document>) args[1];
        // 根据MongoCollection获取实体类
        Class<?> clazz = LogicDeleteHandler.getBeanClass(collection);
        if (Objects.isNull(clazz)) {
            return method.invoke(target, args);
        }
        // 获取实体对象和逻辑删除字段的映射关系
        LogicDeleteResult result = LogicDeleteHandler.mapper().get(clazz);
        if (Objects.isNull(result)) {
            return method.invoke(target, args);
        }
        // 构建update
        Execute execute = (Execute) target;
        Document updateBasic = new Document(result.getColumn(), result.getLogicDeleteValue());
        BasicDBObject update = new BasicDBObject(SpecialConditionEnum.SET.getCondition(), updateBasic);
        UpdateResult updateResult = execute.executeUpdate(
                Collections.singletonList(new MutablePair<>((Bson) args[0], update)), collection
        );
        return new DeleteResult() {
            @Override
            public boolean wasAcknowledged() {
                return updateResult.wasAcknowledged();
            }

            @Override
            public long getDeletedCount() {
                return updateResult.getModifiedCount();
            }
        };
    }

    @Override
    public BoolFunction supplier() {
        return (proxy, target, method, args) -> CollectionLogicDeleteCache.open && method.getName().equals(ExecuteMethodEnum.REMOVE.getMethod());
    }
}

2. 创建逻辑删除配置类LogicConfiguration

import com.anwen.mongo.cache.global.ExecutorReplacerCache;
import com.anwen.mongo.config.MongoPlusAutoConfiguration;
import com.anwen.mongo.logic.replacer.LogicRemoveReplacer;
import com.anwen.mongo.model.LogicProperty;
import org.springframework.context.annotation.Configuration;


/**
 * 逻辑删除配置
 * @author anwen
 */
@Configuration
public class LogicConfiguration {

    private final LogicProperty logicProperty;

    public LogicConfiguration(LogicProperty logicProperty, MongoPlusAutoConfiguration ignoredMongoPlusAutoConfiguration) {
        this.logicProperty = logicProperty;
        setLogic();
    }

    private void setLogic(){
        // 是否开启逻辑删除
        if (logicProperty.getOpen()) {
            // 添加执行替换器缓存组件为CustomLogicRemoveReplacer类
            ExecutorReplacerCache.replacers.removeIf(next -> next instanceof LogicRemoveReplacer);
            ExecutorReplacerCache.replacers.add(new CustomLogicRemoveReplacer());
            // 替换器排序
            ExecutorReplacerCache.sorted();
        }
    }

}

至此 问题解决