Spring Boot HiberNate Web应用程序：如何全局拦截本机和托管实体查询？

Question

是否可以在数据库连接或数据源的底层放置一个钩子，并在执行任何其他查询之前执行一个查询？

你好。

我希望能够在Hibernate状态下拦截所有数据库查询，并在原始查询发送到数据库之前注入执行一个简单的查询：SET @SomeSessionVariable = 123346;。

由于hibernate对其管理的实体使用生命周期，所以我发现实现我所需要的是一个挑战。

在我的Spring启动应用程序中，我想使用触发器审核数据更改。我希望能够将更改与Spring应用程序中当前登录的用户相关联，唯一的方法是将该用户的id连同任何查询一起发送到数据库，但就在查询执行之前，以便触发器可以在当前连接会话中看到变量。

我用的是MySQL，似乎没有内置的方式来做到这一点。Postgresql似乎确实支持向DBMS发送客户机上下文。

应用程序非常庞大，需要重构以手动发送id。

你知道我可以在Hibernate配置中放置全局钩子以能够拦截本机和托管查询的任何其他位置吗？

Answer 1

如果您在应用程序中使用 spring-security 来验证用户身份，则可以根据需要使用 spring-data-jpa 审核系统。

例如，对于< code>Product实体，它可能如下所示:

@Entity
@EntityListeners(AuditingEntityListener.class)
public class Product {
    
    // id and other fields
    
    @Column(name = "created_by")
    @CreatedBy
    private String createdBy;

    @Column(name = "modified_by")
    @LastModifiedBy
    private String modifiedBy;
    
    // getters, setters, etc.
}

您还应该在任何配置类上放置< code>@EnableJpaAuditing注释。

通过这样做，您将告诉 spring-data 在保存或更新操作时自动将created_by和modified_by字段插入数据库关系中，使用存储在 SecurityContext 的身份验证对象中的主体的名称以及调用保存或更新的人员。

您可以通过实现审核员感知来更改此默认行为

public class CustomAuditorAware implements AuditorAware<String> {
    @Override
    public Optional<String> getCurrentAuditor() {
        // retrieve name, id or anything else from your SecurityContext
    }
}

然后，您只需将此< code>AuditorAware注册为bean，并将您的应用程序指向它:

@EnableJpaAuditing(auditorAwareRef = "auditor")
@Configuration
public class ApplicationConfig {
    @Bean
    AuditorAware<String> auditor() {
        return new CustomAuditorAware();
    }
}

请注意，此审核机制仅适用于实体，因此不适用于本机查询。还有其他方法可以实现实体审计 - 看看Baeldung的这篇文章

如果您想在执行原生sql查询之前修改它们，可以使用Hibernate的< code>StatementInspector:

public class CustomStatementInspector implements StatementInspector {
    @Override
    public String inspect(String sql) {
        // check if query is modifying and change sql query 
    }
}

然后，您应该让Spring知道您想使用此检查器，并且在Spring Boot中有一种简单的方法可以做到这一点：

@Configuration
public class ApplicationConfig {
    @Bean
    public HibernatePropertiesCustomizer hibernateCustomizer() {
        return (properties) -> properties.put(AvailableSettings.STATEMENT_INSPECTOR, new CustomStatementInspector());
    }
}

其他配置＜code＞StatementInspector＜/code＞的方法可以在这里找到：如何在Hibernate中配置StatementInspection？