add more parody with asyncio example

mgasner · Nov 4, 2017 · 0459f86 · 0459f86
1 parent f5f2ede
commit 0459f86
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/examples/flask_gevent/flask_app.py b/examples/flask_gevent/flask_app.py
@@ -21,13 +21,15 @@ class RandomType(graphene.ObjectType):
 
 class Subscription(graphene.ObjectType):
 
- count_seconds = graphene.Int()
+ count_seconds = graphene.Int(up_to=graphene.Int())
 
  random_int = graphene.Field(RandomType)
 
 
- def resolve_count_seconds(root, info):
- return Observable.interval(1000).map(lambda i: "{0}".format(i))
+ def resolve_count_seconds(root, info, up_to):
+ return Observable.interval(1000)\
+ .map(lambda i: "{0}".format(i))\
+ .take_while(lambda i: int(i) <= up_to)
 
  def resolve_random_int(root, info):
  import random

diff --git a/graphql_ws/gevent_server.py b/graphql_ws/gevent_server.py
@@ -1,6 +1,7 @@
 import json
 
 from graphql import format_error, graphql
+from graphql.execution.executors.sync import SyncExecutor
 from rx import Observer, Observable
 from .server import BaseWebSocketSubscriptionServer, ConnectionContext, ConnectionClosedException
 from .constants import *