cn.ubibi.jettyboot.framework.rest.RequestHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jettyboot Show documentation
jetty boot Library, simple mvc and data ORM framework
The newest version!
package cn.ubibi.jettyboot.framework.rest;

import cn.ubibi.jettyboot.framework.ioc.ServiceManager;
import cn.ubibi.jettyboot.framework.rest.ifs.ControllerExceptionHandler;
import cn.ubibi.jettyboot.framework.rest.impl.TextRender;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.handler.AbstractHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class RequestHandler extends AbstractHandler {

    private static final Logger LOGGER = LoggerFactory.getLogger(RequestHandler.class);

    private final List controllerHandlers = new ArrayList<>();
    private final List exceptionHandlers = new ArrayList<>();
    private final String my_context;
    public RequestHandler(String my_context) {
        this.my_context = my_context;
    }

    public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {


        // check rest controller handler
        for (ControllerHandler restHandler : controllerHandlers) {
            try {
                if (restHandler.handle(request, response)) {
                    baseRequest.setHandled(true);
                    return;
                }
            } catch (Exception e) {
                try {
                    handleException2(e, request, response);
                } catch (Exception e1) {
                    LOGGER.error("", e1);
                }
                baseRequest.setHandled(true);
                return;
            }
        }


    }

    private void handleException2(Exception e, HttpServletRequest request, HttpServletResponse response) throws Exception {
        boolean isHandled = handleException(e, request, response);
        if (!isHandled) {

            LOGGER.info("", e);

            //如果异常没有被处理
            if (e instanceof IOException) {
                throw (IOException) e;
            } else if (e instanceof ServletException) {
                throw (ServletException) e;
            } else {
                String nextLine = "    \n   ";
                String exMsg = e.toString() + nextLine + e.getMessage() + nextLine + e.getCause();
                new TextRender(exMsg).doRender(request, response);
            }
        }
    }


    private boolean handleException(Exception e, HttpServletRequest request, HttpServletResponse response) throws Exception {

        for (ControllerExceptionHandler exceptionHandler : exceptionHandlers) {

            //可以使用注入Service
            ServiceManager.getInstance().injectDependency(exceptionHandler);

            boolean isHandled = exceptionHandler.handle(e, request, response);
            if (isHandled) {
                return true;
            }
        }
        return false;
    }


    public void addController(String path, Class clazz) throws Exception {
        if (clazz == null) {
            throw new Exception("addController can not null");
        }
        LOGGER.info("addController " + path + "  :  " + clazz.getName());
        String context = this.my_context;
        controllerHandlers.add(new ControllerHandler(context,path, clazz));
    }

    public void addController(String path, Object restController) throws Exception {
        if (restController == null) {
            throw new Exception("addController can not null");
        }
        LOGGER.info("addController " + path + "  :  " + restController.getClass().getName());
        String context = this.my_context;
        controllerHandlers.add(new ControllerHandler(context,path, restController));
    }


    public void addExceptionHandler(ControllerExceptionHandler exceptionHandler) throws Exception {
        if (exceptionHandler == null) {
            throw new Exception("addExceptionHandler can not null");
        }
        LOGGER.info("addExceptionHandler " + exceptionHandler.getClass().getName());
        exceptionHandlers.add(exceptionHandler);
    }


    public List getControllerMethodHandlers() {
        List result = new ArrayList<>();

        for (ControllerHandler controllerHandler : controllerHandlers) {
            List methods = controllerHandler.getControllerMethodList();
            if (methods != null && !methods.isEmpty()) {
                result.addAll(methods);
            }
        }

        return result;

    }

}