Як знайти ідентифікатор потоку в Python

У мене є багатопотокова програма Python та функція утиліти writeLog(message), яка записує часову позначку з наступним повідомленням. На жаль, результат журнального файлу не вказує, який потік генерує яке повідомлення.

Я хотів би writeLog()мати можливість додати щось до повідомлення, щоб визначити, яка нитка викликає його. Очевидно, що я міг би просто змусити потоки передавати цю інформацію, але це було б набагато більше роботи. Чи є якийсь еквівалент потоку, os.getpid()який я міг би використати?

threading.get_ident()твори, або threading.current_thread().ident(або threading.currentThread().identдля Python <2.6).

Використовуючи модуль журналу, ви можете автоматично додавати поточний ідентифікатор потоку у кожну запис журналу. Просто використовуйте один із цих ключів відображення LogRecord у рядку формату реєстратора:

% (нитка) d: ідентифікатор нитки (якщо є).

% (threadName) s: ім'я теми (якщо є).

і налаштувати з ним обробник за замовчуванням:

logging.basicConfig(format="%(threadName)s:%(message)s")

thread.get_ident()Функція повертає довге ціле на Linux. Це насправді не ідентифікатор потоку.

Я використовую цей метод, щоб дійсно отримати ідентифікатор потоку в Linux:

import ctypes
libc = ctypes.cdll.LoadLibrary('libc.so.6')

# System dependent, see e.g. /usr/include/x86_64-linux-gnu/asm/unistd_64.h
SYS_gettid = 186

def getThreadId():
   """Returns OS thread id - Specific to Linux"""
   return libc.syscall(SYS_gettid)

Ця функціональність тепер підтримується Python 3.8+ :)

https://github.com/python/cpython/commit/4959c33d2555b89b494c678d99be81a65ee864b0

https://github.com/python/cpython/pull/11993

Я бачив приклади ідентифікаторів потоку на зразок цього:

class myThread(threading.Thread):
    def __init__(self, threadID, name, counter):
        self.threadID = threadID
        ...

Документи для нарізування модулів також вказують nameатрибут:

...

A thread has a name. 
The name can be passed to the constructor, 
and read or changed through the name attribute.

...

Thread.name

A string used for identification purposes only. 
It has no semantics. Multiple threads may
be given the same name. The initial name is set by the constructor.

Ви можете отримати ідентифікатор поточної запущеної нитки. Ідентифікатор може бути повторно використаний для інших потоків, якщо поточний потік закінчується.

Коли ви створюєте екземпляр Thread, ім’я дається неявно для потоку, що є шаблоном: Thread-number

Ім'я не має значення, і ім'я не повинно бути унікальним. Ідентифікація всіх запущених потоків унікальна.

import threading


def worker():
    print(threading.current_thread().name)
    print(threading.get_ident())


threading.Thread(target=worker).start()
threading.Thread(target=worker, name='foo').start()

Функція threading.current_thread () повертає поточний запущений потік. Цей об’єкт вміщує всю інформацію потоку.

Я створив декілька потоків у Python, надрукував об’єкти потоку та надрукував ідентифікатор за допомогою identзмінної. Я бачу, що всі ідентичні файли однакові:

<Thread(Thread-1, stopped 140500807628544)>
<Thread(Thread-2, started 140500807628544)>
<Thread(Thread-3, started 140500807628544)>

Аналогічно @brucexin, мені потрібно було отримати ідентифікатор потоку на рівні ОС (який! = thread.get_ident()) І використовувати щось на зразок нижче, щоб не залежати від конкретних чисел і бути лише amd64:

---- 8< ---- (xos.pyx)
"""module xos complements standard module os""" 

cdef extern from "<sys/syscall.h>":                                                             
    long syscall(long number, ...)                                                              
    const int SYS_gettid                                                                        

# gettid returns current OS thread identifier.                                                  
def gettid():                                                                                   
    return syscall(SYS_gettid)                                                                  

і

---- 8< ---- (test.py)
import pyximport; pyximport.install()
import xos

...

print 'my tid: %d' % xos.gettid()

це залежить від Cython.