mybatis-plus-advance插件

/* * Copyright (c) 2011-2022, baomidou (jobob@qq.com). * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.baomidou.mybatisplus.advance.injector; import com.baomidou.mybatisplus.core.conditions.AbstractJoinWrapper; import com.baomidou.mybatisplus.core.conditions.AbstractWrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.enums.SqlMethod; import com.baomidou.mybatisplus.core.toolkit.*; import com.baomidou.mybatisplus.core.toolkit.support.ColumnCache; import com.baomidou.mybatisplus.core.toolkit.support.SFunction; import com.baomidou.mybatisplus.extension.conditions.query.BasicJoinQueryWrapper; import com.baomidou.mybatisplus.extension.toolkit.SqlHelper; import lombok.AllArgsConstructor; import lombok.Data; import org.apache.ibatis.session.SqlSession; import org.mybatis.spring.SqlSessionUtils; import java.lang.invoke.CallSite; import java.lang.invoke.LambdaMetafactory; import java.lang.invoke.MethodHandles; import java.lang.invoke.MethodType; import java.lang.reflect.Field; import java.util.Collection; import java.util.HashMap; import java.util.List; import java.util.Map; /** * 通过此类去给wrapper添加参数和执行sql返回结果 * * @author wanglei * @since 2022-03-14 */ public class FuntionTools { /** * 可序列化 */ private static final int FLAG_SERIALIZABLE = 1; private static Map<String, SFunction> functionMap = new HashMap<>(); /** * 把动作 * * @param actions 动作 * @param operator 操作符 比如like * @param po po对象 * @param property 属性名 * @param value 值 */ public static void addAction(List<Action> actions, String operator, Object po, String property, Object value) { actions.add(new Action(po.getClass(), property, operator, value, null, null, null)); } /** * 添加动作 * * @param actions actions * @param operator 操作符 比如like * @param po po对象 * @param property 属性名 * @param minValue 最小值 * @param maxValue 最大值 */ public static void addAction(List<Action> actions, String operator, Object po, String property, Object minValue, Object maxValue) { actions.add(new Action(po.getClass(), property, operator, null, minValue, maxValue, null)); } /** * 构造一个查询条件 - 单表查询 * * @param actions 动作集合 * @param po po * @return QueryWrapper */ public static LambdaQueryWrapper buildQueryWrapper(List<Action> actions, Object po) { LambdaQueryWrapper<?> queryWrapper = new LambdaQueryWrapper(); for (Action action : actions) { // 普通的查询只保留他自己的where条件和select，就算他调用了join也忽略掉 if (action.getModelClass().equals(po.getClass()) && !"select".equals(action.getAction())) { SFunction column = getSFunction(action.getModelClass(), action.getProperty()); buildWhere(queryWrapper, action, column); } else if ("select".equals(action.getAction()) && action.getModelClass().equals(po.getClass())) { SFunction[] columns = new SFunction[action.getProperties().length]; int i = 0; for (String property : action.getProperties()) { columns[i++] = getSFunction(action.getModelClass(), property); } queryWrapper.select(columns); } } return queryWrapper; } /** * 构造where条件 * * @param queryWrapper wapper * @param action 动作 * @param column 列名 */ public static void buildWhere(AbstractWrapper queryWrapper, Action action, Object column) { switch (action.getAction()) { case OperatorConstant.EQ: queryWrapper.eq(column, action.getValue()); break; case OperatorConstant.LT: queryWrapper.lt(column, action.getValue()); break; case OperatorConstant.GT: queryWrapper.gt(column, action.getValue()); break; case OperatorConstant.LE: queryWrapper.le(column, action.getValue()); break; case OperatorConstant.GE: queryWrapper.ge(column, action.getValue()); break; case OperatorConstant.NE: queryWrapper.ne(column, action.getValue()); break; case OperatorConstant.LIKE: queryWrapper.like(column, action.getValue()); break; case OperatorConstant.LIKE_LEFT: queryWrapper.likeLeft(column, action.getValue()); break; case OperatorConstant.LIKE_RIGHT: queryWrapper.likeRight(column, action.getValue()); break; case OperatorConstant.NOT_LIKE: queryWrapper.notLike(column, action.getValue()); break; case OperatorConstant.IS_NULL: queryWrapper.isNull(column); break; case OperatorConstant.NOT_NULL: queryWrapper.isNotNull(column); break; case OperatorConstant.IN: queryWrapper.in(column, (Collection) (action.getValue())); break; case OperatorConstant.NOT_IN: queryWrapper.notIn(column, (Collection) (action.getValue())); break; case OperatorConstant.ORDER_BY_ASC: queryWrapper.orderByAsc(column); break; case OperatorConstant.ORDER_BY_DESC: queryWrapper.orderByDesc(column); break; case OperatorConstant.BETWEEN: queryWrapper.between(column, action.getMin(), action.getMax()); break; case OperatorConstant.NOT_BETWEEN: queryWrapper.notBetween(column, action.getMin(), action.getMax()); break; } } /** * 构造一个查询条件 - 多表查询 * * @param actions 动作集合 * @param po po * @return */ public static BasicJoinQueryWrapper buildJoinWrapper(List<Action> actions, Object po) { BasicJoinQueryWrapper queryWrapper = new BasicJoinQueryWrapper(po.getClass()); // 先搞定join，不然添加条件和select会报错 for (Action action : actions) { if (action.getAction().contains(OperatorConstant.JOIN)) { switch (action.getAction()) { case OperatorConstant.JOIN: queryWrapper.innerJoin(action.getModelClass()); break; case OperatorConstant.LEFT_JOIN: queryWrapper.leftJoin(action.getModelClass()); break; } } } for (Action action : actions) { // 不是select和join动作的话，就是where条件 if (!OperatorConstant.SELECT.equals(action.getAction()) && !action.getAction().contains(OperatorConstant.JOIN)) { BasicJoinQueryWrapper.ModelProperty column = new BasicJoinQueryWrapper.ModelProperty(action.getModelClass(), action.getProperty());