forked from M-Labs/artiq
1
0
Fork 0

frontend: use sipyco SignalHandler (#1063)

This commit is contained in:
Sebastien Bourdeauducq 2022-05-27 15:17:27 +08:00
parent df1b19082c
commit 7cff63e539
4 changed files with 64 additions and 35 deletions

View File

@ -9,6 +9,7 @@ import re
from sipyco.pc_rpc import Server from sipyco.pc_rpc import Server
from sipyco import common_args from sipyco import common_args
from sipyco.logging_tools import log_with_name from sipyco.logging_tools import log_with_name
from sipyco.asyncio_tools import SignalHandler
from artiq.coredevice.comm_mgmt import Request, Reply from artiq.coredevice.comm_mgmt import Request, Reply
@ -77,21 +78,30 @@ def main():
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
try: try:
get_logs_task = asyncio.ensure_future( signal_handler = SignalHandler()
get_logs_sim(args.core_addr) if args.simulation else get_logs(args.core_addr)) signal_handler.setup()
try: try:
server = Server({"corelog": PingTarget()}, None, True) get_logs_task = asyncio.ensure_future(
loop.run_until_complete(server.start(common_args.bind_address_from_args(args), args.port)) get_logs_sim(args.core_addr) if args.simulation else get_logs(args.core_addr))
try: try:
loop.run_until_complete(server.wait_terminate()) server = Server({"corelog": PingTarget()}, None, True)
loop.run_until_complete(server.start(common_args.bind_address_from_args(args), args.port))
try:
_, pending = loop.run_until_complete(asyncio.wait(
[signal_handler.wait_terminate(), server.wait_terminate()],
return_when=asyncio.FIRST_COMPLETED))
for task in pending:
task.cancel()
finally:
loop.run_until_complete(server.stop())
finally: finally:
loop.run_until_complete(server.stop()) get_logs_task.cancel()
try:
loop.run_until_complete(get_logs_task)
except asyncio.CancelledError:
pass
finally: finally:
get_logs_task.cancel() signal_handler.teardown()
try:
loop.run_until_complete(get_logs_task)
except asyncio.CancelledError:
pass
finally: finally:
loop.close() loop.close()

View File

@ -6,7 +6,7 @@ import asyncio
import struct import struct
from enum import Enum from enum import Enum
from sipyco.asyncio_tools import AsyncioServer from sipyco.asyncio_tools import AsyncioServer, SignalHandler
from sipyco.pc_rpc import Server from sipyco.pc_rpc import Server
from sipyco import common_args from sipyco import common_args
@ -198,24 +198,33 @@ def main():
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
try: try:
monitor_mux = MonitorMux() signal_handler = SignalHandler()
comm_moninj = CommMonInj(monitor_mux.monitor_cb, monitor_mux.injection_status_cb) signal_handler.setup()
monitor_mux.comm_moninj = comm_moninj
loop.run_until_complete(comm_moninj.connect(args.core_addr))
try: try:
proxy_server = ProxyServer(monitor_mux) monitor_mux = MonitorMux()
loop.run_until_complete(proxy_server.start(bind_address, args.port_proxy)) comm_moninj = CommMonInj(monitor_mux.monitor_cb, monitor_mux.injection_status_cb)
monitor_mux.comm_moninj = comm_moninj
loop.run_until_complete(comm_moninj.connect(args.core_addr))
try: try:
server = Server({"moninj_proxy": PingTarget()}, None, True) proxy_server = ProxyServer(monitor_mux)
loop.run_until_complete(server.start(bind_address, args.port_control)) loop.run_until_complete(proxy_server.start(bind_address, args.port_proxy))
try: try:
loop.run_until_complete(server.wait_terminate()) server = Server({"moninj_proxy": PingTarget()}, None, True)
loop.run_until_complete(server.start(bind_address, args.port_control))
try:
_, pending = loop.run_until_complete(asyncio.wait(
[signal_handler.wait_terminate(), server.wait_terminate()],
return_when=asyncio.FIRST_COMPLETED))
for task in pending:
task.cancel()
finally:
loop.run_until_complete(server.stop())
finally: finally:
loop.run_until_complete(server.stop()) loop.run_until_complete(proxy_server.stop())
finally: finally:
loop.run_until_complete(proxy_server.stop()) loop.run_until_complete(comm_moninj.close())
finally: finally:
loop.run_until_complete(comm_moninj.close()) signal_handler.teardown()
finally: finally:
loop.close() loop.close()

View File

@ -10,7 +10,7 @@ from sipyco.sync_struct import Publisher
from sipyco.logging_tools import Server as LoggingServer from sipyco.logging_tools import Server as LoggingServer
from sipyco.broadcast import Broadcaster from sipyco.broadcast import Broadcaster
from sipyco import common_args from sipyco import common_args
from sipyco.asyncio_tools import atexit_register_coroutine from sipyco.asyncio_tools import atexit_register_coroutine, SignalHandler
from artiq import __version__ as artiq_version from artiq import __version__ as artiq_version
from artiq.master.log import log_args, init_log from artiq.master.log import log_args, init_log
@ -76,6 +76,9 @@ def main():
log_forwarder = init_log(args) log_forwarder = init_log(args)
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
atexit.register(loop.close) atexit.register(loop.close)
signal_handler = SignalHandler()
signal_handler.setup()
atexit.register(signal_handler.teardown)
bind = common_args.bind_address_from_args(args) bind = common_args.bind_address_from_args(args)
server_broadcast = Broadcaster() server_broadcast = Broadcaster()
@ -155,7 +158,7 @@ def main():
atexit_register_coroutine(server_logging.stop) atexit_register_coroutine(server_logging.stop)
print("ARTIQ master is now ready.") print("ARTIQ master is now ready.")
loop.run_forever() loop.run_until_complete(signal_handler.wait_terminate())
if __name__ == "__main__": if __name__ == "__main__":
main() main()

View File

@ -3,6 +3,8 @@
import argparse import argparse
import asyncio import asyncio
from sipyco.asyncio_tools import SignalHandler
from artiq.coredevice.comm_moninj import * from artiq.coredevice.comm_moninj import *
@ -18,19 +20,24 @@ def get_argparser():
def main(): def main():
args = get_argparser().parse_args() args = get_argparser().parse_args()
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
try: try:
comm = CommMonInj( signal_handler = SignalHandler()
lambda channel, probe, value: print("0x{:06x}: {}".format(channel, value)), signal_handler.setup()
lambda channel, override, value: None)
loop.run_until_complete(comm.connect(args.core_addr))
try: try:
for channel in args.channel: comm = CommMonInj(
comm.monitor_probe(True, channel, 0) lambda channel, probe, value: print("0x{:06x}: {}".format(channel, value)),
loop.run_forever() lambda channel, override, value: None)
loop.run_until_complete(comm.connect(args.core_addr))
try:
for channel in args.channel:
comm.monitor_probe(True, channel, 0)
loop.run_until_complete(signal_handler.wait_terminate())
finally:
loop.run_until_complete(comm.close())
finally: finally:
loop.run_until_complete(comm.close()) signal_handler.teardown()
finally: finally:
loop.close() loop.close()