platformio/platformio-core
1
0
Fork 1
mirror of https://github.com/platformio/platformio-core.git synced 2025-07-29 17:47:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

Setup MultiThreadingStdStreams from child threads

Browse Source
This commit is contained in:
Ivan Kravets
2019-07-02 20:44:33 +03:00
parent c79d5f0cf1
commit 2988724456
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
platformio/commands/home/rpc/handlers/piocore.py
View File

@ -31,7 +31,7 @@ except ImportError:
from threading import get_ident as thread_get_ident
class ThreadSafeStdBuffer(object):
class MultiThreadingStdStream(object):
def __init__(self, parent_stream):
self._buffers = {thread_get_ident(): parent_stream}
@ -66,9 +66,12 @@ class ThreadSafeStdBuffer(object):
class PIOCoreRPC(object):
def __init__(self):
PIOCoreRPC.thread_stdout = ThreadSafeStdBuffer(sys.stdout)
PIOCoreRPC.thread_stderr = ThreadSafeStdBuffer(sys.stderr)
@staticmethod
def setup_multithreading_std_streams():
if isinstance(sys.stdout, MultiThreadingStdStream):
return
PIOCoreRPC.thread_stdout = MultiThreadingStdStream(sys.stdout)
PIOCoreRPC.thread_stderr = MultiThreadingStdStream(sys.stderr)
sys.stdout = PIOCoreRPC.thread_stdout
sys.stderr = PIOCoreRPC.thread_stderr
@ -83,13 +86,17 @@ class PIOCoreRPC(object):
raise jsonrpc.exceptions.JSONRPCDispatchException(
code=4002, message="PIO Core: non-ASCII chars in arguments")
def _call_cli():
with util.cd((options or {}).get("cwd") or os.getcwd()):
PIOCoreRPC.setup_multithreading_std_streams()
cwd = (options or {}).get("cwd") or os.getcwd()
def _call_inline():
with util.cd(cwd):
exit_code = __main__.main(["-c"] + args)
return (PIOCoreRPC.thread_stdout.get_value_and_close(),
PIOCoreRPC.thread_stderr.get_value_and_close(), exit_code)
d = threads.deferToThread(_call_cli)
d = threads.deferToThread(_call_inline)
d.addCallback(PIOCoreRPC._call_callback, "--json-output" in args)
d.addErrback(PIOCoreRPC._call_errback)
return d