Merge in util update
[awesomized/libmemcached] / libtest / wait.h
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * libtest
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 3 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22
23 #pragma once
24
25 #include <unistd.h>
26 #include <string>
27
28 namespace libtest {
29
30 class Wait
31 {
32 public:
33
34 Wait(const std::string &filename, uint32_t timeout= 6) :
35 _successful(false)
36 {
37 uint32_t waited;
38 uint32_t this_wait;
39 uint32_t retry;
40
41 if (filename.empty())
42 {
43 _successful= false;
44 return;
45 }
46
47 for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
48 {
49 if (access(filename.c_str(), R_OK) == 0)
50 {
51 _successful= true;
52 break;
53 }
54 else if (waited >= timeout)
55 {
56 break;
57 }
58
59 this_wait= retry * retry / 3 + 1;
60 sleep(this_wait);
61 #ifdef WIN32
62 sleep(this_wait);
63 #else
64 struct timespec global_sleep_value= { this_wait, 0 };
65 nanosleep(&global_sleep_value, NULL);
66 #endif
67 }
68 }
69
70 bool successful() const
71 {
72 return _successful;
73 }
74
75 private:
76 bool _successful;
77 };
78
79 } // namespace libtest