Merging bzr://gaz.tangent.org/libmemcached/build/ to Build branch
[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 #include <signal.h>
28
29 #include <libtest/dream.h>
30
31 namespace libtest {
32
33 class Wait
34 {
35 public:
36
37 Wait(const std::string &filename, uint32_t timeout= 6) :
38 _successful(false)
39 {
40 uint32_t waited;
41 uint32_t this_wait;
42 uint32_t retry;
43
44 if (filename.empty())
45 {
46 _successful= false;
47 return;
48 }
49
50 for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
51 {
52 if (access(filename.c_str(), R_OK) == 0)
53 {
54 _successful= true;
55 break;
56 }
57 else if (waited >= timeout)
58 {
59 break;
60 }
61
62 this_wait= retry * retry / 3 + 1;
63 libtest::dream(this_wait, 0);
64 }
65 }
66
67 Wait(const pid_t &_pid_arg, uint32_t timeout= 6) :
68 _successful(false)
69 {
70 uint32_t waited;
71 uint32_t this_wait;
72 uint32_t retry;
73
74 for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
75 {
76 if (kill(_pid_arg, 0) == 0)
77 {
78 _successful= true;
79 break;
80 }
81 else if (waited >= timeout)
82 {
83 break;
84 }
85
86 this_wait= retry * retry / 3 + 1;
87 libtest::dream(this_wait, 0);
88 }
89 }
90
91 bool successful() const
92 {
93 return _successful;
94 }
95
96 private:
97 bool _successful;
98 };
99
100 } // namespace libtest