[WebClient] 사용시 주의사항
WebClient 는 Spring 에서 제공하는 RestClient 의 한 종류이다. 과거에 사용되던 RestTemplate 과 비슷한 역할이라고 생각하면 되는데 RestTemplate 는 장기적으로 Deprecated 예정이기 때문에 이제는 WebClient 를 사용해야 한다.
WebClient 의 가장 큰 특징은 RestTemplate 과 다르게 논블로킹을 지원한다는 것이다. 말만 들으면 성능에 굉장한 이득이 있을 것 같지만 논블로킹 방식으로 사용하지 않으면 RestTemplate 을 사용하는 것과 차이가 없다.
WebClient 의 응답값은 Mono 또는 Flux 타입이다. Mono, Flux 는 리액티브 스트림을 구성하는 파이프라인의 일부로 웹플럭스에서는 기본적으로 논블로킹 처리되지만 SpringMVC 환경에서 사용시에는 논블로킹 처리를 수동으로 구성해줘야 한다.
우선 Mono, Flux는 결과를 미래에 알 수 있다는 점에서 Future와 비슷하지만 Future 는 이미 시작되었음을 의미하지만 Mono, Flux 는 시작할 수 있음을 의미한다는 점을 인지하고 있어야 한다.
즉, WebClient의 호출을 단순히 Mono, Flux 을 리턴하도록 하고 끝낸다면 대상 서버를 호출조차 하지 않는다는 말이다.
Mono<Integer> mono = webClient.get().uri("/test1")
.retrieve()
.bodyToMono(Integer.class);
Flux<Integer> flux = webClient.get().uri("/test2")
.retrieve()
.bodyToFlux(Integer.class);
이걸 어떻게 처리할지 추가 정의해야 작업이 시작 되는데 블로킹방식과 논블로킹 방식으로 처리할 수 있고 블로킹 방식은 아래와 같이 사용하면 된다.
int result = webClient.get().uri("/test1")
.retrieve()
.bodyToMono(Integer.class)
.block();
List<Integer> resultList = webClient.get().uri("/test2")
.retrieve()
.bodyToFlux(Integer.class)
.toStream()
.collect(Collectors.toList());
문제는 이는 블로킹 방식으로 RestTemplate 를 사용할 때와 성능상 차이가 없고 성능 향상을 위해서는 논블로킹 방식으로 사용할 필요가 있다.
논블로킹 방식으로 사용하는 방법은 아래와 같다.
AtomicInteger result = new AtomicInteger();
List<Integer> resultList = new ArrayList<>();
webClient.get().uri("/test1")
.retrieve()
.bodyToMono(String.class)
.subscribe(e -> result.set(e));
webClient.get().uri("/test2")
.retrieve()
.bodyToFlux(String.class)
.subscribe(e -> resultList.add(e));
위 코드에서 2번의 WebClient 호출은 논블로킹으로 동작한다. 이에 메인스레드가 종료되면 완료여부와 무관하게 곧바로 종료된다.
해서 메인스레드 입장에서 각 호출의 처리완료를 기다려야 하고 일반적으로 사용되는 방식은 아래와 같이 CountDownLatch 를 이용한 방법이다.
CountDownLatch cdl = new CountDownLatch(2);
AtomicInteger result = new AtomicInteger();
List<Integer> resultList = new ArrayList<>();
webClient.get().uri("/test1")
.retrieve()
.bodyToMono(String.class)
.doOnTerminate(() -> cdl.countDown())
.subscribe(e -> result.set(e));
webClient.get().uri("/test2")
.retrieve()
.bodyToFlux(String.class)
.doOnTerminate(() -> cdl.countDown())
.subscribe(e -> resultList.add(e));
cdl.await();
어차피 처리결과를 기다려야 한다면 블로킹이 발생하게 되는데 무엇이 이득이냐고 생각될 수 있지만 메인스레드 입장에서는 블로킹이지만 각각의 WebClient 호출은 논블로킹으로 호출된다.
이에 각각의 호출이 5초가 소요된다고 가정하면 WebClient 를 블로킹 방식으로 사용한다면 총 처리시간은 10초가 될테지만 위와 같이 논블로킹 방식으로 사용한다면 총 처리시간은 5초가 된다.
참고로 WebClient를 코루틴에서 사용할 경우 Mono와 Flux 를 그대로 사용하진 못하고 변환을 해야 하는데 각각 아래와 같이 대체하면 된다.
bodyToMono == awaitBody
bodyToFlux == bodyToFlow
코루틴에서 역시 메인스레드에서 처리의 완료를 기다려야 하는데 runBlocking 블록을 사용하면 되고 launch, async 블록들을 활용해서 논블로킹 처리가 가능하다.
참조 :