com.github.mybatisintercept.ValueGetterImpl Maven / Gradle / Ivy

Go to download
package com.github.mybatisintercept;

import com.github.mybatisintercept.util.BeanMap;
import com.github.mybatisintercept.util.MybatisUtil;
import com.github.mybatisintercept.util.TypeUtil;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.plugin.Interceptor;

import java.util.Map;

public class ValueGetterImpl implements InterceptContext.ValueGetter {
    private final InterceptContext interceptContext;
    private boolean initMybatisParameter;
    private Map mybatisParameterGetter;

    public ValueGetterImpl(InterceptContext interceptContext) {
        this.interceptContext = interceptContext;
    }

    @Override
    public  T getProviderValue(String name, Class type) {
        Object value = interceptContext.getValueProvider().invokeWithOnBindContext(name, interceptContext);
        return cast(value, type);
    }

    @Override
    public  T getMybatisParameterValue(String name, Class type) {
        if (!initMybatisParameter) {
            this.mybatisParameterGetter = createMybatisParameterGetter(interceptContext.getParameter());
            this.initMybatisParameter = true;
        }
        Object value = mybatisParameterGetter != null && mybatisParameterGetter.containsKey(name) ? mybatisParameterGetter.get(name) : null;
        return cast(value, type);
    }

    @Override
    public  T getMybatisBoundSqlValue(String name, Class type) {
        BoundSql boundSql = MybatisUtil.getBoundSql(interceptContext.getInvocation());
        Object value = boundSql.hasAdditionalParameter(name) ? boundSql.getAdditionalParameter(name) : null;
        return cast(value, type);
    }

    @Override
    public  T getInterceptAttributeValue(String name, Class type) {
        Object value = interceptContext.getAttributeValue(name);
        return cast(value, type);
    }

    @Override
    public Object getCompileValue(String name) {
        return getCompileValue(name, null);
    }

    @Override
    public  T getCompileValue(String name, Class type) {
        T value = getProviderValue(name, type);
        if (value == null) {
            value = getInterceptAttributeValue(name, type);
        }
        return value;
    }

    protected Map createMybatisParameterGetter(Object mybatisParameter) {
        return MybatisUtil.isInstanceofKeyValue(mybatisParameter) ? BeanMap.toMap(mybatisParameter) : null;
    }

    protected  T cast(Object value, Class type) {
        return type == null || type == Object.class ? (T) value : TypeUtil.cast(value, type);
    }

}