从套接字 Python 接收数据时我需要睡眠吗

Question

我是套接字新手。我用 Python 编写了一个简单的“代理”服务器，它只会从远程服务器捕获数据并将其发送到我的客户端（浏览器）。我想知道：有没有办法在没有 time.sleep 的情况下发送 response_text？当我尝试删除 time.sleep(0.5) 时，我只从远程服务器获取一个数据包，因此“Content-lenght”不等于包的长度，并且出现错误（我正在使用 recv()得到一个大小等于 buffer_size 的缓冲区，所以如果服务器数据需要超过一个 4096 字节的包，我需要在下一个包中捕获它）。通过 time.sleep，我可以从远程服务器获取所有数据包，然后我可以将数据发送到我的浏览器。难道我做错了什么？还是我不够了解？有人可以帮忙吗？

代码：

# coding: utf-8
import socket
import sys
import time
from thread import *

max_conn = 5
buffer_size = 4096


def proxy_server(webserver, port, conn, addr, data):
    try:
        remote_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        remote_sock.connect((webserver, port))
        remote_sock.send(data)
        response_text = ''
        while True:
            time.sleep(0.5)
            response = remote_sock.recv(buffer_size)
            response_text += response
            if len(response) < buffer_size:
                remote_sock.close()
                break
        conn.sendall(response_text)
        conn.close()
    except socket.error, msg:
        print 'Proccessing error. Error Code: ' + str(msg[0]) + ', Wiadomość: ' + msg[1]
        remote_sock.close()
        conn.close()
        sys.exit()


def conn_string(conn, data, address):
    header = data.split('\r\n')
    method, address, protocol = header[0].split(' ')
    host_key, host_value = header[1].split(': ')
    proxy_server(host_value, 80, conn, address, data)


def start():
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.bind(('', 8001))
        sock.listen(max_conn)
        print 'Proxy: on'
    except socket.error, msg:
        print 'Failed creating a socket. Error Code: ' + str(msg[0]) + ', Wiadomość: ' + msg[1]
        sys.exit()

    while True:
        try:
            connection, address = sock.accept()
            data = connection.recv(buffer_size)
            # start_new_thread(conn_string, (connection, data, address))
            conn_string(connection, data, address)
        except KeyboardInterrupt:
            sock.close()
            print "Socket closed"
            sys.exit()

if __name__ == "__main__":
    start()

Answer 1

不要使用time.sleep()，它会让你的代理非常变慢，而且效率不高。
您需要将套接字设置为非阻塞模式，并设置超时。
您可以使用 socket.settimeout()
执行此操作 我对你的proxy_server 做了一些修改，现在应该快多了。

def proxy_server(webserver, port, conn, addr, data): 
    try:
        remote_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        remote_sock.connect((webserver, port))
        remote_sock.send(data)
        remote_sock.settimeout(0.5)
        response_text = ''
        while True:
            try : 
                response = remote_sock.recv(buffer_size)
                if len(response) == 0:
                    break
            except : 
                break
            response_text += response
        conn.sendall(response_text)
    except socket.error, msg:
        if str(msg) != 'timed out' : 
            print 'Proccessing error. Error Code: ' + str(msg[0]) + ', Wiadomość: ' + msg[1]
    remote_sock.close()
    conn.close() 

你的代码的其余部分都很好，但你可能想使用多线程
如果您想同时处理多个客户端。