TCP通信--聊天室

聊天室客户端

public class Client {
	/*
	 * 套接字--可以理解为电话
	 * java.net.Socket
	 * 封装了TCP协议，使用它就可以基于TCP协议进行网络通讯
	 * Socket是运行在客户端的
	 */
	private Socket socket;
	
	/**
	 * 构造方法，用来初始化客户端
	 * 实例化Socket的时候需要传入两个参数：
	 * 1.服务端地址：通过IP地址可以找到服务端的计算机
	 * 2.服务端端口：通过端口可以找到服务端计算机上的服务端应用程序
	 * 
	 * 实例化Socket的过程就是连接的过程，若远端计算机没有响应会抛出异常
	 */
	public Client() throws Exception{
		System.out.println("正在连接服务端...");
		socket=new Socket("localhost",8088);//localhost是自己电脑的IP地址，可以改成其他人的IP地址进行连接
		System.out.println("已与服务端建立连接！");
	}
	
	/**
	 * 启动客户端的方法
	 */
	public void start(){
		try {
			Scanner scanner=new Scanner(System.in);
			/*
			 * 先要求用户输入一个昵称
			 */
			String nickName=null;
			while(true){
				System.out.println("请输入用户名");
				nickName=scanner.nextLine();
				if(nickName.length()>0){
					break;
				}
				System.out.println("输入有误！");
			}

			System.out.println("欢迎您"+nickName+"! 开始聊天吧");
			
			/*
			 * Socket提供的方法：OutputStream  getOutputStream
			 * 获取一个字节输出流，通过该流写出的数据会被发送至远端计算机
			 */
			OutputStream out=socket.getOutputStream();
			OutputStreamWriter osw=new OutputStreamWriter(out,"UTF-8");
			PrintWriter pw=new PrintWriter(osw,true);
			
			//先将昵称发送至服务端
			pw.println(nickName);
			
			
			/*
			 * 启动读取服务端发送过来消息的线程
			 */
			ServerHandler handler =new ServerHandler();
			Thread t=new Thread(handler);
			t.start();
			
			while(true){
			/*
			 * 将字符串发送至服务端
			 */
				pw.println(scanner.nextLine());
				
			
			}
			
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args) {
		try {
			Client client=new Client();
			client.start();
		} catch (Exception e) {
			e.printStackTrace();
			System.out.println("客户端启动失败");
		}
	}
	
	/**
	 * 该线程用来读取服务端发送过来的消息，并输出到客户端控制台显示
	 * @author xiaoxiannv
	 *
	 */
	class ServerHandler implements Runnable{
		public void run(){
			try {
				InputStream in=socket.getInputStream();
				InputStreamReader isr=new InputStreamReader(in,"UTF-8");
				BufferedReader br=new BufferedReader(isr);
				String message=null;
				while((message=br.readLine())!=null){
					System.out.println(message);
				}
			} catch (Exception e) {
				
			}
		}
	}
	
}

聊天室服务端

public class Server {
	/*
	 * 运行在服务端的ServerSocket主要负责：
	 * 1.向系统申请服务端口，客户端就是通过这个端口与之连接的
	 * 2.监听申请的服务端口，当一个客户端通过该端口尝试建立连接时，ServerSocket会在
	 * 服务端创建一个Socket与客户端建立连接。
	 */
	private ServerSocket server;
	
	/*
	 * 保存所有客户端输出流的集合
	 */
	private List<PrintWriter> allOut;
	
	/*
	 * 用来初始化服务端
	 */
	public Server() throws Exception{
		/*
		 * 初始化的同时申请服务端口
		 */
		server=new ServerSocket(8088);
		allOut=new ArrayList<PrintWriter>();
	}
	
	/**
	 * 将给定的输出流存入共享集合
	 * @param out
	 */
	private synchronized void addOut(PrintWriter out){
		allOut.add(out);
	}
	
	/**
	 * 将给定的输出流从共享集合中删除
	 * @param out
	 */
	private synchronized void removeOut(PrintWriter out){
		allOut.remove(out);
	}
	
	/**
	 * 将给定的消息发送给所有客户端
	 * @param message
	 */
	private synchronized void sendMessage(String message){
		for(PrintWriter out:allOut){
			out.println(message);
		}
	}
	
	/*
	 * 服务端开始工作的方法
	 */
	public void start(){
		try {
			/*
			 * ServerSocket的accept方法是一个阻塞方法，作用是监听服务端口，直到一个
			 * 客户端连接并创建一个Socket，使用该Socket即可与刚连接的客户端进行交互。
			 */
			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) {
		try {
			Server server=new Server();
			server.start();
		} catch (Exception e) {
			e.printStackTrace();
			System.out.println("服务端启动失败");
		}
	}

	/*
	 * 该线程负责处理一个客户端的交互
	 */
	class ClientHandler implements Runnable{
		/*
		 * 该线程处理的客户端的Socket
		 */
		private Socket socket;
		private String host;//纪录客户端的地址信息
		private String nickName;//该用户的昵称
		
		
		public ClientHandler(Socket socket){
			this.socket=socket;
			/*
			 * 通过Socket可以获取远端计算机的地址信息
			 */
			InetAddress address=socket.getInetAddress();
			host=address.getHostAddress();//获取IP地址
		}
		public void run(){
			PrintWriter pw=null;
			try {
				
				/*
				 * Socket提供的方法InputStream getInputStream()
				 * 该方法可以获取一个输入流，从该流读取的数据就是从远端计算机发送过来的
				 */
				InputStream in=socket.getInputStream();
				InputStreamReader isr=new InputStreamReader(in,"UTF-8");
				BufferedReader br=new BufferedReader(isr);
				
				//首先读取一行字符串为昵称
				nickName=br.readLine();
				sendMessage(nickName+"上线了！");
				
				/*
				 * 通过Socket创建输出流用于将消息发送给客户端
				 */
				OutputStream out=socket.getOutputStream();
				OutputStreamWriter osw=new OutputStreamWriter(out,"UTF-8");
				pw=new PrintWriter(osw,true);
				
				/*
				 * 将该客户端的输出流存入到共享集合中
				 */
				addOut(pw);
				
				
				
				String message=null;
				/*
				 * br.readLine在读取客户端发送过来的消息时，由于客户端断线，而其操作系统的不同，这里读取后
				 * 的结果不同：
				 * 当windows的客户端断开后，br.readLine会抛出异常
				 * 当linux的客户端断开后，br.readLine会返回null
				 */
				while((message=br.readLine())!=null){
					//pw.println(host+"说："+message);
					
					sendMessage(nickName+"说："+message);//广播消息
				}
			} catch (Exception e) {
				
			}finally{
				/*
				 * 处理当前客户端断开后的逻辑
				 */
				//将该客户端的输出流从共享集合中删除 
				removeOut(pw);
				
				sendMessage(nickName+"下线了！");
				try {
					socket.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
				
			}
		}
	}
}