1 /*
2  * Copyright (C) 2017 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 COUNT_METRIC_PRODUCER_H
18 #define COUNT_METRIC_PRODUCER_H
19 
20 #include <unordered_map>
21 
22 #include <android/util/ProtoOutputStream.h>
23 #include <gtest/gtest_prod.h>
24 #include "../anomaly/AnomalyTracker.h"
25 #include "../condition/ConditionTracker.h"
26 #include "../matchers/matcher_util.h"
27 #include "MetricProducer.h"
28 #include "frameworks/base/cmds/statsd/src/statsd_config.pb.h"
29 #include "stats_util.h"
30 
31 namespace android {
32 namespace os {
33 namespace statsd {
34 
35 struct CountBucket {
36     int64_t mBucketStartNs;
37     int64_t mBucketEndNs;
38     int64_t mCount;
39 };
40 
41 class CountMetricProducer : public MetricProducer {
42 public:
43     CountMetricProducer(const ConfigKey& key, const CountMetric& countMetric,
44                         const int conditionIndex, const sp<ConditionWizard>& wizard,
45                         const int64_t timeBaseNs, const int64_t startTimeNs);
46 
47     virtual ~CountMetricProducer();
48 
49 protected:
50     void onMatchedLogEventInternalLocked(
51             const size_t matcherIndex, const MetricDimensionKey& eventKey,
52             const ConditionKey& conditionKey, bool condition,
53             const LogEvent& event) override;
54 
55 private:
56 
57     void onDumpReportLocked(const int64_t dumpTimeNs,
58                             const bool include_current_partial_bucket,
59                             const bool erase_data,
60                             const DumpLatency dumpLatency,
61                             std::set<string> *str_set,
62                             android::util::ProtoOutputStream* protoOutput) override;
63 
64     void clearPastBucketsLocked(const int64_t dumpTimeNs) override;
65 
66     // Internal interface to handle condition change.
67     void onConditionChangedLocked(const bool conditionMet, const int64_t eventTime) override;
68 
69     // Internal interface to handle sliced condition change.
70     void onSlicedConditionMayChangeLocked(bool overallCondition, const int64_t eventTime) override;
71 
72     // Internal function to calculate the current used bytes.
73     size_t byteSizeLocked() const override;
74 
75     void dumpStatesLocked(FILE* out, bool verbose) const override;
76 
77     void dropDataLocked(const int64_t dropTimeNs) override;
78 
79     // Util function to flush the old packet.
80     void flushIfNeededLocked(const int64_t& newEventTime) override;
81 
82     void flushCurrentBucketLocked(const int64_t& eventTimeNs,
83                                   const int64_t& nextBucketStartTimeNs) override;
84 
85     std::unordered_map<MetricDimensionKey, std::vector<CountBucket>> mPastBuckets;
86 
87     // The current bucket (may be a partial bucket).
88     std::shared_ptr<DimToValMap> mCurrentSlicedCounter = std::make_shared<DimToValMap>();
89 
90     // The sum of previous partial buckets in the current full bucket (excluding the current
91     // partial bucket). This is only updated while flushing the current bucket.
92     std::shared_ptr<DimToValMap> mCurrentFullCounters = std::make_shared<DimToValMap>();
93 
94     static const size_t kBucketSize = sizeof(CountBucket{});
95 
96     bool hitGuardRailLocked(const MetricDimensionKey& newKey);
97 
98     FRIEND_TEST(CountMetricProducerTest, TestNonDimensionalEvents);
99     FRIEND_TEST(CountMetricProducerTest, TestEventsWithNonSlicedCondition);
100     FRIEND_TEST(CountMetricProducerTest, TestEventsWithSlicedCondition);
101     FRIEND_TEST(CountMetricProducerTest, TestAnomalyDetectionUnSliced);
102     FRIEND_TEST(CountMetricProducerTest, TestEventWithAppUpgrade);
103     FRIEND_TEST(CountMetricProducerTest, TestEventWithAppUpgradeInNextBucket);
104     FRIEND_TEST(CountMetricProducerTest, TestFirstBucket);
105 };
106 
107 }  // namespace statsd
108 }  // namespace os
109 }  // namespace android
110 #endif  // COUNT_METRIC_PRODUCER_H
111