socket网络套节字---聊天室
一：服务端：
1.创建客户端：
package com.ywh.serversocket; import java.io.InputStream; import java.io.OutputStream; import java.net.Socket; import java.util.Set; public class LoginThread extends Thread { Socket client; boolean bl; //通过构造传入一个客户socket public LoginThread(Socket socket) { client=socket; } @Override public void run() { // TODO Auto-generated method stub try { //如果登录成功才运行发送消息的线程 LoginServer(); if(bl) { new SendThread(client).start(); } } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } } public void LoginServer()throws Exception { //将连接后客户端输入传过来的用户名进行比对确认是否存在此用户 //先把传过来的用户名转换为string字符串 InputStream input=client.getInputStream(); byte[] bytes=new byte[1024]; input.read(bytes); String str=new String(bytes); System.out.println(str); //通过Set的add方法的返回值来确认用户名是否存在， //若用户名存在返回值为false，不存在则将新用户名加入用户名Set为true， boolean b=UserInfo.nickName.add(str); OutputStream output=client.getOutputStream(); if(b) { //当b=true，即新建用户成功时通知所有用户有新用户上线 //告知当前新用户创建成功 output.write("1".getBytes()); //将新用户的socket和用户名存入用户名map UserInfo.userMap.put(client, str); //获取用户socket Set<Socket> users=UserInfo.userMap.keySet(); for(Socket user:users) { OutputStream op=user.getOutputStream(); op.write((str+"已上线").getBytes()); } bl=true; }else { //若创建新用户失败，则返回false output.write("0".getBytes()); LoginServer(); bl=false; } } }
2.创建一个类用于存储登录用户的信息：
package com.ywh.serversocket; import java.net.Socket; import java.util.HashMap; import java.util.HashSet; import java.util.Set; public class UserInfo { //用来存储用户和用户名的Map public static HashMap<Socket, String> userMap=new HashMap<Socket, String>(); //用于存储用户名比对的Set public static Set<String> nickName=new HashSet<String>(); }
3.交互：发送线程
package com.ywh.serversocket; import java.io.IOException; import java.io.InputStream; import java.net.Socket; import java.util.Set; public class SendThread extends Thread { Socket client; public SendThread(Socket socket) { client = socket; } @Override public void run() { // TODO Auto-generated method stub while(true) { try { InputStream input=client.getInputStream(); byte[] bytes=new byte[1024]; input.read(bytes); String str=new String(bytes); Set<Socket> users=UserInfo.userMap.keySet(); for(Socket user:users) { if(user!=client) { user.getOutputStream().write ((UserInfo.userMap.get(client)+":"+str).getBytes()); } } } catch (IOException e) { // TODO Auto-generated catch block Set<Socket> users=UserInfo.userMap.keySet(); for(Socket user:users) { if(user!=client) { try { user.getOutputStream ().write((UserInfo.userMap.get(client)+"下线").getBytes()); } catch (IOException e1) { // TODO Auto-generated catch block e1.printStackTrace(); } } UserInfo.nickName.remove (UserInfo.userMap.get(client)); UserInfo.userMap.remove(client); } } } } }
4.主方法，用于运行
package com.ywh.serversocket; import java.io.IOException; import java.net.ServerSocket; import java.net.Socket; public class Main { public static void main(String[] args) { // TODO Auto-generated method stub try { @SuppressWarnings("resource") ServerSocket serversocket=new ServerSocket(8088); System.out.println("服务器已启动..."); while(true) { Socket socket=serversocket.accept(); new LoginThread(socket).start(); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } }
服务端完成
二：客户端部分
1.客户端发送信息
package com.ywh.clientsocket; import java.io.IOException; import java.io.InputStream; import java.net.Socket; public class ClientReceive extends Thread { Socket client; public ClientReceive(Socket socket) { // TODO Auto-generated constructor stub client=socket; } @Override public void run() { // TODO Auto-generated method stub while(true) { try { InputStream input=client.getInputStream(); byte[] bytes=new byte[1024]; input.read(bytes); System.out.println(new String(bytes)); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } }
2.客户端接受信息
package com.ywh.clientsocket; import java.io.IOException; import java.net.Socket; import java.util.Scanner; public class ClientSend extends Thread { Socket client; public ClientSend(Socket socket) { // TODO Auto-generated constructor stub client=socket; } @Override public void run() { // TODO Auto-generated method stub @SuppressWarnings("resource") Scanner sc=new Scanner(System.in); while(true) { String str=sc.nextLine(); try { client.getOutputStream().write(str.getBytes()); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } }
3.客户端主方法
package com.ywh.clientsocket; import java.io.IOException; import java.io.InputStream; import java.net.Socket; import java.util.Scanner; public class Main { public static String ip = "192.168.10.32"; public static int port = 8088; static Scanner sc = new Scanner(System.in); public static void main(String[] args) { // TODO Auto-generated method stub // System.out.println("请输入IP地址"); // String ip=sc.nextLine().trim(); // System.out.println("请输入端口号"); // int port=sc.nextInt(); try { Socket user = new Socket(ip, port); System.out.println("连接服务器成功，请输入登录昵称"); String nickName = sc.nextLine(); while (nickName.equals("")) { System.out.println("昵称不能为空，请重新输入昵称"); nickName = sc.nextLine(); } Login(user,nickName); new ClientSend(user).start(); new ClientReceive(user).start(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } public static void Login(Socket user, String nickName) { try { user.getOutputStream().write(nickName.getBytes()); InputStream input = user.getInputStream(); byte[] bytes = new byte[1024]; input.read(bytes); String str = new String(bytes).trim(); while (str.equals("0")) { System.out.println("昵称重复，请重新输入昵称"); nickName = sc.nextLine(); while (nickName.equals("")) { System.out.println("昵称不能为空，请重新输入昵称"); nickName = sc.nextLine(); } user.getOutputStream().write(nickName.getBytes()); input = user.getInputStream(); bytes = new byte[1024]; input.read(bytes); str = new String(bytes); } if(str.equals("1")) { System.out.println("登录成功"); }else { System.out.println("服务器错误"); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } }
