3. EchoServer 에서 차단 모드 와 비 차단 모드 를 혼합 합 니 다.

5049 단어 server
이 예 에서 고객 연결 을 받 는 작업 은 하나의 스 레 드 로 이 루어 지고 데 이 터 를 받 고 데 이 터 를 보 내 는 작업 은 다른 스 레 드 로 이 루어 집 니 다. 이것 은 서버 의 병행 성능 을 향상 시 킬 수 있 습 니 다.
      클 라 이언 트 연결 을 받 는 스 레 드 는 차단 모드 에 따라 작업 합 니 다. 클 라 이언 트 연결 을 받 으 면 selector 에 읽 기 준비 와 쓰기 이 벤트 를 등록 합 니 다. 그렇지 않 으 면 클 라 이언 트 의 연결 을 받 을 때 까지 차단 상태 에 들 어 갑 니 다.데 이 터 를 수신 하고 데 이 터 를 보 내 는 스 레 드 는 비 차단 모드 에 따라 작업 하 며 읽 기와 쓰기 가 완 료 된 사건 이 발생 할 때 만 해당 하 는 수신 데이터 와 데 이 터 를 보 내 는 작업 을 수행 합 니 다.
 
package com.test.socket.nio.thread2;
import java.io.*;
import java.nio.*;
import java.nio.channels.*;
import java.nio.charset.*;
import java.net.*;
import java.util.*;

public class EchoServer{
  private Selector selector = null;
  private ServerSocketChannel serverSocketChannel = null;
  private int port = 8000;
  private Charset charset=Charset.forName("GBK");

  public EchoServer()throws IOException{
    selector = Selector.open();
    serverSocketChannel= ServerSocketChannel.open();
    serverSocketChannel.socket().setReuseAddress(true);
    serverSocketChannel.socket().bind(new InetSocketAddress(port));
    System.out.println("     ");
  }

  public void accept(){
      for(;;){
        try{
            SocketChannel socketChannel = serverSocketChannel.accept();
            System.out.println("       ,  :" +
                               socketChannel.socket().getInetAddress() +
                               ":" + socketChannel.socket().getPort());
            socketChannel.configureBlocking(false);

            ByteBuffer buffer = ByteBuffer.allocate(1024);
            synchronized(gate){
                selector.wakeup();
                socketChannel.register(selector,
                                       SelectionKey.OP_READ |
                                       SelectionKey.OP_WRITE, buffer);
            }
        }catch(IOException e){e.printStackTrace();}
      }
  }
  private Object gate=new Object();
  public void service() throws IOException{
    for(;;){
      synchronized(gate){}
      int n = selector.select();

      if(n==0)continue;
      Set readyKeys = selector.selectedKeys();
      Iterator it = readyKeys.iterator();
      while (it.hasNext()){
        SelectionKey key=null;
        try{
            key = (SelectionKey) it.next();
            it.remove();
            if (key.isReadable()) {
                receive(key);
            }
            if (key.isWritable()) {
                send(key);
            }
        }catch(IOException e){
           e.printStackTrace();
           try{
               if(key!=null){
                   key.cancel();
                   key.channel().close();
               }
           }catch(Exception ex){e.printStackTrace();}
        }
      }//#while
    }//#while
  }

  public void send(SelectionKey key)throws IOException{
    ByteBuffer buffer=(ByteBuffer)key.attachment();
    SocketChannel socketChannel=(SocketChannel)key.channel();
    buffer.flip();  //       
    String data=decode(buffer);
    if(data.indexOf("
")==-1)return; String outputData=data.substring(0,data.indexOf("
")+1); System.out.print(outputData); ByteBuffer outputBuffer=encode("echo:"+outputData); while(outputBuffer.hasRemaining()) socketChannel.write(outputBuffer); ByteBuffer temp=encode(outputData); buffer.position(temp.limit()); buffer.compact(); if(outputData.equals("bye\r
")){ key.cancel(); socketChannel.close(); System.out.println(" "); } } public void receive(SelectionKey key)throws IOException{ ByteBuffer buffer=(ByteBuffer)key.attachment(); SocketChannel socketChannel=(SocketChannel)key.channel(); ByteBuffer readBuff= ByteBuffer.allocate(32); socketChannel.read(readBuff); readBuff.flip(); buffer.limit(buffer.capacity()); buffer.put(readBuff); } public String decode(ByteBuffer buffer){ // CharBuffer charBuffer= charset.decode(buffer); return charBuffer.toString(); } public ByteBuffer encode(String str){ // return charset.encode(str); } public static void main(String args[])throws Exception{ final EchoServer server = new EchoServer(); Thread accept=new Thread(){ public void run(){ server.accept(); } }; accept.start(); server.service(); } }

좋은 웹페이지 즐겨찾기