Python与Java接入AI模型的MCP协议的原理与实现

import socket

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

    # 绑定套接字到本地地址和端口
    server_address = ('localhost', 8080)
    print("服务器启动中，监听地址:", server_address)
    server_socket.bind(server_address)

    # 开始监听，最大连接数设为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 = "不支持的命令"

            # 发送响应给客户端
            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("已连接到服务器");

        // 创建输出流，用来发送命令
        PrintWriter out = new PrintWriter(socket.getOutputStream(), true);

        // 创建输入流，用来接收服务器的响应
        BufferedReader in = new BufferedReader(
            new InputStreamReader(socket.getInputStream())
        );

        // 发送命令（可以改成 STATUS 或其他测试）
        String command = "INFERENCE";
        System.out.println("正在发送命令：" + command);
        out.println(command);

        // 接收服务器的响应
        String response = in.readLine();
        System.out.println("收到服务器的响应：" + response);

        // 关闭连接
        socket.close();
        System.out.println("连接已断开");
    }
}