File: test_thread.hpp

package info (click to toggle)
boost1.90 1.90.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 593,168 kB
  • sloc: cpp: 4,190,642; xml: 196,648; python: 34,618; ansic: 23,145; asm: 5,468; sh: 3,776; makefile: 1,162; perl: 1,020; sql: 728; ruby: 676; yacc: 478; java: 77; lisp: 24; csh: 6
file content (83 lines) | stat: -rw-r--r-- 1,962 bytes parent folder | download
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
//  Copyright (c) 2023-2025 Andrey Semashev
//
//  Distributed under the Boost Software License, Version 1.0.
//  See accompanying file LICENSE_1_0.txt or copy at
//  http://www.boost.org/LICENSE_1_0.txt)

#ifndef BOOST_ATOMIC_TEST_THREAD_HPP_INCLUDED_
#define BOOST_ATOMIC_TEST_THREAD_HPP_INCLUDED_

#include <chrono>
#include <thread>
#include <mutex>
#include <condition_variable>

//! Test thread class with the ability to join the thread with a timeout
class test_thread
{
private:
    std::mutex m_mutex;
    std::condition_variable m_cond;
    bool m_finished;
    std::thread m_thread;

public:
    template< typename Func >
    explicit test_thread(Func&& func) :
        m_finished(false),
        m_thread([this, func]()
        {
            try
            {
                func();
            }
            catch (...)
            {
                mark_finished();
                throw;
            }
            mark_finished();
        })
    {
    }

    test_thread(test_thread const&) = delete;
    test_thread& operator= (test_thread const&) = delete;

    void join()
    {
        m_thread.join();
    }

    template< typename Rep, typename Period >
    bool try_join_for(std::chrono::duration< Rep, Period > dur)
    {
        return try_join_until(std::chrono::steady_clock::now() + dur);
    }

    template< typename Clock, typename Duration >
    bool try_join_until(std::chrono::time_point< Clock, Duration > timeout)
    {
        {
            std::unique_lock< std::mutex > lock(m_mutex);
            while (!m_finished)
            {
                if (m_cond.wait_until(lock, timeout) == std::cv_status::timeout)
                    return false;
            }
        }

        join();
        return true;
    }

private:
    void mark_finished()
    {
        std::lock_guard< std::mutex > lock(m_mutex);
        m_finished = true;
        m_cond.notify_all();
    }
};

#endif // BOOST_ATOMIC_TEST_THREAD_HPP_INCLUDED_