pybind11/tests/test_thread.cpp at 366f40d35a80e5fbfe5c9eb20f05bb367136cef7 · pybind/pybind11 · GitHub

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
/*
    tests/test_thread.cpp -- call pybind11 bound methods in threads

    Copyright (c) 2017 Laramie Leavitt (Google LLC) <[email protected]>

    All rights reserved. Use of this source code is governed by a
    BSD-style license that can be found in the LICENSE file.
*/

#include <chrono>
#include <thread>

#include <pybind11/cast.h>
#include <pybind11/pybind11.h>

#include "pybind11_tests.h"


namespace py = pybind11;


namespace {

struct IntStruct {
    int value;
};

}

TEST_SUBMODULE(thread, m) {

    py::class_<IntStruct>(m, "IntStruct")
        .def(py::init([](const int i) { return IntStruct{i}; }));

    // implicitly_convertible uses loader_life_support when an implicit
    // conversion is required in order to llifetime extend the reference.
    py::implicitly_convertible<int, IntStruct>();


    m.def("test", [](const IntStruct& in) {
        IntStruct copy = in;

        {
          py::gil_scoped_release release;
          std::this_thread::sleep_for(std::chrono::milliseconds(5));
        }

        if (in.value != copy.value) {
            throw std::runtime_error("Reference changed!!");
        }
    });

   m.def("test_no_gil", [](const IntStruct& in) {
        IntStruct copy = in;
        std::this_thread::sleep_for(std::chrono::milliseconds(5));
        if (in.value != copy.value) {
            throw std::runtime_error("Reference changed!!");
        }
    },
    py::call_guard<py::gil_scoped_release>());

    // NOTE: std::string_view also uses loader_life_support to ensure that
    // the string contents remain alive, but that's a C++ 17 feature.
}