java socket实现服务端，客户端简单网络通信。Chat

2021-06-14 17:06

阅读：565

标签：ons relative 重要 field values desc 点击事件 utc ane

之前写的实现简单网络通信的代码，有一些严重bug。后面详细写。

根据上次的代码，主要增加了用户注册，登录页面，以及实现了实时显示当前在登录状态的人数。并解决一些上次未发现的bug。(主要功能代码参见之前随笔 https://www.cnblogs.com/yuqingsong-cheng/p/12740307.html)

实现用户注册登录就需要用到数据库，因为我主要在学Sql Server。Sql Server也已支持Linux系统。便先在我的电脑Ubuntu系统下进行安装配置。

链接：https://docs.microsoft.com/zh-cn/sql/linux/quickstart-install-connect-red-hat?view=sql-server-ver15

Sql Server官网有各个系统的安装指导文档，所以按照正常的安装步骤，一切正常安装。

可放到服务器中却出现了问题。阿里云学生服务器是2G内存的（做活动外加学生证，真的很香。但内存有点小了）。sqlserer需要至少2G内存。所以只能放弃SqlServer,转向Mysql。

同样根据MySql的官方指导文档进行安装。但进行远程连接却需要一些“乱七八糟”的配置，于是开始“面向百度连接”，推荐一个解决方案，https://blog.csdn.net/ethan__xu/article/details/89320614 适用于mysql8.0以上版本。

数据库部分解决，开始写关于登录，注册类。登录注册部分新开了一个端口进行socket连接。由于功能较简单，所以只用到了插入，查询语句。

客户端读入用户输入的登录，注册信息，发送至服务端，服务端在连接数据库进行查询/插入操作，将结果发送至客户端。

实例代码

  1 package logindata;
  2 
  3 import java.io.DataInputStream;
  4 import java.io.DataOutputStream;
  5 import java.io.IOException;
  6 import java.net.ServerSocket;
  7 import java.net.Socket;
  8 import java.sql.Connection;
  9 import java.sql.DriverManager;
 10 import java.sql.ResultSet;
 11 import java.sql.SQLException;
 12 import java.sql.Statement;
 13 import java.util.ArrayList;
 14 
 15 public class LoginData implements Runnable{
 16 
 17     static ArrayList loginsocket = new ArrayList();
 18     
 19     public LoginData() { }
 20 
 21     @Override
 22     public void run() {
 23         ServerSocket serverSocket=null;
 24         try {
 25             serverSocket = new ServerSocket(6567);
 26         } catch (IOException e) {
 27             e.printStackTrace();
 28         }
 29         while(true) {
 30             Socket socket=null;
 31             try {
 32                 socket = serverSocket.accept();
 33             } catch (IOException e) {
 34                 // TODO Auto-generated catch block
 35                 e.printStackTrace();
 36             }
 37             loginsocket.add(socket);
 38             
 39             Runnable runnable;
 40             try {
 41                 runnable = new LoginDataIO(socket);
 42                 Thread thread = new Thread(runnable);
 43                 thread.start();
 44             } catch (IOException e) {
 45                 // TODO Auto-generated catch block
 46                 e.printStackTrace();
 47             }
 48         }
 49     }
 50 }
 51 
 52 class LoginDataIO implements Runnable{
 53 
 54     String b="false";
 55     Socket socket;
 56     DataInputStream inputStream;
 57     DataOutputStream outputStream;
 58     public LoginDataIO(Socket soc) throws IOException {
 59         socket = soc;
 60         inputStream = new DataInputStream(socket.getInputStream());
 61         outputStream = new DataOutputStream(socket.getOutputStream());
 62     }
 63     
 64     @Override
 65     public void run() {
 66         String readUTF = null;
 67         String readUTF2 = null;
 68         String readUTF3 = null;
 69         try {
 70             readUTF = inputStream.readUTF();
 71             readUTF2 = inputStream.readUTF();
 72             readUTF3 = inputStream.readUTF();
 73         } catch (IOException e) {
 74             e.printStackTrace();
 75         }
 76         
 77 //        System.out.println(readUTF+readUTF2+readUTF3);
 78         
 79         SqlServerCon serverCon = new SqlServerCon();
 80         try {
 81             //判断连接是登录还是注册，返回值不同。
 82             if(readUTF3.equals("login")) {
 83                 b=serverCon.con(readUTF, readUTF2);
 84                 outputStream.writeUTF(b);
 85             }else {
 86                 String re=serverCon.insert(readUTF, readUTF2);    
 87                 outputStream.writeUTF(re);
 88             }
 89         } catch (SQLException e) {
 90             // TODO Auto-generated catch block
 91             e.printStackTrace();
 92         } catch (IOException e) {
 93             // TODO Auto-generated catch block
 94             e.printStackTrace();
 95         } catch (ClassNotFoundException e) {
 96             // TODO Auto-generated catch block
 97             e.printStackTrace();
 98         }  
 99         
100 //        System.out.println(b);
101     }
102 }
103 
104 
105 class SqlServerCon {
106 
107     public SqlServerCon() {
108         // TODO Auto-generated constructor stub
109     }
110     
111     String name;
112     String password;
113 //    boolean duge = false;
114     String duge = "false";
115 //    String url = "jdbc:sqlserver://127.0.0.1:1433;"
116 //            + "databaseName=TestData;user=sa;password=123456";
117     /**
118      * com.mysql.jdbc.Driver 更换为 com.mysql.cj.jdbc.Driver。
119         MySQL 8.0 以上版本不需要建立 SSL 连接的，需要显示关闭。
120         最后还需要设置 CST。
121      */
122     //连接MySql数据库url格式
123     String url = "jdbc:mysql://127.0.0.1:3306/mytestdata?useSSL=false&serverTimezone=UTC";
124     public String con(String n,String p) throws SQLException, ClassNotFoundException {
125         Class.forName("com.mysql.cj.jdbc.Driver");
126         Connection connection = DriverManager.getConnection(url,"root","uu-7w3yfu?VX");
127 //        System.out.println(connection);
128         
129         Statement statement = connection.createStatement();
130 //        statement.executeUpdate("insert into Data values(‘china‘,‘123456‘)");
131         ResultSet executeQuery = statement.executeQuery("select * from persondata");
132         
133         //登录昵称密码确认
134         while(executeQuery.next()) {
135             name=executeQuery.getString(1).trim();
136             password = executeQuery.getString(2).trim();   //"使用这个方法很重要"  String     trim()      返回值是此字符串的字符串，其中已删除所有前导和尾随空格。
137 //            System.out.println(n.equals(name));
138             if(name.equals(n) && password.equals(p)) {
139                 duge="true";
140                 break;
141             }
142         }
143         statement.close();
144         connection.close();
145 //        System.out.println(duge);
146         return duge;
147     }
148     
149     public String insert(String n,String p) throws SQLException, ClassNotFoundException {
150         boolean b = true;
151         String re = null;
152         Class.forName("com.mysql.cj.jdbc.Driver");
153         Connection connection = DriverManager.getConnection(url,"root","uu-7w3yfu?VX");
154         Statement statement = connection.createStatement();
155         
156         ResultSet executeQuery = statement.executeQuery("select * from persondata");
157         while(executeQuery.next()) {
158             name=executeQuery.getString(1).trim();
159 //            password = executeQuery.getString(2).trim();  
160             if(name.equals(n)) {
161                 b=false;
162                 break;
163             }
164         }
165         
166         //返回登录信息
167         if(b && n.length()!=0 && p.length()!=0) {
168             String in = "insert into persondata "+"values("+"‘"+n+"‘"+","+"‘"+p+"‘"+")";  //这条插入语句写的很捞，但没想到更好的。
169 //            System.out.println(in);
170             statement.executeUpdate(in);
171             statement.close();
172             connection.close();
173             re="注册成功,请返回登录";
174             return re;
175         }else if(n.length()==0 || p.length()==0 ) {
176             re="昵称或密码不能为空，请重新输入";
177             return re;
178         }else {
179             re="已存在该昵称用户，请重新输入或登录";
180             return re;
181         }
182     }
183 }

因为服务端需要放到服务器中，所以就删去了服务端的用户界面。

 1 import file.File;
 2 import logindata.LoginData;
 3 import server.Server;
 4 
 5 public class ServerStart_View {
 6     
 7     private static Server server = new Server();
 8     private static File file = new File();
 9     private static LoginData loginData = new LoginData();
10     public static void main(String [] args) {
11         ServerStart_View frame = new ServerStart_View();
12         server.get(frame);
13         Thread thread = new Thread(server);
14         thread.start();
15         
16         Thread thread2 = new Thread(file);
17         thread2.start();
18         
19         Thread thread3 = new Thread(loginData);
20         thread3.start();
21     }
22     public void setText(String AllName,String string) {
23         System.out.println(AllName+" : "+string);
24     }
25 }

客户端，登录界面与服务带进行socket连接，发送用户信息，并读取返回的信息。

主要代码：

 1 public class Login_View extends JFrame {
 2 
 3     public static String AllName=null;
 4     static Login_View frame;
 5     private JPanel contentPane;
 6     private JTextField textField;
 7     private JTextField textField_1;
 8     JOptionPane optionPane = new JOptionPane();
 9     private final Action action = new SwingAction();
10     private JButton btnNewButton_1;
11     private final Action action_1 = new SwingAction_1();
12     private JLabel lblNewLabel_2;
13 
14     /**
15      * Launch the application.
16      */
17     public static void main(String[] args) {
18         EventQueue.invokeLater(new Runnable() {
19             public void run() {
20                 try {
21                     frame = new Login_View();
22                     frame.setVisible(true);
23                     frame.setDefaultCloseOperation(EXIT_ON_CLOSE);
24                 } catch (Exception e) {
25                     e.printStackTrace();
26                 }
27             }
28         });
29     }
30 
31 ..................
32 ..................
33 ..................
34 
35 private class SwingAction extends AbstractAction {
36         public SwingAction() {
37             putValue(NAME, "登录");
38             putValue(SHORT_DESCRIPTION, "点击登录");
39         }
40         public void actionPerformed(ActionEvent e) {
41             String text = textField.getText();
42             String text2 = textField_1.getText();
43 //            System.out.println(text+text2);
44 //            boolean boo=false;
45             String boo=null;
46             try {
47                 boo = DataJudge.Judge(6567,text,text2,"login");
48             } catch (IOException e1) {
49                 e1.printStackTrace();
50             }
51             if(boo.equals("true")) {
52                 ClientStart_View.main1();
53                 AllName = text;    //保存用户名
54                 frame.dispose();    //void    dispose()    释放此this Window，其子组件和所有其拥有的子级使用的所有本机屏幕资源 。
55             }else {
56                 optionPane.showConfirmDialog
57                 (contentPane, "用户名或密码错误,请再次输入", "登录失败",JOptionPane.OK_CANCEL_OPTION);
58             }
59         }
60     }
61     
62     private class SwingAction_1 extends AbstractAction {
63         public SwingAction_1() {
64             putValue(NAME, "注册");
65             putValue(SHORT_DESCRIPTION, "点击进入注册页面");
66         }
67         public void actionPerformed(ActionEvent e) {
68             Registered_View registered = new Registered_View(Login_View.this);
69             registered.setLocationRelativeTo(rootPane);
70             registered.setVisible(true);
71         }
72     }
73 }

连接服务端：第一次写的时候连接方法是Boolean类型，但只适用于登录的信息判断，当注册时需要判断昵称是否重复，密码昵称是否为空等不同的返回信息，（服务端代码有相应的判断字符串返回，参上）于是该为将连接方法改为String类型。

 1 import java.io.DataInputStream;
 2 import java.io.DataOutputStream;
 3 import java.io.IOException;
 4 import java.net.Socket;
 5 import java.net.UnknownHostException;
 6 
 7 public class DataJudge {
 8 
 9     /*public static boolean Judge(int port,String name,String password,String judge) throws UnknownHostException, IOException {
10         
11         Socket socket = new Socket("127.0.0.1", port);
12         DataInputStream inputStream = new DataInputStream(socket.getInputStream());
13         DataOutputStream outputStream = new DataOutputStream(socket.getOutputStream());
14         
15         outputStream.writeUTF(name);
16         outputStream.writeUTF(password);
17         outputStream.writeUTF(judge);
18         
19         boolean readBoolean = inputStream.readBoolean();
20         
21         outputStream.close();
22         inputStream.close();
23         socket.close();
24         return readBoolean;
25     }*/
26 
27 public static String Judge(int port,String name,String password,String judge) throws UnknownHostException, IOException {
28     
29         //连接服务端数据库部分
30         Socket socket = new Socket("127.0.0.1", port);
31         DataInputStream inputStream = new DataInputStream(socket.getInputStream());
32         DataOutputStream outputStream = new DataOutputStream(socket.getOutputStream());
33         
34         outputStream.writeUTF(name);
35         outputStream.writeUTF(password);
36         outputStream.writeUTF(judge);
37         
38         String read = inputStream.readUTF();
39         
40         //登录是一次性的，所以要及时关闭socket
41         outputStream.close();
42         inputStream.close();
43         socket.close();
44         return read;
45     }
46 }

用户注册界面，主要代码：

 1 public class Registered_View extends JDialog{
 2 //    DataJudge dataJudge = new DataJudge();
 3     private JTextField textField_1;
 4     private JTextField textField;
 5     JLabel lblNewLabel_2;
 6     private final Action action = new SwingAction();
 7     
 8     public Registered_View(JFrame frame) {
 9         super(frame, "", true);   //使注册对话框显示在主面板之上。
10                 .........
11                 .........
12                 .........
13                 .........
14         }  
15       
16         private class SwingAction extends AbstractAction {
17         public SwingAction() {
18             putValue(NAME, "注册");
19             putValue(SHORT_DESCRIPTION, "点击按钮进行注册");
20         }
21         public void actionPerformed(ActionEvent e) {
22             String b=null;  //用于接收服务端返回的注册信息字符串
23             String name = textField.getText();
24             String password = textField_1.getText();
25             try {
26                 b = DataJudge.Judge(6567, name, password, "registered");
27             } catch (IOException e1) {
28                 // TODO Auto-generated catch block
29                 e1.printStackTrace();
30             }
31             
32             lblNewLabel_2.setText(b);
33         }
34     }

用户登录，注册部分至此完毕。

实时显示人数，主要是向客户端返回存储socket对象的泛型数组大小。在当有新的客户端连接之后调用此方法，当有用户断开连接后调用此方法。

 1 public static void SendInfo(String rece, String AllName, String num) throws IOException {
 2         DataOutputStream outputStream = null;
 3         for (Socket Ssocket : Server.socketList) {
 4             outputStream = new DataOutputStream(Ssocket.getOutputStream());
 5             outputStream.writeUTF(num);
 6             outputStream.writeUTF(AllName);
 7             outputStream.writeUTF(rece);
 8             outputStream.flush();
 9         }
10     }

说说Bug

用户每次断开连接之前都没有先进行socket的关闭，服务端也没有移除相应的socket对象，这就导致当服务端再逐个发送至每个客户端，便找不到那个关闭的socket对象，会产生"write error" 。

所以便需要再客户端断开时移除相应的socket对象，查看java API文档，并没有找到在服务端可以判断客户端socket是否关闭的方方法。

技术图片

便想到了之前看的方法。（虽然感觉这样麻烦了一步，但没找到更好的办法）。于是在点击退出按钮，或关闭面板时向服务端发送一个"bye"字符，当服务端读取到此字符时便知道客户端要断开连接了，从而退出循环读取操作，移除对应的socket对象。

 1 面板关闭事件监听
 2 
 3 @Override
 4     public void windowClosing(WindowEvent arg0) {
 5         try {
 6             chat_Client.send("bye");
 7             File_O.file_O.readbye("bye");
 8         } catch (IOException e) {
 9             // TODO Auto-generated catch block
10             e.printStackTrace();
11         }
12     }

 1 退出按钮事件监听
 2 
 3 private class SwingAction extends AbstractAction {
 4         public SwingAction() {
 5             putValue(NAME, "退出");
 6             putValue(SHORT_DESCRIPTION, "关闭程序");
 7         }
 8         public void actionPerformed(ActionEvent e) {
 9             int result=optionPane.showConfirmDialog(contentPane, "是否关闭退出", "退出提醒", JOptionPane.YES_NO_OPTION);
10             if(result==JOptionPane.YES_OPTION) {
11                 try {
12                     chat_Client.send("bye");
13                     File_O.file_O.readbye("bye");
14                     System.exit(EXIT_ON_CLOSE);  //static void    exit?(int status)    终止当前正在运行的Java虚拟机。即终止当前程序，关闭窗口。
15                 } catch (IOException e1) {
16                     e1.printStackTrace();
17                 }
18             }
19         }
20     }

 1 客户端send方法，发送完bye字符后，关闭socket
 2 
 3 //send()方法，发送消息给服务器。 “发送”button 按钮点击事件，调用此方法
 4     public void send(String send) throws IOException {
 5         DataOutputStream stream = new DataOutputStream(socket.getOutputStream());
 6         stream.writeUTF(Login_View.AllName);
 7         stream.writeUTF(send);
 8         
 9         if(send.equals("bye")) {
10             stream.flush();
11             socket.close();
12         }
13     }

 1 服务端读取到bye字符时，移除相应socket对象，退出while循环
 2 
 3 if (rece.equals("bye")) {
 4                             judg = false;
 5                             Server.socketList.remove(socket);
 6                             Server_IO.SendInfo("", "", "" + Server.socketList.size());
 7                             /*
 8                              * for (Socket Ssocket:Server.socketList) { DataOutputStream outputStream = new
 9                              * DataOutputStream(socket.getOutputStream()); outputStream = new
10                              * DataOutputStream(Ssocket.getOutputStream());
11                              * outputStream.writeUTF(""+Server.socketList.size());
12                              * outputStream.writeUTF(""); outputStream.writeUTF("");
13                              * System.out.println("8888888888888888"); outputStream.flush(); }
14                              */
15                             break;
16                         }

文件的流的关闭，移除也是如此，不在赘述。

文件流还有一个问题，正常登录不能进行第二次文件传输。（第一次写的时候可能我只测试了一次，没有找到bug。哈哈哈哈）

解决这个问题耽搁了好久（太cai了，哈哈哈哈）

原来的代码，服务端读取并发送部分（也可参加看之前的随笔）

 1   while((len=input.read(read,0,read.length))>0) {
 2                for(Socket soc:File.socketList_IO) {
 3                       if(soc != socket)
 4                              { 
 5                                  output = new DataOutputStream(soc.getOutputStream());
 6                                  output.writeUTF(name);
 7                                  output.write(read,0,len);
 8                                  output.flush();
 9  //                                System.out.println("开始向客户机转发");
10                              }
11                          }
12  //                        System.out.println("执行");
13  //                        System.out.println(len);
14                      }

read()方法：API文档的介绍

技术图片

当读取到文件末尾时会返回-1,可以看到while循环也是当len等于-1时结束循环，然而事与愿违。在debug时(忘记截图)发现，只要客户端的输出流不关闭，服务端当文件的读取完毕后会一直阻塞在

while((len=input.read(read,0,read.length))>0)，无法退出，从而无法进行下一次读取转发。也无法使用len=-1进行中断break;
修改如下：

 1 int len=0;
 2 while(true) {
 3     len=0;
 4     if(input.available()!=0)
 5        len=input.read(read,0,read.length);
 6     if(len==0) break;
 7     for(Socket soc:File.socketlist_file) {
 8        if(soc != socket)
 9        {
10           output = new DataOutputStream(soc.getOutputStream());
11           output.writeUTF(name);
12           output.write(read,0,len);
13 //        output.flush();
14 //        System.out.println("开始向客户机转发");
15        }
16 //     System.out.println("一次转发"+File.socketlist_file.size());
17     }
18  }