/*
* Copyright 2009 the original author or authors.
*
* 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 org.gradle.api.plugins.jetty.internal;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.LineNumberReader;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import org.mortbay.jetty.Server;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Monitor <p/> Listens for stop commands eg via mvn jetty:stop and causes jetty to stop either by exiting the jvm, or
* by stopping the Server instances. The choice of behaviour is controlled by either passing true (exit jvm) or false
* (stop Servers) in the constructor.
*/
public class Monitor extends Thread {
private static final Logger LOGGER = LoggerFactory.getLogger(Monitor.class);
private String key;
private Server[] servers;
ServerSocket serverSocket;
boolean kill;
public Monitor(int port, String key, Server[] servers, boolean kill) throws IOException {
if (port <= 0) {
throw new IllegalStateException("Bad stop port");
}
if (key == null) {
throw new IllegalStateException("Bad stop key");
}
this.key = key;
this.servers = servers;
this.kill = kill;
setDaemon(true);
setName("StopJettyPluginMonitor");
serverSocket = new ServerSocket(port, 1, InetAddress.getByName("127.0.0.1"));
serverSocket.setReuseAddress(true);
}
public void run() {
while (serverSocket != null) {
Socket socket = null;
try {
socket = serverSocket.accept();
socket.setSoLinger(false, 0);
LineNumberReader lin = new LineNumberReader(new InputStreamReader(socket.getInputStream()));
String key = lin.readLine();
if (!this.key.equals(key)) {
continue;
}
String cmd = lin.readLine();
if ("stop".equals(cmd)) {
try {
socket.close();
} catch (Exception e) {
LOGGER.debug("Exception when stopping server", e);
}
try {
socket.close();
} catch (Exception e) {
LOGGER.debug("Exception when stopping server", e);
}
try {
serverSocket.close();
} catch (Exception e) {
LOGGER.debug("Exception when stopping server", e);
}
serverSocket = null;
if (kill) {
LOGGER.info("Killing Jetty");
System.exit(0);
} else {
for (int i = 0; servers != null && i < servers.length; i++) {
try {
LOGGER.info("Stopping server " + i);
servers[i].stop();
} catch (Exception e) {
LOGGER.error("Exception when stopping server", e);
}
}
}
} else {
LOGGER.info("Unsupported monitor operation");
}
} catch (Exception e) {
LOGGER.error("Exception during monitoring Server", e);
} finally {
if (socket != null) {
try {
socket.close();
} catch (Exception e) {
LOGGER.debug("Exception when stopping server", e);
}
}
socket = null;
}
}
}
}