@Configuration

public class DruidConfig {

    @ConditionalOnClass(DruidDataSource.class)

    @ConditionalOnProperty(name = "spring.datasource.type",

            havingValue = "com.alibaba.druid.pool.DruidDataSource",

            matchIfMissing = true)

    static class Druid extends DruidConfig{

        @Bean

        @ConfigurationProperties("spring.datasource.druid")

        public DruidDataSource dataSource(DataSourceProperties properties){

            DruidDataSource druidDataSource = (DruidDataSource) properties.initializeDataSourceBuilder()

                    .type(DruidDataSource.class).build();

            DatabaseDriver databaseDriver = DatabaseDriver.fromJdbcUrl(properties.determineUrl());

            String validationQuery = databaseDriver.getValidationQuery();

            if(validationQuery != null){

                druidDataSource.setValidationQuery(validationQuery);

            }

            return druidDataSource;

        }

        @Bean

        public WallFilter wallFilter(){

            WallFilter wallFilter = new WallFilter();

            wallFilter.setConfig(wallConfig());

            return wallFilter;

        }

        @Bean

        public WallConfig wallConfig(){

            WallConfig wallConfig = new WallConfig();

            wallConfig.setMultiStatementAllow(true);//允许一次执行多条语句

            wallConfig.setNoneBaseStatementAllow(true);//允许一次执行多条语句

            return wallConfig;

        }

    }

}