[jdk1.4]socket举例--基于nio

直接上代码,后续文章会有io,nio比较..

 

ServerMain.java

package test.server; import java.io.IOException; import java.net.InetSocketAddress; import java.nio.ByteBuffer; import java.nio.CharBuffer; import java.nio.channels.SelectionKey; import java.nio.channels.Selector; import java.nio.channels.ServerSocketChannel; import java.nio.channels.SocketChannel; import java.nio.charset.Charset; import java.nio.charset.CharsetEncoder; import java.util.Iterator; import java.util.Set; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; public class ServerMain { private int num = 0; public static Selector sel; private ServerSocketChannel server; private ExecutorService pool; public ServerMain(int port, int num) throws IOException { sel = Selector.open();// 选择器(多路复用器) pool = Executors.newFixedThreadPool(num); server = ServerSocketChannel.open(); server.socket().bind(new InetSocketAddress(port)); server.configureBlocking(false);// 非阻塞 server.register(sel, SelectionKey.OP_ACCEPT);// 注册 this.run(); } public void run() { System.out.println("listen to port..."); try { for (;;) { sel.select(); //在这里会阻塞 Set<SelectionKey> readyKeys = sel.selectedKeys(); Iterator<SelectionKey> iterator = readyKeys.iterator(); while (iterator.hasNext()){ SelectionKey key = iterator.next(); iterator.remove(); this.process(key); } } } catch (IOException e) { e.printStackTrace(); } } public void process(SelectionKey key) throws IOException { if (key.isAcceptable()) { ServerSocketChannel server = (ServerSocketChannel)key.channel(); SocketChannel client = server.accept(); client.configureBlocking(false); client.register(ServerMain.sel, SelectionKey.OP_WRITE); } else if (key.isWritable()) { System.out.println(++num); CharsetEncoder encoder = Charset.forName("UTF-8").newEncoder(); SocketChannel channel = (SocketChannel) key.channel(); ByteBuffer block = encoder.encode(CharBuffer.wrap("connect to server...")); channel.write(block); if(key.isValid()) key.cancel(); // 注意,注销key否则selector一直选择 pool.execute(new TCPHandler()); } } public static void main(String[] args) throws IOException { new ServerMain(8888,200).run(); } }

ClientMain.java

package test.client; import java.net.InetSocketAddress; import java.nio.ByteBuffer; import java.nio.channels.SelectionKey; import java.nio.channels.Selector; import java.nio.channels.SocketChannel; import java.nio.charset.Charset; import java.nio.charset.CharsetEncoder; import java.util.Iterator; import java.util.Set; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; public class ClientMain implements Runnable { static int SIZE = 10; static InetSocketAddress url = new InetSocketAddress("127.0.0.1", 8888); static CharsetEncoder encoder = Charset.forName("UTF-8").newEncoder(); String msg = ""; @Override public void run() { try{ SocketChannel client = SocketChannel.open(); client.configureBlocking(false); Selector selector = Selector.open(); client.register(selector, SelectionKey.OP_CONNECT); client.connect(url); ByteBuffer buffer = ByteBuffer.allocate(8 * 1024); int total = 0; for (;;) { selector.select(); Set<SelectionKey> set = selector.selectedKeys(); Iterator<SelectionKey> iter = set.iterator(); while (iter.hasNext()) { SelectionKey key = iter.next(); iter.remove(); if (key.isConnectable()) { SocketChannel channel = (SocketChannel) key.channel(); if (channel.isConnectionPending()) channel.finishConnect(); channel.register(selector, SelectionKey.OP_READ); } else if (key.isReadable()) { SocketChannel channel = (SocketChannel) key.channel(); int count = channel.read(buffer); if (count > 0) { total += count; buffer.flip(); while (buffer.remaining() > 0) { byte b = buffer.get(); msg += (char) b; } buffer.clear(); } System.out.println(msg); selector.close(); return; } } } }catch (Exception e) { e.printStackTrace(); } } public static void main(String[] args) { ExecutorService pool = Executors.newFixedThreadPool(200); int i = 2000; while (i-- > 0) { try { pool.execute(new ClientMain()); } catch (Exception e) { e.printStackTrace(); } } pool.shutdown(); } }

src: 点击下载