mirror of
https://github.com/facebookincubator/mvfst.git
synced 2025-04-18 17:24:03 +03:00
63ee419d75
Summary: ``` find quic -name \*.h -o -name \*.cpp | xargs sed -i'' -E "s/#include \"(.*)\"/#include <\1>/" arc lint -a --paths-cmd 'hg files quic/' ``` Reviewed By: mjoras Differential Revision: D34199869 fbshipit-source-id: 3633cb8429b86c03ab367211d6a46dbe4fdd5ff2
48 lines
1.7 KiB
C++
48 lines
1.7 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include <quic/server/SlidingWindowRateLimiter.h>
|
|
|
|
#include <cmath>
|
|
|
|
namespace quic {
|
|
|
|
bool SlidingWindowRateLimiter::check(TimePoint time) {
|
|
// This is the first time point.
|
|
if (!currentWindowStartPoint_) {
|
|
currentWindowStartPoint_ = time;
|
|
}
|
|
auto timeElapsedSinceCurWindow =
|
|
std::chrono::duration_cast<std::chrono::microseconds>(
|
|
time - currentWindowStartPoint_.value());
|
|
// A full window has elapsed.
|
|
if (timeElapsedSinceCurWindow > window_) {
|
|
double windowsElapsed = timeElapsedSinceCurWindow.count() / window_.count();
|
|
currentWindowStartPoint_.value() +=
|
|
std::chrono::microseconds(uint64_t(window_.count() * windowsElapsed));
|
|
// If more than one window has elapsed, there were zero in the previous
|
|
// window.
|
|
countInPrevWindow_ = windowsElapsed == 1 ? countInCurWindow_ : 0;
|
|
countInCurWindow_ = 0;
|
|
timeElapsedSinceCurWindow =
|
|
std::chrono::duration_cast<std::chrono::microseconds>(
|
|
time - currentWindowStartPoint_.value());
|
|
}
|
|
// The weighted count accounts for the "sliding" window by using the
|
|
// previous windows average rate over the time that has elapsed in the
|
|
// current window.
|
|
double weightedCount = countInPrevWindow_ *
|
|
(static_cast<double>((window_ - timeElapsedSinceCurWindow).count()) /
|
|
window_.count()) +
|
|
countInCurWindow_ + 1;
|
|
bool limited = std::trunc(weightedCount) > count_();
|
|
countInCurWindow_ = limited ? countInCurWindow_ : countInCurWindow_ + 1;
|
|
return limited;
|
|
}
|
|
|
|
} // namespace quic
|