105 lines
2.6 KiB
Python
105 lines
2.6 KiB
Python
import t, c
|
||
from stdint import *
|
||
from w32.win32base import *
|
||
from w32.win32process import CreateThread, GetExitCodeThread, GetCurrentThreadId, TlsAlloc, TlsFree, TlsGetValue, TlsSetValue
|
||
from w32.win32sync import WaitForSingleObject
|
||
|
||
# 全部导入锁类，用户也可按需导入
|
||
from lock import Lock
|
||
from event import Event
|
||
from condition import Condition
|
||
from rwlock import RWLock
|
||
|
||
|
||
@t.Object
|
||
class Thread:
|
||
"""线程类，封装 Windows CreateThread。
|
||
|
||
线程回调签名: def thread_func(arg: t.CPtr) -> t.CPtr
|
||
注意: Windows API 限制返回值为 DWORD(32位)，64位指针会被截断。
|
||
如需返回 64 位结果，请通过 arg 参数结构体传递。
|
||
|
||
用法:
|
||
def worker(arg: t.CPtr) -> t.CPtr:
|
||
# 线程逻辑
|
||
return NULL
|
||
|
||
t: Thread = Thread(worker, arg_ptr)
|
||
t.start()
|
||
t.join()
|
||
"""
|
||
handle: HANDLE
|
||
thread_id: ULONG
|
||
target: VOIDPTR
|
||
arg: VOIDPTR
|
||
started: BOOL
|
||
|
||
def __init__(self, target: VOIDPTR, arg: VOIDPTR = NULL):
|
||
self.handle = NULL
|
||
self.thread_id = 0
|
||
self.target = target
|
||
self.arg = arg
|
||
self.started = FALSE
|
||
|
||
def start(self):
|
||
if self.started:
|
||
return
|
||
self.handle = CreateThread(NULL, 0, self.target, self.arg, 0, c.Addr(self.thread_id))
|
||
self.started = TRUE
|
||
|
||
def join(self, timeout_ms: ULONG = INFINITE) -> BOOL:
|
||
if not self.started:
|
||
return FALSE
|
||
result: ULONG = WaitForSingleObject(self.handle, timeout_ms)
|
||
return result == WAIT_OBJECT_0
|
||
|
||
def is_alive(self) -> BOOL:
|
||
if not self.started:
|
||
return FALSE
|
||
result: ULONG = WaitForSingleObject(self.handle, 0)
|
||
return result == WAIT_TIMEOUT
|
||
|
||
def get_exit_code(self) -> ULONG:
|
||
code: ULONG = 0
|
||
GetExitCodeThread(self.handle, c.Addr(code))
|
||
return code
|
||
|
||
def close(self):
|
||
if self.handle != NULL:
|
||
CloseHandle(self.handle)
|
||
self.handle = NULL
|
||
|
||
|
||
@t.Object
|
||
class local:
|
||
"""TLS (Thread Local Storage) 封装。
|
||
|
||
用法:
|
||
tls: local = local()
|
||
tls.set(value_ptr)
|
||
val: t.CPtr = tls.get()
|
||
"""
|
||
index: ULONG
|
||
|
||
def __init__(self):
|
||
self.index = TlsAlloc()
|
||
|
||
def get(self) -> VOIDPTR:
|
||
return TlsGetValue(self.index)
|
||
|
||
def set(self, value: VOIDPTR) -> BOOL:
|
||
return TlsSetValue(self.index, value)
|
||
|
||
def free(self) -> BOOL:
|
||
return TlsFree(self.index)
|
||
|
||
|
||
def get_current_thread_id() -> ULONG:
|
||
"""获取当前线程 ID"""
|
||
return GetCurrentThreadId()
|
||
|
||
|
||
def sleep_ms(ms: ULONG):
|
||
"""让当前线程休眠指定毫秒数"""
|
||
Sleep(ms)
|