当前位置:首页>python>【小沐学 Python】从零写 TCP/UDP 通信程序,看完就会!(八)

【小沐学 Python】从零写 TCP/UDP 通信程序,看完就会!(八)

  • 2026-09-10 03:28:28
【小沐学 Python】从零写 TCP/UDP 通信程序,看完就会!(八)

小沐最近在研究分布式系统的网络通信问题,这天遇到正在下棋的戈戈,便请教道:“戈戈,我最近看了一部《八仙过海》的电视剧,发现八仙各有各的法宝,真是八仙过海各显神通!你说网络通信协议是不是也像八仙一样,各有各的本事?”

戈戈笑着点头:“没错!网络协议确实如此。HTTP像吕洞宾的剑,锋利直接;Socket像铁拐李的葫芦,深不可测;WebSocket像汉钟离的芭蕉扇,扇动风云;gRPC像曹国舅的玉板,高效有序。今天我们就来好好聊聊这四种经典协议!”

小沐兴奋地说:“那还等什么,快给我演示一下Python怎么实现吧!”

一、HTTP/HTTPS协议:吕洞宾的剑

HTTP(HyperText Transfer Protocol)是互联网上最常用的协议之一,它是客户端和服务器之间进行通信的基础。HTTPS则是HTTP的安全版本,通过SSL/TLS加密传输数据,确保通信的安全性。在Python中,我们可以使用内置的urllib库或者强大的requests库来发送HTTP请求,还可以使用Flask快速搭建Web服务进行测试。

安装依赖:

pip install flask requests

编写HTTP服务端代码如下:

# Flask Web服务器
from flask import Flask, request, jsonify

app = Flask(__name__)

@app.route('/api/hello', methods=['GET'])
defhello():
return jsonify({'message': 'Hello from Flask server!'})

@app.route('/api/data', methods=['POST'])
defreceive_data():
    data = request.json
return jsonify({'received': data, 'status': 'success'})

if __name__ == '__main__':
# 启动服务器,访问 http://127.0.0.1:5000/api/hello
    app.run(host='127.0.0.1', port=5000)

代码运行结果如下:

编写HTTP客户端代码如下:

# HTTP客户端请求
import requests

# 发送GET请求测试Flask服务
response = requests.get('http://127.0.0.1:5000/api/hello')
print(f"GET请求状态码: {response.status_code}")
print(f"GET响应内容: {response.json()}")

# 发送POST请求
data = {'name': '小沐', 'message': '测试HTTP协议'}
response = requests.post('http://127.0.0.1:5000/api/data', json=data)
print(f"POST响应内容: {response.json()}")

# HTTPS请求示例
response = requests.get('https://httpbin.org/get', verify=True)
print(f"HTTPS响应状态码: {response.status_code}")

代码运行结果如下:

二、Socket编程:铁拐李的葫芦

Socket是网络通信的底层接口,提供了比HTTP更加灵活的通信方式。TCP(Transmission Control Protocol)提供可靠的面向连接的通信,而UDP(User Datagram Protocol)则提供无连接的快速传输。在Python中,我们可以使用socket库来进行底层网络编程,实现自定义的通信协议。

Socket是Python标准库的一部分,无需额外安装。

编写tcp服务端代码如下:

# Socket服务器端
import socket

deftcp_server():
"""TCP服务器端"""
# 创建TCP套接字
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind(('127.0.0.1', 8888))
    server_socket.listen(5)

    print("TCP服务器正在监听端口8888...")
    print("等待客户端连接...")

# 接受客户端连接
    client_socket, addr = server_socket.accept()
    print(f"收到来自 {addr} 的连接")

# 接收客户端数据
    data = client_socket.recv(1024)
    print(f"收到客户端数据: {data.decode('utf-8')}")

# 发送响应
    response = "Hello from TCP Server!"
    client_socket.send(response.encode('utf-8'))
    print("已发送响应给客户端")

# 关闭连接
    client_socket.close()
    server_socket.close()
    print("服务器已关闭")

if __name__ == "__main__":
    tcp_server()

代码运行结果如下:

编写tcp客户端代码如下:

# Socket客户端
import socket

deftcp_client():
"""TCP客户端"""
# 创建TCP套接字
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接服务器
    server_addr = ('127.0.0.1', 8888)
    client_socket.connect(server_addr)
    print("已连接到服务器")

# 发送数据
    message = "Hello from TCP Client!"
    client_socket.send(message.encode('utf-8'))
    print(f"已发送数据: {message}")

# 接收服务器响应
    response = client_socket.recv(1024)
    print(f"收到服务器响应: {response.decode('utf-8')}")

# 关闭连接
    client_socket.close()
    print("客户端已关闭")

if __name__ == "__main__":
    tcp_client()

代码运行结果如下:

三、WebSocket:汉钟离的芭蕉扇

WebSocket是一种在单个TCP连接上进行全双工通信的协议,非常适合需要实时更新的应用场景,比如在线聊天、股票行情推送等。WebSocket像汉钟离的芭蕉扇,能够扇动实时通信的风云,让客户端和服务器双向通信更加便捷。Python的websockets库提供了简洁而强大的WebSocket实现。

安装websockets:

pip install websockets

编写websocket服务端代码如下:

# WebSocket服务器端
# 需要安装: pip install websockets
import asyncio
import websockets

asyncdefecho(websocket):
"""处理WebSocket连接"""
    print(f"新客户端连接: {websocket.remote_address}")

# 发送欢迎消息
await websocket.send("欢迎连接到WebSocket服务器!")

try:
# 接收并处理消息
asyncfor message in websocket:
            print(f"收到消息: {message}")
# 发送响应
await websocket.send(f"服务器收到: {message}")
except websockets.exceptions.ConnectionClosed:
        print("客户端断开连接")

asyncdefrun_server():
"""运行WebSocket服务器"""
asyncwith websockets.serve(echo, "127.0.0.1", 8765):
        print("WebSocket服务器启动,监听端口8765...")
await asyncio.Future()  # 永远运行

if __name__ == "__main__":
    asyncio.run(run_server())

代码运行结果如下:

编写websocket客户端代码如下:

# WebSocket客户端
import asyncio
import websockets

asyncdefclient():
"""WebSocket客户端"""
    uri = "ws://127.0.0.1:8765"

try:
asyncwith websockets.connect(uri) as websocket:
            print("已连接到服务器")

# 接收服务器欢迎消息
            welcome = await websocket.recv()
            print(f"收到欢迎消息: {welcome}")

# 发送消息
for i in range(3):
await websocket.send(f"客户端消息 {i+1}")
                print(f"已发送消息 {i+1}")

# 接收响应
                response = await websocket.recv()
                print(f"收到响应: {response}")

            print("通信完成")

except Exception as e:
        print(f"连接错误: {e}")

if __name__ == "__main__":
    asyncio.run(client())

代码运行结果如下:

四、gRPC:曹国舅的玉板

gRPC是Google开发的高性能、开源的远程过程调用(RPC)框架,使用Protocol Buffers作为接口定义语言和消息交换格式。它支持多种编程语言,特别适合构建分布式系统和微服务架构。gRPC就像曹国舅的玉板,敲击有序,通信高效,非常适合微服务之间的精确调用。

安装gRPC:

pip install grpcio grpcio-tools

定义通信的服务文件helloworld.proto

// 定义gRPC服务 proto文件: helloworld.proto
syntax = "proto3";

package helloworld;

// 定义服务
service Greeter {
  // 远程调用方法
  rpc SayHello (HelloRequest) returns (HelloReply) {}
  rpc GetUserInfo (UserRequest) returns (UserReply) {}
}

// 请求消息
message HelloRequest {
  string name = 1;
}

message UserRequest {
  int32 user_id = 1;
}

// 响应消息
message HelloReply {
  string message = 1;
}

message UserReply {
  int32 user_id = 1;
  string username = 2;
  string email = 3;
}

编写gRPC服务端代码如下:

# gRPC服务器端
# 需要安装: pip install grpcio grpcio-tools
# 生成Python代码: python -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. helloworld.proto

from concurrent import futures
import grpc
import helloworld_pb2
import helloworld_pb2_grpc

classGreeterServicer(helloworld_pb2_grpc.GreeterServicer):
"""gRPC服务实现"""

defSayHello(self, request, context):
"""处理SayHello请求"""
        message = f"Hello, {request.name}!"
return helloworld_pb2.HelloReply(message=message)

defGetUserInfo(self, request, context):
"""处理GetUserInfo请求"""
# 模拟数据库查询
        user_data = {
1: {"username": "xiaomu", "email": "xiaomu@example.com"},
2: {"username": "gege", "email": "gege@example.com"}
        }
        user_info = user_data.get(request.user_id, {"username": "unknown", "email": "N/A"})
return helloworld_pb2.UserReply(
            user_id=request.user_id,
            username=user_info["username"],
            email=user_info["email"]
        )

defserve():
"""启动gRPC服务器"""
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    helloworld_pb2_grpc.add_GreeterServicer_to_server(GreeterServicer(), server)
    server.add_insecure_port('[::]:50051')
    server.start()
    print("gRPC服务器启动,监听端口50051...")
    server.wait_for_termination()

if __name__ == "__main__":
    serve()

代码运行结果如下:

编写gRPC客户端代码如下:

# gRPC客户端
import grpc
import helloworld_pb2
import helloworld_pb2_grpc

defrun_client():
"""运行gRPC客户端"""
# 连接服务器
    channel = grpc.insecure_channel('localhost:50051')
    stub = helloworld_pb2_grpc.GreeterStub(channel)

# 调用SayHello方法
    print("调用 SayHello 方法...")
    response = stub.SayHello(helloworld_pb2.HelloRequest(name="小沐"))
    print(f"服务器响应: {response.message}")

# 调用GetUserInfo方法
    print("\n调用 GetUserInfo 方法...")
    response = stub.GetUserInfo(helloworld_pb2.UserRequest(user_id=1))
    print(f"用户ID: {response.user_id}")
    print(f"用户名: {response.username}")
    print(f"邮箱: {response.email}")

    channel.close()
    print("\n客户端连接已关闭")

if __name__ == "__main__":
    run_client()

代码运行结果如下:

戈戈收起棋谱,对小沐说:“八仙过海,各显神通。网络协议也是如此,每种协议都有自己的适用场景。HTTP像吕洞宾的剑,适合轻量级的API调用;Socket像铁拐李的葫芦,深藏不露却包罗万象;WebSocket像汉钟离的芭蕉扇,扇动实时通信的风云;gRPC则像曹国舅的玉板,高效有序,是微服务通信的首选。掌握这些,你就能在网络通信的世界里大显神通了!”

结语

如果您觉得这些文字有一点点用处,请给作者点个赞或关个注;╮( ̄▽ ̄)╭
如果您有技术问题探讨,评论处留言。//(ㄒoㄒ)//
谢谢各位童鞋们啦( ´ ▽ ` )ノ ( ´ ▽ `` )っ!
更多精彩文章详见:
CSDN博客:爱看书的小沐

最新文章

随机文章