63 lines
1.4 KiB
C++
63 lines
1.4 KiB
C++
// Copyright 2017 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "time_utils.h"
|
|
|
|
#include <sys/time.h>
|
|
#include <sys/timerfd.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "logging.h"
|
|
|
|
namespace time_utils {
|
|
|
|
uint64_t GetTimestamp() {
|
|
struct timespec ts = {};
|
|
CHECK(clock_gettime(CLOCK_MONOTONIC_COARSE, &ts) == 0);
|
|
return ts.tv_sec * 1000 + ts.tv_nsec / 1000000ul;
|
|
}
|
|
|
|
PeriodicTimer::PeriodicTimer(int interval_ms) : interval_ms_(interval_ms) {
|
|
timer_fd_ = -1;
|
|
}
|
|
|
|
PeriodicTimer::~PeriodicTimer() {
|
|
Stop();
|
|
}
|
|
|
|
void PeriodicTimer::Start() {
|
|
Stop();
|
|
timer_fd_ = timerfd_create(CLOCK_MONOTONIC, 0);
|
|
CHECK(timer_fd_ >= 0);
|
|
int sec = interval_ms_ / 1000;
|
|
int nsec = (interval_ms_ % 1000) * 1000000;
|
|
struct itimerspec ts = {};
|
|
ts.it_value.tv_nsec = nsec;
|
|
ts.it_value.tv_sec = sec;
|
|
ts.it_interval.tv_nsec = nsec;
|
|
ts.it_interval.tv_sec = sec;
|
|
CHECK(timerfd_settime(timer_fd_, 0, &ts, nullptr) == 0);
|
|
}
|
|
|
|
void PeriodicTimer::Stop() {
|
|
if (timer_fd_ < 0)
|
|
return;
|
|
close(timer_fd_);
|
|
timer_fd_ = -1;
|
|
}
|
|
|
|
bool PeriodicTimer::Wait() {
|
|
if (timer_fd_ < 0)
|
|
return false; // Not started yet.
|
|
uint64_t stub = 0;
|
|
int res = read(timer_fd_, &stub, sizeof(stub));
|
|
if (res < 0 && errno == EBADF)
|
|
return false; // Interrupted by Stop().
|
|
CHECK(res > 0);
|
|
return true;
|
|
}
|
|
|
|
} // namespace time_utils
|