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
|
/***************************************************************************
* tests/io/test_cancel.cpp
*
* Part of the STXXL. See http://stxxl.sourceforge.net
*
* Copyright (C) 2009-2011 Johannes Singler <singler@kit.edu>
* Copyright (C) 2009 Andreas Beckmann <beckmann@cs.uni-frankfurt.de>
*
* 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 <stxxl/io>
#include <stxxl/aligned_alloc>
#include <cstring>
//! \example io/test_cancel.cpp
//! This tests the request cancelation mechanisms.
using stxxl::file;
struct print_completion
{
void operator () (stxxl::request* ptr)
{
std::cout << "Request completed: " << ptr << std::endl;
}
};
int main(int argc, char** argv)
{
if (argc < 3)
{
std::cout << "Usage: " << argv[0] << " filetype tempfile" << std::endl;
return -1;
}
const stxxl::uint64 size = 16 * 1024 * 1024, num_blocks = 16;
char* buffer = (char*)stxxl::aligned_alloc<4096>(size);
memset(buffer, 0, size);
stxxl::compat_unique_ptr<stxxl::file>::result file(
stxxl::create_file(
argv[1], argv[2],
stxxl::file::CREAT | stxxl::file::RDWR | stxxl::file::DIRECT)
);
file->set_size(num_blocks * size);
stxxl::request_ptr req[num_blocks];
//without cancelation
std::cout << "Posting " << num_blocks << " requests." << std::endl;
stxxl::stats_data stats1(*stxxl::stats::get_instance());
unsigned i = 0;
for ( ; i < num_blocks; i++)
req[i] = file->awrite(buffer, i * size, size, print_completion());
wait_all(req, num_blocks);
std::cout << stxxl::stats_data(*stxxl::stats::get_instance()) - stats1;
//with cancelation
std::cout << "Posting " << num_blocks << " requests." << std::endl;
stxxl::stats_data stats2(*stxxl::stats::get_instance());
for (unsigned i = 0; i < num_blocks; i++)
req[i] = file->awrite(buffer, i * size, size, print_completion());
//cancel first half
std::cout << "Canceling first " << num_blocks / 2 << " requests." << std::endl;
size_t num_canceled = cancel_all(req, req + num_blocks / 2);
std::cout << "Successfully canceled " << num_canceled << " requests." << std::endl;
//cancel every second in second half
for (unsigned i = num_blocks / 2; i < num_blocks; i += 2)
{
std::cout << "Canceling request " << &(*(req[i])) << std::endl;
if (req[i]->cancel())
std::cout << "Request canceled: " << &(*(req[i])) << std::endl;
else
std::cout << "Request not canceled: " << &(*(req[i])) << std::endl;
}
wait_all(req, num_blocks);
std::cout << stxxl::stats_data(*stxxl::stats::get_instance()) - stats2;
stxxl::aligned_dealloc<4096>(buffer);
return 0;
}
|
