com.cybermkd.route.core.RouteHandler Maven / Gradle / Ivy

Go to download
package com.cybermkd.route.core;

import com.cybermkd.common.http.HttpMethod;
import com.cybermkd.common.http.HttpRequest;
import com.cybermkd.common.http.HttpResponse;
import com.cybermkd.common.http.exception.WebException;
import com.cybermkd.common.http.result.HttpStatus;
import com.cybermkd.log.Logger;
import com.cybermkd.route.handler.Handler;

import java.util.Map;
import java.util.Set;

/**
 * ActionHandler
 */
public final class RouteHandler extends Handler {
    private final static Logger logger = Logger.getLogger(RouteHandler.class);
    private final RouteBuilder routeBuilder;

    public RouteHandler(RouteBuilder routeBuilder) {
        this.routeBuilder = routeBuilder;
    }

    /**
     * handle
     */
    public final void handle(HttpRequest request, HttpResponse response, boolean[] isHandled) {
        RouteMatch routeMatch = null;
        Route route = null;
        RouteInvocation routeInvocation = null;
        //请求的rest路径
        String restPath = request.getRestPath();
        Map>> routesMap = routeBuilder.getRoutesMap();
        Set routesSet;
        String httpMethod = request.getHttpMethod();
        boolean supportMethod = HttpMethod.support(httpMethod);
        //httpmethod区分
        if (supportMethod) {
            if (routesMap.containsKey(httpMethod)) {
                Set>> routesEntrySet = routesMap.get(httpMethod).entrySet();
                //url区分
                for (Map.Entry> routesEntry : routesEntrySet) {
                    if (restPath.startsWith(routesEntry.getKey())) {
                        routesSet = routesEntry.getValue();
                        for (Route r : routesSet) {
                            routeMatch = r.match(request, response);
                            if (routeMatch != null) {
                                route = r;
                                break;
                            }
                        }
                        if (routeMatch != null) {
                            break;
                        }
                    }
                }
            }

            if (routeMatch != null) {
                routeInvocation = new RouteInvocation(route, routeMatch);
            }
        }
        isHandled[0] = true;
        //route
        if (routeInvocation != null) {
            routeInvocation.invoke();
        } else {
            if (!restPath.equals("/") && supportMethod) {
                // no route matched
                throw new WebException(HttpStatus.SERVICE_UNAVAILABLE, "API is unavailable,check request body.");
            } else {
                isHandled[0] = false;
            }
        }
    }
}