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

Go to download
package com.dream.system.antlr.invoker;

import com.dream.antlr.config.Assist;
import com.dream.antlr.exception.AntlrException;
import com.dream.antlr.handler.Handler;
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.sql.ToSQL;
import com.dream.system.antlr.handler.page.PageHandler;
import com.dream.system.config.MethodInfo;

import java.util.List;

public class LimitInvoker extends AbstractInvoker {
    public static final String FUNCTION = "limit";
    PageHandler pageHandler;

    @Override
    public Handler[] handler() {
        return new Handler[]{pageHandler};
    }

    @Override
    protected String invoker(InvokerStatement invokerStatement, Assist assist, ToSQL toSQL, List invokerList) throws AntlrException {
        MethodInfo methodInfo = assist.getCustom(MethodInfo.class);
        pageHandler = new PageHandler(this, methodInfo);
        Statement[] columnList = ((ListColumnStatement) invokerStatement.getParamStatement()).getColumnList();
        pageHandler.setParamList(columnList[1], columnList[2], false);
        String sql = toSQL.toStr(columnList[0], assist, invokerList);
        invokerStatement.replaceWith(columnList[0]);
        return sql;
    }

    @Override
    public Invoker newInstance() {
        return new LimitInvoker();
    }

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