1// Copyright 2014 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "remoting/test/leaky_bucket.h"
6
7#include "base/logging.h"
8
9namespace remoting {
10
11LeakyBucket::LeakyBucket(double depth, double rate)
12    : depth_(depth),
13      rate_(rate),
14      level_(0.0),
15      last_update_(base::TimeTicks::Now()) {
16}
17
18LeakyBucket::~LeakyBucket() {
19}
20
21base::TimeDelta LeakyBucket::AddPacket(int size) {
22  UpdateLevel();
23
24  double new_level = level_ + size;
25  if (new_level > depth_)
26    return base::TimeDelta::Max();
27
28  base::TimeDelta result = base::TimeDelta::FromSecondsD(level_ / rate_);
29  level_ = new_level;
30  return result;
31}
32
33void LeakyBucket::UpdateLevel() {
34  base::TimeTicks now = base::TimeTicks::Now();
35  level_ -= rate_ * (now - last_update_).InSecondsF();
36  if (level_ < 0.0)
37    level_ = 0.0;
38  last_update_ = now;
39}
40
41}  // namespace remoting
42