JavaSocket 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里面读数据并写回到该SocketOutputStream中去,直到这个连接被客户端关闭为止(InputStreamread方法返回-1)。

 

EchoSession需要一个线程来执行,这容易让人联想到用Thread来作为EchoSession的父类。不过,这样做不够灵活,开销也比较大。而选择让EchoSession实现Runnable接口就灵活得多。在接下来的使用Thread PoolEcho 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来处理有可能不能满足我们的要求(想象一下同时有数千个客户端的情况)。这时可以考虑使用JavaNIO API来构建服务器架构,因为NIOIO操作都是非阻塞的,我们只需要很少的Thread就可以充分地利用CPU来处理多个客户端的请求。关于NIO的话题,在这篇文章就不再赘述,希望以后能有机会讨论。 :)