org.jgroups.ping.common.server.UndertowServer Maven / Gradle / Ivy

Go to download
/**
 *  Copyright 2014 Red Hat, Inc.
 *
 *  Red Hat 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.jgroups.ping.common.server;

import java.io.InputStream;

import org.jgroups.JChannel;

import io.undertow.Undertow;
import io.undertow.server.HttpHandler;
import io.undertow.server.HttpServerExchange;

/**
 * @author Ales Justin
 */
public class UndertowServer extends AbstractServer {
    private Undertow server;

    public UndertowServer(int port) {
        super(port);
    }

    public synchronized boolean start(JChannel channel) throws Exception {
        boolean started = false;
        if (server == null) {
            try {
                Undertow.Builder builder = Undertow.builder();
                builder.addHttpListener(port, "0.0.0.0");
                builder.setHandler(new Handler(this));
                server = builder.build();
                server.start();
                started = true;
            } catch (Exception e) {
                server = null;
                throw e;
            }
        }
        addChannel(channel);
        return started;
    }

    public synchronized boolean stop(JChannel channel) {
        boolean stopped = false;
        removeChannel(channel);
        if (server != null && !hasChannels()) {
            try {
                server.stop();
                stopped = true;
            } finally {
                server = null;
            }
        }
        return stopped;
    }

    private class Handler implements HttpHandler {
        private final Server server;

        private Handler(Server server) {
            this.server = server;
        }

        public void handleRequest(HttpServerExchange exchange) throws Exception {
            if(exchange.isInIoThread()) {
                exchange.dispatch(this);
                return;
            }

            exchange.startBlocking();
            String clusterName = exchange.getRequestHeaders().getFirst(CLUSTER_NAME);
            JChannel channel = server.getChannel(clusterName);
            try (InputStream stream = exchange.getInputStream()) {
                handlePingRequest(channel, stream);
            }
        }
    }
}