python多线程超时检测

自定义装饰器——代码超时检测

# -*- coding: utf-8 -*-
"""
@author:T
@email:ttbjsht@163.com
@File:time_out_decorator
@descr:
@Created on:2023/2/13/09:59
@IDE:PyCharm
"""
import threading
import time

# 错误示例 这个版本的代码 即使被检测代码未超时也会打印代码超时

def watchdog(thread, timeout) -> None:
	'''
	检测装饰器装饰函数对象是否超时
	:param thread: 被检测的对象 被检测对象是个可执行函数使用多线程来执行函数的功能 （如果不使用多线程的话会阻塞无法同时检测运行时间 ）
	:param timeout: 定义超时时间
	:return: None
	'''
	time.sleep(timeout)
	if thread.is_alive():  # 如果thread线程 已经执行完毕 则条件成立 执行完毕是线程对象的函数执行完毕 不是线程不存在
		print(f'{thread.name} timeout')
	return


def watchdog_decorator(timeout):
	def decorator(func):
		def wrapper(*args, **kwargs):
			thread = threading.Thread(target=func, args=args, kwargs=kwargs)  # 将被检测的代码创建一个线程
			thread.start()
			watchdog_thread = threading.Thread(target=watchdog, args=(thread, timeout))  # 将检测代码是否超时的函数对象创建一个线程
			watchdog_thread.start()
			# 上面两个线程都是子线程 同时执行
			thread.join()  # 如果watchdog_thread先执行结束也要等 thread执行完毕才能走下面的代码 join是阻塞主线程 执行 thread的，如果没有线程阻塞的话 主线程可能会在thread之前结束掉
			print("主线程运行结束")  #
	
		return wrapper
	
	return decorator

@watchdog_decorator(2)
def test():
	print('test函数开始执行')
	time.sleep(3)
	print('test函数执行结束')


if __name__ == '__main__':
	test()