ece5f705d5
When StatsCompanion informs statsd that an anomaly alarm has fired, statsd will need to check to see whether an anomaly has indeed occurred. This will require determining which anomaly alarms have times in the past (since that indicates that the anomaly did indeed occur). This capability is now added. Test: adb shell data/nativetest64/statsd_test/statsd_test Change-Id: I5464c94634df70832d2723d8c4718277776d58aa
138 lines
4.7 KiB
C++
138 lines
4.7 KiB
C++
/*
|
|
* Copyright (C) 2017 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#define LOG_TAG "AnomalyMonitor"
|
|
#define DEBUG true
|
|
|
|
#include "AnomalyMonitor.h"
|
|
|
|
#include <cutils/log.h>
|
|
|
|
namespace android {
|
|
namespace os {
|
|
namespace statsd {
|
|
|
|
AnomalyMonitor::AnomalyMonitor(uint32_t minDiffToUpdateRegisteredAlarmTimeSec)
|
|
: mRegisteredAlarmTimeSec(0), mMinUpdateTimeSec(minDiffToUpdateRegisteredAlarmTimeSec) {
|
|
}
|
|
|
|
AnomalyMonitor::~AnomalyMonitor() {
|
|
}
|
|
|
|
void AnomalyMonitor::setStatsCompanionService(sp<IStatsCompanionService> statsCompanionService) {
|
|
std::lock_guard<std::mutex> lock(mLock);
|
|
sp<IStatsCompanionService> tmpForLock = mStatsCompanionService;
|
|
mStatsCompanionService = statsCompanionService;
|
|
if (statsCompanionService == nullptr) {
|
|
if (DEBUG) ALOGD("Erasing link to statsCompanionService");
|
|
return;
|
|
}
|
|
if (DEBUG) ALOGD("Creating link to statsCompanionService");
|
|
const sp<const AnomalyAlarm> top = mPq.top();
|
|
if (top != nullptr) {
|
|
updateRegisteredAlarmTime_l(top->timestampSec);
|
|
}
|
|
}
|
|
|
|
void AnomalyMonitor::add(sp<const AnomalyAlarm> alarm) {
|
|
std::lock_guard<std::mutex> lock(mLock);
|
|
if (alarm == nullptr) {
|
|
ALOGW("Asked to add a null alarm.");
|
|
return;
|
|
}
|
|
if (alarm->timestampSec < 1) {
|
|
// forbidden since a timestamp 0 is used to indicate no alarm registered
|
|
ALOGW("Asked to add a 0-time alarm.");
|
|
return;
|
|
}
|
|
// TODO: Ensure that refractory period is respected.
|
|
if (DEBUG) ALOGD("Adding alarm with time %u", alarm->timestampSec);
|
|
mPq.push(alarm);
|
|
if (mRegisteredAlarmTimeSec < 1 ||
|
|
alarm->timestampSec + mMinUpdateTimeSec < mRegisteredAlarmTimeSec) {
|
|
updateRegisteredAlarmTime_l(alarm->timestampSec);
|
|
}
|
|
}
|
|
|
|
void AnomalyMonitor::remove(sp<const AnomalyAlarm> alarm) {
|
|
std::lock_guard<std::mutex> lock(mLock);
|
|
if (alarm == nullptr) {
|
|
ALOGW("Asked to remove a null alarm.");
|
|
return;
|
|
}
|
|
if (DEBUG) ALOGD("Removing alarm with time %u", alarm->timestampSec);
|
|
mPq.remove(alarm);
|
|
if (mPq.empty()) {
|
|
if (DEBUG) ALOGD("Queue is empty. Cancel any alarm.");
|
|
mRegisteredAlarmTimeSec = 0;
|
|
if (mStatsCompanionService != nullptr) {
|
|
mStatsCompanionService->cancelAnomalyAlarm();
|
|
}
|
|
return;
|
|
}
|
|
uint32_t soonestAlarmTimeSec = mPq.top()->timestampSec;
|
|
if (DEBUG) ALOGD("Soonest alarm is %u", soonestAlarmTimeSec);
|
|
if (soonestAlarmTimeSec > mRegisteredAlarmTimeSec + mMinUpdateTimeSec) {
|
|
updateRegisteredAlarmTime_l(soonestAlarmTimeSec);
|
|
}
|
|
}
|
|
|
|
// More efficient than repeatedly calling remove(mPq.top()) since it batches the
|
|
// updates to the registered alarm.
|
|
unordered_set<sp<const AnomalyAlarm>, SpHash<AnomalyAlarm>>
|
|
AnomalyMonitor::popSoonerThan(uint32_t timestampSec) {
|
|
|
|
if (DEBUG) ALOGD("Removing alarms with time <= %u", timestampSec);
|
|
unordered_set<sp<const AnomalyAlarm>, SpHash<AnomalyAlarm>> oldAlarms;
|
|
std::lock_guard<std::mutex> lock(mLock);
|
|
|
|
for (sp<const AnomalyAlarm> t = mPq.top();
|
|
t != nullptr && t->timestampSec <= timestampSec; t = mPq.top()) {
|
|
oldAlarms.insert(t);
|
|
mPq.pop(); // remove t
|
|
}
|
|
// Always update registered alarm time (if anything has changed).
|
|
if (!oldAlarms.empty()) {
|
|
if (mPq.empty()) {
|
|
if (DEBUG) ALOGD("Queue is empty. Cancel any alarm.");
|
|
mRegisteredAlarmTimeSec = 0;
|
|
if (mStatsCompanionService != nullptr) {
|
|
mStatsCompanionService->cancelAnomalyAlarm();
|
|
}
|
|
} else {
|
|
// Always update the registered alarm in this case (unlike remove()).
|
|
updateRegisteredAlarmTime_l(mPq.top()->timestampSec);
|
|
}
|
|
}
|
|
return oldAlarms;
|
|
}
|
|
|
|
void AnomalyMonitor::updateRegisteredAlarmTime_l(uint32_t timestampSec) {
|
|
if (DEBUG) ALOGD("Updating reg alarm time to %u", timestampSec);
|
|
mRegisteredAlarmTimeSec = timestampSec;
|
|
if (mStatsCompanionService != nullptr) {
|
|
mStatsCompanionService->setAnomalyAlarm(secToMs(mRegisteredAlarmTimeSec));
|
|
}
|
|
}
|
|
|
|
int64_t AnomalyMonitor::secToMs(uint32_t timeSec) {
|
|
return ((int64_t)timeSec) * 1000;
|
|
}
|
|
|
|
} // namespace statsd
|
|
} // namespace os
|
|
} // namespace android
|