/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 org.apache.ambari.infra;
import org.apache.ambari.infra.conf.InfraManagerConfig;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.HttpConfiguration;
import org.eclipse.jetty.server.HttpConnectionFactory;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.server.handler.HandlerList;
import org.eclipse.jetty.server.handler.ResourceHandler;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.eclipse.jetty.util.resource.Resource;
import org.eclipse.jetty.util.resource.ResourceCollection;
import org.eclipse.jetty.webapp.WebAppContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.request.RequestContextListener;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import static org.apache.ambari.infra.common.InfraManagerConstants.DEFAULT_PORT;
import static org.apache.ambari.infra.common.InfraManagerConstants.DEFAULT_PROTOCOL;
import static org.apache.ambari.infra.common.InfraManagerConstants.INFRA_MANAGER_SESSION_ID;
import static org.apache.ambari.infra.common.InfraManagerConstants.PROTOCOL_SSL;
import static org.apache.ambari.infra.common.InfraManagerConstants.ROOT_CONTEXT;
import static org.apache.ambari.infra.common.InfraManagerConstants.SESSION_TIMEOUT;
import static org.apache.ambari.infra.common.InfraManagerConstants.WEB_RESOURCE_FOLDER;
public class InfraManager {
private static final Logger LOG = LoggerFactory.getLogger(InfraManager.class);
public static void main(String[] args) {
Options options = new Options();
HelpFormatter helpFormatter = new HelpFormatter();
helpFormatter.setDescPadding(10);
helpFormatter.setWidth(200);
final Option helpOption = Option.builder("h")
.longOpt("help")
.desc("Print commands")
.build();
final Option portOption = Option.builder("p")
.longOpt("port")
.desc("Infra Manager port")
.numberOfArgs(1)
.argName("port_number")
.build();
final Option protocolOption = Option.builder("t")
.longOpt("tls-enabled")
.desc("TLS enabled for Infra Manager")
.build();
options.addOption(helpOption);
options.addOption(portOption);
options.addOption(protocolOption);
try {
CommandLineParser cmdLineParser = new DefaultParser();
CommandLine cli = cmdLineParser.parse(options, args);
int port = cli.hasOption('p') ? Integer.parseInt(cli.getOptionValue('p')) : DEFAULT_PORT;
String protocol = cli.hasOption("t") ? PROTOCOL_SSL : DEFAULT_PROTOCOL;
Server server = buildServer(port, protocol);
HandlerList handlers = new HandlerList();
handlers.addHandler(createSwaggerContext());
handlers.addHandler(createBaseWebappContext());
server.setHandler(handlers);
server.start();
LOG.debug("============================Server Dump=======================================");
LOG.debug(server.dump());
LOG.debug("==============================================================================");
server.join();
} catch (Exception e) {
// TODO
e.printStackTrace();
}
}
private static Server buildServer(int port, String protocol) {
Server server = new Server();
HttpConfiguration httpConfiguration = new HttpConfiguration();
httpConfiguration.setRequestHeaderSize(65535);
// TODO: tls
ServerConnector connector = new ServerConnector(server, new HttpConnectionFactory(httpConfiguration));
connector.setPort(port);
server.setConnectors(new Connector[]{connector});
URI infraManagerURI = URI.create(String.format("%s://0.0.0.0:%s", protocol, String.valueOf(port)));
LOG.info("Starting infra manager URI=" + infraManagerURI);
return server;
}
private static WebAppContext createBaseWebappContext() throws MalformedURLException {
URI webResourceBase = findWebResourceBase();
WebAppContext context = new WebAppContext();
context.setBaseResource(Resource.newResource(webResourceBase));
context.setContextPath(ROOT_CONTEXT);
context.setParentLoaderPriority(true);
// Configure Spring
context.addEventListener(new ContextLoaderListener());
context.addEventListener(new RequestContextListener());
// TODO: security, add: context.addFilter(new FilterHolder(new DelegatingFilterProxy("springSecurityFilterChain")), "/*", EnumSet.allOf(DispatcherType.class));
context.setInitParameter("contextClass", AnnotationConfigWebApplicationContext.class.getName());
context.setInitParameter("contextConfigLocation", InfraManagerConfig.class.getName());
// Configure Jersey
ServletHolder jerseyServlet = context.addServlet(org.glassfish.jersey.servlet.ServletContainer.class, "/api/v1/*");
jerseyServlet.setInitOrder(1);
jerseyServlet.setInitParameter("jersey.config.server.provider.packages","org.apache.ambari.infra.rest,io.swagger.jaxrs.listing");
context.getSessionHandler().getSessionManager().setMaxInactiveInterval(SESSION_TIMEOUT);
context.getSessionHandler().getSessionManager().getSessionCookieConfig().setName(INFRA_MANAGER_SESSION_ID);
return context;
}
private static URI findWebResourceBase() {
URL fileCompleteUrl = Thread.currentThread().getContextClassLoader().getResource(WEB_RESOURCE_FOLDER);
String errorMessage = "Web Resource Folder " + WEB_RESOURCE_FOLDER + " not found in classpath";
if (fileCompleteUrl != null) {
try {
return fileCompleteUrl.toURI().normalize();
} catch (URISyntaxException e) {
LOG.error(errorMessage, e);
System.exit(1);
}
} else {
LOG.error(errorMessage);
System.exit(1);
}
throw new IllegalStateException(errorMessage);
}
private static ServletContextHandler createSwaggerContext() throws URISyntaxException {
ResourceHandler resourceHandler = new ResourceHandler();
ResourceCollection resources = new ResourceCollection(new String[] {
InfraManager.class.getClassLoader()
.getResource("META-INF/resources/webjars/swagger-ui/2.1.0")
.toURI().toString(),
InfraManager.class.getClassLoader()
.getResource("swagger")
.toURI().toString()
});
resourceHandler.setBaseResource(resources);
resourceHandler.setWelcomeFiles(new String[]{"swagger.html"}); // rewrite index.html from swagger-ui webjar
ServletContextHandler context = new ServletContextHandler();
context.setContextPath("/docs/");
context.setHandler(resourceHandler);
return context;
}
}