Java tutorial
/* * JBoss, Home of Professional Open Source. * Copyright 2015 Red Hat, Inc., and individual contributors * as indicated by the @author tags. * * 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 org.jboss.pnc.buildagent.server; import com.fasterxml.jackson.databind.ObjectMapper; import io.undertow.Handlers; import io.undertow.Undertow; import io.undertow.server.HttpHandler; import io.undertow.server.HttpServerExchange; import io.undertow.server.handlers.PathHandler; import io.undertow.server.handlers.ResponseCodeHandler; import io.undertow.servlet.api.DeploymentInfo; import io.undertow.servlet.api.DeploymentManager; import org.jboss.pnc.buildagent.server.servlet.Download; import org.jboss.pnc.buildagent.server.servlet.Terminal; import org.jboss.pnc.buildagent.server.servlet.Upload; import org.jboss.pnc.buildagent.server.servlet.Welcome; import org.jboss.pnc.buildagent.server.termserver.Configurations; import org.jboss.pnc.buildagent.server.termserver.ReadOnlyChannel; import org.jboss.pnc.buildagent.server.termserver.UndertowBootstrap; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import javax.servlet.ServletException; import java.io.IOException; import java.net.URL; import java.util.Enumeration; import java.util.Optional; import java.util.concurrent.ScheduledExecutorService; import java.util.function.Consumer; import java.util.jar.Manifest; import static io.undertow.servlet.Servlets.defaultContainer; import static io.undertow.servlet.Servlets.deployment; import static io.undertow.servlet.Servlets.servlet; /** * @author <a href="mailto:matejonnet@gmail.com">Matej Lazar</a> */ public class BootstrapUndertowBuildAgentHandlers extends UndertowBootstrap { private final String bindPath; Logger log = LoggerFactory.getLogger(BootstrapUndertowBuildAgentHandlers.class); private Undertow server; public BootstrapUndertowBuildAgentHandlers(String host, int port, ScheduledExecutorService executor, String bindPath, Optional<ReadOnlyChannel> ioLoggerChannel) { super(host, port, executor, ioLoggerChannel); this.bindPath = bindPath; } public void bootstrap(final Consumer<Boolean> completionHandler) throws BuildAgentException { String servletPath = bindPath + "/servlet"; String socketPath = bindPath + "/socket"; String httpPath = bindPath + "/"; DeploymentInfo servletBuilder = deployment() .setClassLoader(BootstrapUndertowBuildAgentHandlers.class.getClassLoader()) .setContextPath(servletPath).setDeploymentName("ROOT.war") .addServlets(servlet("WelcomeServlet", Welcome.class).addMapping("/"), servlet("TerminalServlet", Terminal.class).addMapping("/terminal/*"), servlet("UploaderServlet", Upload.class).addMapping("/upload/*"), servlet("DownloaderServlet", Download.class).addMapping("/download/*")); DeploymentManager manager = defaultContainer().addDeployment(servletBuilder); manager.deploy(); HttpHandler servletHandler = null; try { servletHandler = manager.start(); } catch (ServletException e) { throw new BuildAgentException("Cannot deploy servlets.", e); } PathHandler pathHandler = Handlers.path().addPrefixPath(servletPath, servletHandler) .addPrefixPath(socketPath, exchange -> BootstrapUndertowBuildAgentHandlers.this.handleWebSocketRequests(exchange, socketPath)) .addPrefixPath(httpPath, exchange -> BootstrapUndertowBuildAgentHandlers.this .handleHttpRequests(exchange, httpPath)); server = Undertow.builder().addHttpListener(getPort(), getHost()).setHandler(pathHandler).build(); server.start(); completionHandler.accept(true); } private void handleWebSocketRequests(HttpServerExchange exchange, String socketPath) throws Exception { socketPath = stripEndingSlash(socketPath); super.handleWebSocketRequests(exchange, socketPath + Configurations.TERM_PATH, socketPath + Configurations.TERM_PATH_TEXT, socketPath + Configurations.PROCESS_UPDATES_PATH); } private void handleHttpRequests(HttpServerExchange exchange, String httpPath) throws Exception { String requestPath = exchange.getRequestPath(); if (pathMatches(requestPath, httpPath)) { log.debug("Welcome handler requested."); String message = "Welcome to PNC Build Agent (" + getManifestInformation() + ")"; message += "\nVisit /servlet/terminal/ for demo console."; exchange.getResponseSender().send(message); return; } if (pathMatches(requestPath, httpPath + "processes")) { log.debug("Processes handler requested."); getProcessActiveTerms().handleRequest(exchange); return; } ResponseCodeHandler.HANDLE_404.handleRequest(exchange); } private String stripEndingSlash(String requestPath) { if (requestPath.endsWith("/")) { requestPath = requestPath.substring(0, requestPath.length() - 1); } return requestPath; } private boolean pathMatches(String requestPath, String path) { return requestPath.equals(path) || (requestPath + "/").equals(path); } private HttpHandler getProcessActiveTerms() { return exchange -> { ObjectMapper mapper = new ObjectMapper(); String jsonString = mapper.writeValueAsString(getTerms().keySet()); exchange.getResponseSender().send(jsonString); }; } private String getManifestInformation() { String result = ""; try { final Enumeration<URL> resources = Welcome.class.getClassLoader().getResources("META-INF/MANIFEST.MF"); while (resources.hasMoreElements()) { final URL jarUrl = resources.nextElement(); log.trace("Processing jar resource " + jarUrl); if (jarUrl.getFile().contains("build-agent")) { final Manifest manifest = new Manifest(jarUrl.openStream()); result = manifest.getMainAttributes().getValue("Implementation-Version"); result += " ( SHA: " + manifest.getMainAttributes().getValue("Scm-Revision") + " ) "; break; } } } catch (final IOException e) { log.trace("Error retrieving information from manifest", e); } return result; } }