欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Java NIO--聊天室聊天

程序员文章站 2022-07-13 17:01:42
...

最近写了个Java NIO聊天室聊天的程序,NIO学习起来比较困难的,我的代码能给大家起到一个抛砖引玉的作用!

 

服务端:

package test.javanio;

/**
 * @author 
 * @version 
 * CreateTime:2010-12-1 下午05:12:11
 * Description:
 */
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Date;
import java.util.Iterator;
import java.util.logging.Level;
import java.util.logging.Logger;

public class MySocketServer implements Runnable {

	private boolean running;

	private Selector selector;
	String writeMsg;
	StringBuffer sb = new StringBuffer();
	SelectionKey ssckey;

	public MySocketServer() {

		running = true;

	}

	public void init() {
		try {
			selector = Selector.open();
			ServerSocketChannel ssc = ServerSocketChannel.open();
			ssc.configureBlocking(false);
			ssc.socket().bind(new InetSocketAddress(2345));
			ssckey = ssc.register(selector, SelectionKey.OP_ACCEPT);
			System.out.println("server is starting..." + new Date());
		} catch (IOException ex) {
			Logger.getLogger(MySocketServer.class.getName()).log(Level.SEVERE,
					null, ex);
		}
	}

	public static void main(String[] args) {
		MySocketServer server = new MySocketServer();
		new Thread(server).start();

	}

	public void execute() {
		try {
			while (running) {
				int num = selector.select();
				if (num > 0) {
					Iterator<SelectionKey> it = selector.selectedKeys()
							.iterator();
					while (it.hasNext()) {
						SelectionKey key = it.next();
						it.remove();
						if (!key.isValid())
							continue;
						if (key.isAcceptable()) {
							System.out.println("isAcceptable");
							getConn(key);
						} else if (key.isReadable()) {
							System.out.println("isReadable");
							readMsg(key);
						}

						else if (key.isValid() && key.isWritable()) {
							if (writeMsg != null) {
								System.out.println("isWritable");
								writeMsg(key);
							}

						}

						else
							break;

					}

				}
				Thread.yield();
			}

		} catch (IOException ex) {
			Logger.getLogger(MySocketServer.class.getName()).log(Level.SEVERE,
					null, ex);
		}
	}

	private void getConn(SelectionKey key) throws IOException {
		ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
		SocketChannel sc = ssc.accept();
		sc.configureBlocking(false);
		sc.register(selector, SelectionKey.OP_READ);
		System.out.println("build connection :"
				+ sc.socket().getRemoteSocketAddress());
	}

	private void readMsg(SelectionKey key) throws IOException {
		sb.delete(0, sb.length());
		SocketChannel sc = (SocketChannel) key.channel();
		System.out.print(sc.socket().getRemoteSocketAddress() + " ");
		ByteBuffer buffer = ByteBuffer.allocate(1024);
		buffer.clear();
		int len = 0;
		StringBuffer sb = new StringBuffer();
		while ((len = sc.read(buffer)) > 0) {
			buffer.flip();
			sb.append(new String(buffer.array(), 0, len));
		}
		if (sb.length() > 0)
			System.out.println("get from client:" + sb.toString());
		if (sb.toString().trim().toLowerCase().equals("quit")) {
			sc.write(ByteBuffer.wrap("BYE".getBytes()));
			System.out.println("client is closed "
					+ sc.socket().getRemoteSocketAddress());
			key.cancel();
			sc.close();
			sc.socket().close();

		} else {
			String toMsg = sc.socket().getRemoteSocketAddress() + "said:"
					+ sb.toString();
			System.out.println(toMsg);

			writeMsg = toMsg;

			/*
			 * Iterator<SelectionKey> it=key.selector().keys().iterator();
			 * 
			 * while(it.hasNext()){ SelectionKey skey=it.next();
			 * if(skey!=key&&skey!=ssckey){ SocketChannel client=(SocketChannel)
			 * skey.channel(); client.write(ByteBuffer.wrap(toMsg.getBytes()));
			 * }
			 * 
			 * }
			 */

			/*
			 * 
			 * key.attach(toMsg);
			 * key.interestOps(key.interestOps()|SelectionKey.OP_WRITE);
			 */
			Iterator<SelectionKey> it = key.selector().keys().iterator();

			while (it.hasNext()) {
				SelectionKey skey = it.next();
				if (skey != key && skey != ssckey) {
					if (skey.attachment() != null) {
						String str = (String) skey.attachment();
						skey.attach(str + toMsg);
					} else {
						skey.attach(toMsg);
					}
					skey
							.interestOps(skey.interestOps()
									| SelectionKey.OP_WRITE);
				}

			}
			selector.wakeup();// 可有可无

		}

	}

