icu.mhb.mybatisplus.plugln.injector.methods.JoinSelectCount Maven / Gradle / Ivy

Go to download
package icu.mhb.mybatisplus.plugln.injector.methods;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import icu.mhb.mybatisplus.plugln.enums.JoinSqlMethod;
import icu.mhb.mybatisplus.plugln.constant.JoinConstant;
import icu.mhb.mybatisplus.plugln.injector.JoinAbstractMethod;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;

/**
 * @author mahuibo
 * @Title: JoinSelectCount
 * @time 8/27/21 3:14 PM
 */
public class JoinSelectCount extends JoinAbstractMethod {

    public JoinSelectCount(String methodName) {
        super(methodName);
    }
    public JoinSelectCount() {
        super(JoinSqlMethod.JOIN_SELECT_COUNT.name());
    }

    @Override
    public MappedStatement injectMappedStatement(Class mapperClass, Class modelClass, TableInfo tableInfo) {
        JoinSqlMethod sqlMethod = JoinSqlMethod.JOIN_SELECT_COUNT;

        setTableInfo(tableInfo);

        // 转换一下
        modelClass = getTableClass(modelClass);

        String sql = String.format(sqlMethod.getSql(), sqlFirst(), getJoinTableName(), JoinConstant.JOIN_SQL_NAME,
                                   sqlWhereAliasEntityWrapper(true), sqlComment());
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        return this.addSelectMappedStatementForOther(mapperClass, sqlMethod.getMethod(), sqlSource, Integer.class);
    }

}