多线程socket编程--聊天程序

一、好久没写网络编程代码,今天有空,复习一下。下面主要实现服务器端和客户端端通信,服务器端接受一个客户端就启动一个新线程处理,读写也用线程实现,其中读,一个只读10个字节,循环读取输入流。好,废话少说,上代码。

1、服务器端:

  1. package com.ppt.socket;  
  2.   
  3. import java.net.ServerSocket;  
  4. import java.net.Socket;  
  5. /* 
  6.  * 服务端 
  7.  */  
  8. public class Server {  
  9.   
  10.     private int port = 9999;  
  11.     private ServerSocket server;  
  12.     private static Socket socket;  
  13.       
  14.     public Server() {  
  15.         try {  
  16.             init();  
  17.         } catch(Exception e) {  
  18.             e.printStackTrace();  
  19.         }  
  20.     }  
  21.     private void init() throws Exception {  
  22.         server = new ServerSocket(port);  
  23.         System.out.println("server socket is start, port is: " + port);  
  24.         while(true) {  
  25.             socket = server.accept();  
  26.             handle(socket);  
  27.         }  
  28.     }  
  29.     private void handle(Socket socket) throws Exception {  
  30.         String key = socket.getInetAddress().getHostAddress()+":"+socket.getPort();  
  31.         System.out.println("accept a socket: " + key);  
  32.         Thread thread = new Thread(new ThreadSocket(socket));  
  33.         thread.start();  
  34.     }  
  35.       
  36.     public static void main(String[] args) throws Exception {  
  37.         Server server = new Server();  
  38.     }  
  39. }  
package com.ppt.socket;

import java.net.ServerSocket;
import java.net.Socket;
/*
 * 服务端
 */
public class Server {

	private int port = 9999;
	private ServerSocket server;
	private static Socket socket;
	
