Python 多執行緒獲取系統級執行緒 ID (TID) 完整指南

Python 多執行緒獲取系統級執行緒 ID (TID) 完整指南

Python threading.get_ident() 回傳的是 **Python 解釋器內部分配的整數 ID**，非作業系統核心調度的真實執行緒 ID (TID)。在效能分析、系統追蹤 (perf, strace, SystemTap)、綁定 CPU 親和性等場景下，必須獲取 **Linux 核心層面的 TID (即 gettid() 系統呼叫回傳值)**。本文提供完整跨平台解決方案。

核心差異對比

Linux 平台：透過 ctypes 呼叫 libc.gettid()

import ctypes
import threading
import os

# 1. 獲取系統呼叫號 (x86_64 為 186, ARM64 為 178, i386 為 224)
# 建議動態查詢而非硬編碼
def get_syscall_number():
    # 方法一：編譯小程式查詢 (需 gcc)
    # 方法二：查閱 /usr/include/asm/unistd.h 或 /usr/include/asm-generic/unistd.h
    # 方法三：使用 python 內建常數 (Linux 3.3+)
    return 186  # x86_64 常見值

SYS_gettid = get_syscall_number()
libc = ctypes.CDLL('libc.so.6', use_errno=True)

def get_tid():
    """回傳當前執行緒的核心 TID"""
    return libc.syscall(SYS_gettid)

# 測試
def worker():
    print(f"Python ident: {threading.get_ident()}, Kernel TID: {get_tid()}")

threads = [threading.Thread(target=worker) for _ in range(3)]
for t in threads:
    t.start()
for t in threads:
    t.join()

# 輸出範例：
# Python ident: 140234567890112, Kernel TID: 12346
# Python ident: 140234567890113, Kernel TID: 12347
# Python ident: 140234567890114, Kernel TID: 12348

進階：跨架構自動偵測系統呼叫號

import platform
import ctypes

def get_gettid_syscall_nr():
    machine = platform.machine()
    if machine == 'x86_64':
        return 186
    elif machine == 'aarch64':
        return 178
    elif machine in ('i386', 'i686'):
        return 224
    elif machine == 'arm':
        return 319  # ARM EABI
    elif machine == 'ppc64le':
        return 341
    elif machine == 'riscv64':
        return 178
    else:
        raise RuntimeError(f"Unsupported architecture: {machine}")

SYS_gettid = get_gettid_syscall_nr()
libc = ctypes.CDLL('libc.so.6', use_errno=True)
gettid = lambda: libc.syscall(SYS_gettid)

Python 3.8+ 官方支援：threading.get_native_id()

import threading
import sys

if sys.version_info >= (3, 8):
    # 官方 API，跨平台 (Linux: gettid, macOS: pthread_threadid_np, Windows: GetThreadId)
    print(f"Native TID: {threading.get_native_id()}")
else:
    # 回退方案
    pass

強烈建議：Python 3.8+ 專案直接使用 threading.get_native_id()，無需手動 ctypes 綁定。

實戰應用場景

1. 效能剖析對齊：perf record -g -p 產生的報告中執行緒 ID 為 TID，對應 Python 程式碼需映射關係。
2. CPU 親和性綁定：os.sched_setaffinity(tid, {cpu_id}) 需核心 TID (Python 3.3+ os.sched_setaffinity(0, ...) 僅影響當前執行緒)。
3. 系統追蹤關聯：strace -p , SystemTap, eBPF 追蹤特定執行緒。
4. 日誌關聯：記錄 TID 方便與系統日誌、容器監控 (cAdvisor, Prometheus node_exporter) 關聯分析。

macOS / BSD 平台方案

# macOS: pthread_threadid_np(pthread_t, uint64_t*)
import ctypes
import threading

libc = ctypes.CDLL('libc.dylib')
pthread_self = libc.pthread_self
pthread_self.restype = ctypes.c_void_p

pthread_threadid_np = libc.pthread_threadid_np
pthread_threadid_np.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_uint64)]
pthread_threadid_np.restype = ctypes.c_int

def get_tid_macos():
    tid = ctypes.c_uint64()
    pthread_threadid_np(pthread_self(), ctypes.byref(tid))
    return tid.value

Windows 平台方案

# Windows: GetCurrentThreadId() 回傳 DWORD TID
import ctypes
kernel32 = ctypes.windll.kernel32
GetCurrentThreadId = kernel32.GetCurrentThreadId
GetCurrentThreadId.restype = ctypes.c_ulong

def get_tid_windows():
    return GetCurrentThreadId()

完整跨平台封裝模組

# compat_tid.py
import sys
import threading
import ctypes

def get_native_tid():
    """跨平台獲取核心執行緒 ID"""
    if sys.version_info >= (3, 8):
        return threading.get_native_id()
    
    if sys.platform.startswith('linux'):
        # Linux ctypes 方案 (見上文)
        ...
    elif sys.platform == 'darwin':
        # macOS 方案 (見上文)
        ...
    elif sys.platform == 'win32':
        # Windows 方案 (見上文)
        ...
    else:
        raise NotImplementedError(f"Platform {sys.platform} not supported")

# 使用
from compat_tid import get_native_tid
print(f"Current TID: {get_native_tid()}")

注意事項

• threading.get_ident() 在 Python 進程重啟、執行緒池複用時可能重複；TID 由核心全局唯一分配，進程內不重複。
• Linux gettid() 非 POSIX 標準，但已成為事實標準 (Android, 容器運行時均依賴)。
• 容器環境下 TID 在宿主機與容器內命名空間可能不同，注意 pid_namespace 隔離。

結語

Python 3.8+ 內建 threading.get_native_id() 徹底解決了跨平台獲取核心 TID 的歷史痛點。舊版本專案可參考本文 ctypes 方案實現相容封裝。掌握 TID 與 Python 邏輯 ID 的映射關係，是進行深度效能調優、系統級除錯與可觀測性建設的基礎技能。