release 0.0.301 (#44)

* optimize empty return --------- Co-authored-by: nggit <[email protected]>
nggit · Nov 27, 2023 · 654f3a8 · 654f3a8
1 parent 62756ab
commit 654f3a8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='tremolo',
-    version='0.0.310',
+    version='0.0.311',
     license='MIT',
     author='nggit',
     author_email='[email protected]',

diff --git a/tremolo/__init__.py b/tremolo/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '0.0.310'
+__version__ = '0.0.311'
 
 from .tremolo import Tremolo  # noqa: E402
 from . import exceptions  # noqa: E402,F401

diff --git a/tremolo/http_server.py b/tremolo/http_server.py
@@ -232,7 +232,7 @@ async def _handle_response(self, func, options={}):
             if isinstance(data, str):
                 data = data.encode(encoding[0])
 
-            if no_content or data == b'':
+            if no_content:
                 self.response.set_header(b'Connection', b'close')
             else:
                 if self.response.http_chunked:
@@ -265,15 +265,17 @@ async def _handle_response(self, func, options={}):
                 if not isinstance(options, dict):
                     return
 
-            if data == b'' or self.request.method == b'HEAD' or no_content:
+            if self.request.method == b'HEAD' or no_content:
                 await self.response.write(None)
                 return
 
-            self.set_watermarks(high=options['buffer_size'] * 4,
-                                low=options['buffer_size'] // 2)
-            await self.response.write(data,
-                                      rate=options['rate'],
-                                      buffer_size=options['buffer_size'])
+            if data != b'':
+                self.set_watermarks(high=options['buffer_size'] * 4,
+                                    low=options['buffer_size'] // 2)
+                await self.response.write(data,
+                                          rate=options['rate'],
+                                          buffer_size=options['buffer_size'])
+
             await self.response.write(b'', throttle=False)
 
         self.response.close(keepalive=True)