Remove dimension fields in GaugeMetric output (4c31f679) · Commits · e / os / android_frameworks_base

cmds/statsd/src/metrics/GaugeMetricProducer.cpp

+16 −3

Original line number	Original line	Diff line number	Diff line
	@@ -367,14 +367,27 @@ void GaugeMetricProducer::onSlicedConditionMayChangeLocked(bool overallCondition
	}		}

	std::shared_ptr<vector<FieldValue>> GaugeMetricProducer::getGaugeFields(const LogEvent& event) {		std::shared_ptr<vector<FieldValue>> GaugeMetricProducer::getGaugeFields(const LogEvent& event) {
			std::shared_ptr<vector<FieldValue>> gaugeFields;
	if (mFieldMatchers.size() > 0) {		if (mFieldMatchers.size() > 0) {
	std::shared_ptr<vector<FieldValue>> gaugeFields = std::make_shared<vector<FieldValue>>();		gaugeFields = std::make_shared<vector<FieldValue>>();
	filterGaugeValues(mFieldMatchers, event.getValues(), gaugeFields.get());		filterGaugeValues(mFieldMatchers, event.getValues(), gaugeFields.get());
	return gaugeFields;
	} else {		} else {
	return std::make_shared<vector<FieldValue>>(event.getValues());		gaugeFields = std::make_shared<vector<FieldValue>>(event.getValues());
			}
			// Trim all dimension fields from output. Dimensions will appear in output report and will
			// benefit from dictionary encoding. For large pulled atoms, this can give the benefit of
			// optional repeated field.
			for (const auto& field : mDimensionsInWhat) {
			for (auto it = gaugeFields->begin(); it != gaugeFields->end();) {
			if (it->mField.matches(field)) {
			it = gaugeFields->erase(it);
			} else {
			it++;
			}
	}		}
	}		}
			return gaugeFields;
			}

	void GaugeMetricProducer::onDataPulled(const std::vector<std::shared_ptr<LogEvent>>& allData) {		void GaugeMetricProducer::onDataPulled(const std::vector<std::shared_ptr<LogEvent>>& allData) {
	std::lock_guard<std::mutex> lock(mMutex);		std::lock_guard<std::mutex> lock(mMutex);

cmds/statsd/src/metrics/GaugeMetricProducer.h

+1 −0

Original line number	Original line	Diff line number	Diff line
	@@ -175,6 +175,7 @@ private:
	FRIEND_TEST(GaugeMetricProducerTest, TestPulledEventsAnomalyDetection);		FRIEND_TEST(GaugeMetricProducerTest, TestPulledEventsAnomalyDetection);
	FRIEND_TEST(GaugeMetricProducerTest, TestFirstBucket);		FRIEND_TEST(GaugeMetricProducerTest, TestFirstBucket);
	FRIEND_TEST(GaugeMetricProducerTest, TestPullOnTrigger);		FRIEND_TEST(GaugeMetricProducerTest, TestPullOnTrigger);
			FRIEND_TEST(GaugeMetricProducerTest, TestRemoveDimensionInOutput);
	};		};

	} // namespace statsd		} // namespace statsd

cmds/statsd/tests/e2e/GaugeMetric_e2e_pull_test.cpp

+14 −13

cmds/statsd/tests/metrics/GaugeMetricProducer_test.cpp

+105 −3

Original line number	Original line	Diff line number	Diff line
	@@ -95,6 +95,8 @@ TEST(GaugeMetricProducerTest, TestPulledEventsNoCondition) {
	data->clear();		data->clear();
	shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucketStartTimeNs + 10);		shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucketStartTimeNs + 10);
	event->write(3);		event->write(3);
			event->write("some value");
			event->write(11);
	event->init();		event->init();
	data->push_back(event);		data->push_back(event);
	return true;		return true;
	@@ -600,9 +602,109 @@ TEST(GaugeMetricProducerTest, TestPullOnTrigger) {
	EXPECT_EQ(10, it->mValue.int_value);		EXPECT_EQ(10, it->mValue.int_value);
	EXPECT_EQ(1UL, gaugeProducer.mPastBuckets.size());		EXPECT_EQ(1UL, gaugeProducer.mPastBuckets.size());
	EXPECT_EQ(3UL, gaugeProducer.mPastBuckets.begin()->second.back().mGaugeAtoms.size());		EXPECT_EQ(3UL, gaugeProducer.mPastBuckets.begin()->second.back().mGaugeAtoms.size());
	EXPECT_EQ(3, gaugeProducer.mPastBuckets.begin()->second.back().mGaugeAtoms[0].mFields->begin()->mValue.int_value);		EXPECT_EQ(3, gaugeProducer.mPastBuckets.begin()
	EXPECT_EQ(4, gaugeProducer.mPastBuckets.begin()->second.back().mGaugeAtoms[1].mFields->begin()->mValue.int_value);		->second.back()
	EXPECT_EQ(5, gaugeProducer.mPastBuckets.begin()->second.back().mGaugeAtoms[2].mFields->begin()->mValue.int_value);		.mGaugeAtoms[0]
			.mFields->begin()
			->mValue.int_value);
			EXPECT_EQ(4, gaugeProducer.mPastBuckets.begin()
			->second.back()
			.mGaugeAtoms[1]
			.mFields->begin()
			->mValue.int_value);
			EXPECT_EQ(5, gaugeProducer.mPastBuckets.begin()
			->second.back()
			.mGaugeAtoms[2]
			.mFields->begin()
			->mValue.int_value);
			}

			TEST(GaugeMetricProducerTest, TestRemoveDimensionInOutput) {
			GaugeMetric metric;
			metric.set_id(metricId);
			metric.set_bucket(ONE_MINUTE);
			metric.set_sampling_type(GaugeMetric::ALL_CONDITION_CHANGES);
			metric.mutable_gauge_fields_filter()->set_include_all(true);
			auto dimensionMatcher = metric.mutable_dimensions_in_what();
			// use field 1 as dimension.
			dimensionMatcher->set_field(tagId);
			dimensionMatcher->add_child()->set_field(1);

			sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();

			sp<MockStatsPullerManager> pullerManager = new StrictMock<MockStatsPullerManager>();
			EXPECT_CALL(*pullerManager, Pull(tagId, _, _))
			.WillOnce(Invoke([](int tagId, int64_t timeNs,
			vector<std::shared_ptr<LogEvent>>* data) {
			data->clear();
			shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucketStartTimeNs + 3);
			event->write(3);
			event->write(4);
			event->init();
			data->push_back(event);
			return true;
			}))
			.WillOnce(Invoke([](int tagId, int64_t timeNs,
			vector<std::shared_ptr<LogEvent>>* data) {
			data->clear();
			shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucketStartTimeNs + 10);
			event->write(4);
			event->write(5);
			event->init();
			data->push_back(event);
			return true;
			}))
			.WillOnce(Invoke([](int tagId, int64_t timeNs,
			vector<std::shared_ptr<LogEvent>>* data) {
			data->clear();
			shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucketStartTimeNs + 20);
			event->write(4);
			event->write(6);
			event->init();
			data->push_back(event);
			return true;
			}));

			int triggerId = 5;
			GaugeMetricProducer gaugeProducer(kConfigKey, metric, -1 /-1 meaning no condition/, wizard,
			tagId, triggerId, tagId, bucketStartTimeNs, bucketStartTimeNs,
			pullerManager);

			vector<shared_ptr<LogEvent>> allData;
			allData.clear();

			EXPECT_EQ(1UL, gaugeProducer.mCurrentSlicedBucket->size());
			LogEvent trigger(triggerId, bucketStartTimeNs + 10);
			trigger.init();
			gaugeProducer.onMatchedLogEvent(1 /log matcher index/, trigger);
			EXPECT_EQ(2UL, gaugeProducer.mCurrentSlicedBucket->size());
			EXPECT_EQ(1UL, gaugeProducer.mCurrentSlicedBucket->begin()->second.size());
			trigger.setElapsedTimestampNs(bucketStartTimeNs + 20);
			gaugeProducer.onMatchedLogEvent(1 /log matcher index/, trigger);
			EXPECT_EQ(2UL, gaugeProducer.mCurrentSlicedBucket->begin()->second.size());

			allData.clear();
			shared_ptr<LogEvent> event = make_shared<LogEvent>(tagId, bucket2StartTimeNs + 1);
			event->write(4);
			event->write(11);
			event->init();
			allData.push_back(event);

			gaugeProducer.onDataPulled(allData);
			EXPECT_EQ(1UL, gaugeProducer.mCurrentSlicedBucket->size());
			auto it = gaugeProducer.mCurrentSlicedBucket->begin()->second.front().mFields->begin();
			EXPECT_EQ(INT, it->mValue.getType());
			EXPECT_EQ(11, it->mValue.int_value);
			EXPECT_EQ(2UL, gaugeProducer.mPastBuckets.size());
			auto bucketIt = gaugeProducer.mPastBuckets.begin();
			EXPECT_EQ(1UL, bucketIt->second.back().mGaugeAtoms.size());
			EXPECT_EQ(3, bucketIt->first.getDimensionKeyInWhat().getValues().begin()->mValue.int_value);
			EXPECT_EQ(4, bucketIt->second.back().mGaugeAtoms[0].mFields->begin()->mValue.int_value);
			bucketIt++;
			EXPECT_EQ(2UL, bucketIt->second.back().mGaugeAtoms.size());
			EXPECT_EQ(4, bucketIt->first.getDimensionKeyInWhat().getValues().begin()->mValue.int_value);
			EXPECT_EQ(5, bucketIt->second.back().mGaugeAtoms[0].mFields->begin()->mValue.int_value);
			EXPECT_EQ(6, bucketIt->second.back().mGaugeAtoms[1].mFields->begin()->mValue.int_value);
	}		}

	} // namespace statsd		} // namespace statsd

Original line number	Original line	Diff line number	Diff line
	@@ -48,6 +48,7 @@ StatsdConfig CreateStatsdConfig(const GaugeMetric::SamplingType sampling_type) {
	*gaugeMetric->mutable_dimensions_in_what() =		*gaugeMetric->mutable_dimensions_in_what() =
	CreateDimensions(android::util::TEMPERATURE, {2/* sensor name field */ });		CreateDimensions(android::util::TEMPERATURE, {2/* sensor name field */ });
	gaugeMetric->set_bucket(FIVE_MINUTES);		gaugeMetric->set_bucket(FIVE_MINUTES);
			config.set_hash_strings_in_metric_report(false);

	return config;		return config;
	}		}
	@@ -150,7 +151,7 @@ TEST(GaugeMetricE2eTest, TestRandomSamplePulledEvents) {
	EXPECT_EQ(1, data.bucket_info(0).wall_clock_timestamp_nanos_size());		EXPECT_EQ(1, data.bucket_info(0).wall_clock_timestamp_nanos_size());
	EXPECT_EQ(baseTimeNs + 2 * bucketSizeNs, data.bucket_info(0).start_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 2 * bucketSizeNs, data.bucket_info(0).start_bucket_elapsed_nanos());
	EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, data.bucket_info(0).end_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, data.bucket_info(0).end_bucket_elapsed_nanos());
	EXPECT_FALSE(data.bucket_info(0).atom(0).temperature().sensor_name().empty());		EXPECT_TRUE(data.bucket_info(0).atom(0).temperature().sensor_name().empty());
	EXPECT_GT(data.bucket_info(0).atom(0).temperature().temperature_dc(), 0);		EXPECT_GT(data.bucket_info(0).atom(0).temperature().temperature_dc(), 0);

	EXPECT_EQ(1, data.bucket_info(1).atom_size());		EXPECT_EQ(1, data.bucket_info(1).atom_size());
	@@ -159,7 +160,7 @@ TEST(GaugeMetricE2eTest, TestRandomSamplePulledEvents) {
	EXPECT_EQ(configAddedTimeNs + 55, data.bucket_info(0).elapsed_timestamp_nanos(0));		EXPECT_EQ(configAddedTimeNs + 55, data.bucket_info(0).elapsed_timestamp_nanos(0));
	EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, data.bucket_info(1).start_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, data.bucket_info(1).start_bucket_elapsed_nanos());
	EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(1).end_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(1).end_bucket_elapsed_nanos());
	EXPECT_FALSE(data.bucket_info(1).atom(0).temperature().sensor_name().empty());		EXPECT_TRUE(data.bucket_info(1).atom(0).temperature().sensor_name().empty());
	EXPECT_GT(data.bucket_info(1).atom(0).temperature().temperature_dc(), 0);		EXPECT_GT(data.bucket_info(1).atom(0).temperature().temperature_dc(), 0);

	EXPECT_EQ(1, data.bucket_info(2).atom_size());		EXPECT_EQ(1, data.bucket_info(2).atom_size());
	@@ -168,7 +169,7 @@ TEST(GaugeMetricE2eTest, TestRandomSamplePulledEvents) {
	data.bucket_info(2).elapsed_timestamp_nanos(0));		data.bucket_info(2).elapsed_timestamp_nanos(0));
	EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(2).start_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(2).start_bucket_elapsed_nanos());
	EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(2).end_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(2).end_bucket_elapsed_nanos());
	EXPECT_FALSE(data.bucket_info(2).atom(0).temperature().sensor_name().empty());		EXPECT_TRUE(data.bucket_info(2).atom(0).temperature().sensor_name().empty());
	EXPECT_GT(data.bucket_info(2).atom(0).temperature().temperature_dc(), 0);		EXPECT_GT(data.bucket_info(2).atom(0).temperature().temperature_dc(), 0);

	EXPECT_EQ(1, data.bucket_info(3).atom_size());		EXPECT_EQ(1, data.bucket_info(3).atom_size());
	@@ -177,7 +178,7 @@ TEST(GaugeMetricE2eTest, TestRandomSamplePulledEvents) {
	data.bucket_info(3).elapsed_timestamp_nanos(0));		data.bucket_info(3).elapsed_timestamp_nanos(0));
	EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(3).start_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(3).start_bucket_elapsed_nanos());
	EXPECT_EQ(baseTimeNs + 6 * bucketSizeNs, data.bucket_info(3).end_bucket_elapsed_nanos());		EXPECT_EQ(baseTimeNs + 6 * bucketSizeNs, data.bucket_info(3).end_bucket_elapsed_nanos());
	EXPECT_FALSE(data.bucket_info(3).atom(0).temperature().sensor_name().empty());		EXPECT_TRUE(data.bucket_info(3).atom(0).temperature().sensor_name().empty());
	EXPECT_GT(data.bucket_info(3).atom(0).temperature().temperature_dc(), 0);		EXPECT_GT(data.bucket_info(3).atom(0).temperature().temperature_dc(), 0);

	EXPECT_EQ(1, data.bucket_info(4).atom_size());		EXPECT_EQ(1, data.bucket_info(4).atom_size());