Call UVStream.write directly from SSLProtocol

MagicStack · Sep 9, 2024 · 04ec4d5 · 04ec4d5
1 parent ab16cc3
commit 04ec4d5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/uvloop/handles/stream.pxd b/uvloop/handles/stream.pxd
@@ -16,6 +16,8 @@ cdef class UVStream(UVBaseTransport):
         Py_buffer _read_pybuf
         bint _read_pybuf_acquired
 
+    cpdef write(self, object buf)
+
     # All "inline" methods are final
 
     cdef inline _init(self, Loop loop, object protocol, Server server,

diff --git a/uvloop/handles/stream.pyx b/uvloop/handles/stream.pyx
@@ -676,7 +676,7 @@ cdef class UVStream(UVBaseTransport):
             self.__reading,
             id(self))
 
-    def write(self, object buf):
+    cpdef write(self, object buf):
         self._ensure_alive()
 
         if self._eof:

diff --git a/uvloop/sslproto.pyx b/uvloop/sslproto.pyx
@@ -705,7 +705,10 @@ cdef class SSLProtocol:
         if not self._ssl_writing_paused:
             data = self._outgoing_read()
             if len(data):
-                self._transport.write(data)
+                if isinstance(self._transport, UVStream):
+                    (<UVStream>self._transport).write(data)
+                else:
+                    self._transport.write(data)
 
     # Incoming flow