| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef WEBKIT_BROWSER_FILEAPI_LOCAL_FILE_STREAM_WRITER_H_ | 5 #include "storage/browser/fileapi/local_file_stream_writer.h" |
| 6 #define WEBKIT_BROWSER_FILEAPI_LOCAL_FILE_STREAM_WRITER_H_ | |
| 7 | |
| 8 #include <utility> | |
| 9 | |
| 10 #include "base/callback.h" | |
| 11 #include "base/compiler_specific.h" | |
| 12 #include "base/files/file_path.h" | |
| 13 #include "base/gtest_prod_util.h" | |
| 14 #include "base/memory/scoped_ptr.h" | |
| 15 #include "base/memory/weak_ptr.h" | |
| 16 #include "base/task_runner.h" | |
| 17 #include "webkit/browser/fileapi/file_stream_writer.h" | |
| 18 #include "webkit/browser/storage_browser_export.h" | |
| 19 | |
| 20 namespace content { | |
| 21 class LocalFileStreamWriterTest; | |
| 22 } | |
| 23 | |
| 24 namespace net { | |
| 25 class FileStream; | |
| 26 } | |
| 27 | |
| 28 namespace storage { | |
| 29 | |
| 30 // This class is a thin wrapper around net::FileStream for writing local files. | |
| 31 class STORAGE_EXPORT LocalFileStreamWriter | |
| 32 : public NON_EXPORTED_BASE(FileStreamWriter) { | |
| 33 public: | |
| 34 virtual ~LocalFileStreamWriter(); | |
| 35 | |
| 36 // FileStreamWriter overrides. | |
| 37 virtual int Write(net::IOBuffer* buf, int buf_len, | |
| 38 const net::CompletionCallback& callback) OVERRIDE; | |
| 39 virtual int Cancel(const net::CompletionCallback& callback) OVERRIDE; | |
| 40 virtual int Flush(const net::CompletionCallback& callback) OVERRIDE; | |
| 41 | |
| 42 private: | |
| 43 friend class content::LocalFileStreamWriterTest; | |
| 44 friend class FileStreamWriter; | |
| 45 LocalFileStreamWriter(base::TaskRunner* task_runner, | |
| 46 const base::FilePath& file_path, | |
| 47 int64 initial_offset, | |
| 48 OpenOrCreate open_or_create); | |
| 49 | |
| 50 // Opens |file_path_| and if it succeeds, proceeds to InitiateSeek(). | |
| 51 // If failed, the error code is returned by calling |error_callback|. | |
| 52 int InitiateOpen(const net::CompletionCallback& error_callback, | |
| 53 const base::Closure& main_operation); | |
| 54 void DidOpen(const net::CompletionCallback& error_callback, | |
| 55 const base::Closure& main_operation, | |
| 56 int result); | |
| 57 | |
| 58 // Seeks to |initial_offset_| and proceeds to |main_operation| if it succeeds. | |
| 59 // If failed, the error code is returned by calling |error_callback|. | |
| 60 void InitiateSeek(const net::CompletionCallback& error_callback, | |
| 61 const base::Closure& main_operation); | |
| 62 void DidSeek(const net::CompletionCallback& error_callback, | |
| 63 const base::Closure& main_operation, | |
| 64 int64 result); | |
| 65 | |
| 66 // Passed as the |main_operation| of InitiateOpen() function. | |
| 67 void ReadyToWrite(net::IOBuffer* buf, int buf_len, | |
| 68 const net::CompletionCallback& callback); | |
| 69 | |
| 70 // Writes asynchronously to the file. | |
| 71 int InitiateWrite(net::IOBuffer* buf, int buf_len, | |
| 72 const net::CompletionCallback& callback); | |
| 73 void DidWrite(const net::CompletionCallback& callback, int result); | |
| 74 | |
| 75 // Flushes asynchronously to the file. | |
| 76 int InitiateFlush(const net::CompletionCallback& callback); | |
| 77 void DidFlush(const net::CompletionCallback& callback, int result); | |
| 78 | |
| 79 // Stops the in-flight operation and calls |cancel_callback_| if it has been | |
| 80 // set by Cancel() for the current operation. | |
| 81 bool CancelIfRequested(); | |
| 82 | |
| 83 // Initialization parameters. | |
| 84 const base::FilePath file_path_; | |
| 85 OpenOrCreate open_or_create_; | |
| 86 const int64 initial_offset_; | |
| 87 scoped_refptr<base::TaskRunner> task_runner_; | |
| 88 | |
| 89 // Current states of the operation. | |
| 90 bool has_pending_operation_; | |
| 91 scoped_ptr<net::FileStream> stream_impl_; | |
| 92 net::CompletionCallback cancel_callback_; | |
| 93 | |
| 94 base::WeakPtrFactory<LocalFileStreamWriter> weak_factory_; | |
| 95 DISALLOW_COPY_AND_ASSIGN(LocalFileStreamWriter); | |
| 96 }; | |
| 97 | |
| 98 } // namespace storage | |
| 99 | |
| 100 #endif // WEBKIT_BROWSER_FILEAPI_LOCAL_FILE_STREAM_WRITER_H_ | |
| OLD | NEW |