15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/base/rate_counter.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/logging.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)RateCounter::RateCounter(base::TimeDelta time_window) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : time_window_(time_window), 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sum_(0) { 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_GT(time_window.InMilliseconds(), 0); 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)RateCounter::~RateCounter() { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RateCounter::Record(int64 value) { 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Time current_time = CurrentTime(); 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EvictOldDataPoints(current_time); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sum_ += value; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_points_.push(std::make_pair(current_time, value)); 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)double RateCounter::Rate() { 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EvictOldDataPoints(CurrentTime()); 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return sum_ / time_window_.InSecondsF(); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void RateCounter::SetCurrentTimeForTest(base::Time current_time) { 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(current_time >= current_time_for_test_); 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) current_time_for_test_ = current_time; 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void RateCounter::EvictOldDataPoints(base::Time current_time) { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Remove data points outside of the window. 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time window_start = current_time - time_window_; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (!data_points_.empty()) { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (data_points_.front().first > window_start) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sum_ -= data_points_.front().second; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_points_.pop(); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::Time RateCounter::CurrentTime() const { 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (current_time_for_test_ == base::Time()) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::Time::Now(); 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return current_time_for_test_; 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 63