import lombok.extern.slf4j.Slf4j;

import org.springframework.http.HttpStatus;

import org.springframework.http.ResponseEntity;

import org.springframework.web.bind.annotation.ExceptionHandler;

import org.springframework.web.bind.annotation.RestControllerAdvice;

import org.springframework.web.servlet.NoHandlerFoundException;

@Slf4j

@RestControllerAdvice

public class GlobalExceptionHandler {

    /**

     * 处理运行异常

     */

    @ExceptionHandler(RuntimeException.class)

    public ResponseEntity<String> handleRuntimeException(RuntimeException ex) {

        log.error("", ex);

        return new ResponseEntity<>(ex.getMessage(), HttpStatus.BAD_REQUEST);

    }

    /**

     * 用来捕获404，400这种无法到达controller的错误

     *

     * @param ex

     * @return

     * @throws Exception

     */

    @ExceptionHandler(value = Exception.class)

    public ResponseData defaultErrorHandler(Exception ex) throws Exception {

        log.error("", ex);

        ResponseData<Object> result = new ResponseData<Object>();

        result.setMessage(ex.getMessage());

        if (ex instanceof NoHandlerFoundException) {

            result.setCode("404");

        } else {

            result.setCode("500");

        }

        result.setData(null);

        result.setSuccess(false);

        return result;

    }

}