62,614
社区成员
发帖
与我相关
我的任务
分享
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
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.util.Date;
import java.util.Set;
public class NIOClient implements Runnable{
private ByteBuffer sBuffer = ByteBuffer.allocate(1024);
private ByteBuffer rBuffer = ByteBuffer.allocate(1024);
private InetSocketAddress SERVER;
private Selector selector;
private SocketChannel client;
private String receiveText;
private String sendText;
private int count=0;
public NIOClient(int port){
SERVER = new InetSocketAddress("localhost", port);
init();
}
public void init(){
try {
SocketChannel socketChannel = SocketChannel.open();
socketChannel.configureBlocking(false);
selector = Selector.open();
socketChannel.register(selector, SelectionKey.OP_CONNECT);
socketChannel.connect(SERVER);
} catch (Exception e) {
e.printStackTrace();
}
}
private void handle(SelectionKey selectionKey) throws IOException{
if (selectionKey.isConnectable()) {
client = (SocketChannel) selectionKey.channel();
if (client.isConnectionPending()) {
client.finishConnect();
System.out.println("connect success !");
sBuffer.clear();
sBuffer.put((new Date().toLocaleString()+" connected!").getBytes());
sBuffer.flip();
client.write(sBuffer);
new Thread(){
@Override
public void run() {
while(true){
try {
sBuffer.clear();
InputStreamReader input = new InputStreamReader(System.in);
BufferedReader br = new BufferedReader(input);
sendText = br.readLine();
sBuffer.put(sendText.getBytes());
sBuffer.flip();
client.write(sBuffer);
} catch (IOException e) {
e.printStackTrace();
break;
}
}
};
}.start();
}
client.register(selector, SelectionKey.OP_READ);
} else if (selectionKey.isReadable()) {
client = (SocketChannel) selectionKey.channel();
rBuffer.clear();
count=client.read(rBuffer);
if(count>0){
receiveText = new String( rBuffer.array(),0,count);
System.out.println(receiveText);
client = (SocketChannel) selectionKey.channel();
client.register(selector, SelectionKey.OP_READ);
}
}
}
@Override
public void run() {
// TODO Auto-generated method stub
try {
while (true) {
selector.select();
Set<SelectionKey> keySet = selector.selectedKeys();
for(final SelectionKey key : keySet){
handle(key);
};
keySet.clear();
}
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
NIOSServer
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
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.nio.charset.Charset;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
public class NIOSServer implements Runnable{
private int port = 8888;
private Charset cs = Charset.forName("gbk");
private ByteBuffer sBuffer = ByteBuffer.allocate(1024);
private ByteBuffer rBuffer = ByteBuffer.allocate(1024);
private Map<String, SocketChannel> clientsMap = new HashMap<String, SocketChannel>();
private Selector selector;
public NIOSServer(int port){
this.port = port;
try {
init();
} catch (Exception e) {
e.printStackTrace();
}
}
private void init() throws IOException{
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
ServerSocket serverSocket = serverSocketChannel.socket();
serverSocket.bind(new InetSocketAddress(port));
selector = Selector.open();
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
System.out.println("server start on port:"+port);
}
private void listen(){
}
@Override
public void run() {
// TODO Auto-generated method stub
while (true) {
try {
selector.select();
Set<SelectionKey> selectionKeys = selector.selectedKeys();
for(SelectionKey key : selectionKeys){
handle(key);
}
selectionKeys.clear();
} catch (Exception e) {
e.printStackTrace();
break;
}
}
}
private void handle(SelectionKey selectionKey) throws IOException, InterruptedException {
ServerSocketChannel server = null;
SocketChannel client = null;
String receiveText=null;
int count=0;
if (selectionKey.isAcceptable()) {
server = (ServerSocketChannel) selectionKey.channel();
client = server.accept();
//判断client 是否为空
if(client != null){
client.configureBlocking(false);
client.register(selector, SelectionKey.OP_READ);
}
Thread.sleep(10*1000);
} else if (selectionKey.isReadable()) {
client = (SocketChannel) selectionKey.channel();
rBuffer.clear();
count = client.read(rBuffer);
if (count > 0) {
rBuffer.flip();
receiveText = String.valueOf(cs.decode(rBuffer).array());
System.out.println(client.toString()+":"+receiveText);
dispatch(client, receiveText);
client = (SocketChannel) selectionKey.channel();
client.register(selector, SelectionKey.OP_READ);
}
}
}
private void dispatch(SocketChannel client,String info) throws IOException{
Socket s = client.socket();
String name = "["+s.getInetAddress().toString().substring(1)+":"+Integer.toHexString(client.hashCode())+"]";
if(!clientsMap.isEmpty()){
for(Map.Entry<String, SocketChannel> entry : clientsMap.entrySet()){
SocketChannel temp = entry.getValue();
if(!client.equals(temp)){
sBuffer.clear();
sBuffer.put((name+":"+info).getBytes());
sBuffer.flip();
temp.write(sBuffer);
}
}
}
clientsMap.put(name, client);
}
public static void main(String[] args) throws InterruptedException {
//创建NIOSServer
NIOSServer server1 = new NIOSServer(7778);
NIOSServer server2 = new NIOSServer(7777);
new Thread(server1).start();
new Thread(server2).start();
Thread.sleep(10*1000);
//创建NIOClient
NIOClient client1 = new NIOClient(7778);
new Thread(client1).start();
NIOClient client2 = new NIOClient(7777);
new Thread(client2).start();
NIOClient client3 = new NIOClient(7778);
new Thread(client3).start();
NIOClient client4 = new NIOClient(7777);
new Thread(client4).start();
}
}
运行NIOSServer,输出
server start on port:7778
server start on port:7777
connect success !
connect success !
connect success !
connect success !
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7778 remote=/127.0.0.1:52540]:2014-4-11 14:37:51 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7777 remote=/127.0.0.1:52543]:2014-4-11 14:37:51 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7778 remote=/127.0.0.1:52546]:2014-4-11 14:37:51 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7777 remote=/127.0.0.1:52549]:2014-4-11 14:37:51 connected!
[127.0.0.1:5122152b]:2014-4-11 14:37:51 connected!
[127.0.0.1:149be3aa]:2014-4-11 14:37:51 connected!
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
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.nio.charset.Charset;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
public class NIOSServer implements Runnable{
private int port = 8888;
private Charset cs = Charset.forName("gbk");
private static ByteBuffer sBuffer = ByteBuffer.allocate(1024);
private static ByteBuffer rBuffer = ByteBuffer.allocate(1024);
private Map<String, SocketChannel> clientsMap = new HashMap<String, SocketChannel>();
private static Selector selector;
public NIOSServer(int port){
this.port = port;
try {
init();
} catch (Exception e) {
e.printStackTrace();
}
}
private void init() throws IOException{
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
ServerSocket serverSocket = serverSocketChannel.socket();
serverSocket.bind(new InetSocketAddress(port));
selector = Selector.open();
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
System.out.println("server start on port:"+port);
}
//将listen()内的代码放到run()
private void listen(){
}
@Override
public void run() {
// TODO Auto-generated method stub
while (true) {
try {
selector.select();
Set<SelectionKey> selectionKeys = selector.selectedKeys();
for(SelectionKey key : selectionKeys){
handle(key);
}
selectionKeys.clear();
} catch (Exception e) {
e.printStackTrace();
break;
}
}
}
private void handle(SelectionKey selectionKey) throws IOException, InterruptedException {
ServerSocketChannel server = null;
SocketChannel client = null;
String receiveText=null;
int count=0;
if (selectionKey.isAcceptable()) {
server = (ServerSocketChannel) selectionKey.channel();
client = server.accept();
//判断client 是否为空
if(client != null){
client.configureBlocking(false);
client.register(selector, SelectionKey.OP_READ);
}
Thread.sleep(10*1000);
} else if (selectionKey.isReadable()) {
client = (SocketChannel) selectionKey.channel();
rBuffer.clear();
count = client.read(rBuffer);
if (count > 0) {
rBuffer.flip();
receiveText = String.valueOf(cs.decode(rBuffer).array());
System.out.println(client.toString()+":"+receiveText);
dispatch(client, receiveText);
client = (SocketChannel) selectionKey.channel();
client.register(selector, SelectionKey.OP_READ);
}
}
}
private void dispatch(SocketChannel client,String info) throws IOException{
Socket s = client.socket();
String name = "["+s.getInetAddress().toString().substring(1)+":"+Integer.toHexString(client.hashCode())+"]";
if(!clientsMap.isEmpty()){
for(Map.Entry<String, SocketChannel> entry : clientsMap.entrySet()){
SocketChannel temp = entry.getValue();
if(!client.equals(temp)){
sBuffer.clear();
sBuffer.put((name+":"+info).getBytes());
sBuffer.flip();
temp.write(sBuffer);
}
}
}
clientsMap.put(name, client);
}
//删除main函数
}
NIOClient
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
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.util.Date;
import java.util.Set;
public class NIOClient {
private static ByteBuffer sBuffer = ByteBuffer.allocate(1024);
private static ByteBuffer rBuffer = ByteBuffer.allocate(1024);
private InetSocketAddress SERVER;
private static Selector selector;
private static SocketChannel client;
private static String receiveText;
private static String sendText;
private static int count=0;
public NIOClient(int port){
SERVER = new InetSocketAddress("localhost", port);
init();
}
public void init(){
try {
SocketChannel socketChannel = SocketChannel.open();
socketChannel.configureBlocking(false);
selector = Selector.open();
socketChannel.register(selector, SelectionKey.OP_CONNECT);
socketChannel.connect(SERVER);
while (true) {
selector.select();
Set<SelectionKey> keySet = selector.selectedKeys();
for(final SelectionKey key : keySet){
handle(key);
};
keySet.clear();
}
} catch (Exception e) {
e.printStackTrace();
}
}
public static void main(String[] args) throws IOException, InterruptedException {
//创建NIOSServer
NIOSServer server1 = new NIOSServer(7778);
NIOSServer server = new NIOSServer(7777);
new Thread(server1).start();
new Thread(server).start();
Thread.sleep(10*1000);
//创建NIOSServer
NIOClient client = new NIOClient(7777);
}
private void handle(SelectionKey selectionKey) throws IOException{
if (selectionKey.isConnectable()) {
client = (SocketChannel) selectionKey.channel();
if (client.isConnectionPending()) {
client.finishConnect();
System.out.println("connect success !");
sBuffer.clear();
sBuffer.put((new Date().toLocaleString()+" connected!").getBytes());
sBuffer.flip();
client.write(sBuffer);
new Thread(){
@Override
public void run() {
while(true){
try {
sBuffer.clear();
InputStreamReader input = new InputStreamReader(System.in);
BufferedReader br = new BufferedReader(input);
sendText = br.readLine();
sBuffer.put(sendText.getBytes());
sBuffer.flip();
client.write(sBuffer);
} catch (IOException e) {
e.printStackTrace();
break;
}
}
};
}.start();
}
client.register(selector, SelectionKey.OP_READ);
} else if (selectionKey.isReadable()) {
client = (SocketChannel) selectionKey.channel();
rBuffer.clear();
count=client.read(rBuffer);
if(count>0){
receiveText = new String( rBuffer.array(),0,count);
System.out.println(receiveText);
client = (SocketChannel) selectionKey.channel();
client.register(selector, SelectionKey.OP_READ);
}
}
}
}
执行NIOClient,输出
server start on port:7778
server start on port:7777
connect success !
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7777 remote=/127.0.0.1:61140]:2014-4-8 14:19:41 connected!
public static void main(String[] args) throws IOException {
NIOSServer server1 = new NIOSServer(7778);
NIOSServer server = new NIOSServer(7777);
server1.listen(); //这里一直在while(true)。。。。把NIOSServer 改成线程吧
server.listen();
}