	public Server() {
		try {
			init();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	private void init() throws Exception {
		server = new ServerSocket(port);
		System.out.println("server socket is start, port is: " + port);
		while(true) {
			socket = server.accept();
			handle(socket);
		}
	}
	private void handle(Socket socket) throws Exception {
		String key = socket.getInetAddress().getHostAddress()+":"+socket.getPort();
		System.out.println("accept a socket: " + key);
		Thread thread = new Thread(new ThreadSocket(socket));
		thread.start();
	}
	
	public static void main(String[] args) throws Exception {
		Server server = new Server();
	}
}

2、客户端:

  1. package com.ppt.socket;  
  2.   
  3. import java.net.InetSocketAddress;  
  4. import java.net.Socket;  
  5. import java.net.SocketAddress;  
  6.   
  7. /** 
  8.  * 客户端 
  9.  * @author aa 
  10.  * 
  11.  */  
  12. public class Client {  
  13.   
  14.     private String ip = "127.0.0.1";  
  15.     private int port = 9999;  
  16.     private static Socket socket;  
  17.       
  18.       
  19.     public Client() {  
  20.         try {  
  21.             init();  
  22.         } catch(Exception e) {  
  23.             e.printStackTrace();  
  24.         }  
  25.     }  
  26.       
  27.     private void init() throws Exception {  
  28.         System.out.println("init the socket client");  
  29.         socket = new Socket();  
  30.         socket.setSoTimeout(0);  
  31.         SocketAddress addr = new InetSocketAddress(ip, port);  
  32.         socket.connect(addr);  
  33.     }  
  34.     public void handle() throws Exception{  
  35.         Thread threadRead = new Thread(new ThreadRead(socket.getInputStream()),"SERVER_"+socket.getPort());  
  36.         Thread threadwrite = new Thread(new ThreadWrite(socket.getOutputStream()));  
  37.         threadRead.start();  
  38.         threadwrite.start();  
  39.     }  
  40.       
  41.       
  42.     public static void main(String[] args) throws Exception {  
  43.         Client client = new Client();  
  44.         client.handle();  
  45.     }  
  46.   
  47. }  
package com.ppt.socket;

import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.SocketAddress;

/**
 * 客户端
 * @author aa
 *
 */
public class Client {

	private String ip = "127.0.0.1";
	private int port = 9999;
	private static Socket socket;
	
	
	public Client() {
		try {
			init();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	
	private void init() throws Exception {
		System.out.println("init the socket client");
		socket = new Socket();
		socket.setSoTimeout(0);
		SocketAddress addr = new InetSocketAddress(ip, port);
		socket.connect(addr);
	}
	public void handle() throws Exception{
		Thread threadRead = new Thread(new ThreadRead(socket.getInputStream()),"SERVER_"+socket.getPort());
		Thread threadwrite = new Thread(new ThreadWrite(socket.getOutputStream()));
		threadRead.start();
		threadwrite.start();
	}
	
	
	public static void main(String[] args) throws Exception {
		Client client = new Client();
		client.handle();
	}

}

3、服务器处理socket线程:

  1. package com.ppt.socket;  
  2.   
  3. import java.net.Socket;  
  4. /** 
  5.  * 一个socket,一个线程 
  6.  * @author aa 
  7.  * 
  8.  */  
  9. public class ThreadSocket implements Runnable {  
  10.   
  11.     private Socket socket;  
  12.       
  13.     public ThreadSocket(Socket socket) {  
  14.         this.socket = socket;  
  15.     }  
  16.       
  17.     public void run() {  
  18.         try {  
  19.             Thread threadRead = new Thread(new ThreadRead(socket.getInputStream()),"CLIENT_"+socket.getPort());  
  20.             Thread threadwrite = new Thread(new ThreadWrite(socket.getOutputStream()));  
  21.             threadRead.start();  
  22.             threadwrite.start();  
  23.         } catch(Exception e) {  
  24.             e.printStackTrace();  
  25.         }  
  26.     }  
  27. }  
package com.ppt.socket;

import java.net.Socket;
/**
 * 一个socket,一个线程
 * @author aa
 *
 */
public class ThreadSocket implements Runnable {

	private Socket socket;
	
	public ThreadSocket(Socket socket) {
		this.socket = socket;
	}
	
	public void run() {
		try {
			Thread threadRead = new Thread(new ThreadRead(socket.getInputStream()),"CLIENT_"+socket.getPort());
			Thread threadwrite = new Thread(new ThreadWrite(socket.getOutputStream()));
			threadRead.start();
			threadwrite.start();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
}

4、读取输入流线程:

  1. package com.ppt.socket;  
  2.   
  3. import java.io.InputStream;  
  4.   
  5.   
  6. public class ThreadRead implements Runnable {  
  7.   
  8.     private static int HEAD_SIZE = 5;//传输最大字节长度--99999   
  9.     private static int BUFFER_SIZE = 10;//每次读取10个字节,循环读取输入流--测试,所以读10个字节   
  10.     private InputStream is;  
  11.       
  12.     public ThreadRead(InputStream is) {  
  13.         this.is = is;  
  14.     }  
  15.   
  16.     public void run() {  
  17.         try {  
  18.             while(true) {  
  19.                 boolean flag = true;  
  20.                 int msgLength = 0;  
  21.                 int recvSize = -HEAD_SIZE;  
  22.                 byte[] buf = new byte[BUFFER_SIZE];  
  23.                 StringBuffer sb = new StringBuffer();  
  24.                 while(msgLength > recvSize) {  
  25.                     int length = is.read(buf,0,BUFFER_SIZE);  
  26.                     recvSize += length;  
  27.                     if(flag) {  
  28.                         String str = new String(buf,0,HEAD_SIZE);  
  29.                         msgLength = Integer.parseInt(str);  
  30.                         sb.append(new String(buf,HEAD_SIZE,length-HEAD_SIZE));  
  31.                         flag = false;  
  32.                     } else {  
  33.                         sb.append(new String(buf,0,length));  
  34.                     }  
  35.                 }  
  36.                 System.out.println(Thread.currentThread().getName() +": "+ sb.toString());  
  37.             }  
  38.         } catch (Exception e) {  
  39.             e.printStackTrace();  
  40.         }  
  41.           
  42.     }  
  43. }  
package com.ppt.socket;

import java.io.InputStream;


public class ThreadRead implements Runnable {

	private static int HEAD_SIZE = 5;//传输最大字节长度--99999
	private static int BUFFER_SIZE = 10;//每次读取10个字节,循环读取输入流--测试,所以读10个字节
	private InputStream is;
	
	public ThreadRead(InputStream is) {
		this.is = is;
	}

	public void run() {
		try {
			while(true) {
				boolean flag = true;
				int msgLength = 0;
				int recvSize = -HEAD_SIZE;
				byte[] buf = new byte[BUFFER_SIZE];
				StringBuffer sb = new StringBuffer();
				while(msgLength > recvSize) {
					int length = is.read(buf,0,BUFFER_SIZE);
					recvSize += length;
					if(flag) {
						String str = new String(buf,0,HEAD_SIZE);
						msgLength = Integer.parseInt(str);
						sb.append(new String(buf,HEAD_SIZE,length-HEAD_SIZE));
						flag = false;
					} else {
						sb.append(new String(buf,0,length));
					}
				}
				System.out.println(Thread.currentThread().getName() +": "+ sb.toString());
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		
	}
}

5、发送数据线程:

  1. package com.ppt.socket;  
  2.   
  3. import java.io.OutputStream;  
  4. import java.util.Scanner;  
  5.   
  6. public class ThreadWrite implements Runnable {  
  7.   
  8.     private static int HEAD_SIZE = 5;//传输最大字节长度--99999   
  9.     private OutputStream os;  
  10.       
  11.     public ThreadWrite(OutputStream os) {  
  12.         this.os = os;  
  13.     }  
  14.       
  15.     public void run() {  
  16.         try {  
  17.             Scanner scan = new Scanner(System.in);  
  18.             while(true) {  
  19.                 String sendMsg = scan.next();  
  20.                 sendMsg = getMsg(sendMsg);  
  21.                 os.write(sendMsg.getBytes());  
  22.                 os.flush();  
  23.             }  
  24.         } catch (Exception e) {  
  25.             e.printStackTrace();  
  26.         }  
  27.     }  
  28.     /** 
  29.      * @param msg--组转发送字符串,首五个字符是字符串长度,不够五位,用0补足 
  30.      * @return 
  31.      */  
  32.     private static String getMsg(String msg) {  
  33.         int length = String.valueOf(msg.length()).length();  
  34.         String slen = "";  
  35.         if(length < HEAD_SIZE) {  
  36.             StringBuffer sb = new StringBuffer();  
  37.             for(int i=0; i<HEAD_SIZE-length; i++) {  
  38.                 sb.append("0");  
  39.             }  
  40.             sb.append(String.valueOf(msg.length()));  
  41.             return sb.toString()+msg;  
  42.         } else {  
  43.             slen = String.valueOf(length);  
  44.         }  
  45.         return slen+msg;  
  46.     }  
  47. }  
package com.ppt.socket;

import java.io.OutputStream;
import java.util.Scanner;

public class ThreadWrite implements Runnable {

	private static int HEAD_SIZE = 5;//传输最大字节长度--99999
	private OutputStream os;
	
	public ThreadWrite(OutputStream os) {
		this.os = os;
	}
	
	public void run() {
		try {
			Scanner scan = new Scanner(System.in);
			while(true) {
				String sendMsg = scan.next();
				sendMsg = getMsg(sendMsg);
				os.write(sendMsg.getBytes());
				os.flush();
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	/**
	 * @param msg--组转发送字符串,首五个字符是字符串长度,不够五位,用0补足
	 * @return
	 */
	private static String getMsg(String msg) {
		int length = String.valueOf(msg.length()).length();
		String slen = "";
		if(length < HEAD_SIZE) {
			StringBuffer sb = new StringBuffer();
			for(int i=0; i<HEAD_SIZE-length; i++) {
				sb.append("0");
			}
			sb.append(String.valueOf(msg.length()));
			return sb.toString()+msg;
		} else {
			slen = String.valueOf(length);
		}
		return slen+msg;
	}
}

6、至此,一个简单的多线程socket通信就完成了,够简单吧。




  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值