Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 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 SIMPLE_CONDITION_TRACKER_H |
| 18 | #define SIMPLE_CONDITION_TRACKER_H |
| 19 | |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 20 | #include <gtest/gtest_prod.h> |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 21 | #include "ConditionTracker.h" |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 22 | #include "config/ConfigKey.h" |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 23 | #include "frameworks/base/cmds/statsd/src/statsd_config.pb.h" |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 24 | #include "stats_util.h" |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 25 | |
| 26 | namespace android { |
| 27 | namespace os { |
| 28 | namespace statsd { |
| 29 | |
| 30 | class SimpleConditionTracker : public virtual ConditionTracker { |
| 31 | public: |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame^] | 32 | SimpleConditionTracker(const ConfigKey& key, const int64_t& id, const int index, |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 33 | const SimplePredicate& simplePredicate, |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame^] | 34 | const std::unordered_map<int64_t, int>& trackerNameIndexMap); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 35 | |
| 36 | ~SimpleConditionTracker(); |
| 37 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 38 | bool init(const std::vector<Predicate>& allConditionConfig, |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 39 | const std::vector<sp<ConditionTracker>>& allConditionTrackers, |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame^] | 40 | const std::unordered_map<int64_t, int>& conditionIdIndexMap, |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 41 | std::vector<bool>& stack) override; |
| 42 | |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 43 | void evaluateCondition(const LogEvent& event, |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 44 | const std::vector<MatchingState>& eventMatcherValues, |
| 45 | const std::vector<sp<ConditionTracker>>& mAllConditions, |
| 46 | std::vector<ConditionState>& conditionCache, |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 47 | std::vector<bool>& changedCache) override; |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 48 | |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 49 | void isConditionMet(const ConditionKey& conditionParameters, |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 50 | const std::vector<sp<ConditionTracker>>& allConditions, |
Yangster | 7c334a1 | 2017-11-22 14:24:24 -0800 | [diff] [blame] | 51 | std::vector<ConditionState>& conditionCache) const override; |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 52 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 53 | private: |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 54 | const ConfigKey mConfigKey; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 55 | // The index of the LogEventMatcher which defines the start. |
| 56 | int mStartLogMatcherIndex; |
| 57 | |
| 58 | // The index of the LogEventMatcher which defines the end. |
| 59 | int mStopLogMatcherIndex; |
| 60 | |
| 61 | // if the start end needs to be nested. |
| 62 | bool mCountNesting; |
| 63 | |
| 64 | // The index of the LogEventMatcher which defines the stop all. |
| 65 | int mStopAllLogMatcherIndex; |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 66 | |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 67 | ConditionState mInitialValue; |
Yao Chen | 729093d | 2017-10-16 10:33:26 -0700 | [diff] [blame] | 68 | |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 69 | FieldMatcher mOutputDimensions; |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 70 | |
| 71 | std::map<HashableDimensionKey, int> mSlicedConditionState; |
| 72 | |
| 73 | void handleStopAll(std::vector<ConditionState>& conditionCache, |
| 74 | std::vector<bool>& changedCache); |
| 75 | |
| 76 | void handleConditionEvent(const HashableDimensionKey& outputKey, bool matchStart, |
| 77 | std::vector<ConditionState>& conditionCache, |
| 78 | std::vector<bool>& changedCache); |
| 79 | |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 80 | bool hitGuardRail(const HashableDimensionKey& newKey); |
| 81 | |
Yao Chen | 967b205 | 2017-11-07 16:36:43 -0800 | [diff] [blame] | 82 | FRIEND_TEST(SimpleConditionTrackerTest, TestSlicedCondition); |
| 83 | FRIEND_TEST(SimpleConditionTrackerTest, TestSlicedWithNoOutputDim); |
| 84 | FRIEND_TEST(SimpleConditionTrackerTest, TestStopAll); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 85 | }; |
| 86 | |
| 87 | } // namespace statsd |
| 88 | } // namespace os |
| 89 | } // namespace android |
| 90 | |
| 91 | #endif // SIMPLE_CONDITION_TRACKER_H |