org.apache.camel.component.undertow.DefaultUndertowHost Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-undertow Show documentation
There is a newer version: 4.9.0
/*
 * 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.camel.component.undertow;

import java.net.URI;

import jakarta.servlet.ServletException;

import io.undertow.Undertow;
import io.undertow.UndertowOptions;
import io.undertow.server.HttpHandler;
import io.undertow.servlet.Servlets;
import io.undertow.servlet.api.DeploymentInfo;
import io.undertow.servlet.api.DeploymentManager;
import org.apache.camel.component.undertow.handlers.CamelRootHandler;
import org.apache.camel.component.undertow.handlers.NotFoundHandler;
import org.apache.camel.component.undertow.handlers.RestRootHandler;
import org.apache.camel.component.undertow.spi.UndertowSecurityProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * The default UndertowHost which manages standalone Undertow server.
 */
public class DefaultUndertowHost implements UndertowHost {
    private static final Logger LOG = LoggerFactory.getLogger(DefaultUndertowHost.class);

    private final UndertowHostKey key;
    private final UndertowHostOptions options;
    private final CamelRootHandler rootHandler;
    private final RestRootHandler restHandler;
    private Undertow undertow;
    private String hostString;
    private DeploymentManager deploymentManager;

    public DefaultUndertowHost(UndertowHostKey key) {
        this(key, null);
    }

    public DefaultUndertowHost(UndertowHostKey key, UndertowHostOptions options) {
        this.key = key;
        this.options = options;
        this.rootHandler = new CamelRootHandler(new NotFoundHandler());
        this.restHandler = new RestRootHandler();
        this.restHandler.init(key.getPort());
    }

    @Override
    public void validateEndpointURI(URI httpURI) {
        // all URIs are good
    }

    @Override
    public synchronized HttpHandler registerHandler(
            UndertowConsumer consumer, HttpHandlerRegistrationInfo registrationInfo, HttpHandler handler) {
        if (undertow == null) {
            Undertow.Builder builder = Undertow.builder();
            if (key.getSslContext() != null) {
                builder.addHttpsListener(key.getPort(), key.getHost(), key.getSslContext());
            } else {
                builder.addHttpListener(key.getPort(), key.getHost());
            }

            if (options != null) {
                if (options.getIoThreads() != null) {
                    builder.setIoThreads(options.getIoThreads());
                }
                if (options.getWorkerThreads() != null) {
                    builder.setWorkerThreads(options.getWorkerThreads());
                }
                if (options.getBufferSize() != null) {
                    builder.setBufferSize(options.getBufferSize());
                }
                if (options.getDirectBuffers() != null) {
                    builder.setDirectBuffers(options.getDirectBuffers());
                }
                if (options.getHttp2Enabled() != null) {
                    builder.setServerOption(UndertowOptions.ENABLE_HTTP2, options.getHttp2Enabled());
                }
            }

            if (consumer != null && consumer.isRest()) {
                // use the rest handler as its a rest consumer
                undertow = registerHandler(consumer, builder, restHandler);
            } else {
                undertow = registerHandler(consumer, builder, rootHandler);
            }
            LOG.info("Starting Undertow server on {}://{}:{}", key.getSslContext() != null ? "https" : "http", key.getHost(),
                    key.getPort());

            try {
                // If there is an exception while starting up, Undertow wraps it
                // as RuntimeException which leaves the consumer in an inconsistent
                // state as a subsequent start if the route (i.e. manually) won't
                // start the Undertow instance as undertow is not null.
                undertow.start();
            } catch (RuntimeException e) {
                LOG.warn("Failed to start Undertow server on {}://{}:{}, reason: {}",
                        key.getSslContext() != null ? "https" : "http", key.getHost(), key.getPort(), e.getMessage());

                // Cleanup any resource that may have been created during start
                // and reset the instance so a subsequent start will trigger the
                // initialization again.
                undertow.stop();
                undertow = null;

                throw e;
            }
        }
        if (consumer != null && consumer.isRest()) {
            restHandler.addConsumer(consumer);
            return restHandler;
        } else {
            return rootHandler.add(registrationInfo.getUri().getPath(), registrationInfo.getMethodRestrict(),
                    registrationInfo.isMatchOnUriPrefix(), handler);
        }
    }

    private Undertow registerHandler(UndertowConsumer consumer, Undertow.Builder builder, HttpHandler handler) {
        UndertowSecurityProvider securityProvider = consumer == null
                ? null
                : consumer.getEndpoint().getComponent().getSecurityProvider() != null
                        ? consumer.getEndpoint().getComponent().getSecurityProvider()
                : consumer.getEndpoint().getSecurityProvider();
        //if security provider needs servlet context, start empty servlet
        if (securityProvider != null && securityProvider.requireServletContext()) {
            DeploymentInfo deployment = Servlets.deployment()
                    .setContextPath("")
                    .setDisplayName("application")
                    .setDeploymentName("camel-undertow")
                    .setClassLoader(getClass().getClassLoader())
                    //httpHandler for servlet is ignored, camel handler is used instead of it
                    .addOuterHandlerChainWrapper(h -> handler);

            deploymentManager = Servlets.newContainer().addDeployment(deployment);
            deploymentManager.deploy();
            try {
                return builder.setHandler(deploymentManager.start()).build();
            } catch (ServletException e) {
                LOG.warn("Failed to start Undertow server on {}://{}:{}, reason: {}",
                        key.getSslContext() != null ? "https" : "http", key.getHost(), key.getPort(), e.getMessage());

                throw new RuntimeException(e);
            }

        }

        return builder.setHandler(handler).build();
    }

    @Override
    public synchronized void unregisterHandler(UndertowConsumer consumer, HttpHandlerRegistrationInfo registrationInfo) {
        if (undertow == null) {
            return;
        }

        boolean stop;
        if (consumer != null && consumer.isRest()) {
            restHandler.removeConsumer(consumer);
            stop = restHandler.consumers() <= 0;
        } else {
            rootHandler.remove(registrationInfo.getUri().getPath(), registrationInfo.getMethodRestrict(),
                    registrationInfo.isMatchOnUriPrefix());
            stop = rootHandler.isEmpty();
        }
        if (deploymentManager != null) {
            deploymentManager.undeploy();
        }

        if (stop) {
            LOG.info("Stopping Undertow server on {}://{}:{}", key.getSslContext() != null ? "https" : "http", key.getHost(),
                    key.getPort());
            undertow.stop();
            undertow = null;
        }
    }

    @Override
    public String toString() {
        if (hostString == null) {
            hostString = String.format("DefaultUndertowHost[%s://%s:%s]", key.getSslContext() != null ? "https" : "http",
                    key.getHost(), key.getPort());
        }
        return hostString;
    }
}