Count-up/down the dirty count in the usage cache at FileSystemOperation, not at QuotaFU.

webkit/fileapi/file_system_operation.h

 // Copyright (c) 2011 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.
 
 #ifndef WEBKIT_FILEAPI_FILE_SYSTEM_OPERATION_H_
 #define WEBKIT_FILEAPI_FILE_SYSTEM_OPERATION_H_
 
 #include <vector>
 
 #include "base/file_path.h"
 #include "base/file_util_proxy.h"
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_callback_factory.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop_proxy.h"
 #include "base/platform_file.h"
 #include "base/process.h"
 #include "googleurl/src/gurl.h"
+#include "webkit/fileapi/file_system_operation_context.h"
 #include "webkit/fileapi/file_system_types.h"
-#include "webkit/fileapi/file_system_operation_context.h"
 #include "webkit/quota/quota_manager.h"
 
 namespace base {
 class Time;
 }
 
 namespace net {
 class URLRequest;
 class URLRequestContext;
 }  // namespace net
 
 class GURL;
 
 namespace fileapi {
 
 class FileSystemCallbackDispatcher;
 class FileSystemContext;
 class FileWriterDelegate;
 class FileSystemOperationTest;
+class FileSystemQuotaUtil;
 
 // This class is designed to serve one-time file system operation per instance.
 // Only one method(CreateFile, CreateDirectory, Copy, Move, DirectoryExists,
 // GetMetadata, ReadDirectory and Remove) may be called during the lifetime of
 // this object and it should be called no more than once.
 // This class is self-destructed and an instance automatically gets deleted
 // when its operation is finished.
 class FileSystemOperation {
  public:
   // |dispatcher| will be owned by this class.
@@ skipping 34 matching lines @@
       const GURL& path,
       int file_flags,
       base::ProcessHandle peer_handle);
 
   // Try to cancel the current operation [we support cancelling write or
   // truncate only].  Report failure for the current operation, then tell the
   // passed-in operation to report success.
   void Cancel(FileSystemOperation* cancel_operation);
 
  private:
+  class ScopedQuotaUtilHelper {
+   public:
+    explicit ScopedQuotaUtilHelper(FileSystemContext* context,
+                                   const GURL& origin_url,
+                                   FileSystemType type);
+    ~ScopedQuotaUtilHelper();
+
+   private:
+    FileSystemQuotaUtil* quota_util_;
+    const GURL& origin_url_;
+    FileSystemType type_;
+    DISALLOW_COPY_AND_ASSIGN(ScopedQuotaUtilHelper);
+  };
+
   FileSystemContext* file_system_context() const {
     return file_system_operation_context_.file_system_context();
   }
 
   FileSystemOperationContext* file_system_operation_context() {
     return &file_system_operation_context_;
   }
 
   friend class FileSystemOperationTest;
   friend class FileSystemOperationWriteTest;
@@ skipping 127 matching lines @@
 
   // Proxy for calling file_util_proxy methods.
   scoped_refptr<base::MessageLoopProxy> proxy_;
 
   scoped_ptr<FileSystemCallbackDispatcher> dispatcher_;
 
   FileSystemOperationContext file_system_operation_context_;
 
   base::ScopedCallbackFactory<FileSystemOperation> callback_factory_;
 
+  scoped_ptr<ScopedQuotaUtilHelper> quota_util_helper_;
+
   // These are all used only by Write().
   friend class FileWriterDelegate;
   scoped_ptr<FileWriterDelegate> file_writer_delegate_;
   scoped_ptr<net::URLRequest> blob_request_;
   scoped_ptr<FileSystemOperation> cancel_operation_;
 
   // Used only by OpenFile, in order to clone the file handle back to the
   // requesting process.
   base::ProcessHandle peer_handle_;
 
   // Options for CreateFile and CreateDirectory.
   bool exclusive_;
   bool recursive_;
 
   // Options for OpenFile.
   int file_flags_;
 
   // Length to be truncated.
   int64 length_;
 
   DISALLOW_COPY_AND_ASSIGN(FileSystemOperation);
 };
 
 }  // namespace fileapi
 
 #endif  // WEBKIT_FILEAPI_FILE_SYSTEM_OPERATION_H_