运用JAVA的concurrent.ExecutorService线程池实现socket的TCP和UDP连接

最近在项目中可能要用到socket相关的东西来发送消息，所以初步研究了下socket的TCP和UDP实现方式，并且结合java1.5的concurrent.ExecutorService类来实现多线程。

具体实现方式见代码：

一、TCP方式：

1、服务端实现方式：

TCP的服务端实现方式主要用到ServerSocket类，接收等待客户端连接的方法是accept();

代码如下：类SocketServerTCP

 1 private int port = 8823;  2     private ServerSocket serverSocket;  3     private ExecutorService executorService;// 线程池  4     private final int POOL_SIZE = 100;// 单个CPU线程池大小  5   6     public SocketServerTCP() {  7         try {  8             serverSocket = new ServerSocket(port);  9             executorService = Executors.newFixedThreadPool(Runtime.getRuntime() 10                     .availableProcessors() * POOL_SIZE); 11             logger.info("端口号为" + port + "的服务器启动"); 12         } catch (IOException e) { 13             e.printStackTrace(); 14         } 15     } 16  17     public void service() { 18         System.out.println("socket初始化成功！"); 19         logger.info("socket服务端初始化成功！"); 20         while (true) { 21             Socket socket = null; 22             try { 23                 // 接收客户连接,只要客户进行了连接,就会触发accept();从而建立连接 24                 socket = serverSocket.accept(); 25                 executorService.execute(new CallService(socket)); 26             } catch (Exception e) { 27                 e.printStackTrace(); 28             } 29         } 30     }

 CallService类主要作用是接收到消息后，来实现具体业务。

