org.apache.tomee.bootstrap.Server Maven / Gradle / Ivy
/*
* 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.tomee.bootstrap;
import org.apache.catalina.startup.Catalina;
import org.apache.juli.logging.Log;
import org.apache.juli.logging.LogFactory;
import org.apache.openejb.loader.Files;
import org.apache.openejb.loader.IO;
import java.io.File;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.net.URI;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.List;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
public class Server {
private static final Log log = LogFactory.getLog(Server.class);
private final File home;
private final URI uri;
public Server(final File home, final int port) {
this.home = home;
this.uri = URI.create("http://localhost:" + port);
}
public URI getURI() {
return uri;
}
public File getHome() {
return home;
}
private static void cp(final File conf, final String resource) {
try {
final URL url = resolve(resource);
IO.copy(IO.read(url), new File(conf, resource));
} catch (IOException e) {
// todo add more detail
throw new UncheckedIOException(e);
}
}
private static URL resolve(final String resource) throws IOException {
final ClassLoader loader = Thread.currentThread().getContextClassLoader();
final Enumeration resources = loader.getResources("tomee/conf/" + resource);
final List list = Collections.list(resources);
if (list.size() == 0) {
throw new MissingResourceException(resource);
}
if (list.size() == 1) {
return list.get(0);
}
sort(list);
return list.get(0);
}
public static void sort(final List list) {
Collections.sort(list, Server::compare);
}
private static int compare(final URL o1, final URL o2) {
final String a = o1.toExternalForm();
final String b = o2.toExternalForm();
int modifier = 0;
if (a.contains("tomee-bootstrap")) modifier += 1000;
if (b.contains("tomee-bootstrap")) modifier -= 1000;
return a.compareTo(b) + modifier;
}
public static class MissingResourceException extends RuntimeException {
public MissingResourceException(final String message) {
super(message);
}
}
public static Builder builder() {
return new Builder();
}
public static class Builder {
private int httpPort;
private int shutdownPort;
private int ajpPort;
protected final ArrayList> homeConsumers = new ArrayList<>();
protected final ArrayList> builderConsumers = new ArrayList<>();
protected final Archive modifications = Archive.archive();
public Builder httpPort(final int port) {
this.httpPort = port;
return this;
}
public Builder ajpPort(final int port) {
this.ajpPort = port;
return this;
}
public Builder shutdownPort(final int port) {
this.shutdownPort = port;
return this;
}
public Builder add(final String destinationPath, final byte[] bytes) {
modifications.add(destinationPath, bytes);
return this;
}
public Builder add(final String destinationPath, final Supplier content) {
modifications.add(destinationPath, content);
return this;
}
public Builder add(final String destinationPath, final String content) {
modifications.add(destinationPath, content);
return this;
}
public Builder add(final String destinationPath, final File content) {
modifications.add(destinationPath, content);
return this;
}
public Builder add(final String name, final Archive contents) {
modifications.add(name, contents);
return this;
}
public Builder home(final Consumer customization) {
homeConsumers.add(customization);
return this;
}
public Builder and(final Consumer consumer) {
this.builderConsumers.add(consumer);
return this;
}
protected void applyHomeConsumers(final File home) {
// run any customization logic that's been added
for (final Consumer customization : homeConsumers) {
customization.accept(home);
}
}
protected void applyModifications(final File home) {
// copy user files
try {
modifications.toDir(home);
} catch (Exception e) {
throw new IllegalStateException("Failed to apply home modifications to " + home.getAbsolutePath(), e);
}
}
protected void applyBuilderConsumers() {
for (final Consumer consumer : builderConsumers) {
consumer.accept((Builder) this);
}
}
public Server build() {
final long start = System.currentTimeMillis();
applyBuilderConsumers();
final File home = Files.mkdir(Files.tmpdir(), "apache-tomee");
final File conf = Files.mkdir(home, "conf");
final File logs = Files.mkdir(home, "logs");
final File webapps = Files.mkdir(home, "webapps");
cp(conf, "catalina.policy");
cp(conf, "catalina.properties");
cp(conf, "context.xml");
cp(conf, "jaspic-providers.xml");
cp(conf, "jaspic-providers.xsd");
cp(conf, "logging.properties");
cp(conf, "server.xml");
cp(conf, "system.properties");
cp(conf, "tomcat-users.xml");
cp(conf, "tomcat-users.xsd");
cp(conf, "tomee.xml");
cp(conf, "web.xml");
applyModifications(home);
final Iterator ports = Ports.allocate(3).iterator();
final int http = httpPort > 0 ? httpPort : ports.next();
final int shutdown = shutdownPort > 0 ? shutdownPort : ports.next();
final int ajp = ajpPort > 0 ? ajpPort : ports.next();
try { // apply modifications to server.xml
final File serverxml = new File(conf, "server.xml");
final String content = setPorts(http, shutdown, ajp)
.andThen(this::addServerListener)
.andThen(this::setUtilityThreadsAsDaemon)
.apply(IO.slurp(serverxml));
IO.copy(IO.read(content), serverxml);
} catch (final IOException e) {
throw new UncheckedIOException("Unable to modify server.xml", e);
}
applyHomeConsumers(home);
System.setProperty("catalina.home", home.getAbsolutePath());
System.setProperty("catalina.base", home.getAbsolutePath());
final URLClassLoader loader = new URLClassLoader(new URL[0], Server.class.getClassLoader());
final Catalina catalina = new Catalina();
catalina.setParentClassLoader(loader);
catalina.setAwait(false);
catalina.load();
catalina.start();
final long elapsed = System.currentTimeMillis() - start;
final String message = "Full bootstrap in [" + elapsed + "] milliseconds";
log.info(message);
return new Server(home, http);
}
private Function setPorts(final int http, final int shutdown, final int ajp) {
return s -> s.replace("8080", http + "")
.replace("8005", shutdown + "")
.replace("8009", ajp + "");
}
private String setUtilityThreadsAsDaemon(final String serverXml) {
// Normalize by removing any setting of utilityThreadsAsDaemon
// Then explicitly set utilityThreadsAsDaemon to true
return serverXml
.replace("utilityThreadsAsDaemon=\"true\"", "")
.replace("utilityThreadsAsDaemon=\"false\"", "")
.replace("shutdown=\"SHUTDOWN\"", "shutdown=\"SHUTDOWN\" utilityThreadsAsDaemon=\"true\"");
}
private String addServerListener(final String serverXml) {
if (serverXml.contains("\n © 2015 - 2025 Weber Informatics LLC | Privacy Policy