WebFlux + R2DBC 을 사용한 simple application

공식 문서

리엑티브 database driver

1. 의존성 설정

spring initializr 를 사용하여 아래 의존성 추가

Spring Reactive Web
Spring Data R2DBC
H2 Database
Spring REST Docs

2. 스프링 자바 bean 설정

AbstractR2dbcConfiguration 클래스를 상속받아서 bean 설정이 가능하다.

io.r2dbc.spi.ConnectionFactory 스프링을 사용하는 컨테이너와 함께 표준 인터페이스 사용가능.
R2DBC 의 ExceptionTranslator 를 @Repository 기반에 DataAccessException 계층 구조로 변환하는 이점을 제공.

@Configuration
public class DataBaseConfig extends AbstractR2dbcConfiguration {

    @Override
    public ConnectionFactory connectionFactory() {
        ConnectionFactory connectionFactory = ConnectionFactories.get(ConnectionFactoryOptions.builder()
                .option(DRIVER,"h2")
                .option(PROTOCOL, "mem")
                .option(DATABASE,"test")
                .build());

        return connectionFactory;
    }
}

3. R2DBC 드라이버 방언

R2DBC 드라이버는 스프링의 DialectResolver가 SpringFactories 를 사용하여 클래스패스의 방언들을 검색 및 로딩함.

4. R2DBC Repository

@EnableR2dbcRepositories 적용

5. R2DBC Transaction manager

@EnableTransactionManagement 적용 후 ReactiveTransactionManager 을 구현한 R2dbcTransactionManager bean 등록

@Configuration
@EnableR2dbcRepositories
@EnableTransactionManagement // annotation 적용
public class DataBaseConfig extends AbstractR2dbcConfiguration {

    @Override
    public ConnectionFactory connectionFactory() {

        ConnectionFactory connectionFactory = ConnectionFactories.get(ConnectionFactoryOptions.builder()
                .option(DRIVER,"h2")
                .option(PROTOCOL, "mem")
                .option(DATABASE,"test")
                .build());

        return connectionFactory;
    }

    
    @Bean
    public ReactiveTransactionManager transactionManager(ConnectionFactory connectionFactory){
        return new R2dbcTransactionManager(connectionFactory); // bean 등록
    }
}

6. ReactiveCrudRepository 사용

spring-data-repository에서 ReactiveCrudRepository 를 사용하여 r2dbc 사용이 가능함.
현재 r2dbc는 메소드명으로 쿼리가 동적으로 작성되지 않는다.
- @Query를 사용하여 쿼리를 직접 작성해야함.

public interface PersonRepository extends ReactiveCrudRepository<Person, Long> {

    @Query("SELECT * FROM person WHERE name = :name")
    Flux<Person> findByName(String name);
}

7. 테스트를 위한 schema 생성 설정

ConnectionFactoryInitializer 를 사용하여 schema와 초기 데이터를 셋팅 가능함.

@Configuration
public class TestDataBaseConfig {

    @Bean
    public ConnectionFactoryInitializer initializer(ConnectionFactory connectionFactory){
        ConnectionFactoryInitializer initializer = new ConnectionFactoryInitializer();
        initializer.setConnectionFactory(connectionFactory);

        CompositeDatabasePopulator populator = new CompositeDatabasePopulator();
        populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/db-schema.sql")));
        initializer.setDatabasePopulator(populator);

        return initializer;
    }
}

8. 저장 테스트

@Test
@DisplayName("person 저장 테스트")
public void personSaveTest(){

    // given
    PersonDTO dto = PersonDTO.builder()
                             .address("서울시 강북구 수유동")
                             .age(32)
                             .name("kys0213")
                             .build();

    // when
    Mono<Person> personMono = personRepository.save(dto.toEntity());

    // then
    Person person = personMono.block();

    Mono<Person> personMono1 = personRepository.findById(person.getId());

    Person findPerson = personMono1.block();

    assertThat(person).isEqualTo(findPerson);
}

9. jooq 사용해보기

엔티티의 식별자 값과 복잡한 쿼리들 작성의 경우 직접 sql을 작성하는 것 보다 jooq를 사용하여 쿼리를 작성하는것이 조금 더 typesafe 한 구성을 사용할 수 있다고 생각하여 설정.

jooq를 사용하여 쿼리를 생성하고, r2dbc를 사용하여 호출하는 컨셉

참조 : jOOQ/jOOQ#6298

현재 발생한 문제점

1. spring-data-r2dbc에서 ReactiveCrudRepository 사용시

내부에서 SimpleR2dbcRepository 를 사용하여 쿼리를 실행함.
save 메소드 호출시 Entity에 선언된 @Id를 기준으로 값이 null 이면 insert null이 아니면 update 호출.
- PersistentEntityIsNewStrategy 에서 isNew 메소드를 호출하여 최초 저장 여부를 판단한다.
  1. @id 선언된 값이 null 인 경우
  2. @id 선언된 타입이 기본형이 아닐 때, 값이 null 인 경우
  3. @id 가 정수형이고 값이 0 인 경우

@Override
public boolean isNew(Object entity) {

    // @id를 선언한 값을 가져온다.
    Object value = valueLookup.apply(entity); 
    
    // 값이 null 일 경우 true
    if (value == null) { 
        return true;
    }

    // valueType이 null이 아니거나 기본형 타입이 아닌 경우 true
    if (valueType != null && !valueType.isPrimitive()) {
        return value == null;
    }
    
    // 부모가 Number 클래스인 경우 값 비교 
    if (Number.class.isInstance(value)) {
        return ((Number) value).longValue() == 0;
    }
    
    throw new IllegalArgumentException(
            String.format("Could not determine whether %s is new! Unsupported identifier or version property!", entity));
}

schema에 auto_increment 를 추가하여 처리함.
- auto_increment가 아닌 경우 저장하는 메소드를 직접 구현하는게 편하다.
- 비동기로 실행하다보니 디버깅으로 따라가기 너무 힘드네요... 추후 다시 도전을...

KimYongSung/simple-reactive-server