cn.org.atool.fluent.mybatis.segment.HavingBase Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of fluent-mybatis Show documentation
There is a newer version: 1.9.9
package cn.org.atool.fluent.mybatis.segment;

import cn.org.atool.fluent.mybatis.base.crud.IBaseQuery;
import cn.org.atool.fluent.mybatis.base.model.FieldMapping;
import cn.org.atool.fluent.mybatis.base.model.ISqlOp;
import cn.org.atool.fluent.mybatis.functions.IAggregate;
import cn.org.atool.fluent.mybatis.segment.fragment.Column;
import cn.org.atool.fluent.mybatis.segment.fragment.IFragment;

import static cn.org.atool.fluent.mybatis.segment.fragment.Fragments.SEG_COUNT_1;
import static cn.org.atool.fluent.mybatis.segment.fragment.KeyFrag.HAVING;

/**
 * BaseHaving: having设置
 *
 * @author darui.wu 2020/6/21 6:22 下午
 */
@SuppressWarnings({"unchecked", "rawtypes"})
public abstract class HavingBase<
    H extends HavingBase,
    Q extends IBaseQuery
    >
    extends AggregateSegment> {

    protected final HavingOperator operator = new HavingOperator<>(super.getOrigin());

    protected HavingBase(Q query) {
        super(query);
    }

    protected HavingBase(H selector, IAggregate aggregate) {
        super(selector, aggregate);
    }

    /**
     * 设置having条件
     *
     * @param aggregate having 聚合函数
     * @param op        比较操作
     * @param args      参数列表
     * @return Having设置器
     */
    H aggregate(IFragment aggregate, ISqlOp op, Object... args) {
        this.wrapper.data().apply(HAVING, aggregate, op, args);
        return (H) this;
    }

    H apply(IFragment aggregate, ISqlOp op, IFragment func, Object... args) {
        this.wrapper.data().apply(HAVING, aggregate, op, func, args);
        return (H) this;
    }

    /**
     * 执行聚合操作
     *
     * @param column    聚合字段
     * @param aggregate 聚合函数
     * @return Having条件判断
     */
    protected HavingOperator apply(FieldMapping column, IAggregate aggregate) {
        return this.operator.aggregate(Column.set(this.wrapper, column), aggregate);
    }

    /**
     * count(*)
     *
     * @return Having条件判断
     */
    public HavingOperator count() {
        return this.operator.aggregate(null, (c) -> SEG_COUNT_1);
    }

    /**
     * 执行聚合操作
     *
     * @param column 聚合操作, 比如 sum(column) 或者 select中聚合操作的别名
     * @return Having条件判断
     */
    public HavingOperator apply(String column) {
        Column _column = Column.set(this.wrapper, column);
        return this.operator.aggregate(_column,
            this.aggregate == null ? c -> c : this.aggregate);
    }

    @Override
    protected HavingOperator apply() {
        this.apply(this.current, this.aggregate);
        return this.operator;
    }
}