| // Copyright 2016 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 COMPONENTS_BROWSING_DATA_CONTENT_CONDITIONAL_CACHE_COUNTING_HELPER_H_ |
| #define COMPONENTS_BROWSING_DATA_CONTENT_CONDITIONAL_CACHE_COUNTING_HELPER_H_ |
| |
| #include <memory> |
| |
| #include "base/callback_forward.h" |
| #include "net/base/net_errors.h" |
| #include "net/disk_cache/disk_cache.h" |
| |
| namespace content { |
| class StoragePartition; |
| } |
| |
| namespace net { |
| class URLRequestContextGetter; |
| } |
| |
| namespace browsing_data { |
| |
| // Helper to count the size of the http cache data from a StoragePartition. |
| class ConditionalCacheCountingHelper { |
| public: |
| // Returns the number bytes in the selected range and if this value is an |
| // upper estimate. |
| typedef base::Callback<void(int64_t, bool)> CacheCountCallback; |
| |
| static ConditionalCacheCountingHelper* CreateForRange( |
| content::StoragePartition* storage_partition, |
| base::Time begin_time, |
| base::Time end_time); |
| |
| // Count the cache entries according to the specified time range. Destroys |
| // this instance of ConditionalCacheCountingHelper when finished. |
| // Must be called on the UI thread! |
| // |
| // The |completion_callback| will be invoked when the operation completes. |
| base::WeakPtr<ConditionalCacheCountingHelper> CountAndDestroySelfWhenFinished( |
| const CacheCountCallback& result_callback); |
| |
| bool IsFinished(); |
| |
| private: |
| enum class CacheState { |
| NONE, |
| CREATE_MAIN, |
| CREATE_MEDIA, |
| COUNT_MAIN, |
| COUNT_MEDIA, |
| DONE |
| }; |
| |
| friend class base::DeleteHelper<ConditionalCacheCountingHelper>; |
| |
| ConditionalCacheCountingHelper( |
| base::Time begin_time, |
| base::Time end_time, |
| net::URLRequestContextGetter* main_context_getter, |
| net::URLRequestContextGetter* media_context_getter); |
| ~ConditionalCacheCountingHelper(); |
| |
| void Finished(); |
| |
| void CountHttpCacheOnIOThread(); |
| void DoCountCache(int rv); |
| |
| // Stores the cache size computation result before it can be returned |
| // via a callback. This is either the sum of size of the the two cache |
| // backends, or an error code if the calculation failed. |
| int64_t calculation_result_; |
| bool is_upper_limit_; |
| |
| CacheCountCallback result_callback_; |
| const base::Time begin_time_; |
| const base::Time end_time_; |
| |
| bool is_finished_; |
| |
| const scoped_refptr<net::URLRequestContextGetter> main_context_getter_; |
| const scoped_refptr<net::URLRequestContextGetter> media_context_getter_; |
| |
| CacheState next_cache_state_; |
| disk_cache::Backend* cache_; |
| |
| std::unique_ptr<disk_cache::Backend::Iterator> iterator_; |
| |
| base::WeakPtrFactory<ConditionalCacheCountingHelper> weak_ptr_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ConditionalCacheCountingHelper); |
| }; |
| |
| } // namespace browsing_data |
| |
| #endif // COMPONENTS_BROWSING_DATA_CONTENT_CONDITIONAL_CACHE_COUNTING_HELPER_H_ |