Java WallFilter类代码示例

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
@Bean
@ConditionalOnProperty(prefix = DRUID_WALL_FILTER_PREFIX, name = "enabled", havingValue = "true")
@ConfigurationProperties(DRUID_WALL_FILTER_PREFIX)
public WallFilter wallFilter(WallConfig wallConfig) {
    log.debug("druid wall-filter init...");
    WallFilter filter = new WallFilter();
    filter.setConfig(wallConfig);
    return filter;
}
 

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
public AutoGen init(String configPath) throws IOException {
    if(configPath == null || configPath.length() == 0){
        configPath = PathKit.getRootClassPath() + "/defaultRule.json" ;
    }
    config = getConfig(configPath) ;
    if(config == null){
        throw new NullPointerException("No Config.");
    }
    JSONObject dataSourceConfig = config.getJSONObject("dataSource") ;
    dp = new DruidPlugin(
            dataSourceConfig.getString("url"),
            dataSourceConfig.getString("user"),
            dataSourceConfig.getString("pwd")
    );
    dp.addFilter(new StatFilter());
    WallFilter wall = new WallFilter();
    wall.setDbType("mysql");
    WallConfig wallConfig = new WallConfig();
    wallConfig.setFunctionCheck(false);
    wallConfig.setCreateTableAllow(true);
    wallConfig.setCommentAllow(true);
    wallConfig.setMultiStatementAllow(true);
    wall.setConfig(wallConfig);
    dp.addFilter(wall);
    arp = new ActiveRecordPlugin(dp);
    arp.setContainerFactory(new CaseInsensitiveContainerFactory(true));// 大小写不敏感
    arp.setDialect(new MysqlDialect());
    dp.start();
    arp.start();
    resourceLoader = new FileResourceLoader(config.getString("baseTemplatePath"),"utf-8");
    cfg = Configuration.defaultConfiguration();
    gt = new GroupTemplate(resourceLoader, cfg);
    System.out.println("-- init success .");
    return this ;
}
 

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
/**
 * 创建数据源，有前缀
 *
 * @param dataSourceMap
 * @param prefix
 * @return
 * @throws SQLException
 */
private DruidDataSource buildDruidDataSource(Map<String, Object> dataSourceMap, String prefix) throws SQLException {
    DruidDataSource dataSource = new DruidDataSource();
    dataSource.setName(MapUtils.getString(dataSourceMap, prefix + "name"));
    dataSource.setDriverClassName(MapUtils.getString(dataSourceMap, prefix + "driverClass"));
    dataSource.setUrl(MapUtils.getString(dataSourceMap, prefix + "url"));
    dataSource.setUsername(MapUtils.getString(dataSourceMap, prefix + "username"));
    dataSource.setPassword(MapUtils.getString(dataSourceMap, prefix + "password"));

    // 配置过滤器
    WallFilter wallFilter = new WallFilter();
    StatFilter statFilter = new StatFilter();
    statFilter.setSlowSqlMillis(5000);
    statFilter.setLogSlowSql(true);
    statFilter.setMergeSql(true);
    Slf4jLogFilter slf4jLogFilter = new Slf4jLogFilter();
    slf4jLogFilter.setDataSourceLogEnabled(false);
    slf4jLogFilter.setConnectionLogEnabled(false);
    slf4jLogFilter.setConnectionLogErrorEnabled(true);
    slf4jLogFilter.setResultSetLogEnabled(false);
    slf4jLogFilter.setResultSetLogErrorEnabled(true);
    slf4jLogFilter.setStatementLogEnabled(false);
    slf4jLogFilter.setStatementExecutableSqlLogEnable(true);
    slf4jLogFilter.setStatementSqlFormatOption(new SQLUtils.FormatOption(false, false));

    List filterList = new ArrayList();
    filterList.add(wallFilter);
    filterList.add(statFilter);
    filterList.add(slf4jLogFilter);
    dataSource.setProxyFilters(filterList);

    return dataSource;
}
 

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
public WallProviderStatValue getWallStatValue(boolean reset) {
    for (Filter filter : this.filters) {
        if (filter instanceof WallFilter) {
            WallFilter wallFilter = (WallFilter) filter;
            return wallFilter.getProvider().getStatValue(reset);
        }
    }

    return null;
}
 

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
@Override
public void initialize() throws SQLException {
    Prop prop = PropKit.use(DB_CONFIG);
    druidPlugin = new DruidPlugin(prop.get("jdbcUrl"), prop.get("user"), prop.get("password"));

    // StatFilter提供JDBC层的统计信息
    druidPlugin.addFilter(new StatFilter());
    // WallFilter的功能是防御SQL注入攻击
    WallFilter wallFilter = new WallFilter();
    wallFilter.setDbType("mysql");
    druidPlugin.addFilter(wallFilter);
    druidPlugin.start();
}
 

import com.alibaba.druid.wall.WallFilter; //导入依赖的package包/类
/**
 * DruidPlugin
 * @param prop ： property
 * @return
 */
private DruidEncryptPlugin getDruidPlugin(String ds) {
	this.loadPropertyFile();
	String url = this.getProperty(String.format("db.%s.url", ds));
	url = String.format(URL_TEMPLATE, ds, url);
	String endsWith = "?characterEncoding=UTF8&zeroDateTimeBehavior=convertToNull";
	if (!url.endsWith(endsWith)) {
		url += endsWith;
	}
	DruidEncryptPlugin dp = new DruidEncryptPlugin(url,
			this.getProperty(String.format(USER_TEMPLATE, ds)),
			this.getProperty(String.format(PASSWORD_TEMPLATE, ds)));
	dp.setInitialSize(this.getPropertyToInt(String.format(INITSIZE_TEMPLATE, ds)));
	dp.setMaxActive(this.getPropertyToInt(String.format(MAXSIZE_TEMPLATE, ds)));
	dp.addFilter(new StatFilter());
	WallFilter wall = new WallFilter();
	wall.setDbType(ds);
	dp.addFilter(wall);
	
	if (this.geRuned) {
		dp.start();
		BaseModelGeneratorExt baseGe = new BaseModelGeneratorExt(this.getBaseModelPackage(), this.getBaseModelOutDir());
		baseGe.setGenerateTableNameInModel(this.getGeTableNameInModel());
		baseGe.setGenerateTableColumnNameInModel(this.getGeTableColumnName());
		ModelGenerator modelGe = new ModelGenerator(this.getModelPackage(), this.getBaseModelPackage(), this.getModelOutDir());
		modelGe.setGenerateDaoInModel(this.getGeDaoInModel());
		Generator ge = new Generator(dp.getDataSource(), baseGe, modelGe);
		MappingKitGeneratorExt mappingKitGe = new MappingKitGeneratorExt(this.getModelPackage(), this.getModelOutDir());
		if (!JFinalConfigExt.DEFAULT_MAPPINGKIT_CLASS_NAME.equals(this.getMappingKitClassName())) {
			mappingKitGe.setMappingKitClassName(this.getMappingKitClassName());
		}
		mappingKitGe.setGenerateMappingArpKit(this.getGeMappingArpKit());
		mappingKitGe.setGenerateTableMapping(this.getGeTableMapping());
		ge.setMappingKitGenerator(mappingKitGe);
		ge.setGenerateDataDictionary(this.getGeDictionary());
		ge.generate();
	}
	
	return dp;
}