com.dream.system.antlr.invoker.ForEachInvoker Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dream-orm-system Show documentation
The newest version!
package com.dream.system.antlr.invoker;

import com.dream.antlr.config.Assist;
import com.dream.antlr.exception.AntlrException;
import com.dream.antlr.invoker.AbstractInvoker;
import com.dream.antlr.invoker.Invoker;
import com.dream.antlr.smt.InvokerStatement;
import com.dream.antlr.smt.ListColumnStatement;
import com.dream.antlr.smt.Statement;
import com.dream.antlr.smt.SymbolStatement;
import com.dream.antlr.sql.ToSQL;
import com.dream.util.common.ObjectWrapper;

import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Map;

public class ForEachInvoker extends AbstractInvoker {
    public static final String FUNCTION = "foreach";
    private final String cut = ",";
    private final String index = "index";
    private final String item = "item";

    @Override
    public String invoker(InvokerStatement invokerStatement, Assist assist, ToSQL toSQL, List invokerList) throws AntlrException {
        Statement[] columnList = ((ListColumnStatement) invokerStatement.getParamStatement()).getColumnList();
        int len = columnList.length;
        if (len > 2 || len < 1) {
            throw new AntlrException("@foreach参数个数不合法");
        }
        String list = toSQL.toStr(columnList[0], assist, invokerList);
        ObjectWrapper paramWrapper = assist.getCustom(ObjectWrapper.class);
        Object arrayList = paramWrapper.get(list);
        if (arrayList == null) {
            return null;
        }
        boolean isArray = false;
        if (arrayList instanceof Collection || (isArray = arrayList.getClass().isArray())) {
            Collection collection;
            if (isArray) {
                collection = Arrays.asList((Object[]) arrayList);
            } else {
                collection = (Collection) arrayList;
            }
            ListColumnStatement listColumnStatement = new ListColumnStatement(cut);
            if (len == 2) {
                int index = 0;
                Map paramMap = (Map) paramWrapper.getObject();
                Statement[] statements = new Statement[collection.size()];
                for (Object item : collection) {
                    paramMap.put(this.index, index);
                    paramMap.put(this.item, item);
                    String letterItem = toSQL.toStr(columnList[1], assist, invokerList);
                    statements[index] = new SymbolStatement.LetterStatement(letterItem);
                    index++;
                }
                listColumnStatement.add(statements);
                paramMap.remove(this.index);
                paramMap.remove(this.item);
            } else {
                MarkInvoker sqlInvoker = (MarkInvoker) assist.getInvoker(MarkInvoker.FUNCTION, Invoker.DEFAULT_NAMESPACE);
                List paramInfoList = sqlInvoker.getParamInfoList();
                int index = 0;
                Statement[] statements = new Statement[collection.size()];
                for (Object item : collection) {
                    statements[index] = new SymbolStatement.MarkStatement();
                    paramInfoList.add(new MarkInvoker.ParamInfo(list + "[" + index + "]", item));
                    index++;
                }
                listColumnStatement.add(statements);

            }
            listColumnStatement.setParentStatement(invokerStatement.getParentStatement());
            return toSQL.toStr(listColumnStatement, assist, invokerList);
        } else {
            throw new AntlrException("类'" + arrayList.getClass().getName() + "'不是集合或数组类型");
        }
    }

    @Override
    public String function() {
        return FUNCTION;
    }
}