Home | History | Annotate | Download | only in src
      1 /*
      2  * Copyright (C) 2018 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #ifndef THROTTLER_H
     18 #define THROTTLER_H
     19 
     20 #include <utils/RefBase.h>
     21 
     22 #include <unistd.h>
     23 
     24 namespace android {
     25 namespace os {
     26 namespace incidentd {
     27 /**
     28  * This is a size-based throttler which prevents incidentd to take more data.
     29  */
     30 class Throttler : public virtual android::RefBase {
     31 public:
     32     Throttler(size_t limit, int64_t refractoryPeriodMs);
     33     ~Throttler();
     34 
     35     /**
     36      * Asserts this before starting taking report.
     37      */
     38     bool shouldThrottle();
     39 
     40     void addReportSize(size_t reportByteSize);
     41 
     42     void dump(FILE* out);
     43 
     44 private:
     45     const size_t mSizeLimit;
     46     const int64_t mRefractoryPeriodMs;
     47 
     48     size_t mAccumulatedSize;
     49     int64_t mLastRefractoryMs;
     50 };
     51 
     52 }  // namespace incidentd
     53 }  // namespace os
     54 }  // namespace android
     55 
     56 #endif  // THROTTLER_H
     57