Index: mojo/services/test_service/tracked_service.cc |
diff --git a/mojo/services/test_service/tracked_service.cc b/mojo/services/test_service/tracked_service.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a2fedfc9bb6ce6a9dbcf958b8e9511a325f198d3 |
--- /dev/null |
+++ b/mojo/services/test_service/tracked_service.cc |
@@ -0,0 +1,53 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "mojo/services/test_service/tracked_service.h" |
+ |
+#include "base/bind.h" |
+ |
+namespace mojo { |
+namespace test { |
+ |
+TrackedService::TrackedService(TestRequestTrackerPtr tracker, |
+ const std::string& service_name, |
+ const mojo::Callback<void()>& callback) |
+ : id_(0u), |
+ requests_since_upload_(0u), |
+ service_name_(service_name), |
+ tracker_(tracker.Pass()), |
+ tracking_connected_callback_(callback) { |
+ tracker_->SetNameAndReturnId( |
+ service_name, base::Bind(&TrackedService::SetId, base::Unretained(this))); |
+} |
+ |
+TrackedService::~TrackedService() { |
+} |
+ |
+void TrackedService::RecordNewRequest() { |
+ requests_since_upload_++; |
+ if (id_ == 0u) |
+ return; |
+ SendStats(); |
+} |
+ |
+void TrackedService::SendStats() { |
+ ServiceStatsPtr stats(ServiceStats::New()); |
+ stats->num_new_requests = requests_since_upload_; |
+ stats->health = 0.7; |
+ tracker_->RecordStats(id_, stats.Pass()); |
+ requests_since_upload_ = 0u; |
+} |
+ |
+void TrackedService::SetId(uint64_t id) { |
+ assert(id != 0u); |
+ assert(id_ == 0u); |
+ id_ = id; |
+ tracking_connected_callback_.Run(); |
+ if (requests_since_upload_ == 0u) |
+ return; |
+ SendStats(); |
+} |
+ |
+} // namespace test |
+} // namespace mojo |