Применение пользовательских советов по аннотации к репозиторию jpa данных spring

Я работаю над репликацией master-slave mysql. Я использую весенние данные jpa (весенняя загрузка).

Мне нужно было, чтобы все операции записи выполнялись на главном сервере, а операции только для чтения были равномерно распределены между несколькими ведомыми устройствами только для чтения.

Для этого мне нужно:

Используйте специальный драйвер JDBC: com.mysql.jdbc.ReplicationDriver

Установить репликацию: в URL:

spring:
    datasource:
        driverClassName: com.mysql.jdbc.ReplicationDriver
        url: jdbc:mysql:replication://127.0.0.1:3306,127.0.0.1:3307/MyForum?user=root&password=password&autoReconnect=true
        test-on-borrow: true
        validation-query: SELECT 1
    database: MYSQL

Автофиксация должна быть отключена. (*) Соединение должно быть установлено только для чтения.

Чтобы убедиться, что для JDBC Connection установлено значение только для чтения, я создал аннотацию и простой перехватчик AOP.

Аннотация

package com.xyz.forum.replication;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target; 

/**
 * Created by Bhupati Patel on 02/11/15.
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface ReadOnlyConnection {
}

Перехватчик

package com.xyz.forum.replication;

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.Session;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.persistence.EntityManager;

/**
 * Created by Bhupati Patel on 02/11/15.
 */

@Aspect
@Component
public class ConnectionInterceptor {

    private Logger logger;

    public ConnectionInterceptor() {
        logger = LoggerFactory.getLogger(getClass());
        logger.info("ConnectionInterceptor Started");
    }

    @Autowired
    private EntityManager entityManager;

    @Pointcut("@annotation(com.xyz.forum.replication.ReadOnlyConnection)")
    public void inReadOnlyConnection(){}


    @Around("inReadOnlyConnection()")
    public Object proceed(ProceedingJoinPoint pjp) throws Throwable {
        Session session = entityManager.unwrap(Session.class);
        ConnectionReadOnly readOnlyWork = new ConnectionReadOnly();

        try{
            session.doWork(readOnlyWork);
            return pjp.proceed();
        } finally {
            readOnlyWork.switchBack();
        }
    }

}

Ниже приведен мой репозиторий весенних данных.

package com.xyz.forum.repositories;

import com.xyz.forum.entity.Topic;
import org.springframework.data.repository.Repository;

import java.util.List;

/**
 * Created by Bhupati Patel on 16/04/15.
 */
public interface TopicRepository extends Repository<Topic,Integer>{
    Topic save(Topic topic);
    Topic findByTopicIdAndIsDeletedFalse(Integer topicId);
    List<Topic> findByIsDeletedOrderByTopicOrderAsc(Boolean isDelete);

}

Ниже приведен мой класс Manager(Service).

package com.xyz.forum.manager;

import com.xyz.forum.domain.entry.impl.TopicEntry;

import com.xyz.forum.domain.exception.impl.AuthException;

import com.xyz.forum.domain.exception.impl.NotFoundException;
import com.xyz.forum.entity.Topic;
import com.xyz.forum.replication.ReadOnlyConnection;
import com.xyz.forum.repositories.TopicRepository;
import com.xyz.forum.utils.converter.TopicConverter;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;

/**
 * Created by Bhupati Patel on 16/04/15.
 */
@Repository
public class TopicManager {
    @Autowired
    TopicRepository topicRepository;

    @Transactional
    public TopicEntry save(TopicEntry topicEntry) {
        Topic topic = TopicConverter.fromEntryToEntity(topicEntry);
        return TopicConverter.fromEntityToEntry(topicRepository.save(topic));
    }

    @ReadOnlyConnection
    public TopicEntry get(Integer id) {
        Topic topicFromDb = topicRepository.findByTopicIdAndIsDeletedFalse(id);
        if(topicFromDb == null) {
            throw new NotFoundException("Invalid Id", "Topic Id [" + id + "] doesn't exist ");
        }
        return TopicConverter.fromEntityToEntry(topicFromDb);
    }
}

В приведенном выше коде аннотация @ReadOnlyConnection указана на уровне менеджера или сервиса. Приведенные выше фрагменты кода у меня отлично работают. Это тривиальный случай, когда на сервисном уровне я только читаю из подчиненной базы данных и записываю в главную базу данных.

Сказав, что мое фактическое требование состоит в том, что я должен иметь возможность использовать @ReadOnlyConnection на самом уровне репозитория, потому что у меня довольно много бизнес-логики, где я выполняю операции чтения/записи в других классах уровня обслуживания. Поэтому я не могу поместить @ReadOnlyConnection на сервисном уровне.

Я должен быть в состоянии использовать что-то вроде этого

public interface TopicRepository extends Repository<Topic,Integer>{
    Topic save(Topic topic);
    @ReadOnlyConnection
    Topic findByTopicIdAndIsDeletedFalse(Integer topicId);
    @ReadOnlyConnection
    List<Topic> findByIsDeletedOrderByTopicOrderAsc(Boolean isDelete);

}

