cn.mybatisboost.lang.LanguageSqlProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-boost-core Show documentation

The newest version!

package cn.mybatisboost.lang;

import cn.mybatisboost.core.Configuration;
import cn.mybatisboost.core.ConfigurationAware;
import cn.mybatisboost.core.SqlProvider;
import cn.mybatisboost.lang.provider.*;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.reflection.MetaObject;

import java.sql.Connection;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class LanguageSqlProvider implements SqlProvider {

    private Configuration configuration;
    private List providers;

    public LanguageSqlProvider(Configuration configuration) {
        this.configuration = configuration;
        initProviders();
    }

    protected void initProviders() {
        providers = Collections.unmodifiableList(Arrays.asList(new InsertEnhancement(), new UpdateEnhancement(),
                new TableEnhancement(), new NullEnhancement(), new ListParameterEnhancement()));
        for (SqlProvider p : providers) {
            if (p instanceof ConfigurationAware) {
                ((ConfigurationAware) p).setConfiguration(configuration);
            }
        }
    }

    @Override
    public void replace(Connection connection, MetaObject metaObject, MappedStatement mappedStatement, BoundSql boundSql) {
        providers.forEach(p -> p.replace(connection, metaObject, mappedStatement, boundSql));
    }
}