Add UMA metrics and histograms for locally managed users.

chrome/browser/managed_mode/managed_mode.cc

 // Copyright (c) 2012 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 "chrome/browser/managed_mode/managed_mode.h"
 
 #include "base/command_line.h"
+#include "base/metrics/histogram.h"
 #include "base/prefs/public/pref_change_registrar.h"
+#include "base/rand_util.h"
 #include "base/sequenced_task_runner.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
+#include "chrome/browser/history/history.h"
+#include "chrome/browser/history/history_service_factory.h"
+#include "chrome/browser/history/history_types.h"
+#include "chrome/browser/history/url_database.h"
 #include "chrome/browser/managed_mode/managed_mode_site_list.h"
 #include "chrome/browser/managed_mode/managed_mode_url_filter.h"
 #include "chrome/browser/policy/url_blacklist_manager.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/prefs/scoped_user_pref_update.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension_set.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
+#include "content/public/browser/user_metrics.h"
+#include "googleurl/src/gurl.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using content::BrowserThread;
+using content::UserMetricsAction;
 
 // A bridge from ManagedMode (which lives on the UI thread) to
 // ManagedModeURLFilter (which might live on a different thread).
 class ManagedMode::URLFilterContext {
  public:
   explicit URLFilterContext(
       scoped_refptr<base::SequencedTaskRunner> task_runner)
       : task_runner_(task_runner) {}
   ~URLFilterContext() {}
 
@@ skipping 87 matching lines @@
 
 void ManagedMode::InitImpl(Profile* profile) {
   DCHECK(g_browser_process);
   DCHECK(g_browser_process->local_state());
 
   Profile* original_profile = profile->GetOriginalProfile();
   // Set the value directly in the PrefService instead of using
   // CommandLinePrefStore so we can change it at runtime.
   if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kNoManaged)) {
     SetInManagedMode(NULL);
+    content::RecordAction(
+        UserMetricsAction("ManagedMode_StartupNoManagedSwitch"));
   } else if (IsInManagedModeImpl() ||
       CommandLine::ForCurrentProcess()->HasSwitch(switches::kManaged)) {
     SetInManagedMode(original_profile);
+    content::RecordAction(
+        UserMetricsAction("ManagedMode_StartupManagedSwitch"));
   }
+
+  // Only compute these statistics sometimes, because they can be expensive.
+  if (base::RandInt(1, 100) == 50) {
+    base::TimeTicks start_time = base::TimeTicks::Now();
+    base::Time query_start = base::Time::Now() - base::TimeDelta::FromDays(7);
+    CollectDetailedHistoryMetrics(original_profile, query_start, "Weekly");
+    query_start = base::Time::Now() - base::TimeDelta::FromDays(30);
+    CollectDetailedHistoryMetrics(original_profile, query_start, "Monthly");
+    UMA_HISTOGRAM_TIMES("History.DatabaseAdvancedMetricsTime",
+                        base::TimeTicks::Now() - start_time);
+  }
+}
+
+void ManagedMode::CollectDetailedHistoryMetrics(Profile* profile,
+                                                base::Time start_time,
+                                                std::string description) {
+  // We have to collect statistics about the categories here; the history
+  // database dosn't know about managed users or categories. So we gather
+  // statistics about the URLs and hosts here too, since we have the data.
+  HistoryService* const history_service =
+      HistoryServiceFactory::GetForProfile(profile, Profile::IMPLICIT_ACCESS);
+  history::URLDatabase* url_db = history_service ?
+      history_service->InMemoryDatabase() : NULL;
+
+  if (!url_db)
+    return;
+
+  std::string base_metric_name = std::string("History.").append(description);
+
+  history::URLRows urls;
+  url_db->GetUniqueURLsInRange(start_time, base::Time(), 0, &urls);
+  UMA_HISTOGRAM_COUNTS(base_metric_name.append("URLCount"), urls.size());
+
+  std::set<std::string> hosts;
+  std::set<int> categories;
+  for (history::URLRows::iterator row = urls.begin();
+       row != urls.end(); ++row) {
+    GURL url = row->url();
+    hosts.insert(url.host());
+    categories.insert(ManagedMode::GetCategory(url));
+  }
+  UMA_HISTOGRAM_COUNTS_10000(base_metric_name.append("HostCount"),
+                             hosts.size());
+  UMA_HISTOGRAM_CUSTOM_COUNTS(base_metric_name.append("CategoryCount"),
+                              categories.size(), 0, 30, 30);
 }
 
 // static
 bool ManagedMode::IsInManagedMode() {
   return GetInstance()->IsInManagedModeImpl();
 }
 
 bool ManagedMode::IsInManagedModeImpl() const {
   // |g_browser_process| can be NULL during startup.
   if (!g_browser_process)
@@ skipping 154 matching lines @@
   return list->Find(pattern) != list->end();
 }
 
 // static
 scoped_ptr<base::ListValue> ManagedMode::GetBlacklist() {
     return scoped_ptr<base::ListValue>(
         GetInstance()->managed_profile_->GetPrefs()->GetList(
             prefs::kManagedModeBlacklist)->DeepCopy()).Pass();
 }
 
+ // Items not on any list must return -1 (CATEGORY_NOT_ON_LIST in history.js).
+ // Items on a list, but with no category, must return 0 (CATEGORY_OTHER).
+ #define CATEGORY_NOT_ON_LIST -1;
+ #define CATEGORY_OTHER 0;
+ // static
+ int ManagedMode::GetCategory(const GURL& url) {
+   return GetInstance()->GetCategoryImpl(url);
+ }
+
+ int ManagedMode::GetCategoryImpl(const GURL& url) {
+   std::vector<ManagedModeSiteList::Site*> sites;
+   GetURLFilterForUIThreadImpl()->GetSites(url, &sites);
+   if (sites.empty())
+     return CATEGORY_NOT_ON_LIST;
+
+   return (*sites.begin())->category_id;
+ }
+
 std::string ManagedMode::GetDebugPolicyProviderName() const {
   // Save the string space in official builds.
 #ifdef NDEBUG
   NOTREACHED();
   return std::string();
 #else
   return "Managed Mode";
 #endif
 }
 
@@ skipping 203 matching lines @@
       managed_profile_->GetPrefs()->GetList(
           prefs::kManagedModeWhitelist)->DeepCopy());
 }
 
 void ManagedMode::AddURLPatternToManualList(
     bool is_whitelist,
     const std::string& url_pattern) {
   io_url_filter_context_->AddURLPatternToManualList(true, url_pattern);
   ui_url_filter_context_->AddURLPatternToManualList(true, url_pattern);
 }