service
ClientConnectServerThread
package com.hspedu.qqclient.service;
import com.hspedu.qqcommon.Message;
import com.hspedu.qqcommon.MessageType;
import java.io.FileOutputStream;
import java.io.ObjectInputStream;
import java.net.Socket;
/**
*该类为登录后创建的线程，和服务器保持通讯，根据服务器传回的数据进行对应操作
*/
public class ClientConnectServerThread extends Thread {
//该线程需要持有Socket
private Socket socket;
//构造器可以接受一个Socket对象
public ClientConnectServerThread(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
//判断这个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(), true);
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
package com.hspedu.qqclient.service;
import com.hspedu.qqcommon.Message;
import com.hspedu.qqcommon.MessageType;
import java.io.*;
/**
* 该类/对象完成 文件传输服务
*/
public class FileClientService {
/**
*
* @param src 源文件
* @param dest 把该文件传输到对方的哪个目录
* @param senderId 发送用户id
* @param getterId 接收用户id
*/
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.getClientConnectServerThread(senderId).getSocket().getOutputStream());
oos.writeObject(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}
ManageClientConnectServerThread
package com.hspedu.qqclient.service;
import java.util.HashMap;
/**
* 该类管理客户端连接到服务器端的线程的类
*/
public class ManageClientConnectServerThread {
//我们把多个线程放入一个HashMap集合，key 就是用户id, value 就是线程
private static HashMap hm = new HashMap<>();
//将某个线程加入到集合
public static void addClientConnectServerThread(String userId, ClientConnectServerThread clientConnectServerThread) {
hm.put(userId, clientConnectServerThread);
}
//通过userId 可以得到对应线程
public static ClientConnectServerThread getClientConnectServerThread(String userId) {
return hm.get(userId);
}
}
MessageClientService
package com.hspedu.qqclient.service;
import com.hspedu.qqcommon.Message;
import com.hspedu.qqcommon.MessageType;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.util.Date;
/**
* 该类/对象，提供和消息相关的服务方法 发送信息给服务器
*/
public class MessageClientService {
/**
* @param content 内容
* @param senderId 发送者
*/
public void sendMessageToAll(String content, String senderId) {
//构建message
Message message = new Message();
message.setMesType(MessageType.MESSAGE_TO_ALL_MES);//群发消息这种类型
message.setSender(senderId);
message.setContent(content);
message.setSendTime(new Date().toString());//发送时间设置到message对象
System.out.println(senderId + " 对大家说 " + content);
//发送给服务端
try {
ObjectOutputStream oos =
new ObjectOutputStream(ManageClientConnectServerThread.getClientConnectServerThread(senderId).getSocket().getOutputStream());
oos.writeObject(message);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* @param content 内容
* @param senderId 发送用户id
* @param getterId 接收用户id
*/
public void sendMessageToOne(String content, String senderId, String getterId) {
//构建message
Message message = new Message();
message.setMesType(MessageType.MESSAGE_COMM_MES);//普通的聊天消息这种类型
message.setSender(senderId);
message.setGetter(getterId);
message.setContent(content);
message.setSendTime(new Date().toString());//发送时间设置到message对象
System.out.println(senderId + " 对 " + getterId + " 说 " + content);
//发送给服务端
try {
ObjectOutputStream oos =
new ObjectOutputStream(ManageClientConnectServerThread.getClientConnectServerThread(senderId).getSocket().getOutputStream());
oos.writeObject(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}
UserClientService
package com.hspedu.qqclient.service;
import com.hspedu.qqcommon.Message;
import com.hspedu.qqcommon.MessageType;
import com.hspedu.qqcommon.User;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.InetAddress;
import java.net.Socket;
/**
* 该类向服务器发送信息 完成用户登录验证、退出登录和用户注册等功能.
*/
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);
try {
//连接到服务端，发送u对象
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)) {//登录OK
//创建一个和服务器端保持通信的线程-> 创建一个类 ClientConnectServerThread
ClientConnectServerThread clientConnectServerThread =
new ClientConnectServerThread(socket);
//启动客户端的线程
clientConnectServerThread.start();
//这里为了后面客户端的扩展，我们将线程放入到集合管理
ManageClientConnectServerThread.addClientConnectServerThread(userId, clientConnectServerThread);
b = true;
} 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 {
//从管理线程的集合中，通过userId, 得到这个线程对象
ClientConnectServerThread clientConnectServerThread =
ManageClientConnectServerThread.getClientConnectServerThread(u.getUserId());
//通过这个线程得到关联的socket
Socket socket = clientConnectServerThread.getSocket();
//得到当前线程的Socket 对应的 ObjectOutputStream对象
ObjectOutputStream oos = new ObjectOutputStream(socket.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(socket.getOutputStream());
ObjectOutputStream oos =
new ObjectOutputStream(ManageClientConnectServerThread.getClientConnectServerThread(u.getUserId()).getSocket().getOutputStream());
oos.writeObject(message);
System.out.println(u.getUserId() + " 退出系统 ");
System.exit(0);//结束进程
} catch (IOException e) {
e.printStackTrace();
}
}
}
utils
Utility
package com.hspedu.qqclient.utils;
/**
工具类的作用:
处理各种情况的用户输入，并且能够按照程序员的需求，得到用户的控制台输入。
*/
import java.util.Scanner;
/**
*/
public class Utility {
//静态属性。。。
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);//一个整数，长度 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) {
System.out.print("输入长度（不能大于" + limit + "）错误，请重新输入：");
continue;
}
break;
}
return line;
}
}
view
QQView
package com.hspedu.qqclient.view;
import com.hspedu.qqclient.service.FileClientService;
import com.hspedu.qqclient.service.MessageClientService;
import com.hspedu.qqclient.service.UserClientService;
import com.hspedu.qqclient.utils.Utility;
/**
* 登录界面和登录后的界面
*/
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 = Utility.readString(1);
//根据用户的输入，来处理不同的逻辑
switch (key) {
case "1":
System.out.print("请输入用户号: ");
String userId = Utility.readString(50);
System.out.print("请输入密 码: ");
String pwd = Utility.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.print("请输入你的选择: ");
key = Utility.readString(1);
switch (key) {
case "1":
//这里老师准备写一个方法，来获取在线用户列表
userClientService.onlineFriendList();
break;
case "2"://群发消息
System.out.println("请输入想对大家说的话: ");
String s = Utility.readString(100);
messageClientService.sendMessageToAll(s, userId);
break;
case "3"://私发消息
System.out.print("请输入想聊天的用户号(在线): ");
String getterId = Utility.readString(50);
System.out.print("请输入想说的话: ");
String content = Utility.readString(100);
//编写一个方法，将消息发送给服务器端
messageClientService.sendMessageToOne(content, userId, getterId);
break;
case "4"://发文件
System.out.print("请输入你想把文件发送给的用户(在线用户): ");
getterId = Utility.readString(50);
System.out.print("请输入发送文件的路径(形式 d:\\xx.jpg)");
String src = Utility.readString(100);
System.out.print("请输入把文件发送到对应的路径(形式 d:\\yy.jpg)");
String dest = Utility.readString(100);
fileClientService.sendFileToOne(src, dest, userId, getterId);
break;
case "9":
//调用方法，给服务器发送一个退出系统的message
userClientService.logout();
loop = false;
break;
}
}
} else { //登录服务器失败
System.out.println("=========登录失败=========");
}
break;
case "9":
loop = false;
break;
}
}
}
}
Original: https://www.cnblogs.com/wenjie2000/p/16683379.html
Author: 文杰2000
Title: Java学习-第一部分-第三阶段-项目实战：多用户即时通讯系统
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/598844/
转载文章受原作者版权保护。转载请注明原作者出处！