Common error

src/main/java/com/api/controller/AbsctactController.java

@@ -0,0 +1,25 @@
+package com.api.controller;
+
+import com.common.utils.R;
+
+public abstract class AbsctactController {
+    protected R success() {
+        return R.ok();
+    }
+
+    protected R success(String msg) {
+        return R.ok(msg);
+    }
+
+    protected R error() {
+        return R.error();
+    }
+
+    protected R error(String msg) {
+        return R.error(msg);
+    }
+
+    protected R error(String code, String msg) {
+        return R.error(code, msg);
+    }
+}

src/main/java/com/api/controller/ApiLoginController.java

@@ -18,6 +18,7 @@ import com.api.service.UserService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
 import org.springframework.web.bind.annotation.*;
 import springfox.documentation.annotations.ApiIgnore;
 
@@ -31,13 +32,12 @@ import java.util.Map;
 @RestController
 @RequestMapping("/api")
 @Api(tags="登录接口")
-public class ApiLoginController {
+public class ApiLoginController extends AbsctactController {
     @Autowired
     private UserService userService;
     @Autowired
     private TokenService tokenService;
 
-
     @PostMapping("login")
     @ApiOperation("登录")
     public R login(@RequestBody LoginForm form){

src/main/java/com/api/controller/ApiTestController.java

@@ -8,12 +8,15 @@
 
 package com.api.controller;
 
+import com.api.dao.TokenDao;
 import com.common.utils.R;
 import com.api.annotation.Login;
 import com.api.annotation.LoginUser;
 import com.api.entity.UserEntity;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestAttribute;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -28,7 +31,9 @@ import springfox.documentation.annotations.ApiIgnore;
 @RestController
 @RequestMapping("/api/api")
 @Api(tags="测试接口")
-public class ApiTestController {
+public class ApiTestController extends AbsctactController {
+    @Autowired
+    ApplicationContext applicationContext;
 
     @Login
     @GetMapping("userInfo")
@@ -44,10 +49,11 @@ public class ApiTestController {
         return R.ok().put("userId", userId);
     }
 
-    @Login
     @GetMapping("notToken")
     @ApiOperation("忽略Token验证测试")
     public R notToken(){
+        Object beanNamesForType = applicationContext.getBeanProvider(TokenDao.class);
+        TokenDao tokenDao = (TokenDao) beanNamesForType;
         return R.ok().put("msg", "无需token也能访问。。。");
     }
 

src/main/java/com/common/exception/RRExceptionHandler.java

@@ -0,0 +1,132 @@
+package com.common.exception;
+
+import com.alibaba.fastjson.JSONException;
+import com.common.utils.LogUtils;
+import com.common.utils.R;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.dao.DuplicateKeyException;
+import org.springframework.http.converter.HttpMessageConversionException;
+import org.springframework.http.converter.HttpMessageNotReadableException;
+import org.springframework.validation.FieldError;
+import org.springframework.web.HttpRequestMethodNotSupportedException;
+import org.springframework.web.bind.MethodArgumentNotValidException;
+import org.springframework.web.bind.MissingServletRequestParameterException;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.servlet.NoHandlerFoundException;
+
+@Slf4j
+@RestControllerAdvice
+public class RRExceptionHandler {
+    @Autowired
+    private LogUtils logUtils;
+    /**
+     * 处理自定义异常
+     */
+    @ExceptionHandler(RRException.class)
+    public R handleRRException(RRException e){
+        R r;
+//        if (StringUtils.isEmpty(e.getMessage())) {
+//            r = R.error();
+//        }
+        r = R.error(e.getErrno(), e.getErrmsg());
+        logUtils.requestLog(r);
+        return r;
+    }
+
+    @ExceptionHandler(NoHandlerFoundException.class)
+    public R handlerMyNoHandlerFoundException(NoHandlerFoundException e) {
+        R r = R.error("404", "路径不存在，请检查路径是否正确");
+        logUtils.requestLog(r);
+        return r;
+    }
+
+    @ExceptionHandler(DuplicateKeyException.class)
+    public R handleDuplicateKeyException(DuplicateKeyException e){
+        logUtils.requestLog(e);
+        return R.error("数据库中已存在该记录");
+    }
+
+//    @ExceptionHandler(AuthorizationException.class)
+//    public R handleAuthorizationException(AuthorizationException e){
+//        R r = R.error("没有权限，请联系管理员授权");
+//        logUtils.requestLog(r);
+//        return r;
+//    }
+
+    /**
+     * 处理 @Valid 对 @RequestBody 参数验证异常
+     * @param e
+     * @return
+     */
+    @ExceptionHandler(MethodArgumentNotValidException.class)
+    public R handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
+        FieldError fieldError = e.getBindingResult().getFieldError();
+        R r = R.error(fieldError.getField() + fieldError.getDefaultMessage());
+        logUtils.requestLog(r);
+        return r;
+    }
+
+    /**
+     * 参数类型转换错误
+     *
+     * @param exception 错误
+     * @return 错误信息
+     */
+    @ExceptionHandler(HttpMessageConversionException.class)
+    public R parameterTypeException(HttpMessageConversionException exception) {
+        R r = R.error(exception.getCause().getLocalizedMessage());
+        return  r;
+    }
+
+    /**
+     * 处理 @RequestBody 但 body none 异常
+     * @param e
+     * @return
+     */
+    @ExceptionHandler(HttpMessageNotReadableException.class)
+    public R handleHttpMessageNotReadableException(HttpMessageNotReadableException e) {
+        String message = e.getMessage();
+        R r = R.error(message.substring(0, message.indexOf(":")));
+        logUtils.requestLog(message);
+        return r;
+    }
+
+    /**
+     * 处理 @RequestParam required=true 但未收到值异常
+     * @param e
+     * @return
+     */
+    @ExceptionHandler(MissingServletRequestParameterException.class)
+    public R handleMissingServletRequestParameterException(MissingServletRequestParameterException e) {
+        R r = R.error("缺少参数" + e.getParameterName());
+        logUtils.requestLog(r);
+        return r;
+    }
+
+    @ExceptionHandler(JSONException.class)
+    public R handleJSONException(JSONException e){
+//        logUtils.requestLog(e);
+        return R.error("数据格式错误");
+    }
+
+    /**
+     * 处理请求方法不支持异常
+     * @param e
+     * @return
+     */
+    @ExceptionHandler(HttpRequestMethodNotSupportedException.class)
+    public R handleHttpRequestMethodNotSupportedException(HttpRequestMethodNotSupportedException e) {
+        R r = R.error(String.format("当前路径不支持%s请求", e.getMethod()));
+        logUtils.requestLog(r);
+        return r;
+    }
+
+    @ExceptionHandler(Exception.class)
+    public R handleException(Exception e){
+        logUtils.requestLog(e);
+        e.printStackTrace();
+        return R.error();
+    }
+}

src/main/java/com/common/utils/LogUtils.java

@@ -0,0 +1,93 @@
+package com.common.utils;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONException;
+import com.alibaba.fastjson.JSONObject;
+import com.alibaba.fastjson.serializer.SerializerFeature;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.MediaType;
+import org.springframework.stereotype.Component;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.util.*;
+import java.util.stream.Collectors;
+
+@Slf4j
+@Component
+public class LogUtils {
+    public static String newLine = System.getProperty("line.separator");
+
+    private List<String> requestLogCommon() {
+        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        HttpServletRequest request = attributes.getRequest();
+
+        List<String> logList = new LinkedList<>();
+        logList.add("请求日志:");
+        logList.add("url=" + request.getRequestURI());
+        String accessToken = request.getHeader("access_token");
+        logList.add("accessToken=" + accessToken);
+        String method = request.getMethod();
+        String contentType = request.getHeader(HttpHeaders.CONTENT_TYPE);
+        Map<String, String[]> map = request.getParameterMap();
+        Map<String, Object> params = new LinkedHashMap<>();
+        map.forEach((key, value) -> {
+            Object object = value.length == 1 ? value[0] : Arrays.asList(value);
+            params.put(key, object);
+        });
+        logList.add("paramData=" + JSON.toJSONString(params));
+        if (HttpMethod.POST.name().equals(method) && null != contentType &&
+                contentType.startsWith(MediaType.APPLICATION_JSON_VALUE)) {
+            try (BufferedReader br = new BufferedReader(request.getReader())) {
+                String paramData = br.readLine();
+                logList.add("body=" + paramData);
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+        logList.add("clientIp=" + IPUtils.getIpAddr(request));
+        logList.add("HTTP_METHOD=" + method);
+
+        return logList;
+    }
+
+    public void requestLog(Object result, long costTime) {
+        List<String> logList = requestLogCommon();
+        logList.add("返回内容:" + JSON.toJSONString(result, SerializerFeature.WriteDateUseDateFormat));
+        String resultString = JSON.toJSONString(result);
+        try {
+            JSONObject resultJson = JSON.parseObject(resultString);
+            if (resultJson.containsKey("errno")) {
+                Integer errno = resultJson.getInteger("errno");
+                if (errno == 0) {
+                    logList.add(1, "请求开始:");
+                    logList.add("请求耗时:" + costTime);
+                    log.info(logList.stream().collect(Collectors.joining(newLine)));
+                } else {
+                    logList.add(1, "请求错误:");
+                    log.warn(logList.stream().collect(Collectors.joining(newLine)));
+                }
+                return;
+            }
+        } catch (JSONException e) {
+        }
+    }
+
+    public void requestLog(RuntimeException e) {
+        if (e != null) {
+            List<String> logList = requestLogCommon();
+            logList.add(1, "请求异常:");
+            logList.add("异常信息:");
+            log.error(logList.stream().collect(Collectors.joining(newLine)), e);
+        }
+    }
+
+    public void requestLog(Object result) {
+        requestLog(result, 0);
+    }
+}