blob: 2f6371dc12db43f0e03c5256f06d56e6572b6de6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
# Copyright 2012 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
import os
import sys
from .EventLoop import EventLoop
from portage.util._eventloop.asyncio_event_loop import AsyncioEventLoop
_asyncio_enabled = sys.version_info >= (3, 4)
_default_constructor = AsyncioEventLoop if _asyncio_enabled else EventLoop
# If _default_constructor doesn't support multiprocessing,
# then _multiprocessing_constructor is used in subprocesses.
_multiprocessing_constructor = EventLoop
_MAIN_PID = os.getpid()
_instances = {}
def global_event_loop():
"""
Get a global EventLoop (or compatible object) instance which
belongs exclusively to the current process.
"""
pid = os.getpid()
instance = _instances.get(pid)
if instance is not None:
return instance
constructor = _default_constructor
if not constructor.supports_multiprocessing and pid != _MAIN_PID:
constructor = _multiprocessing_constructor
# Use the _asyncio_wrapper attribute, so that unit tests can compare
# the reference to one retured from _wrap_loop(), since they should
# not close the loop if it refers to a global event loop.
instance = constructor()._asyncio_wrapper
_instances[pid] = instance
return instance
|
GitWeb
