NIO 服务器客户端例子

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; } }