聊天室 SWT Socket 简单实现

聊天室 SWT Socket 简单实现

说起来惭愧，都已经大三了，也没写过一篇博客，也就是这次新冠爆发，宅家里宅久了，脑子才醒过来，不能在颓废了，好吧从这第一篇博客开始，每过一段时间就总结一下心得。激励一下自己。

写点什么呢

想了很久，但总要写点东西吧，刚好前几天上网课讲多核程序，我又刚好在想计算机通信的问题，那就花三天写一个简单的聊天室程序吧。想看的朋友可以随便看看，很简陋主要是记录学习。
gitee 聊天室

成果

本是想做一个局域网聊天的工具，多台客户端连一个服务端的C/S结构，因为时间问题，我没有做服务端的界面和数据库，对于数据的储存这一块我还没想好（可能宅久了，脑子瓦特了）。（还有消息提示和警告框也没来的及做，还有头像，文件图片传输问题我以后再加）
好吧，看看这三天的成果。


有点简陋还有好几个bug(头疼)，这几天我好好的思考一下。

简单思路

既然是C/S结构，那么肯定需要一个Server，Server 和 Client 之间的通信用Socket实现（关于Socket 我也才学了两天，不过我发现一篇很棒的文章）
Socket原理实现
大概流程就是Client连接Server，Client 发送请求，Server 接收并分析请求，做出对应的操作并给Client发送响应，Client 接受并分析响应，做出相应的操作。（需要多线程实现。)原理很简单，但实现起来着实让我废了点功夫。果然罗马不是一天建成的。花几天的时间把Socket和SWT学透还是不现实。

贴几段主要代码吧

Server的Main

