com.silentgo.orm.sqlparser.daoresolve.QueryDaoResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of orm Show documentation
There is a newer version: 0.3.2
package com.silentgo.orm.sqlparser.daoresolve;

import com.silentgo.orm.base.BaseDaoDialect;
import com.silentgo.orm.base.BaseTableInfo;
import com.silentgo.orm.base.SQLTool;
import com.silentgo.orm.base.TableModel;
import com.silentgo.orm.sqlparser.annotation.ColumnIgnore;
import com.silentgo.orm.sqlparser.annotation.Query;
import com.silentgo.orm.sqlparser.funcanalyse.DaoKeyWord;
import com.silentgo.utils.Assert;
import com.silentgo.utils.log.Log;
import com.silentgo.utils.log.LogFactory;

import java.lang.annotation.Annotation;
import java.util.*;

/**
 * Project : parent
 * Package : com.silentgo.orm.sqlparser.daoresolve
 *
 * @author teddyzhu
 *         
 *         Created by teddyzhu on 16/9/30.
 */
public class QueryDaoResolver implements DaoResolver {

    private static final Log LOGGER = LogFactory.get();
    @Override
    public boolean handle(String methodName, List parsedMethod, List annotations) {
        return DaoKeyWord.Query.equals(parsedMethod.get(0));
    }

    @Override
    public  SQLTool processSQL(String methodName, Class returnType, Object[] objects, Integer[] objectIndex, List parsedMethod, BaseTableInfo tableInfo, SQLTool sqlTool, List annotations, boolean[] isHandled,
                                                     BaseDaoDialect daoDialect, Map nameObjects) {
        if (isHandled[0]) return sqlTool;
        isHandled[0] = true;
        sqlTool.select(tableInfo.getTableName());
        Integer index = 1;
        String two = DaoResolveKit.getField(parsedMethod, index);
        if (DaoKeyWord.One.equals(two)) {
            sqlTool.limit(1, 1);
        } else if (DaoKeyWord.List.equals(two)) {
            Assert.isTrue(Collection.class.isAssignableFrom(returnType), "Method [" + methodName + "] return type should be collection");
            sqlTool.limitClear();
        }
        boolean needColumns = true;

        Optional opQuery = annotations.stream().filter(annotation -> annotation.annotationType().equals(Query.class)).findFirst();
        if (opQuery.isPresent()) {
            Query query = (Query) opQuery.get();
            needColumns = query.includeAll();
            for (String s : query.value()) {
                String column = s.trim();
                if (tableInfo.getColumnInfo().containsKey(column))
                    sqlTool.selectCol(tableInfo.getColumnInfo().get(column).getSelectFullName());
                else if (tableInfo.getOriginColumn().containsKey(column))
                    sqlTool.selectCol(tableInfo.getOriginColumn().get(column).getSelectFullName());
                else
                    throw new RuntimeException("error query column [" + s + "]");
            }
        }
        if (!needColumns) return sqlTool;

        Optional opColumnIgnore = annotations.stream().filter(annotation -> annotation.annotationType().equals(ColumnIgnore.class)).findFirst();
        if (opColumnIgnore.isPresent()) {
            ColumnIgnore columnIgnore = (ColumnIgnore) opColumnIgnore.get();
            if (columnIgnore.value().length > 0) {
                sqlTool.select(tableInfo.getTableName());
                List ignorelist = Arrays.asList(columnIgnore.value());
                tableInfo.getColumnInfo().entrySet().forEach(column -> {
                    if (!"*".equals(column.getKey()) && !ignorelist.contains(column.getValue().getColumnName())) {
                        sqlTool.selectCol(column.getValue().getSelectFullName());
                    }
                });
            } else {
                sqlTool.select(tableInfo.getTableName(), tableInfo.get("*").getSelectFullName());
            }
        } else {
            sqlTool.select(tableInfo.getTableName(), tableInfo.get("*").getSelectFullName());
        }
        return sqlTool;
    }

}