Spring jpa-перехватчик для производных запросов - PullRequest
Новая
       6

Spring jpa-перехватчик для производных запросов

0 голосов
/ 04 февраля 2020

Я хочу перехватывать производные от Spring jpa запросы перед выполнением. я попытался с EmptyInterceptor и StatementInspector, но они перехватывают запросы, аннотированные @query.

1 Ответ

0 голосов
/ 04 февраля 2020

Я рекомендую net .ttddyy: datasource-proxy: 

implementation 'net.ttddyy:datasource-proxy:1.5.1'

Вот PostProcessor & Interceptor: 

@Component
@Slf4j
public class DatasourceProxyBeanPostProcessor implements BeanPostProcessor {

  @Override
  public Object postProcessAfterInitialization(Object bean, String beanName) {
    if (bean instanceof DataSource && !(bean instanceof ProxyDataSource)) {
      log.info("DataSource bean has been found: " + bean);
      final ProxyFactory factory = new ProxyFactory(bean);
      factory.setProxyTargetClass(true);
      factory.addAdvice(new ProxyDataSourceInterceptor((DataSource) bean));
      return factory.getProxy();
    }
    return bean;
  }

 @Override
 public Object postProcessBeforeInitialization(Object bean, String beanName) {
   return bean;
 }

 private static class ProxyDataSourceInterceptor implements MethodInterceptor {
   private final DataSource dataSource;

    public ProxyDataSourceInterceptor(final DataSource dataSource) {
     this.dataSource = ProxyDataSourceBuilder.create(dataSource)
      .name("MyServiceDS")
      .logQueryBySlf4j(SLF4JLogLevel.DEBUG)
      .multiline()
      .build();
    }

    @Override
    public Object invoke(final MethodInvocation invocation) throws Throwable {
      final Method proxyMethod = ReflectionUtils.findMethod(this.dataSource.getClass(),
          invocation.getMethod().getName());
      if (proxyMethod != null) {
        return proxyMethod.invoke(this.dataSource, invocation.getArguments());
      }
      return invocation.proceed();
    }

  }
}

Для получения дополнительной информации: https://github.com/ttddyy/datasource-proxy

...