HTTP/2 Customizer to WildFly Swarm

HTTP2Customizer.java

package org.wildfly.swarm.undertow.runtime;


import javax.enterprise.inject.Any;
import javax.enterprise.inject.Instance;
import javax.inject.Inject;

import org.wildfly.swarm.config.undertow.Server;
import org.wildfly.swarm.config.undertow.server.HTTPListener;
import org.wildfly.swarm.config.undertow.server.HttpsListener;
import org.wildfly.swarm.spi.api.Customizer;
import org.wildfly.swarm.spi.runtime.annotations.ConfigurationValue;
import org.wildfly.swarm.spi.runtime.annotations.Post;
import org.wildfly.swarm.undertow.UndertowFraction;

/**
 * A {@link Customizer} implementation to enable HTTP/2
 *
 * @author <a href="mailto:[email protected]">George Gastaldi</a>
 */
@Post
public class HTTP2Customizer implements Customizer {

    @Inject
    @ConfigurationValue("swarm.http2.enabled")
    Boolean http2Enabled;

    @Inject
    @Any
    private Instance<UndertowFraction> undertowInstance;

    @Override
    public void customize() {
        UndertowFraction undertowFraction = undertowInstance.get();
        if (undertowFraction.isEnableHttp2() || http2Enabled == null || http2Enabled.booleanValue()) {
            Server server = undertowFraction.subresources().server("default-server");
            for (HTTPListener listener : server.subresources().httpListeners()) {
                listener.redirectSocket("default-https").enableHttp2(true);
            }
            for (HttpsListener listener : server.subresources().httpsListeners()) {
                listener.enableHttp2(true);
            }
        }
    }
}