package com.trydone.test;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Arrays;
public class TestServer {
private static final String MSG = "返回 输出消息 \n";
private static int count = 0;
private static int current = 0;
public static void main(String[] args) throws Exception {
boolean runWait = args != null && args.length > 0;
// runWait=true;
int port = 8989;
ServerSocketChannel ssc = ServerSocketChannel.open();
System.out.println("服务器启动:" + port + (runWait ? " 模拟1秒等待" : ""));
if (args != null)
System.out.println("arguments:" + Arrays.toString(args));
ServerSocket ss = ssc.socket();
ss.bind(new InetSocketAddress(port));
// set no blocking
ssc.configureBlocking(false);
while (true) {
try {
while (current>100) {
synchronized (Thread.currentThread()) {
Thread.currentThread().wait(1);
}
}
SocketChannel sc = ssc.accept();
if (sc == null) {
synchronized (Thread.currentThread()) {
Thread.currentThread().wait(1);
}
} else {
count++;
if (!runWait) {
current++;
processBackRun(sc); //不限制
current--;
} else {
processBack(sc);//限制线程1秒后台开新线程处理
}
}
} catch (Throwable e) {
e.printStackTrace();
}
}
}
private static void processBack(final SocketChannel sc) {
new Thread(new Runnable() {
@Override
public void run() {
try {
current++;
Thread.sleep(1000);//等1秒,测试 1秒并发连接 数用
processBackRun(sc);
} catch (Throwable e) {
e.printStackTrace();
} finally {
current--;
}
}
}).start();
}
private static void processBackRun(SocketChannel sc) throws InterruptedException, IOException {
// current++;
System.out.println("来自：" + sc.socket().getRemoteSocketAddress() + " 连接，总" + count + "个连接，当前连接 " + current + "个");
ByteBuffer readerBuffer = ByteBuffer.allocate(1024);
sc.read(readerBuffer);
readerBuffer.flip();
//output get
ByteBuffer buffer = ByteBuffer.wrap(MSG.getBytes());
if(out(readerBuffer)){
buffer= ByteBuffer.wrap(("总" + count + "个连接，当前连接 " + current + "个").getBytes());
}
buffer.rewind();
sc.write(buffer);
sc.close();
// current--;
}
private static boolean out(ByteBuffer readerBuffer) {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < readerBuffer.limit(); i++) {
char c = (char) readerBuffer.get();
sb.append(new String(new char[]{c}));
}
if(sb.toString().equals("status")){
return true;
}else
System.out.println(sb.toString());
return false;
}
}