public class ServerMain {public static void main(String[] args) {// TODO Auto-generated method stubint i=0;List<User>  ulist = new ArrayList<User>();//List<Integer> user1_friends_id = new ArrayList<Integer>();//List<Integer> user2_friends_id = new ArrayList<Integer>();List<ClientUser> clist = new ArrayList<ClientUser>();User user1 = new User(1001,"zjj1","123456","男");User user2 = new User(1002,"zjj2","654321","女");User user3 = new User(1003,"zjj3","654321","女");User user4 = new User(1004,"zjj4","654321","女");User user5 = new User(1005,"zjj5","654321","女");User user6 = new User(1006,"zjj6","654321","女");/** 初始一下用户*///user1.setFriends_id(user1_friends_id);//user2.setFriends_id(user2_friends_id);ulist.add(user1);ulist.add(user2);ulist.add(user3);ulist.add(user4);ulist.add(user5);ulist.add(user6);try {ServerSocket ss = new ServerSocket(9001);//端口号为 9001 System.out.println("服务器启动");while(true){Socket s = ss.accept();//等待一个客户端连接，否则堵塞i++;ClientUser cuser = new ClientUser("client"+i,s);clist.add(cuser);//将客户端加入 clistSystem.out.println(cuser.getName() + "正在请求");ServerThread sthread = new ServerThread(cuser,clist,ulist);//开始线程监听sthread.start();}} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}}}

就是通过Socket的accept()方法让Server堵塞直到有一个Client接入，然后开启Server的线程监听Client。为了方便管理Client我建立了一个List，这样就能实现对Client to Client的消息传递。

Server的Thread

public class ServerThread extends Thread {private ClientUser cuser;private List<ClientUser> clist;private List<User> ulist;public ServerThread(ClientUser cuser, List<ClientUser> clist, List<User> ulist) {super();this.cuser = cuser;this.clist = clist;this.ulist = ulist;}public void run() {ServerMethod sm =new ServerMethod(cuser,clist,ulist);//sm 为服务器方法类try {while (true) {//一直监听client的消息String msg= cuser.getBr().readLine();//System.out.println(msg);String[] str = msg.split("//");//消息格式为 请求头//请求体switch (str[0]) {case "logout"://登出请求sm.logout_user(cuser.getUid());break;case "register"://注册请求sm.registerUser(str[1],str[2],str[3]);break;case "login"://登录请求sm.loginUser(str[1], str[2]);break;case "send"://转发消息请求sm.sendClientToClient(str[1], str[2]);break;case "refresh_all_user"://刷新请求sm.sendAllUserMessage();break;default:break;}}} catch (Exception e) {System.out.println(cuser.getUid()+"断开");sm.remove(cuser);//client若断开就remove clist} finally {try {cuser.getBr().close();cuser.getSocket().close();} catch (IOException e) {e.printStackTrace();}}}}

就是通过监听Client传过来的消息做出反应。(这里的readLine()会堵塞，所以我中断线程的时候用stop()。虽然不建议用这种方法，但我一时想不到什么解决办法)。

Server 的方法类

public class ServerMethod {private ClientUser cuser;private List<ClientUser> clist;private List<User> ulist;public ServerMethod(ClientUser cuser, List<ClientUser> clist, List<User> ulist) {super();this.cuser = cuser;this.clist = clist;this.ulist = ulist;}//注册新用户public void registerUser(String name, String paw, String sex){int id = ulist.get(ulist.size()-1).getUserid();id++;User user = new User(id,name,paw,sex);ulist.add(user);//System.out.println("添加成功"+user.toString());sendClient("register_success//"+"注册成功你的账号为:"+id+"请牢记!");}//发送所有用户的状态，用于刷新public void sendAllUserMessage(){String st = "send_all_user//";for(User user:ulist){if(user.getUserid()!=cuser.getUid()){st+=user.getUserid()+"//"+user.getUsername()+"//"+user.isOnline()+"//";}}//System.out.println(st);sendClient(st);}//用户登录public int loginUser(String id, String paw){int uid = Integer.parseInt(id); for(User user:ulist){if(user.getUserid()==uid){if(user.getUserpaw().equals(paw)){if(user.isOnline()==true){loginClash(uid,user);return 0;}sendClient("login_success//"+user.getUsername()+"//");user.setOnline(true);cuser.setUid(uid);cuser.setName(user.getUsername());//System.out.println(user.toString());sendAllUserMessage();return 1;}else{sendClient("paw_error//");return 0;}}}sendClient("id_error//");return 0;}//检查用户是否登录冲突public void loginClash(int uid,User user){for(ClientUser cuser:clist){if(cuser.getUid()==uid){try {sendClient("login_clash//");sendClientByClientUser("login_clash//",cuser);user.setOnline(false);cuser.getSocket().close();this.cuser.getSocket().close();remove(cuser);remove(this.cuser);} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}}}}//给client发送响应public void sendClient(String message){try {PrintWriter pw =cuser.getPw();pw.println(message);pw.flush();} catch (Exception e) {e.printStackTrace();}}//用于转发客户端消息public void sendClientToClient(String uid,String message){int id = Integer.parseInt(uid); try {for(ClientUser cuser:clist){if(cuser.getUid()==id){PrintWriter pw =cuser.getPw();pw.println("receive_message//"+this.cuser.getUid()+"//"+this.cuser.getName()+"//"+message);pw.flush();}}} catch (Exception e) {e.printStackTrace();}}//用于发送登录冲突public void sendClientByClientUser(String message,ClientUser cuser){try {PrintWriter pw =cuser.getPw();pw.println(message);pw.flush();} catch (Exception e) {e.printStackTrace();}}//登出public void logout_user(int uid){for(User user:ulist){if(user.getUserid()==uid){user.setOnline(false);}}}public void remove(ClientUser cuser){for(User user:ulist){if(cuser.getUid()==user.getUserid()){user.setOnline(false);System.out.println(user.toString());}}clist.remove(cuser);}}

配合Thread做出反应。

Client的main

public class ClientMain {public static ClientThread ct;public static Display display;public static String[][] all_user ;public static String username;public static void main(String[] args) {// TODO Auto-generated method stubtry {Socket s = new Socket("localhost", 9001);//localhost为Server地址，这为本机ct = new ClientThread(s);//开启线程ct.start();display = new Display();LoginUi lgui = new LoginUi(s,display);//登录界面} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}}}

Client的Thread

public class ClientThread extends Thread{private Socket s;private boolean exit = false;//中断线程标志private boolean success_login = false;//登陆成功标志private boolean success_register = false;//注册成功标志private boolean receive_all_user = false;//获取刷新用户信息标志private List<ReceiveMessage> receive_message;//用于缓存信息private boolean  is_receive = false;//接受信息完成标志private int all_user_num = 0;//用户数private List<ReceiveMessage> local_receive_message;//本地信息public ClientThread(Socket s) {super();this.s = s;}public void run() {receive_message = new ArrayList<ReceiveMessage>();local_receive_message = new ArrayList<ReceiveMessage>();ClientMethod cm = new ClientMethod(s);try {InputStream inputStream = s.getInputStream();InputStreamReader inputStreamReader = new InputStreamReader(inputStream);BufferedReader br = new BufferedReader(inputStreamReader);try {//一直监听Server的响应while (!exit) { String msg=br.readLine();//System.out.println(msg);String[] str = msg.split("//");//响应头//响应体switch (str[0]) {case "register_success":success_register = true;System.out.println(str[1]);break;case "login_success":System.out.println("登录成功");ClientMain.username=str[1];//用户名success_login = true;break;case "paw_error":System.out.println("密码错误");break;case "id_error":System.out.println("账号错误");break;case "login_clash":System.out.println("账号冲突");success_login = false;exit=true;s.close();break;case "receive_message"://if(receive_message!=null && !receive_message.isEmpty())cm.saveUserMessage(str);is_receive = true;break;case "send_all_user":all_user_num = ((str.length-1)/3);cm.getAllUser(str);// System.out.println(all_user_num);receive_all_user = true;break;	default:break;}}} catch (Exception e) {e.printStackTrace();}} catch (IOException e1) {e1.printStackTrace();}}public boolean isReceive_all_user() {return receive_all_user;}public void setReceive_all_user(boolean receive_all_user) {this.receive_all_user = receive_all_user;}public boolean isSuccess_login() {return success_login;}public int getAll_user_num() {return all_user_num;}public void setAll_user_num(int is_online_num) {this.all_user_num = is_online_num;}public List<ReceiveMessage> getReceive_message() {return receive_message;}public void setReceive_message(List<ReceiveMessage> receive_message) {this.receive_message = receive_message;}public boolean is_receive() {return is_receive;}public List<ReceiveMessage> getLocal_receive_message() {return local_receive_message;}public void setLocal_receive_message(List<ReceiveMessage> local_receive_message) {this.local_receive_message = local_receive_message;}public void setIs_receive(boolean is_receive) {this.is_receive = is_receive;}public boolean isSuccess_register() {return success_register;}public void setSuccess_register(boolean success_register) {this.success_register = success_register;}}

用于监听Server的响应。

Client 的方法

public class ClientMethod {private Socket s;public ClientMethod(Socket s) {super();this.s = s;}//发送请求public void sentMessage(String st){BufferedWriter bw;try {bw = new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));InputStream is = new ByteArrayInputStream(st.getBytes());BufferedReader br =new BufferedReader(new InputStreamReader(is,"UTF-8"));String str = br.readLine();bw.write(str);bw.write("\n");bw.flush();}catch(Exception e){System.out.println("服务器异常");}}//获得刷新用户public void getAllUser(String[] str){int t = 1;int i=0, j=0;ClientMain.all_user = new String[ClientMain.ct.getAll_user_num()][3];for( i = 0; i <ClientMain.ct.getAll_user_num(); i++){for( j = 0; j < 3; j++){ClientMain.all_user[i][j] = str[t];//System.out.println(ClientMain.all_user[i][j]);t++;}}}//保存缓存信息public void saveUserMessage(String[] str){boolean exist =false;//判断信息来源用户本地是否存在for(int i=0; i<ClientMain.ct.getReceive_message().size();i++){if(ClientMain.ct.getReceive_message().get(i).getUid().equals(str[1])){ClientMain.ct.getReceive_message().get(i).setMessage(str[3]);ClientMain.ct.getLocal_receive_message().get(i).setMessage(ClientMain.ct.getLocal_receive_message().get(i).getMessage()+ClientMain.ct.getLocal_receive_message().get(i).getUname()+":"+str[3]+"\n");exist = true;}}if(exist == false){ReceiveMessage rm = new ReceiveMessage(str[1],str[2],str[3]);ReceiveMessage lrm = new ReceiveMessage(str[1],str[2],str[2]+":"+str[3]+"\n");ClientMain.ct.getReceive_message().add(rm);ClientMain.ct.getLocal_receive_message().add(lrm);exist = true;}}//关闭线程public void closeThread(){try {ClientMain.ct.stop();//不推荐使用stop,但线程中的readline()用一直堵塞s.close();ClientMain.display.dispose();System.exit(0);} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}ClientMain.display.dispose();}}

总结

三天写成这样对我其实不算轻松。（虽然我自己都觉得很烂）
但总算有点SWT和Socket的收获。
不足：
1.数据存储没有思考好
2.线程规则有点乱
3.代码不规范（老问题了，该找个时间看看规范了）
4.在Server对Client和User有过多的查表操作(太low的操作）可以考虑用Map
5.功能单一
写完项目再总结感觉收获挺多啊，加油吧，大三苦逼仔。