io.joshworks.snappy.handler.HandlerManager Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2017 Josue Gontijo
*
* 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
*
* 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 io.joshworks.snappy.handler;
import io.joshworks.snappy.admin.AdminManager;
import io.joshworks.snappy.metric.RestMetricHandler;
import io.joshworks.snappy.rest.Interceptor;
import io.undertow.Handlers;
import io.undertow.predicate.Predicate;
import io.undertow.predicate.Predicates;
import io.undertow.server.HttpHandler;
import io.undertow.server.RoutingHandler;
import io.undertow.server.handlers.PathTemplateHandler;
import io.undertow.server.handlers.PredicateHandler;
import io.undertow.util.HeaderValues;
import io.undertow.util.Headers;
import java.util.ArrayList;
import java.util.List;
/**
* Created by josh on 3/11/17.
*/
public class HandlerManager {
//chain of responsibility
public HttpHandler createRootHandler(
List mappedEndpoints,
List rootInterceptors,
AdminManager adminManager,
String basePath,
boolean httpMetrics,
boolean httpTracer) {
final List metricsHandlers = new ArrayList<>();
final RoutingHandler routingRestHandler = new TrailingSlashRoutingHandler();
final PathTemplateHandler websocketHandler = Handlers.pathTemplate();
HttpHandler staticHandler = null;
for (MappedEndpoint me : mappedEndpoints) {
//TODO clean this up
//TODO at the moment only rest has metrics
if (MappedEndpoint.Type.REST.equals(me.type)) {
String endpointPath = HandlerUtil.BASE_PATH.equals(basePath) ? me.url : basePath + me.url;
if (httpMetrics) {
RestMetricHandler restMetricHandler = new RestMetricHandler(me.method, endpointPath, me.handler);
metricsHandlers.add(restMetricHandler);
routingRestHandler.add(me.method, endpointPath, restMetricHandler);
} else {
routingRestHandler.add(me.method, endpointPath, me.handler);
}
}
if (MappedEndpoint.Type.MULTIPART.equals(me.type)) {
routingRestHandler.add(me.method, me.url, me.handler);
}
if (MappedEndpoint.Type.SSE.equals(me.type)) {
routingRestHandler.add(me.method, me.url, me.handler);
}
if (MappedEndpoint.Type.WS.equals(me.type)) {
websocketHandler.add(me.url, me.handler);
}
if (MappedEndpoint.Type.STATIC.equals(me.type)) {
staticHandler = me.handler;
}
}
adminManager.registerMetrics(metricsHandlers);
HttpHandler root = resolveHandlers(routingRestHandler, websocketHandler, staticHandler, mappedEndpoints);
if(!rootInterceptors.isEmpty()) {
InterceptorHandler interceptorHandler = new InterceptorHandler(rootInterceptors);
interceptorHandler.setNext(root);
root = interceptorHandler;
}
HttpHandler handler = httpTracer ? Handlers.requestDump(root) : root;
return Handlers.gracefulShutdown(handler);
}
private HttpHandler resolveHandlers(HttpHandler rest, HttpHandler ws, HttpHandler file, List mappedEndpoints) {
PredicateHandler websocketRestResolved = Handlers.predicate(value -> {
HeaderValues upgradeHeader = value.getRequestHeaders().get(Headers.UPGRADE);
return upgradeHeader != null && upgradeHeader.stream().anyMatch(v -> v.equalsIgnoreCase("websocket"));
}, ws, rest);
if (file != null) {
String[] mappedServices = HandlerUtil.removePathTemplate(mappedEndpoints);
Predicate mappedPredicate = Predicates.prefixes(mappedServices);
return Handlers.predicate(mappedPredicate, websocketRestResolved, file);
}
return websocketRestResolved;
}
}