File: spsc_value_stress_test.cpp

package info (click to toggle)
boost1.90 1.90.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 593,156 kB
  • sloc: cpp: 4,190,642; xml: 196,648; python: 34,618; ansic: 23,145; asm: 5,468; sh: 3,776; makefile: 1,161; perl: 1,020; sql: 728; ruby: 676; yacc: 478; java: 77; lisp: 24; csh: 6
file content (91 lines) | stat: -rw-r--r-- 2,167 bytes parent folder | download | duplicates (2)
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
//  Copyright (C) 2011-2024 Tim Blechmann
//
//  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)

#include <boost/lockfree/spsc_value.hpp>

#define BOOST_TEST_MAIN
#ifdef BOOST_LOCKFREE_INCLUDE_TESTS
#    include <boost/test/included/unit_test.hpp>
#else
#    include <boost/test/unit_test.hpp>
#endif

#include <boost/optional.hpp>

#include <atomic>
#include <cstdint>
#include <thread>

// #define BOOST_LOCKFREE_STRESS_TEST

#ifndef BOOST_LOCKFREE_STRESS_TEST
static const uint64_t nodes_per_thread = 100000;
#else
static const uint64_t nodes_per_thread = 100000000;
#endif


BOOST_AUTO_TEST_CASE( spsc_value_stress_test )
{
    boost::lockfree::spsc_value< uint64_t > v;

    std::atomic< bool > done;

    std::thread producer( [ & ] {
        for ( uint64_t i = 0; i != nodes_per_thread; ++i )
            v.write( i );
        done = true;
    } );

    boost::optional< uint64_t > consumed;
    while ( !done.load( std::memory_order_relaxed ) ) {
        uint64_t out;
        bool     read_success = v.read( out );

        if ( !read_success ) {
            std::this_thread::yield();
            continue;
        }

        if ( consumed )
            BOOST_TEST_REQUIRE( out > *consumed );
        consumed = out;
    }

    producer.join();
}

BOOST_AUTO_TEST_CASE( spsc_value_stress_test_allow_multiple_reads )
{
    boost::lockfree::spsc_value< uint64_t, boost::lockfree::allow_multiple_reads< true > > v;

    std::atomic< bool > done;

    std::thread producer( [ & ] {
        for ( uint64_t i = 0; i != nodes_per_thread; ++i ) {
            std::this_thread::yield();
            v.write( i );
        }
        done = true;
    } );

    boost::optional< uint64_t > consumed;
    while ( !done.load( std::memory_order_relaxed ) ) {
        uint64_t out {};
        bool     read_success = v.read( out );

        if ( !read_success ) {
            std::this_thread::yield();
            continue;
        }

        if ( consumed )
            BOOST_TEST_REQUIRE( out >= *consumed );
        consumed = out;
    }

    producer.join();
}