• Home
• com.github.yulichang
• mybatis-plus-join-core
• 0.1.5.2.1
• source code
• MPJLambdaWrapper.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.yulichang.wrapper.MPJLambdaWrapper Maven / Gradle / Ivy

package com.github.yulichang.wrapper;

import com.baomidou.mybatisplus.core.conditions.SharedString;
import com.baomidou.mybatisplus.core.conditions.segments.MergeSegments;
import com.baomidou.mybatisplus.core.toolkit.*;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.github.yulichang.config.ConfigProperties;
import com.github.yulichang.toolkit.Constant;
import com.github.yulichang.toolkit.LambdaUtils;
import com.github.yulichang.toolkit.TableList;
import com.github.yulichang.toolkit.WrapperUtils;
import com.github.yulichang.toolkit.support.ColumnCache;
import com.github.yulichang.wrapper.enums.IfExistsSqlKeyWordEnum;
import com.github.yulichang.wrapper.interfaces.*;
import com.github.yulichang.wrapper.resultmap.Label;
import com.github.yulichang.wrapper.segments.*;
import lombok.Getter;

import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer;
import java.util.stream.Collectors;

/**
 * 参考 {@link com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper}
 * Lambda 语法使用 Wrapper
 *
 * @author yulichang
 */
@SuppressWarnings({"unused", "DuplicatedCode"})
public class MPJLambdaWrapper extends JoinAbstractLambdaWrapper> implements Fill>,
        Query>, QueryLabel>, Chain, SelectWrapper> {

    /**
     * 查询字段 sql
     */
    private SharedString sqlSelect = new SharedString();
    /**
     * 是否 select distinct
     */
    private boolean selectDistinct = false;
    /**
     * 查询的字段
     */
    @Getter
    private final List selectColumns = new ArrayList<>();
    /**
     * 映射关系
     */
    @Getter
    private final List> resultMapMybatisLabel = new ArrayList<>();

    /**
     * union sql
     */
    private SharedString unionSql;

    private List> fill;

    /**
     * 推荐使用 带 class 的构造方法
     */
    public MPJLambdaWrapper() {
        super();
    }

    /**
     * 推荐使用此构造方法
     */
    public MPJLambdaWrapper(Class clazz) {
        super(clazz);
    }

    /**
     * 构造方法
     *
     * @param entity 主表实体
     */
    public MPJLambdaWrapper(T entity) {
        super(entity);
    }

    /**
     * 自定义主表别名
     */
    public MPJLambdaWrapper(String alias) {
        super(alias);
    }

    /**
     * 构造方法
     *
     * @param clazz 主表class类
     * @param alias 主表别名
     */
    public MPJLambdaWrapper(Class clazz, String alias) {
        super(clazz, alias);
    }

    /**
     * 构造方法
     *
     * @param entity 主表实体类
     * @param alias  主表别名
     */
    public MPJLambdaWrapper(T entity, String alias) {
        super(entity, alias);
    }

    /**
     * 不建议直接 new 该实例，使用 JoinWrappers.lambda(UserDO.class)
     */
    protected MPJLambdaWrapper(T entity, Class entityClass, SharedString sqlSelect, AtomicInteger paramNameSeq,
                               Map paramNameValuePairs, MergeSegments mergeSegments, SharedString paramAlias,
                               SharedString lastSql, SharedString sqlComment, SharedString sqlFirst,
                               TableList tableList, Integer index, String keyWord, Class joinClass, String tableName,
                               MBiPredicate IfExists) {
        super.setEntity(entity);
        super.setEntityClass(entityClass);
        this.paramNameSeq = paramNameSeq;
        this.paramNameValuePairs = paramNameValuePairs;
        this.expression = mergeSegments;
        this.sqlSelect = sqlSelect;
        this.paramAlias = paramAlias;
        this.lastSql = lastSql;
        this.sqlComment = sqlComment;
        this.sqlFirst = sqlFirst;
        this.tableList = tableList;
        this.index = index;
        this.keyWord = keyWord;
        this.joinClass = joinClass;
        this.tableName = tableName;
        this.ifExists = IfExists;
    }


    /**
     * sql去重
     * select distinct
     */
    public MPJLambdaWrapper distinct() {
        this.selectDistinct = true;
        return typedThis;
    }


    @Override
    public List getSelectColum() {
        return this.selectColumns;
    }

    @Override
    public void addLabel(Label label, boolean isCollection) {
        if (isCollection) {
            this.resultMapCollection = true;
        }
        this.resultMap = true;
        this.resultMapMybatisLabel.add(label);
    }

    @Override
    public MPJLambdaWrapper getChildren() {
        return typedThis;
    }


    /**
     * 设置查询字段
     *
     * @param columns 字段数组
     * @return children
     */
    @SafeVarargs
    public final  MPJLambdaWrapper select(SFunction... columns) {
        if (ArrayUtils.isNotEmpty(columns)) {
            Class aClass = LambdaUtils.getEntityClass(columns[0]);
            Map cacheMap = ColumnCache.getMapField(aClass);
            for (SFunction s : columns) {
                SelectCache cache = cacheMap.get(LambdaUtils.getName(s));
                getSelectColum().add(new SelectNormal(cache, index, hasAlias, alias));
            }
        }
        return typedThis;
    }

    @Override
    public MPJLambdaWrapper selectAll(Class clazz) {
        return Query.super.selectAll(clazz);
    }

    /**
     * 查询实体类全部字段
     *
     * @param clazz   查询的实体类
     * @param exclude 排除字段
     */
    @Override
    @SafeVarargs
    public final  MPJLambdaWrapper selectAll(Class clazz, SFunction... exclude) {
        return Query.super.selectAll(clazz, exclude);
    }

    /**
     * 查询实体类全部字段
     *
     * @param clazz   查询的实体类
     * @param exclude 排除字段
     */
    @Override
    @SafeVarargs
    public final  MPJLambdaWrapper selectAll(Class clazz, String prefix, SFunction... exclude) {
        return Query.super.selectAll(clazz, prefix, exclude);
    }

    /**
     * 查询主表全部字段
     * 

     * 需要使用 使用 JoinWrappers.lambda(clazz) 或者 new MPJLambdaQueryWrapper<<(clazz) 构造
     *
     * @return children
     */
    public MPJLambdaWrapper selectAll() {
        Assert.notNull(getEntityClass(), "使用 JoinWrappers.lambda(clazz) 或者 new MPJLambdaQueryWrapper<>(clazz)");
        return selectAll(getEntityClass());
    }

    @Override
    public MPJLambdaWrapper selectFunc(String sql, MFunction column, String alias) {
        FuncConsumer funcConsumer = column.apply(new FuncConsumer());
        String formatSql;
        if (ArrayUtils.isEmpty(funcConsumer.getValues())) {
            formatSql = sql;
        } else {
            formatSql = formatSqlMaybeWithParam(sql, null, funcConsumer.getValues());
        }
        getSelectColum().add(new SelectFunc(alias, getIndex(), () -> formatSql,
                funcConsumer.getArgs(), isHasAlias(), getAlias()));
        return typedThis;
    }

    /**
     * 子查询
     */
    public  MPJLambdaWrapper selectSub(Class clazz, Consumer> consumer, SFunction alias) {
        return selectSub(clazz, ConfigProperties.subQueryAlias, consumer, alias);
    }

    public  MPJLambdaWrapper selectSub(Class clazz, String st, Consumer> consumer, SFunction alias) {
        return selectSub(clazz, st, consumer, LambdaUtils.getName(alias));
    }

    /**
     * 子查询
     */
    public  MPJLambdaWrapper selectSub(Class clazz, String st, Consumer> consumer, String alias) {
        MPJLambdaWrapper wrapper = subInstance(clazz, st);
        consumer.accept(wrapper);
        this.selectColumns.add(new SelectSub(() -> WrapperUtils.buildSubSqlByWrapper(clazz, wrapper, alias), hasAlias, this.alias, alias));
        return typedThis;
    }

    /**
     * union
     * 

     * 例： wrapper.union(UserDO.class, union -> union.selectAll(UserDO.class))
     *
     * @param clazz union语句的主表类型
     * @since 1.4.8
     */
    public  MPJLambdaWrapper union(Class clazz, Consumer> consumer) {
        MPJLambdaWrapper unionWrapper = fromInstance(clazz);
        consumer.accept(unionWrapper);

        String sb = " UNION " + WrapperUtils.buildUnionSqlByWrapper(clazz, unionWrapper);

        if (Objects.isNull(unionSql)) {
            unionSql = SharedString.emptyString();
        }
        unionSql.setStringValue(unionSql.getStringValue() + sb);
        return typedThis;
    }

    @Override
    public MPJLambdaWrapper addFill(MConsumer