精弘网络 Java 后端微课 - EP3 异常捕获与日志记录

1
@Slf4j
2
@Component
3
public class AccessLogFilter implements Filter {
4

5
    static {
6
        AnsiOutput.setEnabled(AnsiOutput.Enabled.ALWAYS);
7
    }
8

9
    private static AnsiColor getStatusColor(int status) {
10
        return switch (status / 100) {
11
            case 2 -> AnsiColor.GREEN;
12
            case 4 -> AnsiColor.YELLOW;
13
            case 5 -> AnsiColor.RED;
14
            default -> AnsiColor.DEFAULT;
15
        };
16
    }
17

18
    private static AnsiColor getMethodColor(String method) {
19
        return switch (method) {
20
            case "GET" -> AnsiColor.BLUE;
21
            case "POST" -> AnsiColor.CYAN;
22
            case "PUT" -> AnsiColor.YELLOW;
23
            case "DELETE" -> AnsiColor.RED;
24
            default -> AnsiColor.MAGENTA;
25
        };
26
    }
27

28
    private static String getRemoteAddr(HttpServletRequest request) {
29
        String ip = request.getHeader("x-forwarded-for");
30
        return ip == null ? request.getRemoteAddr() : ip;
31
    }
32

33
    @Override
34
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
35
            throws IOException, ServletException {
36

37
        if (!(request instanceof HttpServletRequest req) || !(response instanceof HttpServletResponse res)) {
38
            chain.doFilter(request, response);
39
            return;
40
        }
41

42
        Instant start = Instant.now();
43
        chain.doFilter(request, response);
44
        Instant end = Instant.now();
45
        long durationMs = Duration.between(start, end).toMillis();
46

47
        int status = res.getStatus();
48
        String method = req.getMethod();
49
        String uri = req.getRequestURI();
50
        String query = req.getQueryString();
51
        String ip = getRemoteAddr(req);
52

53
        // 彩色状态码
54
        String colorStatus = AnsiOutput.toString(
55
                getStatusColor(status),
56
                status,
57
                AnsiColor.DEFAULT
58
        );
59

60
        // 彩色方法
61
        String colorMethod = AnsiOutput.toString(
62
                getMethodColor(method),
63
                method,
64
                AnsiColor.DEFAULT
65
        );
66

67
        log.info("{} | {}ms | {} {} | IP: {}",
68
                colorStatus,
69
                durationMs,
70
                String.format("%-6s", colorMethod),
71
                uri + (query != null ? "?" + query : ""),
72
                ip
73
        );
74
    }
75
}

1
@Getter
2
public class ApiException extends RuntimeException {
3
    private final Integer errorCode;
4
    private final String errorMsg;
5

6
    public ApiException(Integer errorCode, String errorMsg) {
7
        super(errorMsg);
8
        this.errorCode = errorCode;
9
        this.errorMsg = errorMsg;
10
    }
11

12
    public ApiException(Integer errorCode, String errorMsg, Throwable cause) {
13
        super(errorMsg, cause);
14
        this.errorCode = errorCode;
15
        this.errorMsg = errorMsg;
16
    }
17
}

1
@Getter
2
public enum ExceptionEnum {
3
    INVALID_PARAMETER(200000, "参数错误"),
4
    RESOURCE_NOT_FOUND(200001, "资源不存在"),
5
    WRONG_USERNAME_OR_PASSWORD(200002, "用户名或密码错误"),
6
    PERMISSION_NOT_ALLOWED(200003, "权限不足"),
7

8
    NOT_FOUND_ERROR(200404, HttpStatus.NOT_FOUND.getReasonPhrase()),
9
    SERVER_ERROR(200500, "系统错误, 请稍后重试"),
10
    ;
11

12
    private final Integer errorCode;
13
    private final String errorMsg;
14

15
    ExceptionEnum(Integer errorCode, String errorMsg) {
16
        this.errorCode = errorCode;
17
        this.errorMsg = errorMsg;
18
    }
19
}