io.gravitee.gateway.standalone.vertx.VertxReactorConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gravitee-apim-gateway-standalone-container Show documentation
The newest version!
/*
 * Copyright © 2015 The Gravitee team (http://gravitee.io)
 *
 * 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.gravitee.gateway.standalone.vertx;

import static io.gravitee.node.vertx.server.VertxServerOptions.TCP_DEFAULT_PORT;
import static io.gravitee.node.vertx.server.http.VertxHttpServerOptions.HTTP_PREFIX;
import static io.gravitee.node.vertx.server.tcp.VertxTcpServerOptions.TCP_PREFIX;

import io.gravitee.gateway.reactive.reactor.HttpRequestDispatcher;
import io.gravitee.gateway.reactive.reactor.TcpSocketDispatcher;
import io.gravitee.gateway.reactive.standalone.vertx.HttpProtocolVerticle;
import io.gravitee.gateway.reactive.standalone.vertx.TcpProtocolVerticle;
import io.gravitee.node.api.server.DefaultServerManager;
import io.gravitee.node.api.server.ServerManager;
import io.gravitee.node.vertx.server.VertxServer;
import io.gravitee.node.vertx.server.VertxServerFactory;
import io.gravitee.node.vertx.server.VertxServerOptions;
import io.gravitee.node.vertx.server.http.VertxHttpServerOptions;
import io.gravitee.node.vertx.server.tcp.VertxTcpServerOptions;
import io.vertx.core.Vertx;
import java.util.Objects;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Scope;
import org.springframework.core.env.Environment;

/**
 * @author David BRASSELY (david at graviteesource.com)
 * @author GraviteeSource Team
 */
@Configuration
public class VertxReactorConfiguration {

    protected static final String SERVERS_PREFIX = "servers";

    @Bean
    public ServerManager serverManager(
        VertxServerFactory, VertxServerOptions> serverFactory,
        Environment environment
    ) {
        int counter = 0;

        final DefaultServerManager serverManager = new DefaultServerManager();
        if (getServerType(environment, SERVERS_PREFIX + "[" + counter + "]") != null) {
            // There is, at least one server configured in the list.
            String prefix = SERVERS_PREFIX + "[" + counter++ + "]";

            while (getServerType(environment, prefix) != null) {
                String property = getServerType(environment, prefix);
                boolean isTcpServer = Objects.equals(property, TCP_PREFIX);
                final VertxServerOptions options = VertxServerOptions
                    .builder(environment, prefix)
                    .defaultPort(isTcpServer ? TCP_DEFAULT_PORT : 8082)
                    .build();
                if (isTcpServer) {
                    assertTcpOptions(options);
                }
                serverManager.register(serverFactory.create(options));
                prefix = SERVERS_PREFIX + "[" + counter++ + "]";
            }
        } else {
            // No server list configured, fallback to single 'http' server configuration.
            final VertxHttpServerOptions httpOptions = VertxHttpServerOptions
                .builder()
                .defaultPort(8082)
                .prefix(HTTP_PREFIX)
                .environment(environment)
                .id("http")
                .build();

            serverManager.register(serverFactory.create(httpOptions));

            boolean tcpEnabled = environment.getProperty("tcp.enabled", Boolean.class, false);
            if (tcpEnabled) {
                // No server list configured, fallback to single 'tcp' server configuration.
                VertxTcpServerOptions tcpOptions = VertxTcpServerOptions
                    .builder()
                    .defaultPort(VertxServerOptions.TCP_DEFAULT_PORT)
                    .prefix(TCP_PREFIX)
                    .environment(environment)
                    .id("tcp")
                    .build();
                assertTcpOptions(tcpOptions);
                serverManager.register(serverFactory.create(tcpOptions));
            }
        }

        return serverManager;
    }

    private void assertTcpOptions(VertxServerOptions options) {
        if (!(options.isSecured() && options.isSni())) {
            throw new IllegalArgumentException("TCP API requires TLS and SNI to be enabled");
        }
    }

    private static String getServerType(Environment environment, String prefix) {
        return environment.getProperty("%s.type".formatted(prefix));
    }

    @Bean
    @Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
    public HttpProtocolVerticle graviteeVerticle(
        ServerManager serverManager,
        @Qualifier("httpRequestDispatcher") HttpRequestDispatcher requestDispatcher
    ) {
        return new HttpProtocolVerticle(serverManager, requestDispatcher);
    }

    @Bean
    @Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
    public TcpProtocolVerticle graviteeTcpVerticle(
        ServerManager serverManager,
        @Qualifier("tcpSocketDispatcher") TcpSocketDispatcher tcpSocketDispatcher
    ) {
        return new TcpProtocolVerticle(serverManager, tcpSocketDispatcher);
    }

    @Bean
    public VertxEmbeddedContainer container(Vertx vertx) {
        return new VertxEmbeddedContainer(vertx);
    }
}