com.dream.mate.tenant.invoker.TenantGetInvoker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dream-orm-mate Show documentation

The newest version!

package com.dream.mate.tenant.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.sql.ToSQL;
import com.dream.mate.tenant.inject.TenantHandler;

import java.util.List;

public class TenantGetInvoker extends AbstractInvoker {
    public static final String FUNCTION = "dream_mate_tenant_get";
    private TenantHandler tenantHandler;

    public TenantGetInvoker(TenantHandler tenantHandler) {
        this.tenantHandler = tenantHandler;
    }

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

    @Override
    protected String invoker(InvokerStatement invokerStatement, Assist assist, ToSQL toSQL, List invokerList) throws AntlrException {
        Object tenantObject = tenantHandler.getTenantObject();
        if (tenantObject != null) {
            if (tenantObject instanceof Number) {
                return String.valueOf(tenantObject);
            } else if (tenantObject instanceof String) {
                return "'" + tenantObject + "'";
            } else {
                return tenantObject.toString();
            }
        }
        return "NULL";
    }

    @Override
    protected Handler[] handler() {
        return new Handler[0];
    }
}