代码如下：

 1 /**  2  * 功能说明：执行具体业务的线程   3  */  4   class CallService implements Runnable {  5     Logger logger = Logger.getLogger(CallService.class);  6     private Socket socket;  7        8     private BufferedReader is;  9     private PrintWriter os; 10   11     public CallService(Socket socket) { 12         this.socket = socket; 13     } 14  15     public void run() { 16         try { 17             is = new BufferedReader(new InputStreamReader(socket.getInputStream())); 18             os = new PrintWriter(socket.getOutputStream()); 19  20             String msg = is.readLine(); 21  22             System.out.println("接到客户端消息" + msg); 23              24             //返回消息给客户端 25             String responseMsg="服务端返回消息：" + msg; 26             os.println(responseMsg); 27             os.flush(); 28         } catch (IOException e) { 29             e.printStackTrace(); 30         } finally { 31             try { 32                 if (os != null) { 33                     os.close(); 34                 } 35                 if (is != null) { 36                     is.close(); 37                 } 38                 if (socket != null) { 39                     socket.close(); 40                 } 41             } catch (IOException e) { 42                 e.printStackTrace(); 43             } 44         }  45     } 

服务器启动方式：

1 public static void main(String[] args) { 2         new SocketServerTCP().service(); 3     }

 

2、TCP客户端实现代码

客户通过new Socket(ip, port)创建一个socket,并通过PrintWriter来向服务器发送消息

 1 public class SocketClientTCP {  2     public String sendMsg(String ip, int port, String msg) {  3         Socket client = null;  4         try {  5             // 创建Socket  6             client = new Socket(ip, port);  7   8             // 向服务器发送消息  9             PrintWriter out = new PrintWriter(new BufferedWriter( 10                     new OutputStreamWriter(client.getOutputStream())), true); 11             out.println(msg); 12             out.flush(); 13  14             // 接收来自服务器的消息 15             BufferedReader in = new BufferedReader(new InputStreamReader( 16                     client.getInputStream())); 17             String str = in.readLine(); 18             // 关闭流 19             out.close(); 20             in.close(); 21             // 关闭Socket 22             client.close(); 23             return str; 24         } catch (IOException e) { 25             e.printStackTrace(); 26         } 27         return ""; 28     } 29  30     public static void main(String[] args) { 31         SocketClientTCP client = new SocketClientTCP();  32         String str1 = client.sendMsg("127.0.0.1", 8823,"xiaoxi_1234567890A1234567!"); 33         System.out.println("str1=====" + str1); 34     } 35 }

二、UDP实现方式

1、“服务端”实现方式

 

UDP的服务端主要是用到DatagramSocket、DatagramPacket类，通过new DatagramSocket(port)来创建一个socket，并且通过DatagramPacket来接收消息 代码如下：类SocketServerUDP:

 1 private int port = 1114;  2     private DatagramSocket dataSocket;  3     private ExecutorService executorService;// 线程池  4     private final int POOL_SIZE = 100;// 单个CPU线程池大小  5   6     public SocketServerUDP() {  7         try {  8             dataSocket = new DatagramSocket(port);  9             executorService = Executors.newFixedThreadPool(Runtime.getRuntime() 10                     .availableProcessors() * POOL_SIZE); 11             logger.info("端口号为" + port + "的服务器启动"); 12         } catch (IOException e) { 13             e.printStackTrace(); 14         } 15     } 16  17     public void service() { 18         System.out.println("socket初始化成功！"); 19         logger.info("socket服务端初始化成功！"); 20           21         while (true) {  22             try { 23                  byte[] buff = new byte[128];// 传输消息不超过64字   24                  DatagramPacket dataPacket = new DatagramPacket(buff, buff.length);    25                  dataSocket.receive(dataPacket);// 等待接收来自客户端的数据包    26                   27                  executorService.execute(new UdpCallService(dataPacket));//接收到消息后执行具体的业务 28             } catch (Exception e) { 29                 e.printStackTrace(); 30             } 31         } 32     }

 UDPCallService类主要作用是接收到消息后，来实现具体业务。

代码如下：

 1 class UdpCallService implements Runnable {  2     Logger logger = Logger.getLogger(UdpCallService.class);  3     private DatagramPacket packet;     4     private DatagramSocket dataSocket;     5   6     public UdpCallService(DatagramPacket packet) {     7         try {  8             this.packet = packet;     9             // 创建本机可以端口的DatagramSocket    10             dataSocket = new DatagramSocket();    11         } catch (SocketException e) {    12             e.printStackTrace();    13         }    14     }     15  16     public void run() {         17             String str=new String(packet.getData());             18             System.out.println("接收到消息："+str);  19             //返回数据给客户端 20             responeSocket("UDP消息返回："+str);   21     } 22      23     //返回消息给客户端 24     public void responeSocket(String message){ 25          // 构造响应数据包    26         byte[] response = message.toString().getBytes();    27         DatagramPacket dataPacket = new DatagramPacket(response, response.length, packet.getAddress(), packet.getPort());    28         try {// 发送    29             dataSocket.send(dataPacket);    30         } catch (IOException e) {    31             e.printStackTrace();    32         }     33     }

服务器启动方式：

1 public static void main(String[] args) { 2         new SocketServerUDP().service(); 3     }

2、UDP客户端实现方式

      UDP客户端的实现方式与"服务端"实现方式差不多，因为UDP本身就没有很严格的服务端和客户端定义。

代码如下：

 1 public class SocketClientUDP {  2       3     private int PORT;  4   5     public SocketClientUDP(int PORT) {  6         this.PORT=PORT;  7     }  8   9     public String getMessage() { 10         String msg = ""; 11         try { 12  13             String str = "hello world "; 14             DatagramSocket socket = new DatagramSocket(); 15  16             // 发送UPD消息给服务端 17             DatagramPacket requestPacket = new DatagramPacket(str.getBytes(), 18                     str.length(), InetAddress.getLocalHost(), PORT); 19             socket.send(requestPacket); 20  21             // 接收服务端返回的UDP消息 22             byte[] data = new byte[128]; 23             DatagramPacket responsePacket = new DatagramPacket(data,data.length); 24             socket.receive(responsePacket); 25             msg = new String(responsePacket.getData(), 0,responsePacket.getLength()); 26             socket.close(); 27         } catch (UnknownHostException e) { 28             System.err.println("Exception: host could not be found"); 29         } catch (Exception e) { 30             System.err.println("Exception: " + e); 31             e.printStackTrace(); 32         } 33         return msg; 34     } 35      36      public static void main(String[] args){ 37           SocketClientUDP clientUdp = new SocketClientUDP(1114);  38           String str1=   clientUdp.getMessage(); 39           System.out.println("str1====="+str1);  40      }

 

具体的JAVA文件下载地址：

http://files.cnblogs.com/fbsk/Socket_TCP_UDP.zip