package com.firefly.net.tcp;
import static com.firefly.net.tcp.TcpPerformanceParameter.BACKLOG;
import static com.firefly.net.tcp.TcpPerformanceParameter.BANDWIDTH;
import static com.firefly.net.tcp.TcpPerformanceParameter.CONNECTION_TIME;
import static com.firefly.net.tcp.TcpPerformanceParameter.LATENCY;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;
import com.firefly.net.Config;
import com.firefly.net.Decoder;
import com.firefly.net.Encoder;
import com.firefly.net.EventManager;
import com.firefly.net.Handler;
import com.firefly.net.Server;
import com.firefly.net.Worker;
import com.firefly.net.event.DefaultEventManager;
import com.firefly.net.exception.NetException;
import com.firefly.utils.log.Log;
import com.firefly.utils.log.LogFactory;
import com.firefly.utils.time.Millisecond100Clock;
public class TcpServer implements Server {
private static Log log = LogFactory.getInstance().getLog("firefly-system");
private Config config;
private Worker[] workers;
private Thread bossThread;
private boolean start;
public TcpServer() {
}
public TcpServer(Decoder decoder, Encoder encoder, Handler handler) {
config = new Config();
config.setDecoder(decoder);
config.setEncoder(encoder);
config.setHandler(handler);
}
public TcpServer(Decoder decoder, Encoder encoder, Handler handler, int timeout) {
config = new Config();
config.setDecoder(decoder);
config.setEncoder(encoder);
config.setHandler(handler);
config.setTimeout(timeout);
}
@Override
public void setConfig(Config config) {
this.config = config;
}
@Override
public void start(String host, int port) {
if (config == null)
throw new NetException("server config is null");
log.debug(config.toString());
listen(bind(host, port));
}
private ServerSocketChannel bind(String host, int port) {
ServerSocketChannel serverSocketChannel = null;
try {
serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
serverSocketChannel.socket().setReuseAddress(true);
serverSocketChannel.socket().setPerformancePreferences(CONNECTION_TIME, LATENCY, BANDWIDTH);
log.debug("ServerSocket receiveBufferSize: [{}]", serverSocketChannel.socket().getReceiveBufferSize());
serverSocketChannel.socket().bind(new InetSocketAddress(host, port), BACKLOG);
} catch (Exception e) {
log.error("ServerSocket bind error", e);
}
return serverSocketChannel;
}
private void listen(ServerSocketChannel serverSocketChannel) {
EventManager eventManager = new DefaultEventManager(config);
log.info("server worker num: {}", config.getWorkerThreads());
workers = new Worker[config.getWorkerThreads()];
for (int i = 0; i < config.getWorkerThreads(); i++)
workers[i] = new TcpWorker(config, i, eventManager);
Boss boss = null;
try {
boss = new Boss(serverSocketChannel);
} catch (IOException e) {
log.error("Boss create error", e);
}
bossThread = new Thread(boss, config.getServerName());
start = true;
bossThread.start();
}
private final class Boss implements Runnable {
private final Selector selector;
public Boss(ServerSocketChannel serverSocketChannel) throws IOException {
selector = Selector.open();
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
}
@Override
public void run() {
int sessionId = 0;
try {
while (start) {
try {
selector.select(1000);
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while (iterator.hasNext()) {
SelectionKey key = iterator.next();
iterator.remove();
if (key.isAcceptable()) {
ServerSocketChannel server = (ServerSocketChannel) key.channel();
SocketChannel socketChannel = server.accept();
if (socketChannel != null) {
accept(socketChannel, sessionId);
sessionId++;
}
}
}
} catch (ClosedChannelException e) {
// Closed as requested.
break;
} catch (Throwable e) {
log.error("Failed to accept a connection.", e);
}
}
} finally {
try {
selector.close();
} catch (Exception e) {
log.error("Failed to close a selector.", e);
}
}
}
public void accept(SocketChannel socketChannel, int sessionId) {
try {
int workerIndex = Math.abs(sessionId) % workers.length;
log.debug("accept sessionId [{}] and worker index [{}]", sessionId, workerIndex);
workers[workerIndex].registerSelectableChannel(socketChannel, sessionId);
} catch (Exception e) {
log.error("Failed to initialize an accepted socket.", e);
try {
socketChannel.close();
} catch (IOException e1) {
log.error("Failed to close a partially accepted socket.",
e1);
}
}
}
}
@Override
public void shutdown() {
for (Worker worker : workers) {
worker.shutdown();
}
start = false;
Millisecond100Clock.stop();
log.debug("thread {} is shutdown: {}", bossThread.getName(), bossThread.isInterrupted());
}
}