MongoPlus-数据库

package com.anwen.mongo.aggregate.pipeline; import com.anwen.mongo.support.SFunction; import com.mongodb.BasicDBObject; import com.mongodb.client.model.BsonField; import com.mongodb.lang.Nullable; import org.bson.BsonArray; import org.bson.BsonDocument; import org.bson.BsonString; import org.bson.Document; import org.bson.conversions.Bson; import java.util.Arrays; import java.util.List; import static com.mongodb.assertions.Assertions.notNull; import static java.util.stream.Collectors.toList; /** * 构建累加器 * * @author anwen * @date 2024/6/12 下午3:44 */ public final class Accumulators { /** * 获取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的值的总和。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $sum * @since mongodb.driver.manual reference/operator/aggregation/sum/ $sum */ public static <TExpression> BsonField sum(final String fieldName, final TExpression expression) { return accumulatorOperator("$sum", fieldName, expression); } /** * 获取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的值的总和。 * 该累加器会默认count=1 * <p>count: { $sum: 1 }</p> * @param <TExpression> 表达式类型 * @return $sum * @since mongodb.driver.manual reference/operator/aggregation/sum/ $sum */ public static <TExpression> BsonField sum() { return accumulatorOperator("$sum", "count", 1); } /** * 获取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的值的总和。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $sum * @since mongodb.driver.manual reference/operator/aggregation/sum/ $sum */ public static <TExpression,T> BsonField sum(final SFunction<T,?> fieldName, final TExpression expression) { return sum(fieldName.getFieldNameLine(),expression); } /** * 取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的总数。 * @param fieldName 字段名称 * @param expression 表达式 * @author anwen * @date 2024/8/23 10:57 */ public static <TExpression> BsonField count(final String fieldName,final TExpression expression){ return accumulatorOperator("$count",fieldName,expression); } /** * 取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的总数。 * @param fieldName 字段名称 * @param expression 表达式 * @author anwen * @date 2024/8/23 10:57 */ public static <T,TExpression> BsonField count(final SFunction<T,?> fieldName,final TExpression expression){ return count(fieldName.getFieldNameLine(),expression); } /** * 取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的总数。 * 表达式默认为空对象 * @param fieldName 字段名称 * @author anwen * @date 2024/8/23 10:57 */ public static <T,TExpression> BsonField count(final SFunction<T,?> fieldName){ return count(fieldName.getFieldNameLine(),new Document()); } /** * 取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的总数。 * 表达式默认为空对象 * @param fieldName 字段名称 * @author anwen * @date 2024/8/23 10:57 */ public static <TExpression> BsonField count(final String fieldName){ return count(fieldName,new Document()); } /** * 获取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的平均值。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $avg * @since mongodb.driver.manual reference/operator/aggregation/avg/ $avg */ public static <TExpression> BsonField avg(final String fieldName, final TExpression expression) { return accumulatorOperator("$avg", fieldName, expression); } /** * 获取 $group 操作的字段名称，该字段表示将给定表达式应用于组内所有成员时的平均值。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $avg * @since mongodb.driver.manual reference/operator/aggregation/avg/ $avg */ public static <TExpression,T> BsonField avg(final SFunction<T,?> fieldName, final TExpression expression) { return avg(fieldName.getFieldNameLine(),expression); } /** * 获取 $group 操作的字段名称，该字段名称表示应用于组的第一个成员时给定表达式的值。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $first * @since mongodb.driver.manual reference/operator/aggregation/first/ $first */ public static <TExpression> BsonField first(final String fieldName, final TExpression expression) { return accumulatorOperator("$first", fieldName, expression); } /** * 获取 $group 操作的字段名称，该字段名称表示应用于组的第一个成员时给定表达式的值。 * * @param fieldName 字段名称 * @param expression 表达式 * @param <TExpression> 表达式类型 * @return $first * @since mongodb.driver.manual reference/operator/aggregation/first/ $first */ public static <TExpression,T> BsonField first(final SFunction<T,?> fieldName, final TExpression expression) { return first(fieldName.getFieldNameLine(),expression); } /** * 返回计算字段和累加器的组合，该组合生成一个 BSON {@link org.bson.BsonType#ARRAY Array} ，该数组包含给定{@code inExpression}的值，该值是根据预排序组中的前{@code N}个元素计算的，其中N是{@code nExpression}的正整数值。 * * @param fieldName 由累加器计算的字段。 * @param inExpression 输入表达式 * @param nExpression 限制产生的值的数量的表达式 * @param <InExpression> 输入表达式的类型. * @param <NExpression> 限制表达式的类型. * @return 请求的 {@link BsonField}. * @since mongodb.driver.manual reference/operator/aggregation/firstN/ $firstN * @since 4.7 * @since mongodb.server.release 5.2 */ public static <InExpression, NExpression> BsonField firstN( final String fieldName, final NExpression nExpression, final InExpression inExpression) { return pickNAccumulator(notNull("fieldName", fieldName), "$firstN", notNull("inExpression", inExpression), notNull("nExpression", nExpression)); } /** * 返回计算字段和累加器的组合，该组合生成一个 BSON {@link org.bson.BsonType#ARRAY Array} ，该数组包含给定{@code inExpression}的值，该值是根据预排序组中的前{@code N}个元素计算的，其中N是{@code nExpression}的正整数值。 * * @param fieldName 由累加器计算的字段。 * @param inExpression 输入表达式 * @param nExpression 限制产生的值的数量的表达式 * @param <InExpression> 输入表达式的类型. * @param <NExpression> 限制表达式的类型. * @return 请求的 {@link BsonField}. * @since mongodb.driver.manual reference/operator/aggregation/firstN/ $firstN * @since 4.7 * @since mongodb.server.release 5.2 */ public static <InExpression, NExpression,T> BsonField firstN( final SFunction<T,?> fieldName, final NExpression nExpr