[Spring MSA] 04-3. API 게이트웨이

// FilterUtility 클래스는 HTTP 헤더와 관련된 유틸리티 메서드를 제공하는 컴포넌트입니다. @Component public class FilterUtility { // 상수로 정의된 correlation ID의 키 값입니다. public static final String CORRELATION_ID = "eazybank-correlation-id"; // 요청 헤더에서 correlation ID를 추출하는 메서드입니다. public String getCorrelationId(HttpHeaders requestHeaders) { // 헤더에서 correlation ID를 가져옵니다. if (requestHeaders.get(CORRELATION_ID) != null) { List<String> requestHeaderList = requestHeaders.get(CORRELATION_ID); // 첫 번째 correlation ID를 반환합니다. return requestHeaderList.stream().findFirst().get(); } else { // 헤더가 없으면 null을 반환합니다. return null; } } // 주어진 이름과 값으로 요청 헤더를 설정하는 메서드입니다. public ServerWebExchange setRequestHeader(ServerWebExchange exchange, String name, String value) { // 요청 객체를 변경하여 새 헤더를 추가합니다. return exchange.mutate().request(exchange.getRequest().mutate().header(name, value).build()).build(); } // correlation ID를 요청 헤더에 설정하는 메서드입니다. public ServerWebExchange setCorrelationId(ServerWebExchange exchange, String correlationId) { // setRequestHeader를 호출하여 correlation ID를 설정합니다. return this.setRequestHeader(exchange, CORRELATION_ID, correlationId); } }

// 응답 트레이스를 위한 필터 구성을 정의하는 클래스입니다. @Configuration public class ResponseTraceFilter { // 로깅을 위한 Logger 객체입니다. private static final Logger logger = LoggerFactory.getLogger(ResponseTraceFilter.class); // FilterUtility 인스턴스를 자동 주입합니다. @Autowired FilterUtility filterUtility; // 응답 처리 후 실행되는 글로벌 필터를 빈으로 등록합니다. @Bean public GlobalFilter postGlobalFilter() { return (exchange, chain) -> { return chain.filter(exchange).then(Mono.fromRunnable(() -> { // 요청 헤더를 가져옵니다. HttpHeaders requestHeaders = exchange.getRequest().getHeaders(); // correlation ID를 추출합니다. String correlationId = filterUtility.getCorrelationId(requestHeaders); // 로그에 correlation ID를 기록합니다. logger.debug("Updated the correlation id to the outbound headers: {}", correlationId); // 응답 헤더에 correlation ID를 추가합니다. exchange.getResponse().getHeaders().add(filterUtility.CORRELATION_ID, correlationId); })); }; } }

// 요청 트레이스를 위한 글로벌 필터 구현입니다. @Component public class RequestTraceFilter implements GlobalFilter { // 로깅을 위한 Logger 객체입니다. private static final Logger logger = LoggerFactory.getLogger(RequestTraceFilter.class); // FilterUtility 인스턴스를 자동 주입합니다. @Autowired FilterUtility filterUtility; // 요청을 처리하는 필터 메서드입니다. @Override public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { // 요청 헤더를 가져옵니다. HttpHeaders requestHeaders = exchange.getRequest().getHeaders(); // correlation ID가 존재하는지 확인합니다. if (isCorrelationIdPresent(requestHeaders)) { // correlation ID가 있으면 로그에 기록합니다. logger.debug("eazyBank-correlation-id found in RequestTraceFilter : {}", filterUtility.getCorrelationId(requestHeaders)); } else { // correlation ID가 없으면 새로 생성합니다. String correlationID = generateCorrelationId(); // 생성된 correlation ID를 요청 헤더에 설정합니다. exchange = filterUtility.setCorrelationId(exchange, correlationID); // 로그에 새로 생성된 correlation ID를 기록합니다. logger.debug("eazyBank-correlation-id generated in RequestTraceFilter : {}", correlationID); } // 요청을 계속 처리합니다. return chain.filter(exchange); } // 요청 헤더에서 correlation ID의 존재 여부를 확인하는 메서드입니다. private boolean isCorrelationIdPresent(HttpHeaders requestHeaders) { return filterUtility.getCorrelationId(requestHeaders) != null; } // 새로운 correlation ID를 생성하는 메서드입니다. private String generateCorrelationId() { // UUID를 사용하여 유일한 ID를 생성합니다. return java.util.UUID.randomUUID().toString(); } }