bpo-31350: Optimize get_event_loop and _get_running_loop (#3347)

* call remove_done_callback in finally section

* Optimize get_event_loop and _get_running_loop

* rename _loop_pid as loop_pid and add blurb news

* rename _loop_pid as loop_pid and add blurb news

* add back _RunningLoop

* Update 2017-09-05-10-30-48.bpo-31350.dXJ-7N.rst

* Update 2017-09-05-10-30-48.bpo-31350.dXJ-7N.rst
This commit is contained in:
jimmylai 2017-09-05 17:36:59 -07:00 committed by Yury Selivanov
parent 305e56c27a
commit 80bbe6a7b6
2 changed files with 5 additions and 6 deletions

View File

@ -606,8 +606,7 @@ def new_event_loop(self):
# A TLS for the running event loop, used by _get_running_loop.
class _RunningLoop(threading.local):
_loop = None
_pid = None
loop_pid = (None, None)
_running_loop = _RunningLoop()
@ -619,8 +618,8 @@ def _get_running_loop():
This is a low-level function intended to be used by event loops.
This function is thread-specific.
"""
running_loop = _running_loop._loop
if running_loop is not None and _running_loop._pid == os.getpid():
running_loop, pid = _running_loop.loop_pid
if running_loop is not None and pid == os.getpid():
return running_loop
@ -630,8 +629,7 @@ def _set_running_loop(loop):
This is a low-level function intended to be used by event loops.
This function is thread-specific.
"""
_running_loop._pid = os.getpid()
_running_loop._loop = loop
_running_loop.loop_pid = (loop, os.getpid())
def _init_event_loop_policy():

View File

@ -0,0 +1 @@
Micro-optimize :func:`asyncio._get_running_loop` to become up to 10% faster.