TL;DR: 我使用了Hibernate的@Filter,然后创建了一个Aspect来拦截方法
定义了一个基类实体,其结构如下
OwnerAwareEntity.java
import org.hibernate.annotations.Filter;
import org.hibernate.annotations.FilterDef;
import org.hibernate.annotations.ParamDef;
import javax.persistence.Column;
import javax.persistence.MappedSuperclass;
import java.io.Serializable;
@MappedSuperclass
@FilterDef(name = "ownerFilter", parameters = {@ParamDef(name = "ownerRef", type = "long")})
@Filter(name = "ownerFilter", condition = "OWNER_REF = :ownerRef")
public class OwnerAwareEntity implements Serializable{
@Column(name = "OWNER_REF",nullable = true)
private Long ownerRef;
}
我们在这个实体上设置了过滤器。Hibernate的@Filter允许我们设置一个条件,以附加到select where子句中。
接下来,为OwnerAwareEntity类型的实体定义了一个基础存储库。
OwnerAwareRepository.java
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.repository.NoRepositoryBean;
@NoRepositoryBean
public interface OwnerAwareRepository<T, ID extends java.io.Serializable> extends JpaRepository<T, ID> {
}
我将创建一个切面,它将拦截所有扩展OwnerAwareRepository的存储库中的方法。
OwnerFilterAdvisor.java
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.hibernate.Filter;
import org.hibernate.Session;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
@Aspect
@Component
@Slf4j
public class OwnerFilterAdvisor {
@PersistenceContext
private EntityManager entityManager;
@Pointcut("execution(public * com.xyz.app.repository.OwnerAwareRepository+.*(..))")
protected void ownerAwareRepositoryMethod(){
}
@Around(value = "ownerAwareRepositoryMethod()")
public Object enableOwnerFilter(ProceedingJoinPoint joinPoint) throws Throwable{
Session session = null;
try {
session = entityManager.unwrap(Session.class);
Filter filter = session.enableFilter("ownerFilter");
filter.setParameter("ownerRef", getSessionOwnerRef());
} catch (Exception ex) {
log.error("Error enabling ownerFilter : Reason -" +ex.getMessage());
}
Object obj = joinPoint.proceed();
if ( session != null ) {
session.disableFilter("ownerFilter");
}
return obj;
}
private Long getSessionOwnerRef() {
}
}
该顾问被设置为拦截所有继承OwnerAwareRepository类的方法。在拦截时,从entityManager(当前持久化上下文)获取当前hibernate Session,并使用参数值“ownerRef”启用过滤器。
还创建了一个配置文件,以便扫描该顾问。
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
@Configuration
@ComponentScan(basePackages = {"com.xyz.app.advisors"})
public class AOPConfig {
}
一旦这些文件放置好了,您需要为需要拥有者感知的实体完成以下事项:
- 实体需要扩展OwnerAwareEntity
- 实体存储库类需要扩展OwnerAwareRepository
依赖关系
此设置需要在依赖项中添加spring aop。您可以将以下内容添加到pom.xml中。
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
优点
- 适用于所有选择查询(findBy方法,findAll等)
- @Query方法也会被拦截
- 实现简单
注意事项
- 删除或更新的where子句不受此过滤器影响。
- 如果存储库包含保存/更新/删除方法,并且该方法未标记为@Transactional,则拦截器将出错(在这些情况下,您可以捕获并使方法正常进行)