Postgres Hibernate 为行级安全设置会话变量

Question

我无法找到有关我遇到的这个问题的信息。我有兴趣在我的 Postgres 数据库上实现行级安全性，我正在寻找一种能够通过某种形式的拦截器自动设置 postgres 会话变量的方法。现在，我知道通过休眠，您可以使用@Filter 和@FilterDef 进行行级安全，但是我想另外在我的数据库上设置策略。

执行此操作的一个非常简单的方法是在每次查询之前执行 SQL 语句 SET variable=value，尽管我无法找到任何相关信息。

这是在 spring-boot 应用程序上使用的，并且每个请求都可以访问variable 的特定于请求的值。

Answer 1

由于您的应用程序使用 spring，您可以尝试通过以下几种方式之一来完成此操作：

Spring AOP

在这种方法中，您编写了一个建议，要求 Spring 将其应用于特定方法。如果您的方法使用@Transactional 注释，您可以在事务开始后立即将建议应用于那些。

扩展 TransactionManager 实现

假设您的交易使用JpaTransactionManager。

public class SecurityPolicyInjectingJpaTransactionManager extends JpaTransactionManager {
  @Autowired
  private EntityManager entityManager;

  // constructors

  @Override
  protected void prepareSynchronization(DefaultTransactionStatus status, TransactionDefinition definition) {
    super.prepareSynchronization(status, definition);
    if (status.isNewTransaction()) {
      // Use entityManager to execute your database policy param/values

      // I would suggest you also register an after-completion callback synchronization
      // This after-completion would clear all the policy param/values
      // regardless of whether the transaction succeeded or failed 
      // since this happens just before it gets returned to the connection pool
    }
  }
}

现在只需配置您的 JPA 环境以使用您的自定义 JpaTransactionManager 类。

---

可能还有其他人，但这是我探索过的两个。