14e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin/* 24e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * Copyright (C) 2018 The Android Open Source Project 34e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * 44e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * Licensed under the Apache License, Version 2.0 (the "License"); 54e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * you may not use this file except in compliance with the License. 64e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * You may obtain a copy of the License at 74e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * 84e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * http://www.apache.org/licenses/LICENSE-2.0 94e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * 104e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * Unless required by applicable law or agreed to in writing, software 114e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * distributed under the License is distributed on an "AS IS" BASIS, 124e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 134e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * See the License for the specific language governing permissions and 144e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin * limitations under the License. 154e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin */ 164e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin#define DEBUG false 174e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin#include "Log.h" 184e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 194e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin#include "Throttler.h" 204e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 214e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin#include <utils/SystemClock.h> 224e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 236cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jinnamespace android { 246cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jinnamespace os { 256cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jinnamespace incidentd { 266cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jin 274e843106a6f510a7d77fedfbeaec8e95b57905b7Yi JinThrottler::Throttler(size_t limit, int64_t refractoryPeriodMs) 284e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin : mSizeLimit(limit), 294e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mRefractoryPeriodMs(refractoryPeriodMs), 304e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mAccumulatedSize(0), 314e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mLastRefractoryMs(android::elapsedRealtime()) {} 324e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 334e843106a6f510a7d77fedfbeaec8e95b57905b7Yi JinThrottler::~Throttler() {} 344e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 354e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jinbool Throttler::shouldThrottle() { 364e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin int64_t now = android::elapsedRealtime(); 374e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin if (now > mRefractoryPeriodMs + mLastRefractoryMs) { 384e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mLastRefractoryMs = now; 394e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mAccumulatedSize = 0; 404e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin } 414e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin return mAccumulatedSize > mSizeLimit; 424e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin} 434e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 444e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jinvoid Throttler::addReportSize(size_t reportByteSize) { 454e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin VLOG("The current request took %d bytes to dropbox", (int)reportByteSize); 464e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin mAccumulatedSize += reportByteSize; 474e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin} 484e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin 494e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jinvoid Throttler::dump(FILE* out) { 504e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin fprintf(out, "mSizeLimit=%d\n", (int)mSizeLimit); 514e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin fprintf(out, "mAccumulatedSize=%d\n", (int)mAccumulatedSize); 524e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin fprintf(out, "mRefractoryPeriodMs=%d\n", (int)mRefractoryPeriodMs); 534e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin fprintf(out, "mLastRefractoryMs=%d\n", (int)mLastRefractoryMs); 544e843106a6f510a7d77fedfbeaec8e95b57905b7Yi Jin} 556cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jin 566cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jin} // namespace incidentd 576cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jin} // namespace os 586cacbcbf436be744a34f7ea0d4f838ff97757446Yi Jin} // namespace android