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
|
// SPDX-License-Identifier: BSD-3-Clause
/* Copyright 2019-2020, Intel Corporation */
#include "unittest.hpp"
#include <libpmemobj++/detail/enumerable_thread_specific.hpp>
#include <libpmemobj++/make_persistent.hpp>
namespace nvobj = pmem::obj;
using container_type = pmem::detail::enumerable_thread_specific<size_t>;
struct root {
nvobj::persistent_ptr<container_type> pptr;
};
void
test(nvobj::pool<struct root> &pop)
{
// Adding more concurrency will increase DRD test time
const size_t concurrency = 16;
auto tls = pop.root()->pptr;
UT_ASSERT(tls != nullptr);
parallel_exec(concurrency, [&](size_t thread_index) {
tls->local() = 99;
pop.persist(&tls->local(), sizeof(tls->local()));
});
UT_ASSERT(tls->size() <= concurrency);
container_type &ref = *tls;
for (auto &e : ref) {
UT_ASSERT(e == 99);
}
}
static void
test(int argc, char *argv[])
{
if (argc < 2) {
UT_FATAL("usage: %s file-name", argv[0]);
}
auto path = argv[1];
auto pop = nvobj::pool<root>::create(
path, "TLSTest: enumerable_thread_specific_iterators",
PMEMOBJ_MIN_POOL, S_IWUSR | S_IRUSR);
auto r = pop.root();
try {
nvobj::transaction::run(pop, [&] {
r->pptr = nvobj::make_persistent<container_type>();
});
test(pop);
nvobj::transaction::run(pop, [&] {
nvobj::delete_persistent<container_type>(r->pptr);
});
} catch (std::exception &e) {
UT_FATALexc(e);
}
pop.close();
}
int
main(int argc, char *argv[])
{
return run_test([&] { test(argc, argv); });
}
|