Add stream decorator for HTTPMethodView

docs/sanic/streaming.md

@@ -35,16 +35,31 @@ async def index(request):
 
 Sanic allows you to get request data by stream, as below. When the request ends, `request.stream.get()` returns `None`.
 
-```
+```python
 from sanic import Sanic
 from sanic.views import CompositionView
+from sanic.views import HTTPMethodView
+from sanic.views import stream as stream_decorator
 from sanic.blueprints import Blueprint
 from sanic.response import stream, text
 
 bp = Blueprint('blueprint_request_stream')
 app = Sanic('request_stream', is_request_stream=True)
 
 
+class SimpleView(HTTPMethodView):
+
+    @stream_decorator
+    async def post(self, request):
+        result = ''
+        while True:
+            body = await request.stream.get()
+            if body is None:
+                break
+            result += body.decode('utf-8')
+        return text(result)
+
+
 @app.stream('/stream')
 async def handler(request):
     async def streaming(response):
@@ -83,6 +98,7 @@ async def post_handler(request):
     return text(result)
 
 app.blueprint(bp)
+app.add_route(SimpleView.as_view(), '/method_view')
 view = CompositionView()
 view.add(['POST'], post_handler, stream=True)
 app.add_route(view, '/composition_view')

examples/request_stream/client.py

@@ -6,5 +6,5 @@
 for i in range(1, 250000):
     data += str(i)
 
-r = requests.post('http://127.0.0.1:8000/bp_stream', data=data)
+r = requests.post('http://127.0.0.1:8000/method_view', data=data)
 print(r.text)

examples/request_stream/server.py

@@ -1,12 +1,27 @@
 from sanic import Sanic
 from sanic.views import CompositionView
+from sanic.views import HTTPMethodView
+from sanic.views import stream as stream_decorator
 from sanic.blueprints import Blueprint
 from sanic.response import stream, text
 
 bp = Blueprint('blueprint_request_stream')
 app = Sanic('request_stream')
 
 
+class SimpleView(HTTPMethodView):
+
+    @stream_decorator
+    async def post(self, request):
+        result = ''
+        while True:
+            body = await request.stream.get()
+            if body is None:
+                break
+            result += body.decode('utf-8')
+        return text(result)
+
+
 @app.stream('/stream')
 async def handler(request):
     async def streaming(response):
@@ -45,6 +60,7 @@ async def post_handler(request):
     return text(result)
 
 app.blueprint(bp)
+app.add_route(SimpleView.as_view(), '/method_view')
 view = CompositionView()
 view.add(['POST'], post_handler, stream=True)
 app.add_route(view, '/composition_view')

sanic/router.py

@@ -352,4 +352,7 @@ def is_stream_handler(self, request):
         :return: bool
         """
         handler = self.get(request)[0]
+        if (hasattr(handler, 'view_class') and
+                hasattr(handler.view_class, request.method.lower())):
+            handler = getattr(handler.view_class, request.method.lower())
         return hasattr(handler, 'is_stream') and handler.is_stream

sanic/views.py

@@ -64,6 +64,11 @@ def view(*args, **kwargs):
         return view
 
 
+def stream(func):
+    func.is_stream = True
+    return func
+
+
 class CompositionView:
     """Simple method-function mapped view for the sanic.
     You can add handler functions to methods (get, post, put, patch, delete)

tests/test_request_stream.py

@@ -2,6 +2,7 @@
 from sanic.blueprints import Blueprint
 from sanic.views import CompositionView
 from sanic.views import HTTPMethodView
+from sanic.views import stream as stream_decorator
 from sanic.response import stream, text
 
 bp = Blueprint('test_blueprint_request_stream')
@@ -13,6 +14,16 @@ class SimpleView(HTTPMethodView):
     def get(self, request):
         return text('OK')
 
+    @stream_decorator
+    async def post(self, request):
+        result = ''
+        while True:
+            body = await request.stream.get()
+            if body is None:
+                break
+            result += body.decode('utf-8')
+        return text(result)
+
 
 @app.stream('/stream')
 async def handler(request):
@@ -31,7 +42,7 @@ async def get(request):
 
 
 @bp.stream('/bp_stream')
-async def bp_handler(request):
+async def bp_stream(request):
     result = ''
     while True:
         body = await request.stream.get()
@@ -41,6 +52,11 @@ async def bp_handler(request):
     return text(result)
 
 
+@bp.get('/bp_get')
+async def bp_get(request):
+    return text('OK')
+
+
 def get_handler(request):
     return text('OK')
 
@@ -73,6 +89,10 @@ def test_request_stream():
     assert response.status == 200
     assert response.text == 'OK'
 
+    request, response = app.test_client.post('/method_view', data=data)
+    assert response.status == 200
+    assert response.text == data
+
     request, response = app.test_client.get('/composition_view')
     assert response.status == 200
     assert response.text == 'OK'
@@ -89,6 +109,10 @@ def test_request_stream():
     assert response.status == 200
     assert response.text == data
 
+    request, response = app.test_client.get('/bp_get')
+    assert response.status == 200
+    assert response.text == 'OK'
+
     request, response = app.test_client.post('/bp_stream', data=data)
     assert response.status == 200
     assert response.text == data