All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.anwen.mongo.conditions.aggregate.AggregateChainWrapper Maven / Gradle / Ivy

There is a newer version: 2.1.6.1
Show newest version
package com.anwen.mongo.conditions.aggregate;

import com.anwen.mongo.conditions.accumulator.Accumulator;
import com.anwen.mongo.conditions.interfaces.aggregate.Aggregate;
import com.anwen.mongo.conditions.interfaces.aggregate.pipeline.AddFields;
import com.anwen.mongo.conditions.interfaces.aggregate.pipeline.Let;
import com.anwen.mongo.conditions.interfaces.aggregate.pipeline.Projection;
import com.anwen.mongo.conditions.interfaces.aggregate.pipeline.ReplaceRoot;
import com.anwen.mongo.conditions.interfaces.condition.Order;
import com.anwen.mongo.conditions.query.QueryChainWrapper;
import com.anwen.mongo.constant.SqlOperationConstant;
import com.anwen.mongo.enums.*;
import com.anwen.mongo.model.AggregateBasicDBObject;
import com.anwen.mongo.model.BaseAggregate;
import com.anwen.mongo.model.FuncGroupField;
import com.anwen.mongo.model.GroupField;
import com.anwen.mongo.strategy.aggregate.impl.*;
import com.anwen.mongo.support.SFunction;
import com.mongodb.BasicDBObject;
import com.mongodb.client.model.Collation;
import com.mongodb.client.model.CollationStrength;
import org.bson.BsonValue;
import org.bson.conversions.Bson;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author JiaChaoYang
 **/
public class AggregateChainWrapper implements Aggregate {

    List baseAggregateList = new ArrayList<>();

    List basicDBObjectList = new ArrayList<>();

    BasicDBObject optionsBasicDBObject = new BasicDBObject();

    protected final Children typedThis = (Children) this;

    private Integer aggregateOrder = 0;

