content/browser/webui/url_data_manager.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/webui/url_data_manager.h"
   6
   7 #include <vector>
   8
   9 #include "base/bind.h"
  10 #include "base/lazy_instance.h"
  11 #include "base/memory/ref_counted_memory.h"
  12 #include "base/message_loop/message_loop.h"
  13 #include "base/strings/string_util.h"
  14 #include "base/synchronization/lock.h"
  15 #include "content/browser/resource_context_impl.h"
  16 #include "content/browser/webui/url_data_manager_backend.h"
  17 #include "content/browser/webui/url_data_source_impl.h"
  18 #include "content/browser/webui/web_ui_data_source_impl.h"
  19 #include "content/public/browser/browser_context.h"
  20 #include "content/public/browser/browser_thread.h"
  21 #include "content/public/browser/url_data_source.h"
  22
  23 namespace content {
  24 namespace {
  25
  26 const char kURLDataManagerKeyName[] = "url_data_manager";
  27
  28 base::LazyInstance<base::Lock>::Leaky g_delete_lock = LAZY_INSTANCE_INITIALIZER;
  29
  30 URLDataManager* GetFromBrowserContext(BrowserContext* context) {
  31   if (!context->GetUserData(kURLDataManagerKeyName)) {
  32     context->SetUserData(kURLDataManagerKeyName, new URLDataManager(context));
  33   }
  34   return static_cast<URLDataManager*>(
  35       context->GetUserData(kURLDataManagerKeyName));
  36 }
  37
  38 // Invoked on the IO thread to do the actual adding of the DataSource.
  39 static void AddDataSourceOnIOThread(
  40     ResourceContext* resource_context,
  41     scoped_refptr<URLDataSourceImpl> data_source) {
  42   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  43   GetURLDataManagerForResourceContext(resource_context)->AddDataSource(
  44       data_source.get());
  45 }
  46
  47 }  // namespace
  48
  49 // static
  50 URLDataManager::URLDataSources* URLDataManager::data_sources_ = NULL;
  51
  52 URLDataManager::URLDataManager(BrowserContext* browser_context)
  53     : browser_context_(browser_context) {
  54 }
  55
  56 URLDataManager::~URLDataManager() {
  57 }
  58
  59 void URLDataManager::AddDataSource(URLDataSourceImpl* source) {
  60   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  61   BrowserThread::PostTask(
  62       BrowserThread::IO, FROM_HERE,
  63       base::Bind(&AddDataSourceOnIOThread,
  64                  browser_context_->GetResourceContext(),
  65                  make_scoped_refptr(source)));
  66 }
  67
  68 // static
  69 void URLDataManager::DeleteDataSources() {
  70   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  71   URLDataSources sources;
  72   {
  73     base::AutoLock lock(g_delete_lock.Get());
  74     if (!data_sources_)
  75       return;
  76     data_sources_->swap(sources);
  77   }
  78   for (size_t i = 0; i < sources.size(); ++i)
  79     delete sources[i];
  80 }
  81
  82 // static
  83 void URLDataManager::DeleteDataSource(const URLDataSourceImpl* data_source) {
  84   // Invoked when a DataSource is no longer referenced and needs to be deleted.
  85   if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
  86     // We're on the UI thread, delete right away.
  87     delete data_source;
  88     return;
  89   }
  90
  91   // We're not on the UI thread, add the DataSource to the list of DataSources
  92   // to delete.
  93   bool schedule_delete = false;
  94   {
  95     base::AutoLock lock(g_delete_lock.Get());
  96     if (!data_sources_)
  97       data_sources_ = new URLDataSources();
  98     schedule_delete = data_sources_->empty();
  99     data_sources_->push_back(data_source);
 100   }
 101   if (schedule_delete) {
 102     // Schedule a task to delete the DataSource back on the UI thread.
 103     BrowserThread::PostTask(
 104         BrowserThread::UI, FROM_HERE,
 105         base::Bind(&URLDataManager::DeleteDataSources));
 106   }
 107 }
 108
 109 // static
 110 void URLDataManager::AddDataSource(BrowserContext* browser_context,
 111                                    URLDataSource* source) {
 112   GetFromBrowserContext(browser_context)->
 113       AddDataSource(new URLDataSourceImpl(source->GetSource(), source));
 114 }
 115
 116 // static
 117 void URLDataManager::AddWebUIDataSource(BrowserContext* browser_context,
 118                                         WebUIDataSource* source) {
 119   WebUIDataSourceImpl* impl = static_cast<WebUIDataSourceImpl*>(source);
 120   GetFromBrowserContext(browser_context)->AddDataSource(impl);
 121 }
 122
 123 // static
 124 bool URLDataManager::IsScheduledForDeletion(
 125     const URLDataSourceImpl* data_source) {
 126   base::AutoLock lock(g_delete_lock.Get());
 127   if (!data_sources_)
 128     return false;
 129   return std::find(data_sources_->begin(), data_sources_->end(), data_source) !=
 130       data_sources_->end();
 131 }
 132
 133 }  // namespace content