import socket

def start_server():
    # 創(chuàng)建 TCP/IP 套接字
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 綁定套接字到本地地址和端口
    server_address = ('localhost', 8080)
    print("服務器啟動中，監(jiān)聽地址:", server_address)
    server_socket.bind(server_address)

    # 開始監(jiān)聽，最大連接數設為1
    server_socket.listen(1)
    print("服務器已啟動，等待客戶端連接...")

    while True:
        # 等待客戶端連接
        connection, client_address = server_socket.accept()
        try:
            print("客戶端已連接:", client_address)

            # 接收數據
            data = connection.recv(1024).decode('utf-8')
            print("收到命令:", data)

            # 處理命令，并生成響應
            if data == "INFERENCE":
                response = "模型推理完成"
            elif data == "STATUS":
                response = "模型正在運行"
            else:
                response = "不支持的命令"

            # 發(fā)送響應給客戶端
            connection.sendall(response.encode('utf-8'))
        finally:
            # 關閉連接
            connection.close()
            print("連接已關閉")

if __name__ == "__main__":
    start_server()

import java.io.*;
import java.net.*;

public class AIModelClient {
    public static void main(String[] args) throws IOException {
        // 連接本地主機的8080端口
        Socket socket = new Socket("localhost", 8080);
        System.out.println("已連接到服務器");

        // 創(chuàng)建輸出流，用來發(fā)送命令
        PrintWriter out = new PrintWriter(socket.getOutputStream(), true);

        // 創(chuàng)建輸入流，用來接收服務器的響應
        BufferedReader in = new BufferedReader(
            new InputStreamReader(socket.getInputStream())
        );

        // 發(fā)送命令（可以改成 STATUS 或其他測試）
        String command = "INFERENCE";
        System.out.println("正在發(fā)送命令：" + command);
        out.println(command);

        // 接收服務器的響應
        String response = in.readLine();
        System.out.println("收到服務器的響應：" + response);

        // 關閉連接
        socket.close();
        System.out.println("連接已斷開");
    }
}