Created
February 8, 2022 15:30
-
-
Save Linkit123/2c87ffcef6471f7c54a9c9068e406933 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| package com.dvtt.demo.coredemo.filter; | |
| import com.dvtt.demo.coredemo.thread.ThreadContextKeeper; | |
| import com.dvtt.demo.coredemo.wrappers.BufferedRequestWrapper; | |
| import com.dvtt.demo.coredemo.wrappers.BufferedResponseWrapper; | |
| import com.dvtt.demo.utils.CoreUtils; | |
| import lombok.extern.slf4j.Slf4j; | |
| import org.apache.commons.lang3.ObjectUtils; | |
| import org.apache.commons.lang3.StringUtils; | |
| import org.slf4j.MDC; | |
| import org.springframework.stereotype.Component; | |
| import org.springframework.web.filter.OncePerRequestFilter; | |
| import javax.servlet.FilterChain; | |
| import javax.servlet.ServletException; | |
| import javax.servlet.http.HttpServletRequest; | |
| import javax.servlet.http.HttpServletResponse; | |
| import java.io.IOException; | |
| import java.util.Map; | |
| import java.util.UUID; | |
| import static com.dvtt.demo.utils.CoreUtils.TRACING_ID; | |
| /** | |
| * Created by linhtn on 1/21/2022. | |
| */ | |
| @Component | |
| @Slf4j | |
| public class CustomFilter extends OncePerRequestFilter { | |
| @Override | |
| protected void doFilterInternal(HttpServletRequest httpServletRequest, | |
| HttpServletResponse httpServletResponse, | |
| FilterChain filterChain) throws IOException, ServletException { | |
| var tracingId = UUID.randomUUID().toString(); | |
| MDC.put(TRACING_ID, tracingId); | |
| try { | |
| Map<String, String> requestParam = CoreUtils.getRequestParam(httpServletRequest); | |
| BufferedRequestWrapper bufferedRequest = new BufferedRequestWrapper( | |
| httpServletRequest); | |
| BufferedResponseWrapper bufferedResponse = new BufferedResponseWrapper( | |
| httpServletResponse); | |
| if (!ObjectUtils.isEmpty(bufferedRequest.getHeader(TRACING_ID))) { | |
| tracingId = bufferedRequest.getHeader(TRACING_ID); | |
| } | |
| var requestAttributes = ThreadContextKeeper.getRequestAttributes(); | |
| requestAttributes.setTracingId(tracingId); | |
| requestAttributes.setAcceptLanguage("vi-VN"); | |
| if (!StringUtils.isEmpty(bufferedRequest.getHeader("Accept-Language"))) { | |
| requestAttributes.setAcceptLanguage(bufferedRequest.getHeader("Accept-Language")); | |
| } | |
| var requestBody = bufferedRequest.getRequestBody(); | |
| var logMessage = new StringBuilder().append("[REQ_LOG] ====> ") | |
| .append("[STATUS CODE: ").append(bufferedResponse.getStatus()).append("]") | |
| .append(", [URI: ").append(CoreUtils.getFullURL(bufferedRequest)).append("]") | |
| .append(", [REQUEST HEADERS: ").append(CoreUtils.buildHeadersMap(bufferedRequest)).append("]") | |
| .append(", [REQUEST CONTENT TYPE: ").append(bufferedRequest.getContentType()).append("]") | |
| .append(", [REMOTE ADDRESS: ").append(bufferedRequest.getRemoteAddr()).append("]") | |
| .append(", [METHOD: ").append(bufferedRequest.getMethod()).append("]"); | |
| if (!ObjectUtils.isEmpty(requestBody)) { | |
| logMessage.append(", [REQUEST BODY: ").append(requestBody).append("]"); | |
| } | |
| filterChain.doFilter(bufferedRequest, bufferedResponse); | |
| var responseBody = bufferedResponse.getContent(); | |
| if (!ObjectUtils.isEmpty(requestParam)) { | |
| logMessage.append(", [REQUEST PARAMS: ").append(requestParam).append("]"); | |
| } | |
| if (!ObjectUtils.isEmpty(responseBody)) { | |
| logMessage.append(", [RESPONSE BODY: ").append(responseBody).append("]"); | |
| } | |
| log.info(logMessage.toString()); | |
| } catch (Throwable a) { | |
| log.error(a.getMessage()); | |
| } finally { | |
| MDC.remove(TRACING_ID); | |
| } | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment