37. socketserver模块

一、socketserver模块

  SocketServer 是标准库中的一个高级模块，它的目标是简化很多样板代码，它们是创建网络客户端和服务器所必须的代码。这个模块中有为你创建的各种各样的类。

二、基于TCP协议通信

2.1、创建socketserver的TCP服务器

from socketserver import ThreadingTCPServer,StreamRequestHandler
from time import ctime

HOST = "127.0.0.1"
PORT = 8085
ADDRESS = (HOST, PORT)


class MyRequestHandler(StreamRequestHandler):
    def handle(self):
        # 如果是tcp协议，self.request --> connection
        print("...connected from: ",self.client_address)

        while True:  # 通信循环
            try:
                data = self.request.recv(1024)                              # 服务端接收消息，单次最大接收为1024个字节
                # 在Linux系统中，一旦data收到空，意味着是一种异常的行为：客户端非法断开连接
                if not data:                                                # 适用于Linux系统
                    break
                print("收到客户端%s数据：%s" % (self.client_address,data.decode("utf-8")))
                self.request.send(f"[{ctime()}] {data}".encode("utf-8"))    # 服务端发送消息
            except ConnectionResetError:                                    # 适用于Windows系统
                break

        self.request.close()


tcp_server = ThreadingTCPServer(ADDRESS,MyRequestHandler)
print("waiting for connection...")
tcp_server.serve_forever()

  我们得到请求处理程序，作为 socketserver 中 StreamRequestHandler 的一个子类，并重写了它的 handler() 方法，该方法在基类 Request 中默认情况下没有任何行为。当接收一个来自客户端的消息时，它就会调用 handler() 方法。

2.2、创建socketserver的TCP客户端

from socket import socket
from socket import AF_INET, SOCK_STREAM

HOST = "127.0.0.1"
PORT = 8085
ADDRESS = (HOST,PORT)

tcp_client = socket(AF_INET, SOCK_STREAM)                   # 创建客户端套接字
tcp_client.connect(ADDRESS)                                 # 尝试连接服务器

while True:                                                 # 通信循环
    data = input(">>>: ").strip()
    if not data:                                            # 按空格、回车等键结束连接
        break
    tcp_client.send(data.encode("utf-8"))                   # 客户端发送数据

    data = tcp_client.recv(1024)                            # 客户端接收数据，单次最大接收为1024个字节
    if not data:
        break
    print("收到服务端返回的数据：%s" % data.decode("utf-8"))

tcp_client.close()                                          # 关闭客户端套接字

三、基于UDP协议通信

3.1、创建socketserver的UDP服务器

from socketserver import ThreadingUDPServer,BaseRequestHandler
from time import ctime

HOST = "127.0.0.1"
PORT = 8086
ADDRESS = (HOST, PORT)


class MyRequestHandler(BaseRequestHandler):
    def handle(self):
        data = self.request[0]
        server = self.request[1]

        print("收到客户端%s数据：%s" % (self.client_address, data))
        server.sendto(f"[{ctime()}] {data}".encode("utf-8"), self.client_address)  # 服务端发送消息


udp_server = ThreadingUDPServer(ADDRESS,MyRequestHandler)
print("waiting for connection...")
udp_server.serve_forever()

3.2、创建socketserver的UDP客户端

from socket import socket
from socket import AF_INET, SOCK_DGRAM

HOST = "127.0.0.1"
PORT = 8086
ADDRESS = (HOST, PORT)

udp_client = socket(AF_INET,SOCK_DGRAM)                     # 创建服务器套接字

while True:                                                 # 通信循环
    data = input(">>>: ").strip()
    udp_client.sendto(data.encode("utf-8"),ADDRESS)         # 客户端发送数据

    data,server_addr = udp_client.recvfrom(1024)            # 客户端接收数据，单次最大接收为1024个字节
    print("收到服务端返回的数据：%s" % data.decode("utf-8"))

udp_client.close()                                          # 关闭客户端套接字