jenkins-promote-staging-trunk-libmemcached-16
[awesomized/libmemcached] / libtest / timer.hpp
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 #pragma once
38
39 #include <ctime>
40 #include <ostream>
41
42 #ifdef __MACH__
43 # include <mach/clock.h>
44 # include <mach/mach.h>
45 #else
46 # include <sys/time.h>
47 #endif
48
49
50 namespace libtest {
51
52 class Timer {
53 public:
54
55 Timer()
56 {
57 _begin.tv_sec= 0;
58 _begin.tv_nsec= 0;
59 _end.tv_sec= 0;
60 _end.tv_nsec= 0;
61 }
62
63 void reset()
64 {
65 _end.tv_sec= 0;
66 _end.tv_nsec= 0;
67 _time(_begin);
68 }
69
70 void sample()
71 {
72 _time(_end);
73 }
74
75 void difference(struct timespec& arg) const
76 {
77 if ((_end.tv_nsec -_begin.tv_nsec) < 0)
78 {
79 arg.tv_sec= _end.tv_sec -_begin.tv_sec-1;
80 arg.tv_nsec= 1000000000 +_end.tv_nsec -_begin.tv_nsec;
81
82 }
83 else
84 {
85 arg.tv_sec= _end.tv_sec -_begin.tv_sec;
86 arg.tv_nsec= _end.tv_nsec -_begin.tv_nsec;
87 }
88 }
89
90 private:
91 void _time(struct timespec& ts)
92 {
93 #ifdef __MACH__ // OSX lacks clock_gettime()
94 clock_serv_t _clock_serv;
95 mach_timespec_t _mach_timespec;
96 host_get_clock_service(mach_host_self(), CALENDAR_CLOCK, &_clock_serv);
97 clock_get_time(_clock_serv, &_mach_timespec);
98 mach_port_deallocate(mach_task_self(), _clock_serv);
99 ts.tv_sec= _mach_timespec.tv_sec;
100 ts.tv_nsec= _mach_timespec.tv_nsec;
101 #else
102 clock_gettime(CLOCK_REALTIME, &ts);
103 #endif
104 }
105
106 private:
107 struct timespec _begin;
108 struct timespec _end;
109 };
110
111 std::ostream& operator<<(std::ostream& output, const libtest::Timer& arg);
112
113 } // namespace libtest