wbtiger
/
foundationdb
mirror of https://github.com/apple/foundationdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

prevent int overflow

This commit is contained in:
Markus Pilman 2022-09-19 14:01:12 -06:00
parent bd8347d92e
commit 441ea31fca
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
flow/include/flow/IRateControl.h
View File

@ -53,7 +53,13 @@ public:
// Replenish budget based on time since last update
double ts = now();
// returnUnused happens to do exactly what we want here
returnUnused((ts - m_last_update) / m_seconds * m_limit);
auto unused = double(m_limit) * (ts - m_last_update) / m_seconds;
if (unused >= double(std::numeric_limits<int>::max())) {
// prevent int overflow
m_budget = m_limit;
} else {
returnUnused(int(std::min(unused, double(m_limit))));
}
m_last_update = ts;
m_budget -= n;
// If budget is still >= 0 then it's safe to use the allowance right now.