| Index: remoting/test/leaky_bucket.cc
|
| diff --git a/remoting/test/leaky_bucket.cc b/remoting/test/leaky_bucket.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ee8a62b06de136d82c2a040c5abb5706f2d9f06e
|
| --- /dev/null
|
| +++ b/remoting/test/leaky_bucket.cc
|
| @@ -0,0 +1,41 @@
|
| +// Copyright 2014 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 "remoting/test/leaky_bucket.h"
|
| +
|
| +#include "base/logging.h"
|
| +
|
| +namespace remoting {
|
| +
|
| +LeakyBucket::LeakyBucket(double depth, double rate)
|
| + : depth_(depth),
|
| + rate_(rate),
|
| + level_(0.0),
|
| + last_update_(base::TimeTicks::Now()) {
|
| +}
|
| +
|
| +LeakyBucket::~LeakyBucket() {
|
| +}
|
| +
|
| +base::TimeDelta LeakyBucket::AddPacket(int size) {
|
| + UpdateLevel();
|
| +
|
| + double new_level = level_ + size;
|
| + if (new_level > depth_)
|
| + return base::TimeDelta::Max();
|
| +
|
| + base::TimeDelta result = base::TimeDelta::FromSecondsD(level_ / rate_);
|
| + level_ = new_level;
|
| + return result;
|
| +}
|
| +
|
| +void LeakyBucket::UpdateLevel() {
|
| + base::TimeTicks now = base::TimeTicks::Now();
|
| + level_ -= rate_ * (now - last_update_).InSecondsF();
|
| + if (level_ < 0.0)
|
| + level_ = 0.0;
|
| + last_update_ = now;
|
| +}
|
| +
|
| +} // namespace remoting
|
|
|