Python36 和套接字

Question

所以我使用 socket.connec 连接到 IRC 聊天

我通过 socket.send 传递我的变量来登录

登录成功，然后我使用 while true 循环 Socket.recv(1024)

如果我只是不断打印响应一切看起来都很好，但是假设我想添加到字符串的末尾...我注意到 socket.recv 并不总是得到完整的消息（最多只能抓取 1024正如预期的那样）并且消息的其余部分在循环的下一次迭代中。

这使得它无法逐行处理反馈。

有没有更好的方法来不断读取数据而不会被中继？是否可以在收到响应之前确定响应的大小，以便动态设置缓冲区？

Answer 1

TCP 是一种基于流的协议。缓冲接收到的字节，只从流中提取完整的消息。

对于完整的行，在缓冲区中查找换行符。

示例服务器：

import socket

class Client:

    def __init__(self,socket):
        self.socket = socket
        self.buffer = b''

    def getline(self):
        # if there is no complete line in buffer,
        # add to buffer until there is one.
        while b'\n' not in self.buffer:
            data = self.socket.recv(1024)
            if not data:
                # socket was closed
                return ''
            self.buffer += data

        # break the buffer on the first newline.
        # note: partition(n) return "left of n","n","right of n"
        line,newline,self.buffer = self.buffer.partition(b'\n')
        return line + newline

srv = socket.socket()
srv.bind(('',5000))
srv.listen(1)
conn,where = srv.accept()
client = Client(conn)
print(f'Client connected on {where}')
while True:
    line = client.getline()
    if not line:
        break
    print(line)

示例客户端：

s=socket()
s.connect(('127.0.0.1',5000))
s.sendall(b'line one\nline two\nline three\nincomplete')
s.close()

服务器输出：

Client connected on ('127.0.0.1', 2667)
b'line one\n'
b'line two\n'
b'line three\n'