com.anwen.mongo.mapper.MongoPlusMapMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongo-plus-core Show documentation
mongo-plus核心代码
There is a newer version: 2.1.6.1
package com.anwen.mongo.mapper;

import com.anwen.mongo.conditions.aggregate.AggregateChainWrapper;
import com.anwen.mongo.conditions.inject.aggregate.LambdaAggregateChainInjectWrapper;
import com.anwen.mongo.conditions.inject.query.LambdaQueryChainInjectWrapper;
import com.anwen.mongo.conditions.inject.update.LambdaUpdateChainInjectWrapper;
import com.anwen.mongo.conditions.interfaces.Inject.InjectQuery;
import com.anwen.mongo.conditions.interfaces.condition.CompareCondition;
import com.anwen.mongo.conditions.query.QueryChainWrapper;
import com.anwen.mongo.conditions.update.UpdateChainWrapper;
import com.anwen.mongo.execute.ExecutorFactory;
import com.anwen.mongo.manager.MongoPlusClient;
import com.anwen.mongo.model.PageParam;
import com.anwen.mongo.model.PageResult;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.CreateIndexOptions;
import com.mongodb.client.model.DropIndexOptions;
import com.mongodb.client.model.IndexModel;
import com.mongodb.client.model.IndexOptions;
import org.bson.Document;
import org.bson.conversions.Bson;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;

import static com.anwen.mongo.toolkit.StringPool.EMPTY;

/**
 * @author JiaChaoYang
 **/

public class MongoPlusMapMapper implements InjectQuery {
    private final ExecutorFactory factory;

    private final MongoPlusClient mongoPlusClient;

    public ExecutorFactory getFactory() {
        return factory;
    }

    public MongoPlusClient getMongoPlusClient() {
        return mongoPlusClient;
    }

    public MongoPlusMapMapper(MongoPlusClient mongoPlusClient) {
        this.mongoPlusClient = mongoPlusClient;
        factory = new ExecutorFactory(mongoPlusClient);
    }

    /**
     * 获取当前操作对象的连接，以便使用MongoDriver的语法
     * @author JiaChaoYang
     * @date 2023/11/15 13:43
     */
    public MongoCollection getMongoCollection(String database,String collectionName){
        return mongoPlusClient.getCollection(database,collectionName);
    }

    /**
     * 获取当前操作对象的连接，以便使用MongoDriver的语法
     * @author JiaChaoYang
     * @date 2023/11/15 13:43
     */
    public MongoCollection getMongoCollection(String collectionName){
        return mongoPlusClient.getCollection(EMPTY,collectionName);
    }

    public LambdaQueryChainInjectWrapper lambdaQuery(){
        return new LambdaQueryChainInjectWrapper(factory);
    }

    public LambdaAggregateChainInjectWrapper lambdaAggregate(){
        return new LambdaAggregateChainInjectWrapper(factory);
    }

    public LambdaUpdateChainInjectWrapper lambdaUpdate(){
        return new LambdaUpdateChainInjectWrapper(factory);
    }

    @Override
    public List> list(String collectionName) {
        return list(EMPTY,collectionName);
    }

    @Override
    public List> list(String database, String collectionName) {
        return factory.getInjectExecute(database).list(collectionName);
    }

