【发布时间】:2019-06-04 10:42:36
【问题描述】:
我正在构建一个运行另一个 Python 程序的看门狗计时器,如果它无法从任何线程中找到签入,则关闭整个程序。这样它最终将能够控制所需的通信端口。定时器代码如下:
from multiprocessing import Process, Queue
from time import sleep
from copy import deepcopy
PATH_TO_FILE = r'.\test_program.py'
WATCHDOG_TIMEOUT = 2
class Watchdog:
def __init__(self, filepath, timeout):
self.filepath = filepath
self.timeout = timeout
self.threadIdQ = Queue()
self.knownThreads = {}
def start(self):
threadIdQ = self.threadIdQ
process = Process(target = self._executeFile)
process.start()
try:
while True:
unaccountedThreads = deepcopy(self.knownThreads)
# Empty queue since last wake. Add new thread IDs to knownThreads, and account for all known thread IDs
# in queue
while not threadIdQ.empty():
threadId = threadIdQ.get()
if threadId in self.knownThreads:
unaccountedThreads.pop(threadId, None)
else:
print('New threadId < {} > discovered'.format(threadId))
self.knownThreads[threadId] = False
# If there is a known thread that is unaccounted for, then it has either hung or crashed.
# Shut everything down.
if len(unaccountedThreads) > 0:
print('The following threads are unaccounted for:\n')
for threadId in unaccountedThreads:
print(threadId)
print('\nShutting down!!!')
break
else:
print('No unaccounted threads...')
sleep(self.timeout)
# Account for any exceptions thrown in the watchdog timer itself
except:
process.terminate()
raise
process.terminate()
def _executeFile(self):
with open(self.filepath, 'r') as f:
exec(f.read(), {'wdQueue' : self.threadIdQ})
if __name__ == '__main__':
wd = Watchdog(PATH_TO_FILE, WATCHDOG_TIMEOUT)
wd.start()
我还有一个小程序可以测试看门狗功能
from time import sleep
from threading import Thread
from queue import SimpleQueue
Q_TO_Q_DELAY = 0.013
class QToQ:
def __init__(self, processQueue, threadQueue):
self.processQueue = processQueue
self.threadQueue = threadQueue
Thread(name='queueToQueue', target=self._run).start()
def _run(self):
pQ = self.processQueue
tQ = self.threadQueue
while True:
while not tQ.empty():
sleep(Q_TO_Q_DELAY)
pQ.put(tQ.get())
def fastThread(q):
while True:
print('Fast thread, checking in!')
q.put('fastID')
sleep(0.5)
def slowThread(q):
while True:
print('Slow thread, checking in...')
q.put('slowID')
sleep(1.5)
def hangThread(q):
print('Hanging thread, checked in')
q.put('hangID')
while True:
pass
print('Hello! I am a program that spawns threads!\n\n')
threadQ = SimpleQueue()
Thread(name='fastThread', target=fastThread, args=(threadQ,)).start()
Thread(name='slowThread', target=slowThread, args=(threadQ,)).start()
Thread(name='hangThread', target=hangThread, args=(threadQ,)).start()
QToQ(wdQueue, threadQ)
如您所见,我需要将线程放入 queue.Queue,而一个单独的对象将 queue.Queue 的输出缓慢地馈送到多处理队列中。相反,如果我将线程直接放入多处理队列,或者在放入之间没有 QToQ 对象休眠,则多处理队列将锁定,并且在看门狗端看起来总是空的。
现在,由于多处理队列应该是线程和进程安全的,我只能假设我在实现中搞砸了。我的解决方案似乎奏效了,但也感觉很老套,我觉得我应该修复它。
如果重要的话,我正在使用 Python 3.7.2。
【问题讨论】:
-
我使用的是 Python 3.6 (
from queue import Queue),我不能让它失败。我可以直接通过wdQueue或通过Queue使用QToQ发送ID,而Watchdog 总是接听hangID并关闭程序。 -
更新虽然没有
hangThread它总是在某个时候关闭。 -
您的
except想成为finally。
标签: python multithreading multiprocessing message-queue python-multiprocessing