From 5c3cda0d1a850a1a9b43892f48376b8876bd5863 Mon Sep 17 00:00:00 2001 From: Victor Stinner Date: Wed, 25 Mar 2020 21:23:53 +0100 Subject: bpo-39947: Add PyThreadState_GetID() function (GH-19163) Add PyThreadState_GetID() function: get the unique identifier of a Python thread state. --- Doc/c-api/init.rst | 9 +++++++++ Doc/whatsnew/3.9.rst | 2 ++ Include/pystate.h | 1 + Misc/NEWS.d/next/C API/2020-03-25-19-44-55.bpo-39947.2OxvPt.rst | 2 ++ Modules/_asynciomodule.c | 5 +++-- Python/pystate.c | 8 ++++++++ 6 files changed, 25 insertions(+), 2 deletions(-) create mode 100644 Misc/NEWS.d/next/C API/2020-03-25-19-44-55.bpo-39947.2OxvPt.rst diff --git a/Doc/c-api/init.rst b/Doc/c-api/init.rst index 294c1b9..f78ab99 100644 --- a/Doc/c-api/init.rst +++ b/Doc/c-api/init.rst @@ -1084,6 +1084,15 @@ All of the following functions must be called after :c:func:`Py_Initialize`. .. versionadded:: 3.9 +.. c:function:: uint64_t PyThreadState_GetID(PyThreadState *tstate) + + Get the unique thread state identifier of the Python thread state *tstate*. + + *tstate* must not be ``NULL``. + + .. versionadded:: 3.9 + + .. c:function:: PyInterpreterState* PyThreadState_GetInterpreter(PyThreadState *tstate) Get the interpreter of the Python thread state *tstate*. diff --git a/Doc/whatsnew/3.9.rst b/Doc/whatsnew/3.9.rst index b11c29b..778e443 100644 --- a/Doc/whatsnew/3.9.rst +++ b/Doc/whatsnew/3.9.rst @@ -429,6 +429,8 @@ Build and C API Changes :c:func:`PyInterpreterState_Get` functions to get the interpreter. New :c:func:`PyThreadState_GetFrame` function to get the current frame of a Python thread state. + New :c:func:`PyThreadState_GetID` function: get the unique identifier of a + Python thread state. (Contributed by Victor Stinner in :issue:`39947`.) * Add ``--with-platlibdir`` option to the ``configure`` script: name of the diff --git a/Include/pystate.h b/Include/pystate.h index 5866fef..c46d3fe 100644 --- a/Include/pystate.h +++ b/Include/pystate.h @@ -91,6 +91,7 @@ PyAPI_FUNC(int) PyThreadState_SetAsyncExc(unsigned long, PyObject *); /* New in 3.9 */ PyAPI_FUNC(PyInterpreterState*) PyThreadState_GetInterpreter(PyThreadState *tstate); PyAPI_FUNC(struct _frame*) PyThreadState_GetFrame(PyThreadState *tstate); +PyAPI_FUNC(uint64_t) PyThreadState_GetID(PyThreadState *tstate); #endif typedef diff --git a/Misc/NEWS.d/next/C API/2020-03-25-19-44-55.bpo-39947.2OxvPt.rst b/Misc/NEWS.d/next/C API/2020-03-25-19-44-55.bpo-39947.2OxvPt.rst new file mode 100644 index 0000000..e9910a5 --- /dev/null +++ b/Misc/NEWS.d/next/C API/2020-03-25-19-44-55.bpo-39947.2OxvPt.rst @@ -0,0 +1,2 @@ +Add :c:func:`PyThreadState_GetID` function: get the unique identifier of a +Python thread state. diff --git a/Modules/_asynciomodule.c b/Modules/_asynciomodule.c index 1092b08..c10866a 100644 --- a/Modules/_asynciomodule.c +++ b/Modules/_asynciomodule.c @@ -231,7 +231,8 @@ get_running_loop(PyObject **loop) PyObject *rl; PyThreadState *ts = PyThreadState_Get(); - if (ts->id == cached_running_holder_tsid && cached_running_holder != NULL) { + uint64_t ts_id = PyThreadState_GetID(ts); + if (ts_id == cached_running_holder_tsid && cached_running_holder != NULL) { // Fast path, check the cache. rl = cached_running_holder; // borrowed } @@ -253,7 +254,7 @@ get_running_loop(PyObject **loop) } cached_running_holder = rl; // borrowed - cached_running_holder_tsid = ts->id; + cached_running_holder_tsid = ts_id; } assert(Py_IS_TYPE(rl, &PyRunningLoopHolder_Type)); diff --git a/Python/pystate.c b/Python/pystate.c index 66a1d3b..246665b 100644 --- a/Python/pystate.c +++ b/Python/pystate.c @@ -1021,6 +1021,14 @@ PyThreadState_GetFrame(PyThreadState *tstate) } +uint64_t +PyThreadState_GetID(PyThreadState *tstate) +{ + assert(tstate != NULL); + return tstate->id; +} + + /* Asynchronously raise an exception in a thread. Requested by Just van Rossum and Alex Martelli. To prevent naive misuse, you must write your own extension -- cgit v0.12