com.github.jonbonazza.puni.core.mux.DefaultMuxer.java Source code

Introduction

Here is the source code for com.github.jonbonazza.puni.core.mux.DefaultMuxer.java

Source

/*
 * Copyright 2014 Jon Bonazza
 *
 * Jon Bonazza licenses this file to you 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:
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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.github.jonbonazza.puni.core.mux;

import com.github.jonbonazza.puni.core.handlers.HttpHandler;
import com.github.jonbonazza.puni.core.requests.HttpRequest;
import com.google.common.annotations.VisibleForTesting;
import io.netty.handler.codec.http.HttpMethod;

import java.util.HashMap;
import java.util.Map;
import java.util.regex.Pattern;

/**
 * Default {@link Muxer} implementation using Regex.
 */
public class DefaultMuxer implements Muxer {

    private Map<HttpMethod, Map<String, HttpHandler>> methodMap = new HashMap<>();

    public DefaultMuxer() {
        methodMap.put(HttpMethod.CONNECT, new HashMap<>());
        methodMap.put(HttpMethod.DELETE, new HashMap<>());
        methodMap.put(HttpMethod.GET, new HashMap<>());
        methodMap.put(HttpMethod.HEAD, new HashMap<>());
        methodMap.put(HttpMethod.OPTIONS, new HashMap<>());
        methodMap.put(HttpMethod.PATCH, new HashMap<>());
        methodMap.put(HttpMethod.POST, new HashMap<>());
        methodMap.put(HttpMethod.PUT, new HashMap<>());
        methodMap.put(HttpMethod.TRACE, new HashMap<>());
    }

    @VisibleForTesting
    protected DefaultMuxer(Map<HttpMethod, Map<String, HttpHandler>> methodMap) {
        this.methodMap = methodMap;
    }

    /**
     * Registers handler for muxing at resource path and method method.
     * @param method The HTTP method that handler should be tied to.
     * @param path The resource that handler should be tied to.
     * @param handler The {@link com.github.jonbonazza.puni.core.handlers.HttpHandler} that should handle requests at method and path.
     */
    public void handle(HttpMethod method, String path, HttpHandler handler) {
        methodMap.get(method).put(path, handler);
    }

    /**
     * Uses Regex to mux a request with resource resource and method method.
     * @param request The request to mux.
     * @return The {@link com.github.jonbonazza.puni.core.handlers.HttpHandler} that should handle the request.
     */
    public HttpHandler mux(HttpRequest request) {
        Map<String, HttpHandler> handlerMap = methodMap.get(request.getMethod());
        for (Map.Entry<String, HttpHandler> entry : handlerMap.entrySet()) {
            if (Pattern.matches(entry.getKey(), request.getUri().split("\\?")[0]))
                return entry.getValue();
        }

        return null;
    }

    @VisibleForTesting
    protected Map<HttpMethod, Map<String, HttpHandler>> getMethodMap() {
        return methodMap;
    }
}