def _run(name, queue, options):
"""
The actual process that runs the separate controller instance.
:param name: name of the process
:param queue: Queue of the binding parent.
:param options: Custom Options
:type name: str
"""
from pyplanet.core.instance import Controller
from pyplanet.utils.log import initiate_logger, QueueHandler
import logging
# Tokio Asyncio (EXPERIMENTAL).
if 'tokio' in options and options['tokio'] is True:
import asyncio
import tokio
policy = tokio.TokioLoopPolicy()
asyncio.set_event_loop_policy(policy)
asyncio.set_event_loop(tokio.new_event_loop())
logging.warning('Using experimental Tokio Asyncio Loop!')
# Logging to queue.
if multiprocessing.get_start_method() != 'fork': # pragma: no cover
initiate_logger()
root_logger = logging.getLogger()
formatter = ColoredFormatter(
'%(log_color)s%(levelname)-8s%(reset)s %(yellow)s[%(threadName)s][%(name)s]%(reset)s %(blue)s%(message)s'
)
queue_handler = QueueHandler(queue)
queue_handler.setFormatter(formatter)
root_logger.addHandler(queue_handler)
logging.getLogger(__name__).info(
'Starting pool process for \'{}\'...'.format(name))
# Setting thread name to our process name.
threading.main_thread().setName(name)
# Initiate instance.
instance = Controller.prepare(name).instance
instance._queue = queue
# Start and loop instance.
instance.start()
def loop(request, loop_type):
old_loop = asyncio.get_event_loop()
asyncio.set_event_loop(None)
if loop_type == 'uvloop':
loop = uvloop.new_event_loop()
elif loop_type == 'tokio':
import tokio
policy = tokio.TokioLoopPolicy()
asyncio.set_event_loop_policy(policy)
loop = tokio.new_event_loop()
else:
loop = asyncio.new_event_loop()
yield loop
loop.close()
asyncio.set_event_loop(old_loop)
gc.collect()
parser.add_argument('--uvloop', default=False, action='store_true')
parser.add_argument('--tokio', default=False, action='store_true')
parser.add_argument('--addr', default='127.0.0.1:25000', type=str)
parser.add_argument('--print', default=False, action='store_true')
args = parser.parse_args()
if args.uvloop:
import uvloop
loop = uvloop.new_event_loop()
print('using UVLoop')
elif args.tokio:
import tokio
policy = tokio.EventLoopPolicy()
asyncio.set_event_loop_policy(policy)
loop = tokio.new_event_loop()
print('using tokio loop')
else:
loop = asyncio.new_event_loop()
print('using asyncio loop')
asyncio.set_event_loop(loop)
loop.set_debug(False)
if args.print:
PRINT = 1
if hasattr(loop, 'print_debug_info'):
loop.create_task(print_debug(loop))
PRINT = 0