Druid连接池的自定义过滤功能实现方法

package com.example;

import com.alibaba.druid.filter.FilterAdapter;
import com.alibaba.druid.filter.FilterChain;
import com.alibaba.druid.proxy.jdbc.StatementProxy;

import java.sql.SQLException;

public class LogFilter extends FilterAdapter {
    @Override
    public void statement_executeBefore(FilterChain chain, StatementProxy statement) throws SQLException {
        String sql = statement.getBatchSql();
        
        // 这里可以根据实际需求判断是否需要记录日志
        if (shouldLog(sql)) {
            logSql(sql);
        }
        chain.statement_executeBefore(statement);
    }

    private boolean shouldLog(String sql) {
        // 实现判断逻辑，如检查是否包含特定表名
        return sql.contains("sensitive_table");
    }

    private void logSql(String sql) {
        // 实际的日志记录逻辑
        System.out.println("Executing SQL: " + sql);
    }
}

import com.alibaba.druid.filter.FilterEventAdapter;
import com.alibaba.druid.proxy.jdbc.StatementProxy;
import com.alibaba.druid.proxy.jdbc.ResultSetProxy;

public class CustomDruidFilter extends FilterEventAdapter {

    @Override
    protected void statementExecuteBefore(StatementProxy statement, String sql) {
        // 记录SQL执行前的日志
        logger.info("Before SQL Execute: " + sql);
    }

    @Override
    protected void statementExecuteAfter(StatementProxy statement, String sql, boolean result) {
        // 记录SQL执行后的日志
        logger.info("After SQL Execute: " + sql + " , result: " + result);
    }

    @Override
    protected void resultSetOpenAfter(ResultSetProxy resultSet) {
        // 当结果集打开后的处理
        logger.info("ResultSet Opened: " + resultSet.getRawObject());
    }

    @Override
    protected void resultSetCloseAfter(ResultSetProxy resultSet) {
        // 当结果集关闭后的处理
        logger.info("ResultSet Closed: " + resultSet.getRawObject());
    }

    // ... 其他事件处理 ...
}

<bean id="dataSource" class="com.alibaba.druid.pool.DruidDataSource" ...>
    <property name="proxyFilters">
        <list>
            <ref bean="logFilter"/>
        </list>
    </property>
</bean>

<bean id="logFilter" class="com.example.LogFilter"/>

import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.util.Collections;

@Configuration
public class DruidConfig {

    @Bean
    public DruidDataSource dataSource() {
        DruidDataSource dataSource = new DruidDataSource();
        // 配置数据源参数
        // ...
        // 添加自定义过滤器
        dataSource.setProxyFilters(Collections.singletonList(new LogFilter()));
        return dataSource;
    }
}

spring.datasource.druid.url=jdbc:mysql://localhost:3306/yourdb
spring.datasource.druid.username=yourusername
spring.datasource.druid.password=yourpassword
# 启用配置过滤器
spring.datasource.druid.filter.config.enabled=true
# 其他Druid配置...

spring:
  datasource:
    druid:
      url: jdbc:mysql://localhost:3306/yourdb
      username: yourusername
      password: yourpassword
      filter:
        config:
          enabled: true
      # 其他Druid配置...