com.erigir.wrench.ape.http.ApeErrorHandlerControl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wrench-ape Show documentation
A library to simplify API construction atop Spring MVC
There is a newer version: 2.2.16+16
package com.erigir.wrench.ape.http;

import com.erigir.wrench.ape.exception.ApeExceptionWriter;
import com.erigir.wrench.ape.exception.DataValidationException;
import com.erigir.wrench.ape.exception.NoSuchResourceException;
import com.erigir.wrench.aws.sns.ServerErrorNotifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.security.Principal;
import java.util.Collections;
import java.util.Map;
import java.util.TreeMap;

/**
 * Created by cweiss on 8/22/15.
 */
@Controller
public class ApeErrorHandlerControl {
  private static final Logger LOG = LoggerFactory.getLogger(ApeErrorHandlerControl.class);
  @Resource(name = "apeExceptionWriter")
  private ApeExceptionWriter apeExceptionWriter;
  // Autowired here since there might not be one
  @Autowired(required = false)
  private ServerErrorNotifier serverErrorNotifier;

  @RequestMapping(value = "/ErrorHandler")
  public void
  errorHandler(HttpServletRequest req, HttpServletResponse resp) {
    Exception main = (Exception) req.getAttribute("javax.servlet.error.exception");
    int currentStatus = resp.getStatus();
    if (main == null) {
      if (currentStatus == 400) {
        // This is typically caused by Spring not being able to parse the supplied
        // body into the needed object.  Fake that instead
        LOG.warn("No exception and status 400 - treating as spring parse error");
        main = new DataValidationException(Collections.singletonMap("request-body", "bad or mismatched json"));
      } else {
        LOG.warn("Someone requested the error handler directly.  Faking a 404");
        main = new NoSuchResourceException();
      }
    }

    try {
      if (serverErrorNotifier != null) {
        buildAndSendErrorReport(req);
      }
      // Now output json
      apeExceptionWriter.writeExceptionToResponse(req, resp, main);

      //req.removeAttribute("javax.servlet.error.exception");
    } catch (Exception e) {
      LOG.error("Well this is really bad, got an exception trying to handle the error page!:" + e, e);
    }
  }

  private void buildAndSendErrorReport(HttpServletRequest req) {
    LOG.info("Building error report");
    try {
      Map other = new TreeMap();
      Principal principal = req.getUserPrincipal();
      if (principal != null) {
        other.put("Logged In Principal", principal);
      }
      serverErrorNotifier.reportError(req, other);
    } catch (Throwable t) {
      LOG.error("Didn't send report due to error", t);
    }
  }

  public void setServerErrorNotifier(ServerErrorNotifier serverErrorNotifier) {
    this.serverErrorNotifier = serverErrorNotifier;
  }

  public void setApeExceptionWriter(ApeExceptionWriter apeExceptionWriter) {
    this.apeExceptionWriter = apeExceptionWriter;
  }
}