Java的Socket API提供了一个很方便的对象接口进行网络编程。本文用一个简单的TCP Echo Server做例子,演示了如何使用Java完成一个网络服务器。 
  
用作例子的TCP Echo Server是按以下方式工作的: 
当一个客户端通过TCP连接到服务器后,客户端可以通过这个连接发送数据到服务端,而服务端接收到数据后会把这些数据用同一个TCP连接发送回客户端。服务端会一直保持这个连接直到客户端关闭它为止。 
  
因为服务器需要能同时处理多个客户端,我们先选用一个常见的多线程服务模型: 
让一个Thread负责监听服务端口,当有新的连接建立的时候,这个监听的Thread会为这个连接创建一个新的Thread来处理它。这样,服务器可以接受多个连接,并让多个Thread来分别处理它们。 
  
以下是相应的服务端程序: 
| 
 public class EchoServer implements Runnable { 
     
    public void run() { 
       try { 
           ServerSocket svr = new ServerSocket(7); 
           while (true) { 
              Socket sock = svr.accept(); 
              new Thread(new EchoSession(sock)).start(); 
           } 
       } catch (IOException ex) { 
           throw new ExceptionAdapter(ex); 
       } 
    } 
}  |   
  
这段代码先创建了一个ServerSocket的对象并让其监听在TCP端口7上,然后在一个循环中用accept()方法接收新的连接,并创建处理这一连接的Thread。实际处理每个客户端连接的逻辑包含在EchoSession这个类里面。 
  
在以上代码中使用了ExceptionAdapter这个类,它的作用是把一个checked Exception包装成RuntimeException。详细的说明可以参考避免在Java中使用Checked Exception 一文。 
  
以下是EchoSession的代码: 
| 
 public class EchoSession implements Runnable { 
     
    public EchoSession(Socket s) { 
       _sock = s; 
    } 
     
    public void run() { 
       try { 
           try { 
              InputStream input = _sock.getInputStream(); 
              OutputStream output = _sock.getOutputStream(); 
              byte [] buf = new byte [128];              
              while (true) { 
                  int count = input.read(buf); 
                  if (count == -1) 
                     break; 
                  output.write(buf, 0 , count); 
              } 
           } finally { 
              _sock.close(); 
           } 
       } catch (IOException ex) { 
           throw new ExceptionAdapter(ex);     
       } 
    } 
     
    protected Socket _sock = null; 
}  |   
  
EchoSession接受一个Socket对象作为构造参数,在其run()方法中,它不停的从这个Socket对象的InputStream里面读数据并写回到该Socket的OutputStream中去,直到这个连接被客户端关闭为止(InputStream的read方法返回-1)。 
  
EchoSession需要一个线程来执行,这容易让人联想到用Thread来作为EchoSession的父类。不过,这样做不够灵活,开销也比较大。而选择让EchoSession实现Runnable接口就灵活得多。在接下来的使用Thread Pool的Echo Server中可以看到这一点。 
  
以上已经是一个完整的TCP Echo Server,不过随着客户不停的连接和断开,这个服务器会不停的产生和消除线程,而这两个都是比较‘昂贵’的操作。为了避免这种消耗,可以考虑采用Thread Pool的机制。 
  
使用在一个简单的Thread缓冲池的实现一文中Thread Pool的实现,可以对EchoServer作如下修改(EchoSession无需做修改): 
| 
 public class EchoServer implements Runnable { 
     
    public void run() { 
       try { 
           ServerSocket svr = new ServerSocket(7); 
            
           // 初始化Thread Pool 
           SyncQueue queue = new SyncQueue(10); 
           for (int i = 0; i < 10; i ++) { 
              new Thread(new Worker(queue)).start(); 
           } 
  
           while (true) { 
              Socket sock = svr.accept(); 
              // 把任务放入Thread Pool 
              queue.put(new EchoSession(sock)); 
           } 
       } catch (IOException ex) { 
           throw new ExceptionAdapter(ex); 
       } 
    } 
}  |   
  
这里可以看出让EchoSession实现Runnable接口的灵活性,无需修改它就可以在Thread Pool里使用。 
  
在这个例子里使用的Thread Pool比较简单,没有动态调整Thread数量的功能,所以这个Echo Server最多只能同时服务10个客户端。然而通过重载SyncQueue,我们可以很方便地加入这个功能以突破这个限制。 
  
在对网络服务器的性能以及并发度要求很高的时候,让每个客户端由一个专门的Thread来处理有可能不能满足我们的要求(想象一下同时有数千个客户端的情况)。这时可以考虑使用Java的NIO API来构建服务器架构,因为NIO中IO操作都是非阻塞的,我们只需要很少的Thread就可以充分地利用CPU来处理多个客户端的请求。关于NIO的话题,在这篇文章就不再赘述,希望以后能有机会讨论。 :)  
 
  |