chrome/browser/chromeos/file_system_provider/throttled_file_system.cc

   1 // Copyright 2015 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 "chrome/browser/chromeos/file_system_provider/throttled_file_system.h"
   6
   7 #include <limits>
   8 #include <vector>
   9
  10 #include "base/files/file.h"
  11 #include "chrome/browser/chromeos/file_system_provider/queue.h"
  12
  13 namespace chromeos {
  14 namespace file_system_provider {
  15
  16 ThrottledFileSystem::ThrottledFileSystem(
  17     scoped_ptr<ProvidedFileSystemInterface> file_system)
  18     : file_system_(file_system.Pass()), weak_ptr_factory_(this) {
  19   const int opened_files_limit =
  20       file_system_->GetFileSystemInfo().opened_files_limit();
  21   open_queue_.reset(opened_files_limit
  22                         ? new Queue(static_cast<size_t>(opened_files_limit))
  23                         : new Queue(std::numeric_limits<size_t>::max()));
  24 }
  25
  26 ThrottledFileSystem::~ThrottledFileSystem() {
  27 }
  28
  29 AbortCallback ThrottledFileSystem::RequestUnmount(
  30     const storage::AsyncFileUtil::StatusCallback& callback) {
  31   return file_system_->RequestUnmount(callback);
  32 }
  33
  34 AbortCallback ThrottledFileSystem::GetMetadata(
  35     const base::FilePath& entry_path,
  36     MetadataFieldMask fields,
  37     const GetMetadataCallback& callback) {
  38   return file_system_->GetMetadata(entry_path, fields, callback);
  39 }
  40
  41 AbortCallback ThrottledFileSystem::ReadDirectory(
  42     const base::FilePath& directory_path,
  43     const storage::AsyncFileUtil::ReadDirectoryCallback& callback) {
  44   return file_system_->ReadDirectory(directory_path, callback);
  45 }
  46
  47 AbortCallback ThrottledFileSystem::ReadFile(
  48     int file_handle,
  49     net::IOBuffer* buffer,
  50     int64 offset,
  51     int length,
  52     const ReadChunkReceivedCallback& callback) {
  53   return file_system_->ReadFile(file_handle, buffer, offset, length, callback);
  54 }
  55
  56 AbortCallback ThrottledFileSystem::OpenFile(const base::FilePath& file_path,
  57                                             OpenFileMode mode,
  58                                             const OpenFileCallback& callback) {
  59   const size_t task_token = open_queue_->NewToken();
  60   open_queue_->Enqueue(
  61       task_token,
  62       base::Bind(
  63           &ProvidedFileSystemInterface::OpenFile,
  64           base::Unretained(file_system_.get()),  // Outlives the queue.
  65           file_path, mode,
  66           base::Bind(&ThrottledFileSystem::OnOpenFileCompleted,
  67                      weak_ptr_factory_.GetWeakPtr(), task_token, callback)));
  68   return base::Bind(&ThrottledFileSystem::Abort, weak_ptr_factory_.GetWeakPtr(),
  69                     task_token);
  70 }
  71
  72 AbortCallback ThrottledFileSystem::CloseFile(
  73     int file_handle,
  74     const storage::AsyncFileUtil::StatusCallback& callback) {
  75   return file_system_->CloseFile(
  76       file_handle,
  77       base::Bind(&ThrottledFileSystem::OnCloseFileCompleted,
  78                  weak_ptr_factory_.GetWeakPtr(), file_handle, callback));
  79 }
  80
  81 AbortCallback ThrottledFileSystem::CreateDirectory(
  82     const base::FilePath& directory_path,
  83     bool recursive,
  84     const storage::AsyncFileUtil::StatusCallback& callback) {
  85   return file_system_->CreateDirectory(directory_path, recursive, callback);
  86 }
  87
  88 AbortCallback ThrottledFileSystem::DeleteEntry(
  89     const base::FilePath& entry_path,
  90     bool recursive,
  91     const storage::AsyncFileUtil::StatusCallback& callback) {
  92   return file_system_->DeleteEntry(entry_path, recursive, callback);
  93 }
  94
  95 AbortCallback ThrottledFileSystem::CreateFile(
  96     const base::FilePath& file_path,
  97     const storage::AsyncFileUtil::StatusCallback& callback) {
  98   return file_system_->CreateFile(file_path, callback);
  99 }
 100
 101 AbortCallback ThrottledFileSystem::CopyEntry(
 102     const base::FilePath& source_path,
 103     const base::FilePath& target_path,
 104     const storage::AsyncFileUtil::StatusCallback& callback) {
 105   return file_system_->CopyEntry(source_path, target_path, callback);
 106 }
 107
 108 AbortCallback ThrottledFileSystem::WriteFile(
 109     int file_handle,
 110     net::IOBuffer* buffer,
 111     int64 offset,
 112     int length,
 113     const storage::AsyncFileUtil::StatusCallback& callback) {
 114   return file_system_->WriteFile(file_handle, buffer, offset, length, callback);
 115 }
 116
 117 AbortCallback ThrottledFileSystem::MoveEntry(
 118     const base::FilePath& source_path,
 119     const base::FilePath& target_path,
 120     const storage::AsyncFileUtil::StatusCallback& callback) {
 121   return file_system_->MoveEntry(source_path, target_path, callback);
 122 }
 123
 124 AbortCallback ThrottledFileSystem::Truncate(
 125     const base::FilePath& file_path,
 126     int64 length,
 127     const storage::AsyncFileUtil::StatusCallback& callback) {
 128   return file_system_->Truncate(file_path, length, callback);
 129 }
 130
 131 AbortCallback ThrottledFileSystem::AddWatcher(
 132     const GURL& origin,
 133     const base::FilePath& entry_path,
 134     bool recursive,
 135     bool persistent,
 136     const storage::AsyncFileUtil::StatusCallback& callback,
 137     const storage::WatcherManager::NotificationCallback&
 138         notification_callback) {
 139   return file_system_->AddWatcher(origin, entry_path, recursive, persistent,
 140                                   callback, notification_callback);
 141 }
 142
 143 void ThrottledFileSystem::RemoveWatcher(
 144     const GURL& origin,
 145     const base::FilePath& entry_path,
 146     bool recursive,
 147     const storage::AsyncFileUtil::StatusCallback& callback) {
 148   file_system_->RemoveWatcher(origin, entry_path, recursive, callback);
 149 }
 150
 151 const ProvidedFileSystemInfo& ThrottledFileSystem::GetFileSystemInfo() const {
 152   return file_system_->GetFileSystemInfo();
 153 }
 154
 155 RequestManager* ThrottledFileSystem::GetRequestManager() {
 156   return file_system_->GetRequestManager();
 157 }
 158
 159 Watchers* ThrottledFileSystem::GetWatchers() {
 160   return file_system_->GetWatchers();
 161 }
 162
 163 const OpenedFiles& ThrottledFileSystem::GetOpenedFiles() const {
 164   return file_system_->GetOpenedFiles();
 165 }
 166
 167 void ThrottledFileSystem::AddObserver(ProvidedFileSystemObserver* observer) {
 168   file_system_->AddObserver(observer);
 169 }
 170
 171 void ThrottledFileSystem::RemoveObserver(ProvidedFileSystemObserver* observer) {
 172   file_system_->RemoveObserver(observer);
 173 }
 174
 175 void ThrottledFileSystem::Notify(
 176     const base::FilePath& entry_path,
 177     bool recursive,
 178     storage::WatcherManager::ChangeType change_type,
 179     scoped_ptr<ProvidedFileSystemObserver::Changes> changes,
 180     const std::string& tag,
 181     const storage::AsyncFileUtil::StatusCallback& callback) {
 182   return file_system_->Notify(entry_path, recursive, change_type,
 183                               changes.Pass(), tag, callback);
 184 }
 185
 186 base::WeakPtr<ProvidedFileSystemInterface> ThrottledFileSystem::GetWeakPtr() {
 187   return weak_ptr_factory_.GetWeakPtr();
 188 }
 189
 190 void ThrottledFileSystem::Abort(int queue_token) {
 191   open_queue_->Abort(queue_token);
 192 }
 193
 194 void ThrottledFileSystem::OnOpenFileCompleted(int queue_token,
 195                                               const OpenFileCallback& callback,
 196                                               int file_handle,
 197                                               base::File::Error result) {
 198   // If the file is opened successfully then hold the queue token until the file
 199   // is closed.
 200   if (result == base::File::FILE_OK)
 201     opened_files_[file_handle] = queue_token;
 202   else
 203     open_queue_->Complete(queue_token);
 204
 205   callback.Run(file_handle, result);
 206 }
 207
 208 void ThrottledFileSystem::OnCloseFileCompleted(
 209     int file_handle,
 210     const storage::AsyncFileUtil::StatusCallback& callback,
 211     base::File::Error result) {
 212   // Closing is always final. Even if an error happened, the file is considered
 213   // closed on the C++ side. Release the task from the queue, so other files
 214   // which are enqueued can be opened.
 215   const auto it = opened_files_.find(file_handle);
 216   DCHECK(it != opened_files_.end());
 217
 218   const int queue_token = it->second;
 219   open_queue_->Complete(queue_token);
 220   opened_files_.erase(file_handle);
 221
 222   callback.Run(result);
 223 }
 224
 225 }  // namespace file_system_provider
 226 }  // namespace chromeos