    @Override
    public List> list(String collectionName, QueryChainWrapper,?> queryChainWrapper) {
        return list(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    public List> list(String database, String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).list(collectionName,queryChainWrapper.getCompareList(),queryChainWrapper.getOrderList(),queryChainWrapper.getProjectionList(),queryChainWrapper.getBasicDBObjectList());
    }

    @Override
    public List> aggregateList(String collectionName, AggregateChainWrapper, ?> queryChainWrapper) {
        return aggregateList(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    public List> aggregateList(String database, String collectionName, AggregateChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).aggregateList(collectionName, queryChainWrapper.getBaseAggregateList(), queryChainWrapper.getBasicDBObjectList(),queryChainWrapper.getOptionsBasicDBObject());
    }

    @Override
    public Map one(String collectionName, QueryChainWrapper,?> queryChainWrapper) {
        return one(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    public Map one(String database, String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).one(collectionName,queryChainWrapper.getCompareList(),queryChainWrapper.getProjectionList(),queryChainWrapper.getBasicDBObjectList());
    }

    @Override
    @Deprecated
    public Map limitOne(String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return limitOne(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    @Deprecated
    public Map limitOne(String database, String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).limitOne(collectionName,queryChainWrapper.getCompareList(),queryChainWrapper.getProjectionList(),queryChainWrapper.getBasicDBObjectList());
    }

    @Override
    public PageResult> page(String collectionName, PageParam pageParam) {
        return page(EMPTY,collectionName,pageParam);
    }

    @Override
    public PageResult> page(String database, String collectionName, PageParam pageParam) {
        return factory.getInjectExecute(database).page(collectionName,null,null,null,null,pageParam.getPageNum(),pageParam.getPageSize());
    }

    @Override
    public PageResult> page(String collectionName, PageParam pageParam, QueryChainWrapper,?> queryChainWrapper) {
        return page(EMPTY,collectionName,pageParam,queryChainWrapper);
    }

    @Override
    public PageResult> page(String database, String collectionName, PageParam pageParam, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).page(collectionName,queryChainWrapper.getCompareList(),queryChainWrapper.getOrderList(),queryChainWrapper.getProjectionList(),queryChainWrapper.getBasicDBObjectList(),pageParam.getPageNum(),pageParam.getPageSize());
    }

    @Override
    public PageResult> page(String collectionName, Integer pageNum, Integer pageSize) {
        return page(EMPTY,collectionName,pageNum,pageSize);
    }

    @Override
    public PageResult> page(String database, String collectionName, Integer pageNum, Integer pageSize) {
        return page(collectionName,new PageParam(pageNum,pageSize));
    }

    @Override
    public PageResult> page(String collectionName, Integer pageNum, Integer pageSize, QueryChainWrapper, ?> queryChainWrapper) {
        return page(EMPTY,collectionName,pageNum,pageSize,queryChainWrapper);
    }

    @Override
    public PageResult> page(String database, String collectionName, Integer pageNum, Integer pageSize, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).page(collectionName,queryChainWrapper.getCompareList(),queryChainWrapper.getOrderList(),queryChainWrapper.getProjectionList(),queryChainWrapper.getBasicDBObjectList(),pageNum,pageSize);
    }

    @Override
    public Map getById(String collectionName , Serializable id) {
        return getById(EMPTY,collectionName,id);
    }

    @Override
    public Map getById(String database, String collectionName, Serializable id) {
        return factory.getInjectExecute(database).getById(collectionName,id);
    }

    @Override
    public List> getByIds(String collectionName , Collection ids) {
        return getByIds(EMPTY,collectionName,ids);
    }

    @Override
    public List> getByIds(String database, String collectionName, Collection ids) {
        return factory.getInjectExecute(database).getByIds(collectionName,ids);
    }

    @Override
    public Boolean save(String collectionName, Map entityMap) {
        return save(EMPTY,collectionName,entityMap);
    }

    @Override
    public Boolean save(String database, String collectionName, Map entityMap) {
        return factory.getInjectExecute(database).save(collectionName,entityMap);
    }

    @Override
    public Boolean saveBatch(String collectionName, Collection> entityMapList) {
        return saveBatch(EMPTY,collectionName,entityMapList);
    }

    @Override
    public Boolean saveBatch(String database, String collectionName, Collection> entityMapList) {
        return factory.getInjectExecute(database).saveBatch(collectionName,entityMapList);
    }

    @Override
    public Boolean saveOrUpdate(String collectionName, Map entityMap) {
        return saveOrUpdate(EMPTY,collectionName,entityMap);
    }

    @Override
    public Boolean saveOrUpdate(String database, String collectionName, Map entityMap) {
        return factory.getInjectExecute(database).saveOrUpdate(collectionName,entityMap);
    }

    @Override
    public Boolean saveOrUpdateWrapper(String collectionName, Map entityMap, QueryChainWrapper, ?> queryChainWrapper) {
        return saveOrUpdateWrapper(EMPTY,collectionName,entityMap,queryChainWrapper);
    }

    @Override
    public Boolean saveOrUpdateWrapper(String database, String collectionName, Map entityMap, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).saveOrUpdateWrapper(collectionName,entityMap,queryChainWrapper.getCompareList());
    }

    @Override
    public Boolean saveOrUpdateBatch(String collectionName, Collection> entityMapList) {
        return saveOrUpdateBatch(EMPTY,collectionName,entityMapList);
    }

    @Override
    public Boolean saveOrUpdateBatch(String database, String collectionName, Collection> entityMapList) {
        return factory.getInjectExecute(database).saveOrUpdateBatch(collectionName,entityMapList);
    }

    @Override
    public Boolean updateById(String collectionName, Map entityMap) {
        return updateById(EMPTY,collectionName,entityMap);
    }

    @Override
    public Boolean updateById(String database, String collectionName, Map entityMap) {
        return factory.getInjectExecute(database).updateById(collectionName,entityMap);
    }

    @Override
    public Boolean updateBatchByIds(String collectionName, Collection> entityMapList) {
        return updateBatchByIds(EMPTY,collectionName,entityMapList);
    }

    @Override
    public Boolean updateBatchByIds(String database, String collectionName, Collection> entityMapList) {
        return factory.getInjectExecute(database).updateBatchByIds(collectionName,entityMapList);
    }

    @Override
    public Boolean updateByColumn(String collectionName, Map entityMap, String column) {
        return updateByColumn(EMPTY,collectionName,entityMap,column);
    }

    @Override
    public Boolean updateByColumn(String database, String collectionName, Map entityMap, String column) {
        return factory.getInjectExecute(database).updateByColumn(collectionName,entityMap,column);
    }

    @Override
    public Boolean removeById(String collectionName, Serializable id) {
        return removeById(EMPTY,collectionName,id);
    }

    @Override
    public Boolean removeById(String database, String collectionName, Serializable id) {
        return factory.getInjectExecute(database).removeById(collectionName,id);
    }

    @Override
    public Boolean removeByColumn(String collectionName, String column, String value) {
        return removeByColumn(EMPTY,collectionName,column,value);
    }

    @Override
    public Boolean removeByColumn(String database, String collectionName, String column, String value) {
        return factory.getInjectExecute(database).removeByColumn(collectionName,column,value);
    }

    @Override
    public Boolean removeBatchByIds(String collectionName, Collection idList) {
        return removeBatchByIds(EMPTY,collectionName,idList);
    }

    @Override
    public Boolean removeBatchByIds(String database, String collectionName, Collection idList) {
        return factory.getInjectExecute(database).removeBatchByIds(collectionName,idList);
    }

    @Override
    public long count(String collectionName, QueryChainWrapper,?> queryChainWrapper) {
        return count(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    public long count(String database, String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).count(collectionName,queryChainWrapper.getCompareList());
    }

    @Override
    public List> getByColumn(String collectionName,String field, Object fieldValue) {
        return getByColumn(EMPTY,collectionName,field,fieldValue);
    }

    @Override
    public List> getByColumn(String database, String collectionName, String field, Object fieldValue) {
        return factory.getInjectExecute(database).getByColumn(collectionName,field,fieldValue);
    }

    @Override
    public Boolean remove(String collectionName, UpdateChainWrapper, ?> updateChainWrapper) {
        return remove(EMPTY,collectionName,updateChainWrapper);
    }

    @Override
    public Boolean remove(String database, String collectionName, UpdateChainWrapper, ?> updateChainWrapper) {
        return factory.getInjectExecute(database).remove(collectionName,updateChainWrapper.getCompareList());
    }

    @Override
    public Boolean update(String collectionName, UpdateChainWrapper, ?> updateChainWrapper) {
        return update(EMPTY,collectionName,updateChainWrapper);
    }

    @Override
    public Boolean update(String database, String collectionName, UpdateChainWrapper, ?> updateChainWrapper) {
        List compareConditionList = new ArrayList<>();
        compareConditionList.addAll(updateChainWrapper.getCompareList());
        compareConditionList.addAll(updateChainWrapper.getUpdateCompareList());
        return factory.getInjectExecute(database).update(collectionName,compareConditionList);
    }

    @Override
    public List> queryCommand(String collectionName,String command) {
        return queryCommand(EMPTY,collectionName,command);
    }

    @Override
    public List> queryCommand(String database, String collectionName, String command) {
        return factory.getInjectExecute(database).queryCommand(collectionName,command);
    }

    @Override
    public Boolean exist(String collectionName, Serializable id) {
        return exist(EMPTY,collectionName,id);
    }

    @Override
    public Boolean exist(String database, String collectionName, Serializable id) {
        return factory.getInjectExecute(database).isExist(collectionName,id);
    }

    @Override
    public Boolean exist(String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return exist(EMPTY,collectionName,queryChainWrapper);
    }

    @Override
    public Boolean exist(String database, String collectionName, QueryChainWrapper, ?> queryChainWrapper) {
        return factory.getInjectExecute(database).isExist(collectionName,queryChainWrapper);
    }

    @Override
    public Boolean update(String collectionName, Map entityMap, QueryChainWrapper, ?> queryChainWrapper) {
        return update(EMPTY,collectionName,entityMap,queryChainWrapper);
    }

    @Override
    public Boolean update(String database, String collectionName, Map entityMap, QueryChainWrapper, ?> queryChainWrapper) {
        return null;
    }

    @Override
    public String createIndex(String collectionName,Bson bson) {
        return createIndex(EMPTY,collectionName,bson);
    }

    @Override
    public String createIndex(String database, String collectionName, Bson bson) {
        return factory.getInjectExecute(database).createIndex(collectionName,bson);
    }

    @Override
    public String createIndex(String collectionName,Bson bson, IndexOptions indexOptions) {
        return createIndex(EMPTY,collectionName,bson,indexOptions);
    }

    @Override
    public String createIndex(String database, String collectionName, Bson bson, IndexOptions indexOptions) {
        return factory.getInjectExecute(database).createIndex(collectionName,bson,indexOptions);
    }

    @Override
    public List createIndexes(String collectionName,List indexes) {
        return createIndexes(EMPTY,collectionName,indexes);
    }

    @Override
    public List createIndexes(String database, String collectionName, List indexes) {
        return factory.getInjectExecute(database).createIndexes(collectionName,indexes);
    }

    @Override
    public List createIndexes(String collectionName,List indexes, CreateIndexOptions createIndexOptions) {
        return createIndexes(EMPTY,collectionName,indexes,createIndexOptions);
    }

    @Override
    public List createIndexes(String database, String collectionName, List indexes, CreateIndexOptions createIndexOptions) {
        return factory.getInjectExecute(database).createIndexes(collectionName,indexes,createIndexOptions);
    }

    @Override
    public List listIndexes(String collectionName) {
        return listIndexes(EMPTY,collectionName);
    }

    @Override
    public List listIndexes(String database, String collectionName) {
        return factory.getInjectExecute(database).listIndexes(collectionName);
    }

    @Override
    public void dropIndex(String collectionName,String indexName) {
        dropIndex(EMPTY,collectionName,indexName);
    }

    @Override
    public void dropIndex(String database, String collectionName, String indexName) {
        factory.getInjectExecute(database).dropIndex(collectionName,indexName);
    }

    @Override
    public void dropIndex(String collectionName,String indexName, DropIndexOptions dropIndexOptions) {
        dropIndex(EMPTY,collectionName,indexName,dropIndexOptions);
    }

    @Override
    public void dropIndex(String database, String collectionName, String indexName, DropIndexOptions dropIndexOptions) {
        factory.getInjectExecute(database).dropIndex(collectionName,indexName,dropIndexOptions);
    }

    @Override
    public void dropIndex(String collectionName,Bson keys) {
        dropIndex(EMPTY,collectionName,keys);
    }

    @Override
    public void dropIndex(String database, String collectionName, Bson keys) {
        factory.getInjectExecute(database).dropIndex(collectionName,keys);
    }

    @Override
    public void dropIndex(String collectionName,Bson keys, DropIndexOptions dropIndexOptions) {
        dropIndex(EMPTY,collectionName,keys,dropIndexOptions);
    }

    @Override
    public void dropIndex(String database, String collectionName, Bson keys, DropIndexOptions dropIndexOptions) {
        factory.getInjectExecute(database).dropIndex(collectionName,keys,dropIndexOptions);
    }

    @Override
    public void dropIndexes(String collectionName) {
        dropIndexes(EMPTY,collectionName);
    }

    @Override
    public void dropIndexes(String database, String collectionName) {
        factory.getInjectExecute(database).dropIndexes(collectionName);
    }

    @Override
    public void dropIndexes(String collectionName,DropIndexOptions dropIndexOptions) {
        dropIndexes(EMPTY,collectionName,dropIndexOptions);
    }

    @Override
    public void dropIndexes(String database, String collectionName, DropIndexOptions dropIndexOptions) {
        factory.getInjectExecute(database).dropIndexes(collectionName,dropIndexOptions);
    }

    @Override
    public long count(String collectionName) {
        return count(EMPTY,collectionName);
    }

    @Override
    public long count(String database, String collectionName) {
        return factory.getInjectExecute(database).count(collectionName);
    }
}