feat: 添加忽略统一返回格式的注解判断逻辑

Author: cadecode

simple-common/src/main/java/top/cadecode/common/exception/SimpleExceptionHandler.java

@@ -70,19 +70,20 @@ public Map<String, Object> getErrorAttributes(WebRequest webRequest, ErrorAttrib
         // 获取请求路径
         int status = (int) errorAttributes.get("status");
         String path = (String) errorAttributes.get("path");
-        MapBuilder<String, Object> resMapBuilder = MapUtil.create().add("path", path);
         // 处理 404
         if (status == 404) {
-            return resMapBuilder
+            return MapUtil.create()
                     .add("code", ResCode.REQ_PATH_NOT_EXIST.getCode())
                     .add("reason", ResCode.REQ_PATH_NOT_EXIST.getReason())
+                    .add("path", path)
                     .asMap();
         }
         // 处理 405
         if (status == 405) {
-            return resMapBuilder
+            return MapUtil.create()
                     .add("code", ResCode.REQ_METHOD_INVALID.getCode())
                     .add("reason", ResCode.REQ_METHOD_INVALID.getReason())
+                    .add("path", path)
                     .asMap();
         }
         return errorAttributes;

simple-common/src/main/java/top/cadecode/common/response/SimpleRes.java

@@ -36,6 +36,7 @@ public static <T> SimpleRes<T> ok(T data) {
         ResCode success = ResCode.SUCCESS;
         res.setCode(success.getCode());
         res.setReason(success.getReason());
+        res.setData(data);
         return res;
 
     }

simple-common/src/main/java/top/cadecode/common/response/SimpleResAdvice.java

@@ -5,15 +5,11 @@
 import org.springframework.http.server.ServerHttpRequest;
 import org.springframework.http.server.ServerHttpResponse;
 import org.springframework.web.bind.annotation.ControllerAdvice;
-import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
 import top.cadecode.common.constant.ResCode;
 import top.cadecode.common.exception.SimpleException;
 import top.cadecode.common.util.JsonUtil;
 
-import java.lang.annotation.Annotation;
-import java.lang.reflect.AnnotatedElement;
-
 /**
  * @author Cade Li
  * @date 2021/7/16
@@ -46,6 +42,11 @@ public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType
             // String 类型的 Body 需要返回 String 类型，否则报转换错误
             return JsonUtil.objToStr(SimpleRes.ok(body).path(path));
         }
+        // 判断是否有忽略格式化注解，有则直接返回
+        SimpleResIgnore resIgnore = returnType.getMethodAnnotation(SimpleResIgnore.class);
+        if (resIgnore != null && resIgnore.value()) {
+            return body;
+        }
 
         return SimpleRes.ok(body).path(path);
     }