运用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文件下载地址：