	public void run() {
		init();
		execute();
	}

	private void writeMsg(SelectionKey key) throws IOException {

		System.out.println("++++enter write+++");
		SocketChannel sc = (SocketChannel) key.channel();
		String str = (String) key.attachment();

		sc.write(ByteBuffer.wrap(str.getBytes()));
		key.interestOps(SelectionKey.OP_READ);
	}
}

 

 

 

客户端:

package test.javanio;

/**
 * @author 
 * @version 
 * CreateTime:2010-12-1 下午05:12:46
 * Description:
 */
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.Currency.*;

public class MySocketClient implements Runnable {
	Selector selector;

	boolean running;

	SocketChannel sc;

	public MySocketClient() {
		running = true;

	}

	public void init() {
		try {
			sc = SocketChannel.open();
			sc.configureBlocking(false);
			sc.connect(new InetSocketAddress("localhost", 2345));

		} catch (IOException ex) {
			Logger.getLogger(MySocketClient.class.getName()).log(Level.SEVERE,
					null, ex);
		}
	}

	public static void main(String[] args) {

		MySocketClient client = new MySocketClient();
		new Thread(client).start();
	}

	public void execute() {

		int num = 0;
		try {
			while (!sc.finishConnect()) {
			}
		} catch (IOException ex) {
			Logger.getLogger(MySocketClient.class.getName()).log(Level.SEVERE,
					null, ex);
		}

		ReadKeyBoard rkb = new ReadKeyBoard();
		new Thread(rkb).start();//启动一个线程向socket中进行写入数据
                //该线程从socket中读取数据
		while (running) {
			try {

				ByteBuffer buffer = ByteBuffer.allocate(1024);
				buffer.clear();

				StringBuffer sb = new StringBuffer();
				Thread.sleep(500);

				while ((num = sc.read(buffer)) > 0) {
					sb.append(new String(buffer.array(), 0, num));
					buffer.clear();
				}
				if (sb.length() > 0)
					System.out.println(sb.toString());
				if (sb.toString().toLowerCase().trim().equals("bye")) {
					System.out.println("closed....");

					sc.close();
					sc.socket().close();
					rkb.close();
					running = false;
				}
			} catch (InterruptedException ex) {
				Logger.getLogger(MySocketClient.class.getName()).log(
						Level.SEVERE, null, ex);
			} catch (IOException ex) {
				Logger.getLogger(MySocketClient.class.getName()).log(
						Level.SEVERE, null, ex);
			}
		}

	}

	public void run() {
		init();
		execute();
	}

	class ReadKeyBoard implements Runnable {

		boolean running2 = true;

		public ReadKeyBoard() {

		}

		public void close() {
			running2 = false;
		}

		public void run() {
			BufferedReader reader = new BufferedReader(new InputStreamReader(
					System.in));
			while (running2) {
				try {
					System.out.println("enter some commands:");
					String str = reader.readLine();
					sc.write(ByteBuffer.wrap(str.getBytes()));

				} catch (IOException ex) {
					Logger.getLogger(ReadKeyBoard.class.getName()).log(
							Level.SEVERE, null, ex);
				}
			}

		}

	}
}

 

相关标签: 网络编程 NIO