Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

fix uvloop being ignored

Browse source
This commit is contained in:
Jack Robison 2019-06-28 13:47:02 -04:00
parent 8b0d96acba
commit d5183123fa
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
torba/torba/server/env.py
View file

@ -33,7 +33,7 @@ class Env:
self.allow_root = self.boolean('ALLOW_ROOT', False) self.allow_root = self.boolean('ALLOW_ROOT', False)
self.host = self.default('HOST', 'localhost') self.host = self.default('HOST', 'localhost')
self.rpc_host = self.default('RPC_HOST', 'localhost') self.rpc_host = self.default('RPC_HOST', 'localhost')
self.loop_policy = self.event_loop_policy() self.loop_policy = self.set_event_loop_policy()
self.obsolete(['UTXO_MB', 'HIST_MB', 'NETWORK']) self.obsolete(['UTXO_MB', 'HIST_MB', 'NETWORK'])
self.db_dir = self.required('DB_DIRECTORY') self.db_dir = self.required('DB_DIRECTORY')
self.db_engine = self.default('DB_ENGINE', 'leveldb') self.db_engine = self.default('DB_ENGINE', 'leveldb')
@ -129,14 +129,18 @@ class Env:
raise cls.Error('remove obsolete environment variables {}' raise cls.Error('remove obsolete environment variables {}'
.format(bad)) .format(bad))
def event_loop_policy(self): def set_event_loop_policy(self):
policy = self.default('EVENT_LOOP_POLICY', None) policy_name = self.default('EVENT_LOOP_POLICY', None)
if policy is None: if not policy_name:
return None import asyncio
if policy == 'uvloop': return asyncio.get_event_loop_policy()
elif policy_name == 'uvloop':
import uvloop import uvloop
return uvloop.EventLoopPolicy() import asyncio
raise self.Error('unknown event loop policy "{}"'.format(policy)) loop_policy = uvloop.EventLoopPolicy()
asyncio.set_event_loop_policy(loop_policy)
return loop_policy
raise self.Error('unknown event loop policy "{}"'.format(policy_name))
def cs_host(self, *, for_rpc): def cs_host(self, *, for_rpc):
"""Returns the 'host' argument to pass to asyncio's create_server """Returns the 'host' argument to pass to asyncio's create_server