停止 Python 多处理 BaseManager 'serve_forever' 服务器？

Question

我在 Python 3.6 中有以下设置用于进程间通信：

from multiprocessing.managers import BaseManager
class MyManager(BaseManager): pass
MyManager.register('get_instance', callable=lambda:my_instance)
m = MyManager(address=('', 50000), authkey=b'key')
s = m.get_server()
s.serve_forever()

为了避免阻塞大型应用程序的消息循环，我使用了一个线程来包含此设置。 所以s.serve_forever()实际上在线程的run函数中run 。

这是根据文档完成的。 设置本身与调用共享实例的客户经理配合得很好。

但是，我发现无法停止这个“serve_forever”服务器。 由于文档没有提到这一点，我检查了源代码。 有一个stop_event据说我可以set退出循环。 但它不起作用，因为accepter守护进程/线程仍在运行。 而且我不能在服务器对象上调用shutdown ，因为我没有套接字对象c 。

那么如何关闭这个服务器呢？

附言。 使用BaseManager.start()对我来说并不是一个真正的选择，因为我的服务器共享一个异步消息循环，只能由启动进程访问。 BaseManager.start()生成一个不再有权访问消息循环的新进程。 另一方面， get_server().serve_forever()在调用进程中运行。

Answer 1

在服务器中试试这个：

import threading

s = m.get_server()
stop_timer = threading.Timer(1, lambda:s.stop_event.set())
MyManager.register('stop', callable=lambda:stop_timer.start())
s.serve_forever()

在客户端：

MyManager.register('stop')
m.stop()

更新：

我通过使用threading.Timer延迟stop_event.set()解决了超时问题

旧：

但是，跨机器，你会遇到很长的超时，因为stop()方法无法接收数据，这个答案谈到了 - 但我无法让它工作：

正确断开多处理远程管理器

我是多处理管理员的新手，如果找到更好的解决方案，我会更新这个答案。 （更新：发现，更新）

Answer 2

你可以包装/使用管理器启动/关闭以避免永远运行服务器（start（）为你启动服务器）：

m = MyManager(address=('', 50000), authkey=b'key')
m.start()
# do something
m.shutdown()

Answer 3

这是阅读 multiprocessing.managers 源代码后的 hack，从 _finalize_manager() 中剥离，基本上创建到服务器的连接并发送关闭消息

from multiprocessing.managers import dispatch,listener_client

_Client = listener_client['pickle'][1]
# address and authkey same as when started the manager
conn = _Client(address=('127.0.0.1', 50000), authkey=b'key')
dispatch(conn, None, 'shutdown')
conn.close()