用于从 Twisted Python 中的传输读取的模式

Question

在Twisted Python中，数据被写入协议的传输，但通过覆盖dataReceived方法接收。是否有从传输中读取的模式？这在使用inlineCallbacks实现状态时会很有帮助

例如：

class SomeProtocol(Protocol):
    @defer.inlineCallbacks
    def login(self):
        self.transport.write('login')
        resp = yield self.transport.read(5, timeout=1) # this doesn't exist
        if resp != 'user:':
            raise SomeException()
        self.transport.write('admin')
        resp = transport.read(9, timeout=1)
        if resp != 'password:':
            raise SomeException()
        self.transport.write('hunter2')
        # ... etc

Answer 1

多年来，已经有几次尝试实现这样的API。 没有一个获得任何牵引力。 我认为它们在这一点上都被抛弃了。

原则上，这并不难实现。 您只是将 dataReceived 回调(推送样式 API(转换为拉取样式 API。

在实践中，生成的代码是脆弱的，并且往往包含更多的错误。

我认为您要解决的问题是dataReceived是用于解析字节流的非常低级的原语。

对此有许多可能的解决方案。 您可以尝试构建一个更高级别的基于协议的工具，该工具了解协议的各个方面(这基本上是 Twisted 中所有协议实现所做的(。 您还可以查看诸如tube之类的第三方库(它为处理字节流提供了不同的抽象(。

Answer 2

我最终维护了一个延迟回调列表，并在数据到达时缓冲传入数据，直到它满足列表中第一个延迟所需的数据长度。

class SomeProtocol(Protocol):
    # initialise self.buf and self.readers in __init__
    def deferred_read(self, count, timeout=None):
        """Return a deferred that fires when data becomes available"""
        d = defer.Deferred()
        reader = [d, count]
        timeout_cb = None
        if timeout is not None:
            timeout_cb = self.reactor.callLater(timeout, self.deferred_read_timeout, reader)
        reader.append(timeout_cb)
        self.readers.append(reader)
        self.check_readers()
        return d
    def deferred_read_timeout(self, reader):
        """Timeout this reader and check if others now match"""
        d, count, timeout_cb = reader
        self.readers.remove(reader)
        d.errback(TimeoutException()) # defined elsewhere
        self.check_readers()
    def check_readers(self):
        """Check if there is enough data to satisfy first reader"""
        try:
            while 1:
                reader = self.readers[0]
                d, count, timeout_cb = reader
                if len(self.buf) < count:
                    break
                data = self.buf[:count]
                self.buf = self.buf[count:]
                self.readers.remove(reader)
                try:
                    timeout_cb.cancel()
                except: pass
                d.callback(data)
        except IndexError: pass
    def dataReceived(self, data):
        self.buf += data
        self.check_readers()

它当前要求计数为非零。最好将其扩展为支持返回读取缓冲区中当前的任何内容，以及使用超时但不计数的读取，以便返回超时后缓冲区中的任何内容。