top.summerboot.orm.association.util.QueryUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of orm-summer-boot-starter Show documentation
mybatis-plus功能扩展
The newest version!
package top.summerboot.orm.association.util;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.activerecord.Model;
import top.summerboot.orm.association.annotation.*;
import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * @author xieshuang
 * @date 2020-11-18 15:36
 */
@Slf4j
public class QueryUtil {

    public static  String generateSql(Class vClass, Object query){
        String tableName;
        TableName annotation = ReflexUtil.getAnnotation(vClass, TableName.class, Model.class);
        if (annotation == null){
            tableName = StrUtil.toUnderlineCase(vClass.getSimpleName());
        }else {
            tableName = annotation.value();
        }
        Map allFieldMap = ReflexUtil.getAllFieldMap(vClass, Model.class);
        StringBuilder stringBuilder = new StringBuilder();
        stringBuilder.append("select DISTINCT ");
        boolean one = true;
        for (Map.Entry stringFieldEntry : allFieldMap.entrySet()) {
            JoinSelect joinSelect = stringFieldEntry.getValue().getAnnotation(JoinSelect.class);
            if (joinSelect != null){
                continue;
            }
            Select select = stringFieldEntry.getValue().getAnnotation(Select.class);
            TableField tableField = stringFieldEntry.getValue().getAnnotation(TableField.class);
            if (select != null){
                if (!select.exist()){
                    continue;
                }
                String field = select.value();
                if (one){
                    stringBuilder.append(field);
                    one = false;
                }else {
                    stringBuilder.append(" ,");
                    stringBuilder.append(field);
                }
            }else if (tableField != null){
                if (!tableField.exist()){
                    continue;
                }
                String field = tableField.value();
                if (!tableField.value().contains(".")){
                    field = tableName + "." + field;
                }
                if (one){
                    stringBuilder.append(field);
                    one = false;
                }else {
                    stringBuilder.append(" ,");
                    stringBuilder.append(field);
                }
            }else {
                if (one){
                    one = false;
                }else {
                    stringBuilder.append(" ,");
                }
                stringBuilder.append(tableName);
                stringBuilder.append(".");
                stringBuilder.append(StrUtil.toUnderlineCase(stringFieldEntry.getKey()));
            }
        }
        stringBuilder.append(" from ");
        stringBuilder.append(tableName);
        stringBuilder.append("\n");
        setJoinExpression(stringBuilder, query);
        return stringBuilder.toString();
    }

    public static void setJoinExpression(StringBuilder stringBuilder, Object query) {
        if (query != null){
            List stringList = new ArrayList<>();
            Class dtoClass = query.getClass();
            Field[] declaredFields = ReflexUtil.getAllFields(dtoClass, null);
            for (Field field : declaredFields) {
                //打开私有访问
                field.setAccessible(true);
                Object value = null;
                try {
                    value = field.get(query);
                } catch (IllegalAccessException e) {
                    log.error("通过反射获取属性值出错：" + e);
                }
                JoinExpression joinExpression = field.getAnnotation(JoinExpression.class);
                if (joinExpression != null && (value != null || joinExpression.allJoin())) {
                    if (!stringList.contains(joinExpression.value())){
                        stringBuilder.append(joinExpression.value());
                        stringBuilder.append("\n");
                        stringList.add(joinExpression.value());
                    }
                }
            }
        }
    }
}