repository 拦截sql

如果你使用的是Spring Data JPA，你可以使用Spring提供的JPA拦截器来拦截SQL语句。JPA拦截器是用于拦截JPA执行过程中所有方法调用的组件。通过实现org.hibernate.Interceptor接口，你可以自定义JPA执行过程中的行为。

在拦截器中，你可以使用org.hibernate.SQLQuery来获取当前执行的SQL语句，然后对其进行修改或记录。下面是一个简单的拦截器示例，它记录了执行的SQL语句：

public class SqlInterceptor extends EmptyInterceptor {

    @Override
    public String onPrepareStatement(String sql) {
        // 记录或修改SQL语句
        System.out.println("执行SQL：" + sql);
        return super.onPrepareStatement(sql);
    }

}

你可以在Spring Data JPA的repository中配置此拦截器，使其生效。例如，假设你有一个UserRepository类，你可以像这样配置它：

@Repository
public interface UserRepository extends JpaRepository<User, Long> {

    @Query(value = "SELECT * FROM users WHERE id = ?1", nativeQuery = true)
    User findById(long id);

    @Query(value = "INSERT INTO users(name, age) VALUES(?1, ?2)", nativeQuery = true)
    @Modifying
    void insertUser(String name, int age);

    @Query(value = "UPDATE users SET name=?1, age=?2 WHERE id=?3", nativeQuery = true)
    @Modifying
    void updateUser(String name, int age, long id);

    @Query(value = "DELETE FROM users WHERE id=?1", nativeQuery = true)
    @Modifying
    void deleteUser(long id);
    
}

在这个例子中，我们使用@Query注释和nativeQuery=true属性来执行原生SQL查询。你可以在拦截器中拦截这些语句，例如：

public class SqlInterceptor extends EmptyInterceptor {

    @Override
    public String onPrepareStatement(String sql) {
        // 记录或修改SQL语句
        System.out.println("执行SQL：" + sql);
        return super.onPrepareStatement(sql);
    }

    @Override
    public void afterTransactionCompletion(Transaction tx) {
        // 清除线程上下文中的拦截器
        TransactionSynchronizationManager.unbindResource(this);
        super.afterTransactionCompletion(tx);
    }

}

你可以将拦截器添加到Spring上下文中，以便在任何使用UserRepository的地方都可以拦截SQL语句。例如，在Spring Boot中，你可以将拦截器添加到应用程序上下文中：

@Configuration
public class JpaConfig {

    @Autowired
    private EntityManagerFactory entityManagerFactory;

    @Bean
    public SqlInterceptor sqlInterceptor() {
        return new SqlInterceptor();
    }

    @Bean
    public JpaTransactionManager transactionManager() {
        JpaTransactionManager transactionManager = new JpaTransactionManager();
        transactionManager.setEntityManagerFactory(entityManagerFactory);
        return transactionManager;
    }

    @Bean
    public PersistenceExceptionTranslationPostProcessor exceptionTranslation() {
        return new PersistenceExceptionTranslationPostProcessor();
    }

    @PostConstruct
    public void registerInterceptor() {
        SessionFactory sessionFactory = entityManagerFactory.unwrap(SessionFactory.class);
        Interceptor[] interceptors = {sqlInterceptor()};
        sessionFactory.setInterceptors(interceptors);
    }

}

这将确保在任何使用UserRepository的地方，SqlInterceptor将拦截所有SQL语句并记录它们。