blob: de27b9acd666b40497118b014ffd67f1eab8c9d7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
/*
* Ceph - scalable distributed file system
*
* Copyright (C) 2017 SUSE LLC
*
* This is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software
* Foundation. See file COPYING.
*
*/
#include "Python.h"
#include "common/debug.h"
#define dout_context g_ceph_context
#define dout_subsys ceph_subsys_mgr
#undef dout_prefix
#define dout_prefix *_dout << "mgr " << __func__ << " "
#include "Gil.h"
SafeThreadState::SafeThreadState(PyThreadState *ts_)
: ts(ts_)
{
ceph_assert(ts != nullptr);
thread = pthread_self();
}
Gil::Gil(SafeThreadState &ts, bool new_thread) : pThreadState(ts)
{
// Acquire the GIL, set the current thread state
PyEval_RestoreThread(pThreadState.ts);
dout(25) << "GIL acquired for thread state " << pThreadState.ts << dendl;
//
// If called from a separate OS thread (i.e. a thread not created
// by Python, that does't already have a python thread state that
// was created when that thread was active), we need to manually
// create and switch to a python thread state specifically for this
// OS thread.
//
// Note that instead of requring the caller to set new_thread == true
// when calling this from a separate OS thread, we could figure out
// if this was necessary automatically, as follows:
//
// if (pThreadState->thread_id != PyThread_get_thread_ident()) {
//
// However, this means we're accessing pThreadState->thread_id, but
// the Python C API docs say that "The only public data member is
// PyInterpreterState *interp", i.e. doing this would violate
// something that's meant to be a black box.
//
if (new_thread) {
pNewThreadState = PyThreadState_New(pThreadState.ts->interp);
PyThreadState_Swap(pNewThreadState);
dout(20) << "Switched to new thread state " << pNewThreadState << dendl;
} else {
ceph_assert(pthread_self() == pThreadState.thread);
}
}
Gil::~Gil()
{
if (pNewThreadState != nullptr) {
dout(20) << "Destroying new thread state " << pNewThreadState << dendl;
PyThreadState_Swap(pThreadState.ts);
PyThreadState_Clear(pNewThreadState);
PyThreadState_Delete(pNewThreadState);
}
// Release the GIL, reset the thread state to NULL
PyEval_SaveThread();
dout(25) << "GIL released for thread state " << pThreadState.ts << dendl;
}
without_gil_t::without_gil_t()
{
assert(PyGILState_Check());
release_gil();
}
without_gil_t::~without_gil_t()
{
if (save) {
acquire_gil();
}
}
void without_gil_t::release_gil()
{
save = PyEval_SaveThread();
}
void without_gil_t::acquire_gil()
{
assert(save);
PyEval_RestoreThread(save);
save = nullptr;
}
with_gil_t::with_gil_t(without_gil_t& allow_threads)
: allow_threads{allow_threads}
{
allow_threads.acquire_gil();
}
with_gil_t::~with_gil_t()
{
allow_threads.release_gil();
}
|
