使用(Java)变量或方法的结果替换到mapper文件中

使用(Java)变量或方法的结果替换到mapper文件中

方法

mapper.xml

<sql id="selectUserVo">
    select
    <bind name="initSql" value="'u.id,u.name,u.password,u.sex'"/>
    /*  会调用com.rikylinz.utils.SqlUtils类的resultFields方法 传入params参数和initSql参数 返回最新sql字段列表 */
    ${@com.rikylinz.utils.SqlUtils@resultFields(params,initSql)}
    from user u
</sql>

注意：params为对应 select组件的入参参数，initSql是自己定义的变量(可以根据自身进行优化)

java代码

/**
 * @author rikylinz
 * @description SQL中相关操作工具类
 * @date 2022/11/11 11:44
 */
public class SqlUtils {

    /**
     *  查询语句 返回字段列表
     *  与 mapper.xml中的 ${@com.rikylinz.utils.SqlUtils@resultFields(params,initSql)}格式配合使用
     * @param params        入参params
     * @param initSql       查询语句的完整字段列表
     * @return 最新字段列表
     */
    public static String resultFields(Map<String, Object> params, String initSql){
        List<String> includeFields = (List<String>) params.get("includeFields");
        List<String> excludeFields = (List<String>) params.get("excludeFields");

        String newSql = Stream.of(initSql.split(",")).filter(field -> {
            if(includeFields != null && !includeFields.isEmpty()){
                String newField;
                if(field.contains(".")){
                    newField = field.substring(field.indexOf(".") + 1);
                }else{
                    newField = field;
                }
                return includeFields.contains(newField);
            }
            return true;
        }).filter(field -> {
            if(excludeFields != null && !excludeFields.isEmpty()){
                String newField;
                if(field.contains(".")){
                    newField = field.substring(field.indexOf(".") + 1);
                }else{
                    newField = field;
                }
                return !excludeFields.contains(newField);
            }
            return true;
        }).collect(Collectors.joining(","));
        return newSql;
    }
}

扩展：

• 可以放入if标签的test属性中，其他类似,

  <if test="@com.rikylinz.utils.SqlUtils@resultFields(params)">
  </if>
  

变量

推测与方法使用一致，但未进行测试

以取别名为例

mapper.xml

<select>
	select count(*) from user ${@com.rikylinz.utils.SqlUtils@alias}
</select>

java代码

/**
 * @author rikylinz
 * @description SQL中相关操作工具类
 * @date 2022/11/11 11:44
 */
public class SqlUtils {

    public static final String alias = "u";
}