Routers.java example

Explorer
blade-master
/**
 * Copyright (c) 2015, biezhi 王爵 (biezhi.me@gmail.com)
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.blade.mvc.route;

import com.blade.exception.BladeException;
import com.blade.kit.Assert;
import com.blade.kit.CollectionKit;
import com.blade.kit.StringKit;
import com.blade.kit.reflect.ReflectKit;
import com.blade.mvc.http.HttpMethod;
import com.blade.mvc.http.Request;
import com.blade.mvc.http.Response;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;

/**
 * Registration, management route
 *
 * @author <a href="mailto:biezhi.me@gmail.com" target="_blank">biezhi</a>
 * @since 1.5
 */
public class Routers {

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

    private Map<String, Route> routes = null;
    private Map<String, Route> interceptors = null;
    private static final String METHOD_NAME = "handle";

    public Routers() {
        this.routes = CollectionKit.newConcurrentHashMap();
        this.interceptors = CollectionKit.newConcurrentHashMap();
    }

    public Map<String, Route> getRoutes() {
        return routes;
    }

    public Map<String, Route> getInterceptors() {
        return interceptors;
    }

    public void addRoute(Route route) {
        String path = route.getPath();
        HttpMethod httpMethod = route.getHttpMethod();
        String key = path + "#" + httpMethod.toString();

        // existent
        if (null != this.routes.get(key)) {
            LOGGER.warn("\tRoute {} -> {} has exist", path, httpMethod.toString());
        }

        if (httpMethod == HttpMethod.BEFORE || httpMethod == HttpMethod.AFTER) {
            if (null != this.interceptors.get(key)) {
                LOGGER.warn("\tInterceptor {} -> {} has exist", path, httpMethod.toString());
            }
            this.interceptors.put(key, route);
            LOGGER.debug("Add Interceptor => {}", route);
        } else {
            this.routes.put(key, route);
            LOGGER.debug("Add Route => {}", route);
        }
    }

    public void addRoutes(List<Route> routes) {
        Assert.notNull(routes);
        routes.forEach(this::addRoute);
    }

    public void addRoute(HttpMethod httpMethod, String path, com.blade.mvc.handler.RouteHandler handler, String methodName) throws NoSuchMethodException {
        Class<?> handleType = handler.getClass();
        Method method = handleType.getMethod(methodName, Request.class, Response.class);
        addRoute(httpMethod, path, handler, com.blade.mvc.handler.RouteHandler.class, method);
    }

    public void addRoute(HttpMethod httpMethod, String path, Object controller, Class<?> controllerType, Method method) {

        Assert.notBlank(path);

        String key = path + "#" + httpMethod.toString();
        // existent
        if (null != this.routes.get(key)) {
            LOGGER.warn("\tRoute {} -> {} has exist", path, httpMethod.toString());
        }

        Route route = new Route(httpMethod, path, controller, controllerType, method);
        if (httpMethod == HttpMethod.BEFORE || httpMethod == HttpMethod.AFTER) {
            if (null != this.interceptors.get(key)) {
                LOGGER.warn("\tInterceptor {} -> {} has exist", path, httpMethod.toString());
            }
            this.interceptors.put(key, route);
            LOGGER.info("Add Interceptor: {}", route);
        } else {
            this.routes.put(key, route);
            LOGGER.info("Add Route => {}", route);
        }

    }

    public void route(String path, com.blade.mvc.handler.RouteHandler handler, HttpMethod httpMethod) {
        try {
            addRoute(httpMethod, path, handler, METHOD_NAME);
        } catch (NoSuchMethodException e) {
            throw new BladeException(e);
        }
    }

    public void route(String[] paths, com.blade.mvc.handler.RouteHandler handler, HttpMethod httpMethod) {
        for (String path : paths) {
            route(path, handler, httpMethod);
        }
    }

    private Map<String, Method[]> classMethosPool = CollectionKit.newConcurrentHashMap(16);
    private Map<Class<?>, Object> controllerPool = CollectionKit.newConcurrentHashMap(16);

    public void route(String path, Class<?> clazz, String methodName) {
        Assert.notNull(methodName, "Method name not is null");
        HttpMethod httpMethod = HttpMethod.ALL;
        if (methodName.contains(":")) {
            String[] methodArr = methodName.split(":");
            httpMethod = HttpMethod.valueOf(methodArr[0].toUpperCase());
            methodName = methodArr[1];
        }
        this.route(path, clazz, methodName, httpMethod);
    }

    public void route(String path, Class<?> clazz, String methodName, HttpMethod httpMethod) {
        try {
            Assert.notNull(path, "Route path not is null!");
            Assert.notNull(clazz, "Class Type not is null!");
            Assert.notNull(methodName, "Method name not is null");
            Assert.notNull(httpMethod, "Request Method not is null");

            Method[] methods = classMethosPool.get(clazz.getName());
            if (null == methods) {
                methods = clazz.getMethods();
                classMethosPool.put(clazz.getName(), methods);
            }
            if (null != methods) {
                for (Method method : methods) {
                    if (method.getName().equals(methodName)) {
                        Object controller = controllerPool.get(clazz);
                        if (null == controller) {
                            controller = ReflectKit.newInstance(clazz);
                            controllerPool.put(clazz, controller);
                        }
                        addRoute(httpMethod, path, controller, clazz, method);
                    }
                }
            }
        } catch (Exception e) {
            LOGGER.error("", e);
        }
    }

    public void buildRoute(String path, Class<?> clazz, Method method, HttpMethod httpMethod) {
        addRoute(httpMethod, path, null, clazz, method);
    }

    public void delRoute(String key) {
        if (StringKit.isNotBlank(key)) {
            routes.remove(key);
        }
    }

}