Java实现TCP通信，写入数据库

Ipkiss_Studio

已于 2022-11-24 16:04:03 修改

阅读量1.1k

点赞数 4

文章标签：服务器 java

于 2022-10-11 22:39:19 首次发布

本文链接：https://blog.csdn.net/Ipkiss_Yongheng/article/details/127272781

版权

毕设专栏收录该内容

14 篇文章 1 订阅

订阅专栏

准备工作

检查

这里的hosts配置文件内容一般不用修改。我是因为之前玩tomcat的时候，动过这里，测试的时候一直没有效果，才回过头来检查。

jar包

Java操作mysql数据库的jar包。也可以用其他的数据库，添加对应的jar包即可。

https://mvnrepository.com/artifact/mysql/mysql-connector-java/8.0.28

先下载到本地，之后会用到。

服务器

/**
 * @author WuYongheng
 * @date 2022/10/11
 * @description
 */

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import java.util.ArrayList;

class ServerThread extends Thread {
    Socket socket;
    ArrayList<Socket> lst;

    public ServerThread() {

    }

    public ServerThread(Socket socket, ArrayList<Socket> lst) {
        this.socket = socket;
        this.lst = lst;
    }

    public ServerThread(Socket socket) {
        this.socket = socket;
    }

    public void run() {
        byte b[] = new byte[300];
        try {
            // jdbc
            Class.forName("com.mysql.cj.jdbc.Driver");
            String url = "jdbc:mysql://localhost:3306/chat";
            Connection conn = DriverManager.getConnection(url, "root", "123456");
            Statement st = conn.createStatement();

            while (true) {//服务器可以一直收到消息
                int len = socket.getInputStream().read(b);
                String s = new String(b, 0, len);
                String sql = "insert into msginfo (msg) values('" + s + "')";
                st.executeUpdate(sql);//调用函数执行insert语句
                //将收到的消息想其他客户端发送
                for (Socket skt : lst) {
				if(skt==socket)//加这一行，自己的客户端就不发了，向其他的客户端发
						continue;
                    skt.getOutputStream().write(b, 0, len);
                }
                System.out.println(s);
            }

            // System.out.println(new String(b, 0, len));
        } catch (Exception e) {

        }
    }
}

class Server {
    public static void main(String[] args) throws IOException {
        if (args.length != 1) {
            System.out.println("使用方法： java 程序  端口");
            return;
        }
        int port = 0;
        try {
            port = Integer.parseInt(args[0]);
            if (port < 5555 || port > 9999) {
                System.err.println("无效的端口，端口必须在5555~9999之间");
                return;
            }
        } catch (Exception e) {
            System.err.println("无效的端口，端口必须在5555~9999之间");
            return;
        }

        ServerSocket server = new ServerSocket(port);
        ArrayList<Socket> lst = new ArrayList<Socket>();

        while (true) {
            Socket socket = server.accept();
            lst.add(socket);
            new ServerThread(socket, lst).start();//socket,lst！！
        }

//		byte b[] = new byte[300];
//		int len = socket.getInputStream().read(b);
//		System.out.println(new String(b, 0, len));
//
//		System.in.read();

    }
}

客户端

/**
 * @author WuYongheng
 * @date 2022/10/11
 * @description
 */

import java.io.IOException;
import java.net.Socket;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

class ClientThread extends Thread {
    Socket socket;

    public ClientThread() {

    }

    public ClientThread(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {
        byte[] b = new byte[300];
        try {
            while (true) {
                int len = socket.getInputStream().read(b);
                String s = new String(b, 0, len);
                System.out.println("\t\t\t\t\t^_^" +"收到："+ s);
            }
        } catch (Exception e) {

        }

    }
}

class Client {
    /**
     * 检查ip地址是否正确？使用正则表达式
     * @param ip
     * @return
     */
    static boolean isIp(String ip) {
        String str = ip;
        String pattern = "((2(5[0-5]|[0-4]\\d))|[0-1]?\\d{1,2})(\\.((2(5[0-5]|[0-4]\\d))|[0-1]?\\d{1,2})){3}";
        Pattern r = Pattern.compile(pattern);
        Matcher m = r.matcher(str);
        boolean flag = m.matches();

        return flag;
    }

    public static void main(String[] args) throws IOException {
        if (args.length != 2) {
            System.out.println("使用方法： java  程序  服务器IP  端口");
            return;
        }
        if (!isIp(args[0])) {
            System.out.println("无效的ip地址格式");
            return;
        }
        int port = 0;
        try {
            port = Integer.parseInt(args[1]);
            if (port < 5555 || port > 9999) {
                System.err.println("无效的端口，端口必须在5555~9999之间");
                return;
            }
        } catch (Exception e) {
            System.err.println("无效的端口，端口必须在5555~9999之间");
            return;
        }

        Socket socket = new Socket(args[0], port);//两个参数，ip地址，端口
        //创建一个线程负责读服务器发来的消息，这样实现了客户端，服务器同时接收，发送信息
        new ClientThread(socket).start();
        String s = null;
        byte[] b = new byte[300];
        while (true) {//死循环，客户端一直发消息
            System.out.print("chat>");
            int len = System.in.read(b);
            len = len - 1;//处理用户直接回车

            if (len == 0) {
                continue;
            }
            if (b[len - 1] == 13) {
                len = len - 1;
            }
            if (len == 0) {
                continue;
            }
            s = new String(b, 0, len);
            if (s.contentEquals("exit") || s.contentEquals("quit")) {//处理用户想退出程序
                System.out.println("欢迎再次使用");
                return;
            }
            socket.getOutputStream().write(b, 0, len);

        }

//		socket.getOutputStream().write("hello".getBytes());
//		System.in.read();

    }

}