Python学习第36天（socketserver模块导入）

　　今天的内容其实不多，主要是因为后面讲到底层部分涉及了一些Linux知识，我这部分没看，后面的内容就没法听了，所以就先复习一下昨天关于粘包的处理方式的知识，自己重新写了一下这部分内容，然后大致的看了一下socketserver模块如何实现并发的一个过程，明天正式开始Linux的学习，已经到了不得不进行的程度了。

一、粘包问题的解决

low版处理：

服务端：

import socket
import subprocess
ip_port = ('127.0.0.1',8080)
backlog = 5
buffer_size = 1024

tcp_server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)   # 用于解决time out占用问题
tcp_server.bind(ip_port)
tcp_server.listen(backlog)

while True:
    conn,addr = tcp_server.accept()
    print('客户端地址是：',addr)
    while True:
        msg = conn.recv(buffer_size)
        if not msg:
            break
        res = subprocess.Popen(msg.decode('utf-8'),shell = True,
                               stdin = subprocess.PIPE,
                               stderr = subprocess.PIPE,
                               stdout = subprocess.PIPE)
        err = res.stderr.read()
        if err:    # 表示收到了报错信息的情况下
            ret = err
        else:
            ret =res.stdout.read()
        data_lengeth = len(ret)
        conn.send(str(data_lengeth).encode('utf-8'))
        data = conn.recv(buffer_size).decode('utf-8')   # 客户反馈收到的确认信息
        if data == 'recv_ready':
            conn.sendall(ret)
    conn.close()

客户端：

import socket
ip_port = ('127.0.0.1',8080)
buffer_size = 1024

tcp_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_client.connect(ip_port)

while True:
    msg = input('>>>').strip()
    if len(msg) == 0:
        continue
    if msg == 'quit':
        break

    tcp_client.send(msg.encode('utf-8'))
    length = int(tcp_client.recv(buffer_size).decode('utf-8'))
    tcp_client.send('recv_ready'.encode('utf-8'))

    send_size = 0
    data = ''
    while send_size < length:
        data += tcp_client.recv(buffer_size)
        send_size = len(data)

    print(data.decode('utf-8'))

相对于昨天的还是进行了一些优化的，

然后是利用stuck模块实现的操作

服务端：

from socket import *
import subprocess
import struct
ip_port=('127.0.0.1',8080)
back_log=5
buffer_size=1024

tcp_server=socket(AF_INET,SOCK_STREAM)
tcp_server.bind(ip_port)
tcp_server.listen(back_log)

while True:
    conn,addr=tcp_server.accept()
    print('新的客户端链接',addr)
    while True:
        #收
        try:
            cmd=conn.recv(buffer_size)
            if not cmd:break
            print('收到客户端的命令',cmd)

            #执行命令，得到命令的运行结果cmd_res
            res=subprocess.Popen(cmd.decode('utf-8'),shell=True,
                                 stderr=subprocess.PIPE,
                                 stdout=subprocess.PIPE,
                                 stdin=subprocess.PIPE)
            err=res.stderr.read()
            if err:
                cmd_res=err
            else:
                cmd_res=res.stdout.read()

            #发
            if not cmd_res:
                cmd_res='执行成功'.encode('gbk')

            length=len(cmd_res)

            data_length=struct.pack('i',length)
            conn.send(data_length)
            conn.send(cmd_res)
        except Exception as e:
            print(e)
            break

客户端：

from socket import *
import struct
from functools import partial
ip_port=('127.0.0.1',8080)
back_log=5
buffer_size=1024

tcp_client=socket(AF_INET,SOCK_STREAM)
tcp_client.connect(ip_port)

while True:
    cmd=input('>>: ').strip()
    if not cmd:continue
    if cmd == 'quit':break

    tcp_client.send(cmd.encode('utf-8'))


    #解决粘包
    length_data=tcp_client.recv(4)
    length=struct.unpack('i',length_data)[0]

    recv_size=0
    recv_data=b''
    while recv_size < length:
        recv_data+=tcp_client.recv(buffer_size)
        recv_size=len(recv_data)
    print('命令的执行结果是 ',recv_data.decode('gbk'))
tcp_client.close()

这样看起来就清晰的多了

然后就要开始今天内容了，上面这些操作很显然存在一个问题，在基于tcp的时候每次只能进行一个连接，那么这样就造成了很不现实的问题，一次只能服务一个客户端，基于此种需求，这个时候就引入了socketserver模块

先看一下，下面这个图就是大致的服务器socketserver模块的书写形式吧

下面实际测试的部分：

服务端

import socket
import socketserver

class MyServer(socketserver.BaseRequestHandler):
    def handle(self):
        print('链接信息：',self.request)
        # 相当于之前的conn
        print('地址信息：',self.client_address)
        # 相当于之前的addr
        while True:
            try:
            #收消息
                data=self.request.recv(1024)
                if not data:break
                print('收到客户端的消息是',data,self.client_address)

                #发消息
                self.request.sendall(data.upper())

            except Exception as e:
                print(e)
                break

if __name__ == '__main__':
    s=socketserver.ThreadingTCPServer(('127.0.0.1',8080),MyServer) #多线程
    # s=socketserver.ForkingTCPServer(('127.0.0.1',8080),MyServer) #多进程

    # self.server_address = server_address
    # self.RequestHandlerClass = RequestHandlerClass
    # print(s.server_address)
    # print(s.RequestHandlerClass)
    # print(MyServer)
    # print(s.socket)
    # print(s.server_address)
    s.serve_forever()

客户端：

from socket import *
ip_port=('127.0.0.1',8080)
buffer_size=1024

udp_client=socket(AF_INET,SOCK_DGRAM) #数据报

while True:
    msg=input('>>: ').strip()
    udp_client.sendto(msg.encode('utf-8'),ip_port)

    data,addr=udp_client.recvfrom(buffer_size)
    # print(data.decode('utf-8'))
    print(data)

通过socketserver模块的导入，我们在服务端建立了一个新的类MyServer，而每个与其连接的客户端都是由MyServer生成的实例化对象，而收发消息的过程则会通过调用类的handle方法来实现。

Linux真的不学不行了，明天正式开始Linux的内容更新。

Python学习第36天（socketserver模块导入）

猜你喜欢