File: equal_range_transparent.pass.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 (120 lines) | stat: -rw-r--r-- 2,605 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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Copyright 2020, Intel Corporation
//
// Modified to test pmem::obj containers
//

// <map>

// class map

// template<typename K>
//         pair<iterator,iterator>             equal_range(const K& x); // C++14
// template<typename K>
//         pair<const_iterator,const_iterator> equal_range(const K& x) const;
//         // C++14

#include "unittest.hpp"

#include <libpmemobj++/experimental/concurrent_map.hpp>
#include <libpmemobj++/make_persistent.hpp>
#include <libpmemobj++/persistent_ptr.hpp>
#include <libpmemobj++/pool.hpp>
#include <libpmemobj++/transaction.hpp>

struct Comp;
namespace nvobj = pmem::obj;
namespace nvobjex = pmem::obj::experimental;
using C = nvobjex::concurrent_map<std::pair<int, int>, int, Comp>;

struct root {
	nvobj::persistent_ptr<C> s;
};

struct Comp {
	using is_transparent = void;

	bool
	operator()(const std::pair<int, int> &lhs,
		   const std::pair<int, int> &rhs) const
	{
		return lhs < rhs;
	}

	bool
	operator()(const std::pair<int, int> &lhs, int rhs) const
	{
		return lhs.first < rhs;
	}

	bool
	operator()(int lhs, const std::pair<int, int> &rhs) const
	{
		return lhs < rhs.first;
	}
};

int
run(pmem::obj::pool<root> &pop)
{
	auto r = pop.root();
	pmem::obj::transaction::run(pop, [&] {
		r->s = nvobj::make_persistent<C>(
			std::initializer_list<typename C::value_type>{
				{{2, 1}, 1},
				{{1, 2}, 2},
				{{1, 3}, 3},
				{{1, 4}, 4},
				{{2, 2}, 5}});
	});
	auto &s = *r->s;
	auto er = s.equal_range(1);
	long nels = 0;

	for (auto it = er.first; it != er.second; it++) {
		UT_ASSERTeq(it->first.first, 1);
		nels++;
	}

	UT_ASSERTeq(nels, 3);
	pmem::obj::transaction::run(pop,
				    [&] { nvobj::delete_persistent<C>(r->s); });
	return 0;
}

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

	const char *path = argv[1];

	pmem::obj::pool<root> pop;
	try {
		pop = pmem::obj::pool<root>::create(
			path, "equal_range_transparent.pass", PMEMOBJ_MIN_POOL,
			S_IWUSR | S_IRUSR);
	} catch (...) {
		UT_FATAL("!pmemobj_create: %s", path);
	}
	try {
		run(pop);
		pop.close();
	} catch (std::exception &e) {
		UT_FATALexc(e);
	}
}

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