Например, аннотация Spring @Transactional или @Modifying или @Query. Ниже приведен пример того, что я имею в виду.

    public interface AnswerRepository extends Repository<Answer,Integer> {
    @Transactional
    Answer save(Answer answer);

    @Transactional
    @Modifying
    @Query("update Answer ans set ans.isDeleted = 1, ans.deletedBy = :deletedBy, ans.deletedOn = :deletedOn " +
            "where ans.questionId = :questionId and ans.isDeleted = 0")
    void softDeleteBulkAnswers(@Param("deletedBy") String deletedBy, @Param("deletedOn") Date deletedOn,
                               @Param("questionId") Integer questionId);
}

Я новичок в мире aspectj и aop, я пробовал довольно много регулярных выражений pointcut в ConnectionInterceptor, но ни одно из них не сработало. Я пробовал это с давних пор, но пока не повезло.

Как выполнить поставленную задачу.


person Bhupati Patel    schedule 04.11.2015    source источник
comment
Вы добавили @EnableAspectJAutoProxy в свою конфигурацию?   -  person px5x2    schedule 04.11.2015
comment
Да. Я сделал . Я упомянул Выше фрагменты кода отлично работает для меня. Я не поместил здесь весь код. Я поместил только важный фрагмент кода.   -  person Bhupati Patel    schedule 04.11.2015
comment
Привет, Бхупати, ты смог достичь своей цели, указав аннотацию @Transactional внутри классов репозитория ?? Если да, то какие изменения вы сделали. Не могли бы вы опубликовать решение здесь, чтобы помочь другим (например, мне: P). Спасибо   -  person Ankur Garg    schedule 16.11.2015
comment
Привет, @AnkurGarg, я опубликовал свой ответ. Надеюсь, это поможет вам.   -  person Bhupati Patel    schedule 17.11.2015


Ответы (2)


Я не мог найти обходной путь для моей пользовательской аннотации @ReadOnlyConnection (например, @Transactional) на уровне метода, но у меня сработал небольшой черт.

Я вставляю фрагмент кода ниже.

@Aspect
@Component
@EnableAspectJAutoProxy
public class ConnectionInterceptor {

    private Logger logger;
    private static final String JPA_PREFIX = "findBy";
    private static final String CUSTOM_PREFIX = "read";

    public ConnectionInterceptor() {
        logger = LoggerFactory.getLogger(getClass());
        logger.info("ConnectionInterceptor Started");
    }

    @Autowired
    private EntityManager entityManager;

    @Pointcut("this(org.springframework.data.repository.Repository)")
    public void inRepositoryLayer() {}

    @Around("inRepositoryLayer()")
    public Object proceed(ProceedingJoinPoint pjp) throws Throwable {
        String methodName = pjp.getSignature().getName();
        if (StringUtils.startsWith(methodName, JPA_PREFIX) || StringUtils.startsWith(methodName, CUSTOM_PREFIX)) {
            System.out.println("I'm there!" );
            Session session = entityManager.unwrap(Session.class);
            ConnectionReadOnly readOnlyWork = new ConnectionReadOnly();

            try{
                session.doWork(readOnlyWork);
                return pjp.proceed();
            } finally {
                readOnlyWork.switchBack();
            }
        }
        return pjp.proceed();
    }
}

Итак, в приведенном выше коде я использую pointcut, как показано ниже.

@Pointcut("this(org.springframework.data.repository.Repository)")
public void inRepositoryLayer() {}

и что он делает

любая точка соединения (выполнение метода только в Spring AOP), где прокси реализует интерфейс репозитория

Вы можете посмотреть его на http://docs.spring.io/spring/docs/current/spring-framework-reference/html/aop.html

Теперь все мои методы запроса на чтение репозитория начинаются либо с префикса «findByXXX» (метод для чтения spring-data-jpa по умолчанию) или «readXXX» (пользовательский метод чтения с аннотацией @Query), который в моих методах выполнения соответствует вышеуказанному pointcut. В соответствии с моим требованием я устанавливаю значение JDBC Connection только для чтения.

Session session = entityManager.unwrap(Session.class);
ConnectionReadOnly readOnlyWork = new ConnectionReadOnly();

И мой ConnectionReadOnly выглядит следующим образом

package com.xyz.forum.replication;

import org.hibernate.jdbc.Work;

import java.sql.Connection;
import java.sql.SQLException;

/**
 * Created by Bhupati Patel on 04/11/15.
 */
public class ConnectionReadOnly implements Work {

    private Connection connection;
    private boolean autoCommit;
    private boolean readOnly;

    @Override
    public void execute(Connection connection) throws SQLException {
        this.connection = connection;
        this.autoCommit = connection.getAutoCommit();
        this.readOnly = connection.isReadOnly();
        connection.setAutoCommit(false);
        connection.setReadOnly(true);
    }

    //method to restore the connection state before intercepted
    public void switchBack() throws SQLException{
        connection.setAutoCommit(autoCommit);
        connection.setReadOnly(readOnly);
    }
}

Таким образом, вышеуказанные настройки работают для моего требования.

person Bhupati Patel    schedule 17.11.2015

кажется, что @Pointcut && @Around следует объявить следующим образом:

@Pointcut(value = "execution(public * *(..))")
public void anyPublicMethod() {
}

@Around("@annotation(readOnlyConnection)")
person axel    schedule 14.09.2020