欢迎使用SOCKET聊天室

欢迎使用SOCKET聊天室

/**

• 聊天室服务端
• @author soft01

*/

public class Server {
	/*
	 * 运行在服务端的ServerSocket主要有两个作用：
	 * 1：向系统申请服务端口（客户端就是通过这个端口与服务端程序建立连接的）
	 * 2：监听端口，一旦一个客户端建立连接，那么就会自动创建一个Socket与该客户端通讯。
	 */
	private ServerSocket server;
	/*
	 * 该数组用于保存所有ClientHandler对应客户端的输出流，便于这些ClientHandler转发消息
	 */
	//private PrintWriter[] allOut = {};
	private Collection<PrintWriter> allOut = new ArrayList<>();
	
	public Server() {
		try {
			System.out.println("正在启动服务端");
			server = new ServerSocket(8088);
			System.out.println("服务端启动完毕");
			
		}catch(Exception e) {
			e.printStackTrace();
		}
		
	}
	
	public void start() {
		try {
			/*
			 * ServerSocket提供了接受客户端连接的方法：
			 * Socket accept()
			 * 这是一个阻塞方法，调用后服务端开始等待，
			 * 直到一个客户端通过这个端口与服务端建立连接为止，
			 * 这时该方法会返回一个Socket实例，通过这个实例就可以与该客户通讯了。。
			 * 
			 * 多次调用accept方法可以等待多个客户端的连接。
			 */
			while(true) {
			System.out.println("等待客户端连接...");
			Socket socket = server.accept();
			System.out.println("一个客户端连接了");
			ClientHandler handler = new ClientHandler(socket);
			Thread t = new Thread(handler);
			t.start();
			
			//启动一个线程来处理
			
			}
		}catch(Exception e) {
			e.printStackTrace();
		}
		
	}
	
	public static void main(String[] args) {
		Server server = new Server();
		server.start();
	}
	/*
	 * 该线程任务是负责与指定客户端进行交互
	 */
	private class ClientHandler implements Runnable{
		private Socket socket;
		//客户端地址信息
		private String host;
		
		public ClientHandler(Socket socket) {
			this.socket = socket;
			
			//通过socket获取远端计算机地址信息
			InetAddress address = socket.getInetAddress();
			//获取远端计算机地址信息IP的字符串格式
			host = address.getHostAddress();
		}
		public void run() {
			PrintWriter pw = null;
			try {
				/*
				 * 通过Socket获取输入流，可以读取远端计算机发送过来的字节数据。
				 */
				InputStream in = socket.getInputStream();
				InputStreamReader isr = new InputStreamReader(in,"utf-8");
				BufferedReader br = new BufferedReader(isr);
				
				/*
				 * 通过socket获取输出流，以便将消息发送给客户端
				 */
				OutputStream out = socket.getOutputStream();
				OutputStreamWriter osw = new OutputStreamWriter(out);
				BufferedWriter bw = new BufferedWriter(osw);
				 pw = new PrintWriter(bw,true);
				/*
				 * 将当前ClientHandler对应客户端的输出流放入共享数组allOut中，以便其他的ClientHandler
				 * 也可以访问到，从而给这个客户端转发消息
				 */
				 
				 /*
				  * 多个线程（ClientHandler)不能同时对allOut数组扩容添元素，会出现并发安全问题。
				  */
				 synchronized(allOut) {
//					//1     对allOut扩容
//						allOut = Arrays.copyOf(allOut, allOut.length+1);
//						//2     将输出流存入到数组中
//						allOut[allOut.length-1] = pw;
					 allOut.add(pw);
				 }
				
				System.out.println(host+"上线了，当前在线人数："+allOut.size());
				System.out.println("length:"+allOut.size());
				
				
				String message = null;
				while((message = br.readLine())!=null) {//读取一行到换行符位置停止
					System.out.println(host+"说："+message);
					//将消息发送给客户端
					synchronized(allOut) {
					for(PrintWriter o  : allOut) {
					o.println(host+"说："+message);	
					}
				}	
				}
			}catch(Exception e) {
				e.printStackTrace();
			}finally {
				//处理客户端断开连接后的操作
				
				//先将当前客户端输出流从共享数组中删除
				synchronized(allOut) {
//					for (int i = 0; i < allOut.length; i++) {
//						if(allOut[i]==pw) {
//							allOut[i]=allOut[allOut.length-1];
//							allOut=Arrays.copyOf(allOut, allOut.length-1);
//							break;
//					}
//					}
					allOut.remove(pw);
				}
				
				System.out.println(host+"下线了，当前在线人数："+allOut.size());
				try {
					socket.close();
				}catch(IOException e) {
					
				}
			}
			
			
		}
	}
	

}

/**

• 聊天室客户端
• @author soft01

*/

public class Client {
	/*
	 * java.net.socket套接字
	 * Socket 封装了TCP连接和通讯的细节，使用它可以通过两条流的读写完成与远端计算机的数据交互。
	 */
	
	private Socket socket;
	/**
	 * 构造方法，用来初始化客户端
	 * @param args
	 */
	Scanner scan = new Scanner(System.in);
	public Client() {
		try {
			/*
			 * 实例化Socket时需要传入两个参数：
			 * 1:服务端计算机的IP地址
			 * 2:服务段应用程序打开的服务端口
			 * 我们通过IP可以找到网络上的服务端计算机，通过端口连接到该机器上的服务端应用程序。
			 * 
			 * 实例化Socket的过程就是连接的过程，若服务器端没有响应这里就会抛出异常。
			 */
			System.out.println("正在连接服务端...");
			socket = new Socket("localhost",8088);//本机地址，（连接别人需要别人IP地址）
			System.out.println("已连接");
			
		}catch(Exception e) {
			e.printStackTrace();//捕获异常
		}
		
	}
	
	
	/**
	 * 客户端开始工作的方法
	 * @param args
	 */
	public void start() {
		try {
			//启动用于读取服务端发送过来消息的线程
			ServerHandler handler = new ServerHandler();
			Thread t = new Thread(handler);
			t.start();
			/*
			 * Socket提供的方法：
			 * OutputStream getOutputStream()
			 * 获取的字节输出流写出的字节会通过网络发送给远端的计算机
			 */
			OutputStream out = socket.getOutputStream();
			OutputStreamWriter osw = new OutputStreamWriter(out,"utf-8");
			BufferedWriter bw = new BufferedWriter(osw);
			PrintWriter  pw = new PrintWriter(bw,true);
			
		
			
			while(true) {
				
				System.out.println("请输入内容");
				String line = scan.nextLine();
				pw.println(line);
				} 
		}catch(Exception e) {
			e.printStackTrace();
		}
	}
	
	
	
	public static void main(String[] args) {
		Client client = new Client();
		client.start();
		
	}
/*
 * 该线程负责读取服务端发送过来的消息
 */
	private class ServerHandler implements Runnable{
		public void run() {
			try {
				
				//创建输入流读取服务端发送过来的消息
				InputStream in = socket.getInputStream();
				InputStreamReader read = new InputStreamReader(in,"utf-8");
				BufferedReader br = new BufferedReader(read);
				//循环读取服务端发送过来的消息
				String message = null;
				while((message=br.readLine())!=null) {
					System.out.println(message);
				}
				
			}catch(Exception e) {
				
			}
		}
	}
	
	
}