File: self_relative_ptr_atomic_pmem.cpp

package info (click to toggle)
libpmemobj-cpp 1.12-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 8,388 kB
  • sloc: cpp: 136,076; sh: 1,022; perl: 381; ansic: 163; makefile: 13
file content (94 lines) | stat: -rw-r--r-- 2,066 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
84
85
86
87
88
89
90
91
92
93
94
// SPDX-License-Identifier: BSD-3-Clause
/* Copyright 2020, Intel Corporation */

#include "thread_helpers.hpp"
#include "unittest.hpp"

#include <libpmemobj++/experimental/atomic_self_relative_ptr.hpp>
#include <libpmemobj++/experimental/self_relative_ptr.hpp>
#include <libpmemobj++/make_persistent.hpp>
#include <libpmemobj++/make_persistent_array_atomic.hpp>
#include <libpmemobj++/make_persistent_atomic.hpp>
#include <libpmemobj++/p.hpp>
#include <libpmemobj++/persistent_ptr.hpp>
#include <libpmemobj++/persistent_ptr_base.hpp>
#include <libpmemobj++/pool.hpp>
#include <libpmemobj++/transaction.hpp>

#define LAYOUT "cpp"

namespace nvobj = pmem::obj;

template <typename T>
using self_relative_ptr = pmem::obj::experimental::self_relative_ptr<T>;
template <typename T>
using atomic_ptr = std::atomic<self_relative_ptr<T>>;

constexpr int ARR_SIZE = 10000;

struct root {
	atomic_ptr<nvobj::p<int>[ARR_SIZE]> parr;
	atomic_ptr<int> ptr;
};

namespace
{

void
test_ptr_transactional(nvobj::pool<root> &pop)
{
	auto r = pop.root();
	try {
		nvobj::transaction::run(pop, [&] {
			UT_ASSERT(r->ptr.load() == nullptr);
			nvobj::transaction::snapshot<atomic_ptr<int>>(&r->ptr);
			r->ptr = nvobj::make_persistent<int>();
		});
	} catch (...) {
		UT_ASSERT(0);
	}

	UT_ASSERT(r->ptr.load() != nullptr);

	try {
		nvobj::transaction::run(pop, [&] {
			nvobj::delete_persistent<int>(r->ptr.load());
			nvobj::transaction::snapshot<atomic_ptr<int>>(&r->ptr);
			r->ptr.store(nullptr);
		});
	} catch (...) {
		UT_ASSERT(0);
	}

	UT_ASSERT(r->ptr.load() == nullptr);
}

} /* namespace */

static void
test(int argc, char *argv[])
{
	if (argc != 2)
		UT_FATAL("usage: %s file-name", argv[0]);

	const char *path = argv[1];

	nvobj::pool<root> pop;

	try {
		pop = nvobj::pool<root>::create(path, LAYOUT, PMEMOBJ_MIN_POOL,
						S_IWUSR | S_IRUSR);
	} catch (pmem::pool_error &pe) {
		UT_FATAL("!pool::create: %s %s", pe.what(), path);
	}

	test_ptr_transactional(pop);

	pop.close();
}

int
main(int argc, char *argv[])
{
	return run_test([&] { test(argc, argv); });
}