Java高级Day40-QQ项目全代码
114.多用户通信系统(QQ)项目
QQServer包
===============//ManageClientThread//=============== public class ManageClientThread { //返回 public static HashMap<String, ServerConnectClientThread> getHm(){ return hm; } private static HashMap<String,ServerConnectClientThread> hm = new HashMap<>(); //添加线程对象到 hm 集合 public static void addClientThread(String userId, ServerConnectClientThread serverConnectClientThread) { hm.put(userId, serverConnectClientThread); } //根据userId 返回ServerConnectClientThread线程 public static ServerConnectClientThread getServerConnectClientThread(String userId){ return hm.get(userId); } //增加一个方法,从集合中,移除某个线程对象 public static void removeServerConnectClientThread(String userId){ hm.remove(userId); } //这里编写方法,可以返回在线用户列表 public static String getOnlineUsers(){ //集合遍历,遍历hashmap的key Iterator<String> iterator = hm.keySet().iterator(); String onlineUserList = ""; while (iterator.hasNext()){ onlineUserList += iterator.next().toString() + " "; } return onlineUserList; } } ===============//Message//=============== public class Message implements Serializable { private static final long serialVersionUID = 1L; private String sender;//发送方 private String getter;//接收方 private String content;//消息内容 private String sendTime;//发送时间 private String mesType;//消息类型[可以在接口中定义消息类型] //进行扩展 和文件相关的成员 private byte[] fileBytes; private int fileLen = 0; private String dest;//将文件传输到那里 private String src;//源文件路径 public byte[] getFileBytes() { return fileBytes; } public void setFileBytes(byte[] fileBytes) { this.fileBytes = fileBytes; } public int getFileLen() { return fileLen; } public void setFileLen(int fileLen) { this.fileLen = fileLen; } public String getDest() { return dest; } public void setDest(String dest) { this.dest = dest; } public String getSrc() { return src; } public void setSrc(String src) { this.src = src; } public String getMesType() { return mesType; } public void setMesType(String mesType) { this.mesType = mesType; } public String getSender() { return sender; } public void setSender(String sender) { this.sender = sender; } public String getGetter() { return getter; } public void setGetter(String getter) { this.getter = getter; } public String getContent() { return content; } public void setContent(String content) { this.content = content; } public String getSendTime() { return sendTime; } public void setSendTime(String sendTime) { this.sendTime = sendTime; } } ===============//MessageType//=============== public interface MessageType { //1.在接口中定义了一些常量 //2.不同的常量表示不同的消息类型 String MESSAGE_LOGIN_SUCCEED = "1";//表示登录成功 String MESSAGE_LOGIN_FAIL = "2";//表示登录失败 String MESSAGE_COMM_MES = "3";//普通信息包 String MESSAGE_GET_ONLINE_FRIEND = "4";//要求返回在线用户列表 String MESSAGE_RET_ONLINE_FRIEND = "5";//返回在线用户列表 String MESSAGE_CLIENT_EXIT = "6";//客户端请求退出 String MESSAGE_TO_ALL_MES = "7";//群发消息 String MESSAGE_FILE_MES = "8";//文件消息 } ===============//QQServer//=============== public class QQServer { private ServerSocket ss = null; //创建一个集合,存放多个用户,如果是这些用户登录,就认为是合法的 private static HashMap<String, User> validUsers = new HashMap<>(); static {//在静态代码块初始化validUsers validUsers.put("100", new User("100", "123456")); validUsers.put("200", new User("200", "123456")); validUsers.put("300", new User("300", "123456")); validUsers.put("至尊宝", new User("至尊宝", "123456")); validUsers.put("ZhangKewei", new User("ZhangKewei", "20040426")); } //验证用户是否有效的方法 private boolean checkUser(String userId, String passwd) { User user = validUsers.get(userId); if (user == null) {//说明userId没有存在validUsers 的 key 中 return false; } if (!(user.getPasswd().equals(passwd))) {//userId正确,密码错误 return false; } return true; } public static void main(String[] args) { new QQServer(); } public QQServer() { //注意:端口可以写在配置文件 try { System.out.println("服务端在9999端口监听..."); ss = new ServerSocket(9999); while (true) {//当和某个客户端连接后,会继续监听,因为while循环 Socket socket = ss.accept(); //得到Socket关联的对象输入流 ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); //得到socket关联的对象输出流 ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream()); User u = (User) ois.readObject();//读取客户端发送的User对象 //创建一个Message对象,准备回复客户端 Message message = new Message(); //验证 if (checkUser(u.getUserID(), u.getPasswd())){//合法用户 message.setMesType(MessageType.MESSAGE_LOGIN_SUCCEED); //将message对象回复 oos.writeObject(message); //创建一个线程,和客户端保持通信,该线程需要持有socket对象 ServerConnectClientThread serverConnectClientThread = new ServerConnectClientThread(socket, u.getUserID()); //启动该线程 serverConnectClientThread.start(); //把该线程对象,放入到一个集合中,进行管理 ManageClientThread.addClientThread(u.getUserID(), serverConnectClientThread); } else {//登陆失败 System.out.println("用户 id=" + u.getUserID() + " pwd=" + u.getPasswd() + "验证失败"); message.setMesType(MessageType.MESSAGE_LOGIN_FAIL); oos.writeObject(message); //关闭socket socket.close(); } } } catch (Exception e) { e.printStackTrace(); } finally { //如果服务器退出了while,说明服务器端不在监听,因此需要关闭ServerSocket try{ ss.close(); } catch (IOException e){ e.printStackTrace(); } } } } ===============//ServerConnectClientThread//=============== public class ServerConnectClientThread extends Thread { private Socket socket; private String userId;//连接到服务端端的用户id public ServerConnectClientThread(Socket socket, String userId) { this.socket = socket; this.userId = userId; } public Socket getSocket() { return socket; } @Override public void run() {//这里的线程处于run状态,可以发送接收信息 while (true) { try { System.out.println("服务器端和客户端" + userId + "保持通讯,读取数据..."); ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); Message message = (Message) ois.readObject(); //后面会使用message,根据message的类型,做相应的业务处理 if (message.getMesType().equals(MessageType.MESSAGE_GET_ONLINE_FRIEND)){ //客户端要在线用户列表 System.out.println(message.getSender() + "要在线用户列表"); String onlineUsers = ManageClientThread.getOnlineUsers(); //返回message //构建一个message对象,返回给客户端 Message message2 = new Message(); message2.setMesType(MessageType.MESSAGE_RET_ONLINE_FRIEND); message2.setContent(onlineUsers); message2.setGetter(message.getSender()); //返回给客户端 ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream()); oos.writeObject(message2); } else if (message.getMesType().equals(MessageType.MESSAGE_COMM_MES)){ //根据message获取getter id,在得到对应的线程 ServerConnectClientThread serverConnectClientThread = ManageClientThread.getServerConnectClientThread(message.getGetter()); //得到对应的socket的对象输出流,将message对象转发给指定的客户 ObjectOutputStream oos = new ObjectOutputStream(serverConnectClientThread.getSocket().getOutputStream()); oos.writeObject(message);//转发,提示如果客户不在线,可以保存到数据库,这样就可以实现离线留言 } else if (message.getMesType().equals(MessageType.MESSAGE_TO_ALL_MES)){ //需要遍历 管理线程的合集,把所有的线程的socket得到,然后把message进行转发 HashMap<String, ServerConnectClientThread> hm = ManageClientThread.getHm(); Iterator<String> iterator = hm.keySet().iterator(); while (iterator.hasNext()){ //取出在线用户的ID String onLineUserId = iterator.next().toString(); if (!onLineUserId.equals(message.getSender())){//排除群发消息的用户 //进行转发message ObjectOutputStream oos = new ObjectOutputStream(hm.get(onLineUserId).getSocket().getOutputStream()); oos.writeObject(message); } } } else if (message.getMesType().equals(MessageType.MESSAGE_FILE_MES)) { //根据getterid 获取到对应的线程,将message对象转发 ServerConnectClientThread serverConnectClientThread = ManageClientThread.getServerConnectClientThread(message.getGetter()); ObjectOutputStream oos = new ObjectOutputStream(serverConnectClientThread.getSocket().getOutputStream()); //转发 oos.writeObject(message); } else if (message.getMesType().equals(MessageType.MESSAGE_CLIENT_EXIT)){//客户端退出 System.out.println(message.getSender() + "退出"); //将这个客户端对应的线程,从集合中删除 ManageClientThread.removeServerConnectClientThread(message.getSender()); socket.close();//关闭连接 //退出线程 break; } else { System.out.println("其他类型的message暂不处理"); } } catch (Exception e) { e.printStackTrace(); } } } } ===============//User//=============== public class User implements Serializable { private static final long serialVersionUID = 1L; private String userID;//用户ID private String passwd;//用户密码 public User(){} public User (String userID, String passwd) { this.userID = userID; this.passwd = passwd; } public String getUserID() { return userID; } public void setUserID(String userID) { this.userID = userID; } public String getPasswd() { return passwd; } public void setPasswd(String passwd) { this.passwd = passwd; } }
QQClientService包
===============//ClientConncetServerThread//=============== public class ClientConncetServerThread extends Thread { //该线程需要持有Socket private Socket socket; //构造器可以接收一个Socket对象 public ClientConncetServerThread(Socket socket){ this.socket = socket; } @Override public void run() { //因为Thread需要在后台和服务器通信,所以我们要一个while循环 while (true) { try { System.out.println("客户端线程,等待读取从服务器端发送的信息"); ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); //如果服务器没有发送Message对象,线程会阻塞在这里 Message message = (Message) ois.readObject(); //判断这个message类型,然后做相应的业务处理 //如果是读取到的是 服务端返回的在线用户列表 if (message.getMesType().equals(MessageType.MESSAGE_RET_ONLINE_FRIEND)) { //取出在线列表信息,并显示 String[] onlineUsers = message.getContent().split(" "); System.out.println("\n=====当前在线用户列表====="); for (int i = 0; i < onlineUsers.length; i ++){ System.out.println("用户: " + onlineUsers[i]); } } else if (message.getMesType().equals(MessageType.MESSAGE_COMM_MES)){//普通的聊天消息 //把从服务器转发的消息,显示到控制台前即可 System.out.println("\n" + message.getSender() + " 对 " + message.getGetter() + " 说: " + message.getContent()); } else if (message.getMesType().equals(MessageType.MESSAGE_TO_ALL_MES)){ //显示在客户端的控制台 System.out.println("\n" + message.getSender() + " 对大家说: " + message.getContent()); } else if (message.getMesType().equals(MessageType.MESSAGE_FILE_MES)) { //如果是文件消息 System.out.println("\n" + message.getSender() + " 给 " + message.getGetter() + " 发文件: " + message.getSrc() + " 到我的电脑目录 " + message.getDest()); //取出message的文件字节数组,通过文件输出流写入到磁盘 FileOutputStream fileOutputStream = new FileOutputStream(message.getDest()); fileOutputStream.write(message.getFileBytes()); fileOutputStream.close(); System.out.println("\n 保存文件成功~"); } else { System.out.println("是其他类型的message,暂不处理..."); } } catch (Exception e) { e.printStackTrace(); } } } //为了更方便得到Socket public Socket getSocket() { return socket; } } ===============//FileClientService//=============== public class FileClientService { public void sendFileToOne(String src, String dest, String senderId, String getterId){ //读取src文件 --> message Message message = new Message(); message.setMesType(MessageType.MESSAGE_FILE_MES); message.setSender(senderId); message.setGetter(getterId); message.setSrc(src); message.setDest(dest); //需要将文件进行读取 FileInputStream fileInputStream = null; byte[] fileBytes = new byte[(int) new File(src).length()]; try{ fileInputStream = new FileInputStream(src); fileInputStream.read(fileBytes);//将src文件读入到程序的字节数组 //将文件的字节数设置到message对象 message.setFileBytes(fileBytes); } catch (Exception e) { e.printStackTrace(); }finally { //关闭 if (fileInputStream != null) { try{ fileInputStream.close(); } catch (IOException e) { e.printStackTrace(); } } } //提示信息 System.out.println("\n" + senderId + " 给 " + getterId + " 发送文件: " + src + " 到对的电脑的目录: " + dest); //发送 try{ ObjectOutputStream oos = new ObjectOutputStream(ManageClientConnectServerThread.getClientConncetServerThread(senderId).getSocket().getOutputStream()); } catch (IOException e) { e.printStackTrace(); } } } ===============//ManageClientConnectServerThread//=============== public class ManageClientConnectServerThread { //我们把多个线程放入一个HashMap集合,key 就是用户id;value 就是一个线程 private static HashMap<String, ClientConncetServerThread> hm = new HashMap<>(); //将某个线程加入到集合 public static void addClientConnectServerThread(String userId, ClientConncetServerThread clientConncetServerThread){ hm.put(userId, clientConncetServerThread); } //通过userId,可以得到对应的线程 public static ClientConncetServerThread getClientConncetServerThread(String userId){ return hm.get(userId); } } ===============//UserClientService//=============== public class UserClientService { //因为可能在其他地方要使用User信息,所以做成属性 private User u = new User(); //因为Socket在其它地方也有可能使用,因此也做成属性 private Socket socket; //根据userId 和 pwd 到服务器验证用户是否合法 public boolean checkUser(String userId, String pwd) { boolean b = false; //创建User对象 u.setUserID(userId); u.setPasswd(pwd); //连接到服务端,发送u对象 try { socket = new Socket(InetAddress.getByName("127.0.0.1"), 9999); //得到ObjectOutputStream对象 ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream()); oos.writeObject(u);//发送User对象 //读取从服务端回复的Message对象 ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); Message ms = (Message) ois.readObject(); if (ms.getMesType().equals(MessageType.MESSAGE_LOGIN_SUCCEED)){//登录成功 b = true; //创建一个和服务器端保持通信的线程 --> 创建一个类 ClientConnectServerThread ClientConncetServerThread clientConncetServerThread = new ClientConncetServerThread(socket); //启动客户端的线程 clientConncetServerThread.start(); //这里为了后面客户端的扩展,我们将线程放入到集合管理 ManageClientConnectServerThread.addClientConnectServerThread(userId, clientConncetServerThread); }else { //如果登陆失败,我们就不能启动和服务器通信的线程,关闭Socket socket.close(); } } catch (Exception e){ e.printStackTrace(); } return b; } //向服务器请求在线用户列表 public void onlineFriendList() { //发送一个Message,类型MESSAGE_GET_ONLINE_FRIEND Message message = new Message(); message.setMesType(MessageType.MESSAGE_GET_ONLINE_FRIEND); message.setSender(u.getUserID()); //发送给服务器 try{ //应该得到当前线程的Sock 对应的 ObjectOutputStream对象 ObjectOutputStream oos = new ObjectOutputStream (ManageClientConnectServerThread.getClientConncetServerThread(u.getUserID()).getSocket().getOutputStream()); oos.writeObject(message);//发送一个message对象,向服务端要求在线用户列表 } catch (IOException e) { e.printStackTrace(); } } //编写方法,退出客户端,并给服务端发送一个退出系统的message对象 public void logout() { Message message = new Message(); message.setMesType(MessageType.MESSAGE_CLIENT_EXIT); message.setSender(u.getUserID());//一定要指定我是哪个客户端id //发送message try{ ObjectOutputStream oos = new ObjectOutputStream(ManageClientConnectServerThread.getClientConncetServerThread(u.getUserID()).getSocket().getOutputStream()); oos.writeObject(message); System.out.println(u.getUserID() + " 退出系统"); System.exit(0);//结束进程 } catch (IOException e) { e.printStackTrace(); } } }
QQClient.view包
===============//QQView//=============== public class QQView { private boolean loop = true;//控制是否显示菜单 private String key = "";//接收用户的键盘输入 private UserClientService userClientService = new UserClientService();//对象是用于登录服务器/注册用户 private MessageClientService messageClientService = new MessageClientService();//对象用户私聊/群聊 private FileClientService fileClientService = new FileClientService();//该对象用于传输文件 public static void main(String[] args) { new QQView().mainMenu(); System.out.println("客户端退出系统..."); } //显示主菜单 private void mainMenu(){ while (loop){ System.out.println("===========欢迎登录网络通信系统=========="); System.out.println("\t\t 1 登陆系统"); System.out.println("\t\t 9 退出系统"); System.out.print("请输入你的选择:"); key = utils.readString(1); //根据用户的输入,来处理不同的逻辑 switch (key){ case "1": System.out.print("请输入用户号:"); String userID = utils.readString(50); System.out.print("请输入密 码:"); String pwd = utils.readString(50); //需要到服务端验证该用户是否合法 //这里有很多代码,所以在这里编写一个UserClientService if (userClientService.checkUser(userID, pwd)){ System.out.println("===========欢迎(用户"+ userID +"登陆成功)=========="); //进入二级菜单 while (loop){ System.out.println("\n===========网络通信二级菜单(用户"+ userID +")=========="); System.out.println("\t\t 1 显示在线用户列表"); System.out.println("\t\t 2 群发消息"); System.out.println("\t\t 3 私聊消息"); System.out.println("\t\t 4 发送文件"); System.out.println("\t\t 9 退出系统"); System.out.println("请输入你的选择:"); key = utils.readString(1); switch (key){ case "1": userClientService.onlineFriendList(); break; case "2": System.out.println("请输入想对大家说的话"); String s = utils.readString(100); messageClientService.sendMessageToAll(s, userID); //调用一个方法,将消息封装成message对象,发送给服务端 break; case "3": System.out.println("请输入想聊天的用户号(在线):"); String getterID = utils.readString(50); System.out.println("请输入想说的话"); String content = utils.readString(100); //编写一个方法,将消息发送给服务端 messageClientService.sendMessageToOne(content,userID,getterID); System.out.println("\t\t 3 私聊消息"); break; case "4": System.out.println("请输入你想把文件发送给的用户(在线):"); getterID = utils.readString(50); System.out.println("请输入发送文件的路径(形式 d:\\xx.jpg)"); String src = utils.readString(100); System.out.println("请输入把文件发送到对方的路径(形式 d:\\xx.jpg)"); String dest = utils.readString(100); fileClientService.sendFileToOne(src,dest,userID,getterID); break; case "9": loop = false; break; } } }else {//登录服务器失败 System.out.println("==========登陆服务器失败=========="); } break; case "9": //调用方法,给服务器发送一个退出系统的message loop = false; break; } } } }
qqframe包
public class QQFrame { public static void main(String[] args) { new QQServer(); } }
QQUtils
public class utils { //静态属性。。。 private static Scanner scanner = new Scanner(System.in); /** * 功能:读取键盘输入的一个菜单选项,值:1——5的范围 * @return 1——5 */ public static char readMenuSelection() { char c; for (; ; ) { String str = readKeyBoard(1, false);//包含一个字符的字符串 c = str.charAt(0);//将字符串转换成字符char类型 if (c != '1' && c != '2' && c != '3' && c != '4' && c != '5') { System.out.print("选择错误,请重新输入:"); } else break; } return c; } /** * 功能:读取键盘输入的一个字符 * @return 一个字符 */ public static char readChar() { String str = readKeyBoard(1, false);//就是一个字符 return str.charAt(0); } /** * 功能:读取键盘输入的一个字符,如果直接按回车,则返回指定的默认值;否则返回输入的那个字符 * @param defaultValue 指定的默认值 * @return 默认值或输入的字符 */ public static char readChar(char defaultValue) { String str = readKeyBoard(1, true);//要么是空字符串,要么是一个字符 return (str.length() == 0) ? defaultValue : str.charAt(0); } /** * 功能:读取键盘输入的整型,长度小于2位 * @return 整数 */ public static int readInt() { int n; for (; ; ) { String str = readKeyBoard(10, false);//一个整数,长度<=10位 try { n = Integer.parseInt(str);//将字符串转换成整数 break; } catch (NumberFormatException e) { System.out.print("数字输入错误,请重新输入:"); } } return n; } /** * 功能:读取键盘输入的 整数或默认值,如果直接回车,则返回默认值,否则返回输入的整数 * @param defaultValue 指定的默认值 * @return 整数或默认值 */ public static int readInt(int defaultValue) { int n; for (; ; ) { String str = readKeyBoard(10, true); if (str.equals("")) { return defaultValue; } //异常处理... try { n = Integer.parseInt(str); break; } catch (NumberFormatException e) { System.out.print("数字输入错误,请重新输入:"); } } return n; } /** * 功能:读取键盘输入的指定长度的字符串 * @param limit 限制的长度 * @return 指定长度的字符串 */ public static String readString(int limit) { return readKeyBoard(limit, false); } /** * 功能:读取键盘输入的指定长度的字符串或默认值,如果直接回车,返回默认值,否则返回字符串 * @param limit 限制的长度 * @param defaultValue 指定的默认值 * @return 指定长度的字符串 */ public static String readString(int limit, String defaultValue) { String str = readKeyBoard(limit, true); return str.equals("")? defaultValue : str; } /** * 功能:读取键盘输入的确认选项,Y或N * 将小的功能,封装到一个方法中. * @return Y或N */ public static char readConfirmSelection() { System.out.println("请输入你的选择(Y/N): 请小心选择"); char c; for (; ; ) {//无限循环 //在这里,将接受到字符,转成了大写字母 //y => Y n=>N String str = readKeyBoard(1, false).toUpperCase(); c = str.charAt(0); if (c == 'Y' || c == 'N') { break; } else { System.out.print("选择错误,请重新输入:"); } } return c; } /** * 功能: 读取一个字符串 * @param limit 读取的长度 * @param blankReturn 如果为true ,表示 可以读空字符串。 * 如果为false表示 不能读空字符串。 * * 如果输入为空,或者输入大于limit的长度,就会提示重新输入。 * @return */ private static String readKeyBoard(int limit, boolean blankReturn) { //定义了字符串 String line = ""; //scanner.hasNextLine() 判断有没有下一行 while (scanner.hasNextLine()) { line = scanner.nextLine();//读取这一行 //如果line.length=0, 即用户没有输入任何内容,直接回车 if (line.length() == 0) { if (blankReturn) return line;//如果blankReturn=true,可以返回空串 else continue; //如果blankReturn=false,不接受空串,必须输入内容 } //如果用户输入的内容大于了 limit,就提示重写输入 //如果用户如的内容 >0 <= limit ,我就接受 if (line.length() < 1 || line.length() > limit) { System.out.print("输入长度(不能大于" + limit + ")错误,请重新输入:"); continue; } break; } return line; } }