com.minlia.cloud.exception.handler.RestErrorEndpoint Maven / Gradle / Ivy

Go to download
package com.minlia.cloud.exception.handler;

import com.google.common.base.CaseFormat;
import com.minlia.cloud.body.StatefulBody;
import com.minlia.cloud.body.impl.FailureResponseBody;
import com.minlia.cloud.code.ApiCode;
import com.minlia.cloud.constant.Constants.LanguageTypes;
import com.minlia.cloud.i18n.Lang;
import com.minlia.cloud.utils.ApiPreconditions;
import com.minlia.cloud.utils.Environments;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.web.ErrorAttributes;
import org.springframework.boot.autoconfigure.web.ErrorController;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * Based on the helpful answer at http://stackoverflow.com/q/25356781/56285,
 * with error details in response body added.
 * 添加404 JSON输出
 */
@RestController
public class RestErrorEndpoint implements ErrorController {

  private static final String PATH = "/error";


  @Autowired
  private ErrorAttributes errorAttributes;

  //    @RequestMapping(value = PATH)
  @RequestMapping(value = PATH, produces = {MediaType.APPLICATION_JSON_VALUE})
  StatefulBody error(HttpServletRequest request, HttpServletResponse response) {
    // Appropriate HTTP response code (e.g. 404 or 500) is automatically set by Spring.
    // Here we just define response body.
//        return new ErrorJson(response.getStatus(), getErrorAttributes(request, debug));

    response.setStatus(200);
    if (!Environments.isProduction()) {
      return buildBody(request, false);
    } else {
      return buildBody(request, false);
    }
  }


  private StatefulBody buildBody(HttpServletRequest request, Boolean includeStackTrace) {
    Map errorAttributes = getErrorAttributes(request, includeStackTrace);
    Integer status = (Integer) errorAttributes.get("status");
    String path = (String) errorAttributes.get("path");
    String messageFound = (String) errorAttributes.get("message");

    Integer messageStatus=extractStatus(messageFound,status);

    if (HttpStatus.NOT_FOUND.value() != status) {
      String localizedMessage = Lang.get(toKeyFormat(messageFound));
      if (!org.springframework.util.StringUtils.isEmpty(localizedMessage)) {
        messageFound = localizedMessage;
      }
    }else{
      //强制触发抛出异常
      ApiPreconditions.is(true, ApiCode.NOT_FOUND);
    }

    String message = messageFound;
    String trace = "";
//        if(!StringUtils.isEmpty(path)){
//            message=String.format("Requested path %s with result %s",path,messageFound);
//        }
    if (includeStackTrace) {
      trace = (String) errorAttributes.get("trace");
      if (!StringUtils.isEmpty(trace)) {
        message += String.format(" with trace %s", trace);
      }
    }


    return FailureResponseBody.builder().code(messageStatus).status(0).message(message).build();
  }


  private Integer extractStatus(String message,Integer status){
      if(message.startsWith(LanguageTypes.ExceptionsApiCode.name())){
        int index=LanguageTypes.ExceptionsApiCode.name().length();
        String message2=message.substring(index);
        return Integer.valueOf(message2);
      }else{
        return status;
      }

  }
  private String toKeyFormat(String code) {
    String DOT = ".";
    code = CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, code);
    code = code.replaceAll("_", DOT);
    code = code.replaceAll("(\\d+)", ".$1");
    return code;
  }

  @Override
  public String getErrorPath() {
    return PATH;
  }

  private Map getErrorAttributes(HttpServletRequest request,
      boolean includeStackTrace) {
    RequestAttributes requestAttributes = new ServletRequestAttributes(request);
    return errorAttributes.getErrorAttributes(requestAttributes, includeStackTrace);
  }

}