docs: cmake build
[m6w6/libmemcached] / libtest / timer.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #include "libtest/yatlcon.h"
38
39 #include <libtest/timer.hpp>
40
41 #include <ctime>
42 #include <iomanip>
43
44 #ifdef __MACH__
45 # include <mach/clock.h>
46 # include <mach/mach.h>
47 #else
48 # include <sys/time.h>
49 #endif
50
51 namespace libtest {
52
53 Timer::Timer()
54 {
55 _begin.tv_sec= 0;
56 _begin.tv_nsec= 0;
57 _end.tv_sec= 0;
58 _end.tv_nsec= 0;
59 }
60
61 void Timer::reset()
62 {
63 _end.tv_sec= 0;
64 _end.tv_nsec= 0;
65 _time(_begin);
66 }
67
68 void Timer::sample()
69 {
70 _time(_end);
71 }
72
73 void Timer::offset(int64_t minutes_arg, int64_t seconds_arg, int64_t nanoseconds)
74 {
75 reset();
76 _end= _begin;
77 _end.tv_sec+= (minutes_arg * 60) +seconds_arg;
78 _end.tv_nsec+= nanoseconds;
79 }
80
81 int64_t Timer::minutes()
82 {
83 struct timespec result;
84 difference(result);
85 return int64_t(result.tv_sec / 60);
86 }
87
88 uint64_t Timer::elapsed_milliseconds() const
89 {
90 struct timespec temp;
91 difference(temp);
92
93 return temp.tv_sec*1000 +temp.tv_nsec/1000000;
94 }
95
96 void Timer::difference(struct timespec& arg) const
97 {
98 if ((_end.tv_nsec -_begin.tv_nsec) < 0)
99 {
100 arg.tv_sec= _end.tv_sec -_begin.tv_sec -1;
101 arg.tv_nsec= 1000000000 +_end.tv_nsec -_begin.tv_nsec;
102
103 }
104 else
105 {
106 arg.tv_sec= _end.tv_sec -_begin.tv_sec;
107 arg.tv_nsec= _end.tv_nsec -_begin.tv_nsec;
108 }
109 }
110
111 void Timer::_time(struct timespec& ts)
112 {
113 #ifdef __MACH__ // OSX lacks clock_gettime()
114 clock_serv_t _clock_serv;
115 mach_timespec_t _mach_timespec;
116 host_get_clock_service(mach_host_self(), CALENDAR_CLOCK, &_clock_serv);
117 clock_get_time(_clock_serv, &_mach_timespec);
118 mach_port_deallocate(mach_task_self(), _clock_serv);
119 ts.tv_sec= _mach_timespec.tv_sec;
120 ts.tv_nsec= _mach_timespec.tv_nsec;
121 #elif defined(_WIN32)
122 ts.tv_sec= time(NULL);
123 ts.tv_nsec= 0;
124 #else
125 clock_gettime(CLOCK_REALTIME, &ts);
126 #endif
127 }
128
129 std::ostream& operator<<(std::ostream& output, const libtest::Timer& arg)
130 {
131 struct timespec temp;
132 arg.difference(temp);
133
134 if (temp.tv_sec > 60)
135 {
136 output << temp.tv_sec / 60;
137 output << "." << temp.tv_sec % 60;
138 }
139 else
140 {
141 output << temp.tv_sec;
142 }
143
144 output << ":";
145 output << std::setfill('0') << std::setw(9) << temp.tv_nsec;
146
147 return output;
148 }
149
150 } // namespace libtest