    @Override
    public Children match(QueryChainWrapper queryChainWrapper) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.MATCH.getType(), new MatchConcretePipeline(queryChainWrapper.getCompareList(),queryChainWrapper.getBasicDBObjectList()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children match(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.MATCH.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children match(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.MATCH.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children project(Projection... projection) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(projection),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children project(List projectionList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(projectionList),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children projectDisplay(SFunction... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (SFunction sFunction : column) {
                        add(new Projection(sFunction.getFieldNameLine(), ProjectionEnum.DISPLAY.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children projectDisplay(String... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (String col : column) {
                        add(new Projection(col, ProjectionEnum.DISPLAY.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children projectNone(SFunction... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (SFunction sFunction : column) {
                        add(new Projection(sFunction.getFieldNameLine(), ProjectionEnum.NONE.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children projectNone(String... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (String col : column) {
                        add(new Projection(col, ProjectionEnum.NONE.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children project(boolean displayId, Projection... projection) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(displayId, projection),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children projectDisplay(boolean displayId, SFunction... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (SFunction sFunction : column) {
                        add(new Projection(sFunction.getFieldNameLine(), ProjectionEnum.DISPLAY.getValue()));
                    }
                    if (!displayId) {
                        add(new Projection(SqlOperationConstant._ID, ProjectionEnum.NONE.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children projectDisplay(boolean displayId, String... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (String col : column) {
                        add(new Projection(col, ProjectionEnum.DISPLAY.getValue()));
                    }
                    if (!displayId) {
                        add(new Projection(SqlOperationConstant._ID, ProjectionEnum.NONE.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children projectNone(boolean displayId, SFunction... column) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                new ArrayList() {{
                    for (SFunction sFunction : column) {
                        add(new Projection(sFunction.getFieldNameLine(), ProjectionEnum.NONE.getValue()));
                    }
                    if (!displayId) {
                        add(new Projection(SqlOperationConstant._ID, ProjectionEnum.NONE.getValue()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children projectNone(boolean displayId, String... column) {
        this.baseAggregateList.add(
                new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new ProjectConcretePipeline(
                        new ArrayList() {{
                            for (String col : column) {
                                add(new Projection(col, ProjectionEnum.NONE.getValue()));
                            }
                            if (!displayId) {
                                add(new Projection(SqlOperationConstant._ID, ProjectionEnum.NONE.getValue()));
                            }
                        }}),
                        getNextAggregateOrder()
                )
        );
        return typedThis;
    }


    @Override
    public Children project(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children project(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.PROJECT.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sort(Order... orders) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new SortConcretePipeline(orders),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sort(List orderList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new SortConcretePipeline(orderList),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children sortAsc(SFunction... field) {
        this.baseAggregateList.add(new BaseAggregate(
                AggregateTypeEnum.SORT.getType(),
                new SortConcretePipeline(
                        new ArrayList() {{
                            for (SFunction sFunction : field) {
                                add(new Order(OrderEnum.ORDER_BY.getFlag(), sFunction.getFieldNameLine()));
                            }
                        }}
                ),
                getNextAggregateOrder()
        ));
        return typedThis;
    }

    @Override
    public Children sortAsc(String... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new SortConcretePipeline(
            new ArrayList(){{
                for (String col : field) {
                    add(new Order(OrderEnum.ORDER_BY.getFlag(), col));
                }
            }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children sortDesc(SFunction... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new SortConcretePipeline(
                new ArrayList(){{
                    for (SFunction sFunction : field) {
                        add(new Order(OrderEnum.ORDER_BY_DESC.getFlag(),sFunction.getFieldNameLine()));
                    }
                }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sortDesc(String... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new SortConcretePipeline(
            new ArrayList(){{
                for (String col : field) {
                    add(new Order(OrderEnum.ORDER_BY_DESC.getFlag(), col));
                }
            }}
        ),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sort(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sort(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SORT.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children limit(long limit) {
        this.basicDBObjectList.add(new AggregateBasicDBObject(AggregateTypeEnum.LIMIT.getType(),limit,getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children skip(long skip) {
        this.basicDBObjectList.add(new AggregateBasicDBObject(AggregateTypeEnum.SKIP.getType(),skip,getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }},true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(Accumulator... _id) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList<>(Arrays.asList(_id))),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, Accumulator accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, Accumulator accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, accumulator,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, Accumulator accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, Accumulator accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, accumulator,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, Accumulator... accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, Accumulator... accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, true,accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, Accumulator... accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, Accumulator... accumulator) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, true,accumulator),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, List accumulatorList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), accumulatorList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, List accumulatorList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, accumulatorList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, List accumulatorList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, accumulatorList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, List accumulatorList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, accumulatorList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, String resultMappingField, String operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), resultMappingField, operator, field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, String resultMappingField, String operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, resultMappingField, operator, field,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, String resultMappingField, String operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField, operator, field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, String resultMappingField, String operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField, operator, field,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, String resultMappingField, GroupTypeEnum operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), resultMappingField, operator.getOperator(), field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, String resultMappingField, GroupTypeEnum operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, resultMappingField, operator.getOperator(), field,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, String resultMappingField, GroupTypeEnum operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField, operator.getOperator(), field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, String resultMappingField, GroupTypeEnum operator, Object field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField, operator.getOperator(), field,true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, SFunction resultMappingField, String operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), resultMappingField.getFieldNameLine(), operator, field.getFieldNameLine()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, SFunction resultMappingField, String operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, resultMappingField.getFieldNameLine(), operator, field.getFieldNameLine(),true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, SFunction resultMappingField, String operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField.getFieldNameLine(), operator, field.getFieldNameLine()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, SFunction resultMappingField, String operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField.getFieldNameLine(), operator, field.getFieldNameLine(),true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(String _id, SFunction resultMappingField, GroupTypeEnum operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField.getFieldNameLine(), operator.getOperator(), field.getFieldNameLine()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(List _id, SFunction resultMappingField, GroupTypeEnum operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id, resultMappingField.getFieldNameLine(), operator.getOperator(), field.getFieldNameLine(),true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(SFunction _id, SFunction resultMappingField, GroupTypeEnum operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(_id.getFieldNameLine(), resultMappingField.getFieldNameLine(), operator.getOperator(), field.getFieldNameLine()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children groupFunc(List> _id, SFunction resultMappingField, GroupTypeEnum operator, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new GroupConcretePipeline(new ArrayList(){{
            _id.forEach(funcGroupField -> {
                add(new GroupField(funcGroupField.getGroupField(),funcGroupField.getField()));
            });
        }}, resultMappingField.getFieldNameLine(), operator.getOperator(), field.getFieldNameLine(),true),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children group(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.GROUP.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children lookup(String from, String localField, String foreignField, String as) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.LOOKUP.getType(), new LookupConcretePipeline(new BasicDBObject(){{
            put("from",from);
            put("localField",localField);
            put("foreignField",foreignField);
            put("as",as);
        }}),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children lookup(String from, List letList, AggregateChainWrapper pipeline, String as) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.LOOKUP.getType(), new LookupConcretePipeline(new BasicDBObject(){{
            put("from",from);
            put("pipeline",new ArrayList(){{
                pipeline.getBaseAggregateList().forEach(baseAggregate -> add(new BasicDBObject("$"+baseAggregate.getType(),baseAggregate.getPipelineStrategy().buildAggregate())));
                addAll(pipeline.getBasicDBObjectList());
            }});
            put("as",as);
        }}),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children lookup(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.LOOKUP.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children lookup(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.LOOKUP.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(String resultMappingField, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(new AddFields(resultMappingField,field.getFieldNameLine())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(SFunction resultMappingField, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(new AddFields(resultMappingField.getFieldNameLine(),field.getFieldNameLine())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(SFunction resultMappingField, String field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(new AddFields(resultMappingField.getFieldNameLine(),field)),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(String resultMappingField, String field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(new AddFields(resultMappingField,field)),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(AddFields... addFields) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(addFields),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(List addFieldsList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new AddFieldsConcretePipeline(addFieldsList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(BasicDBObject basicDBObject) {
        baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children addFields(Bson bson) {
        baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.ADD_FIELDS.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new UnwindConcretePipeline(field.getFieldNameLine(),false),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(Boolean preserveNullAndEmptyArrays, SFunction field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new UnwindConcretePipeline(field.getFieldNameLine(),preserveNullAndEmptyArrays),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(String field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new UnwindConcretePipeline(field,false),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(Boolean preserveNullAndEmptyArrays, String field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new UnwindConcretePipeline(field,preserveNullAndEmptyArrays),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unwind(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children sample(long size) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.SAMPLE.getType(), new SampleConcretePipeline(size),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children replaceRoot(SFunction... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(false,field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(ReplaceRoot... replaceRoot) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(false,replaceRoot),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(List replaceRootList) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(false,replaceRootList),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(String... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(false,field),getNextAggregateOrder()));
        return typedThis;
    }

    @SafeVarargs
    @Override
    public final Children replaceRoot(Boolean reserveOriginalDocument, SFunction... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(reserveOriginalDocument,field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(Boolean reserveOriginalDocument, String... field) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new RootConcretePipelineReplace(reserveOriginalDocument,field),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children replaceRoot(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.REPLACE_ROOT.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unionWith(String collectionName) {
        this.basicDBObjectList.add(new AggregateBasicDBObject(AggregateTypeEnum.UNION_WITH.getType(),collectionName,getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unionWith(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children unionWith(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.UNWIND.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children out(String coll) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.OUT.getType(), new OutConcretePipeline(null,coll),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children out(String db, String coll) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.OUT.getType(), new OutConcretePipeline(db,coll),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children out(BasicDBObject basicDBObject) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.OUT.getType(), new DefaultConcretePipeline(basicDBObject),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children out(Bson bson) {
        this.baseAggregateList.add(new BaseAggregate(AggregateTypeEnum.OUT.getType(), new DefaultConcretePipeline(BasicDBObject.parse(bson.toBsonDocument().toJson())),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children custom(BasicDBObject basicDBObject) {
        this.basicDBObjectList.add(new AggregateBasicDBObject(basicDBObject,getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children custom(Bson bson) {
        this.basicDBObjectList.add(new AggregateBasicDBObject(BasicDBObject.parse(bson.toBsonDocument().toJson()),getNextAggregateOrder()));
        return typedThis;
    }

    @Override
    public Children allowDiskUse(boolean allowDiskUse) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.ALLOW_DISK_USE.getOptions(), allowDiskUse);
        return typedThis;
    }

    @Override
    public Children batchSize(Integer size) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.BATCH_SIZE.getOptions(), size);
        return typedThis;
    }

    @Override
    public Children collation(CollationStrength collationStrength) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.COLLATION.getOptions(), Collation.builder().collationStrength(collationStrength).build().asDocument());
        return typedThis;
    }

    @Override
    public Children maxTimeMS(long time) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.MAX_TIME_MS.getOptions(), time);
        return typedThis;
    }

    @Override
    public Children maxAwaitTimeMS(long maxAwaitTime) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.MAX_AWAIT_TIME_MS.getOptions(), maxAwaitTime);
        return typedThis;
    }

    @Override
    public Children bypassDocumentValidation(boolean bypassDocumentValidation) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.BYPASS_DOCUMENT_VALIDATION.getOptions(), bypassDocumentValidation);
        return typedThis;
    }

    @Override
    public Children comment(BsonValue comment) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.COMMENT.getOptions(), comment);
        return typedThis;
    }

    @Override
    public Children comment(String comment) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.COMMENT_STR.getOptions(), comment);
        return typedThis;
    }

    @Override
    public Children hint(Bson hint) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.HINT.getOptions(), hint);
        return typedThis;
    }

    @Override
    public Children hint(String hint) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.HINT_STR.getOptions(), hint);
        return typedThis;
    }

    @Override
    public Children let(Bson variables) {
        this.optionsBasicDBObject.append(AggregateOptionsEnum.LET.getOptions(), variables);
        return typedThis;
    }

    public Integer getNextAggregateOrder(){
        return ++aggregateOrder;
    }

    public List getBaseAggregateList() {
        return baseAggregateList;
    }

    public List getBasicDBObjectList() {
        return basicDBObjectList;
    }

    public BasicDBObject getOptionsBasicDBObject() {
        return optionsBasicDBObject;
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy