[Device Discovery] Make DialRegistry a Singleton

chrome/browser/extensions/api/dial/dial_api.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/extensions/api/dial/dial_api.h"
 
 #include <stddef.h>
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/time/time.h"
 #include "chrome/browser/extensions/api/dial/dial_api_factory.h"
 #include "chrome/browser/media/router/discovery/dial/device_description_fetcher.h"
+#include "chrome/browser/media/router/discovery/dial/dial_registry.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/extensions/api/dial.h"
 #include "content/public/browser/browser_thread.h"
 #include "extensions/browser/event_router.h"
 #include "extensions/browser/extension_system.h"
 #include "url/gurl.h"
 
 using base::TimeDelta;
 using content::BrowserThread;
 using media_router::DeviceDescriptionFetcher;
 using media_router::DialDeviceData;
 using media_router::DialDeviceDescriptionData;
 using media_router::DialRegistry;
 
 namespace extensions {
 
-namespace {
-
-// How often to poll for devices.
-const int kDialRefreshIntervalSecs = 120;
-
-// We prune a device if it does not respond after this time.
-const int kDialExpirationSecs = 240;
-
-// The maximum number of devices retained at once in the registry.
-const size_t kDialMaxDevices = 256;
-
-}  // namespace
-
 DialAPI::DialAPI(Profile* profile)
     : RefcountedKeyedService(
           BrowserThread::GetTaskRunnerForThread(BrowserThread::IO)),
-      profile_(profile) {
+      profile_(profile),
+      dial_registry_(nullptr) {
   EventRouter::Get(profile)->RegisterObserver(
       this, api::dial::OnDeviceList::kEventName);
 }
 
-DialAPI::~DialAPI() {}
+DialAPI::~DialAPI() {
+  // TODO(zhaobin): Call dial_registry_->UnregisterObserver() instead. In
+  // current implementation, UnregistryObserver() does not StopDiscovery() and
+  // causes crash in ~DialRegistry(). May keep a listener count and
+  // Register/UnregisterObserver as needed.
+  if (dial_registry_)
+    dial_registry_->StopPeriodicDiscovery();
+}
 
 DialRegistry* DialAPI::dial_registry() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
-  if (!dial_registry_.get()) {
-    dial_registry_.reset(new DialRegistry(
-        TimeDelta::FromSeconds(kDialRefreshIntervalSecs),
-        TimeDelta::FromSeconds(kDialExpirationSecs), kDialMaxDevices));
+  if (!dial_registry_) {
+    dial_registry_ = media_router::DialRegistry::GetInstance();
     dial_registry_->RegisterObserver(this);
     if (test_device_data_) {
       dial_registry_->AddDeviceForTest(*test_device_data_);
     }
   }
-  return dial_registry_.get();
+  return dial_registry_;
 }
 
 void DialAPI::OnListenerAdded(const EventListenerInfo& details) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(&DialAPI::NotifyListenerAddedOnIOThread, this));
 }
 
 void DialAPI::OnListenerRemoved(const EventListenerInfo& details) {
@@ skipping 189 matching lines @@
 void DialFetchDeviceDescriptionFunction::OnFetchError(
     const std::string& message) {
   // Destroy the DeviceDescriptionFetcher since it still contains a reference
   // to |this| in its un-invoked callback.
   device_description_fetcher_.reset();
   SetError(message);
   SendResponse(false);
 }
 
 }  // namespace extensions