twistedutil.protocol.GreenTransport: allow write_event be customized

This commit is contained in:
Denis Bilenko
2008-12-18 19:26:25 +06:00
parent 2ea50a1757
commit 27693e7215

View File

@@ -78,10 +78,10 @@ class GreenTransportBase(object):
if self.transportBufferSize is not None: if self.transportBufferSize is not None:
transport.bufferSize = self.transportBufferSize transport.bufferSize = self.transportBufferSize
self.init_transport_producer(transport) self.init_transport_producer(transport)
ev = event() if self.write_event is None:
ev.send(1) self.write_event = event()
transport.registerProducer(Producer2Event(ev), True) self.write_event.send(1)
self.write_event = ev transport.registerProducer(Producer2Event(self.write_event), True)
self.transport = transport self.transport = transport
class Protocol(twistedProtocol): class Protocol(twistedProtocol):