]> git.ipfire.org Git - thirdparty/httpx.git/commitdiff
Tweak GatewayServer example
authorTom Christie <tom@tomchristie.com>
Sat, 6 Apr 2019 12:44:35 +0000 (13:44 +0100)
committerTom Christie <tom@tomchristie.com>
Sat, 6 Apr 2019 12:44:35 +0000 (13:44 +0100)
README.md

index 63eb03ddb79b37c0d791f2524e7bc7e3349d38a8..7aad675510d6c03af8a84e6bdb9d22e84423d85a 100644 (file)
--- a/README.md
+++ b/README.md
@@ -116,14 +116,7 @@ class GatewayServer:
         if query:
             url += '?' + query.decode()
 
-        async def body():
-            nonlocal receive
-
-            while True:
-                message = await receive()
-                yield message.get('body', b'')
-                if not message.get('more_body', False):
-                    break
+        body = self.stream_body(receive)
 
         response = await self.http.request(
             method, url, headers=headers, body=body, stream=True
@@ -142,6 +135,12 @@ class GatewayServer:
             })
         await send({'type': 'http.response.body'})
 
+    async def stream_body(self, receive):
+        while True:
+            message = await receive()
+            yield message.get('body', b'')
+            if not message.get('more_body', False):
+                break
 
 app = GatewayServer('http://example.org')
 ```