77 lines
1.9 KiB
Java
77 lines
1.9 KiB
Java
package com.dayrain.component;
|
|
|
|
import com.sun.net.httpserver.HttpServer;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import java.io.IOException;
|
|
import java.net.InetSocketAddress;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
/**
|
|
* 服务
|
|
* @author peng
|
|
* @date 2021/10/25
|
|
*/
|
|
public class Server implements Runnable {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(Server.class);
|
|
|
|
private final ServerConfig serverConfig;
|
|
|
|
private HttpServer httpServer;
|
|
|
|
public Server(ServerConfig serverConfig) {
|
|
this.serverConfig = serverConfig;
|
|
}
|
|
|
|
|
|
@Override
|
|
public void run() {
|
|
start();
|
|
System.out.println("【" + serverConfig.getServerName() + "】服务已开启...");
|
|
}
|
|
|
|
public synchronized void start() {
|
|
try {
|
|
this.httpServer = HttpServer.create(new InetSocketAddress(serverConfig.getPort()), 0);
|
|
httpServer.setExecutor(Executors.newCachedThreadPool());
|
|
for (ServerUrl serverUrl : serverConfig.getServerUrls()) {
|
|
addContext(serverUrl);
|
|
}
|
|
httpServer.start();
|
|
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
logger.error(e.getMessage());
|
|
}
|
|
|
|
}
|
|
|
|
public synchronized void stop() {
|
|
if (httpServer != null) {
|
|
System.out.println("【" + serverConfig.getServerName() + "】服务已关闭...");
|
|
httpServer.stop(0);
|
|
}
|
|
}
|
|
|
|
public synchronized void restart() {
|
|
stop();
|
|
try {
|
|
TimeUnit.SECONDS.sleep(2);
|
|
} catch (InterruptedException e) {
|
|
e.printStackTrace();
|
|
}
|
|
start();
|
|
}
|
|
|
|
public synchronized void addContext(ServerUrl serverUrl) {
|
|
httpServer.createContext(serverUrl.getUrl(), new RequestHandler(serverUrl));
|
|
}
|
|
|
|
public void removeContext(String url) {
|
|
httpServer.removeContext(url);
|
|
}
|
|
}
|