storage/browser/quota/quota_task.cc

   1 // Copyright 2013 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 "storage/browser/quota/quota_task.h"
   6
   7 #include <algorithm>
   8 #include <functional>
   9
  10 #include "base/bind.h"
  11 #include "base/location.h"
  12 #include "base/message_loop/message_loop.h"
  13 #include "base/thread_task_runner_handle.h"
  14
  15 using base::TaskRunner;
  16
  17 namespace storage {
  18
  19 // QuotaTask ---------------------------------------------------------------
  20
  21 QuotaTask::~QuotaTask() {
  22 }
  23
  24 void QuotaTask::Start() {
  25   DCHECK(observer_);
  26   observer()->RegisterTask(this);
  27   Run();
  28 }
  29
  30 QuotaTask::QuotaTask(QuotaTaskObserver* observer)
  31     : observer_(observer),
  32       original_task_runner_(base::ThreadTaskRunnerHandle::Get()),
  33       delete_scheduled_(false) {
  34 }
  35
  36 void QuotaTask::CallCompleted() {
  37   DCHECK(original_task_runner_->BelongsToCurrentThread());
  38   if (observer_) {
  39     observer_->UnregisterTask(this);
  40     Completed();
  41   }
  42 }
  43
  44 void QuotaTask::Abort() {
  45   DCHECK(original_task_runner_->BelongsToCurrentThread());
  46   observer_ = NULL;
  47   Aborted();
  48 }
  49
  50 void QuotaTask::DeleteSoon() {
  51   DCHECK(original_task_runner_->BelongsToCurrentThread());
  52   if (delete_scheduled_)
  53     return;
  54   delete_scheduled_ = true;
  55   base::MessageLoop::current()->DeleteSoon(FROM_HERE, this);
  56 }
  57
  58 // QuotaTaskObserver -------------------------------------------------------
  59
  60 QuotaTaskObserver::~QuotaTaskObserver() {
  61   std::for_each(running_quota_tasks_.begin(),
  62                 running_quota_tasks_.end(),
  63                 std::mem_fun(&QuotaTask::Abort));
  64 }
  65
  66 QuotaTaskObserver::QuotaTaskObserver() {
  67 }
  68
  69 void QuotaTaskObserver::RegisterTask(QuotaTask* task) {
  70   running_quota_tasks_.insert(task);
  71 }
  72
  73 void QuotaTaskObserver::UnregisterTask(QuotaTask* task) {
  74   DCHECK(running_quota_tasks_.find(task) != running_quota_tasks_.end());
  75   running_quota_tasks_.erase(task);
  76 }
  77
  78 }  // namespace storage