59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
"""
|
|
writen by shenjackyuanjie
|
|
mail: 3695888@qq.com
|
|
github: @shenjackyuanjie
|
|
gitee: @shenjackyuanjie
|
|
"""
|
|
|
|
import threading
|
|
|
|
from typing import Union
|
|
from threading import Lock
|
|
|
|
from Difficult_Rocket import DR_option, crash
|
|
from Difficult_Rocket.exception.threading import LockTimeOutError
|
|
|
|
|
|
class Threads(threading.Thread):
|
|
def run(self):
|
|
if DR_option.record_thread:
|
|
crash.all_thread.append(self)
|
|
super().run()
|
|
|
|
|
|
class ThreadLock:
|
|
|
|
def __init__(self, the_lock: Lock, time_out: Union[float, int] = 1 / 60) -> None:
|
|
self.lock = the_lock
|
|
self.time_out = time_out
|
|
|
|
def __enter__(self):
|
|
self.lock.acquire(timeout=self.time_out)
|
|
if not self.lock.locked():
|
|
raise LockTimeOutError(f'Lock time Out with {self.time_out}')
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
if self.lock.locked():
|
|
self.lock.release()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from Difficult_Rocket.exception import TestError
|
|
|
|
thread_lock = Lock()
|
|
|
|
test_lock = ThreadLock(thread_lock)
|
|
with test_lock:
|
|
print('do some thing')
|
|
...
|
|
with test_lock:
|
|
print('do some error')
|
|
raise TestError('ah lock test')
|