Chromium Code Reviews| Index: content/browser/cache_storage/cache_storage_cache.cc |
| diff --git a/content/browser/cache_storage/cache_storage_cache.cc b/content/browser/cache_storage/cache_storage_cache.cc |
| index 6ca5786b882581b11095984e4565ee02f7e691fc..ff8778cd65481f3d95a5edf427ff2441ce054d95 100644 |
| --- a/content/browser/cache_storage/cache_storage_cache.cc |
| +++ b/content/browser/cache_storage/cache_storage_cache.cc |
| @@ -24,6 +24,7 @@ |
| #include "content/browser/cache_storage/cache_storage.pb.h" |
| #include "content/browser/cache_storage/cache_storage_blob_to_disk_cache.h" |
| #include "content/browser/cache_storage/cache_storage_cache_handle.h" |
| +#include "content/browser/cache_storage/cache_storage_cache_observer.h" |
| #include "content/browser/cache_storage/cache_storage_scheduler.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/common/referrer.h" |
| @@ -265,8 +266,8 @@ std::unique_ptr<CacheStorageCache> CacheStorageCache::CreateMemoryCache( |
| CacheStorageCache* cache = |
| new CacheStorageCache(origin, cache_name, base::FilePath(), cache_storage, |
| std::move(request_context_getter), |
| - std::move(quota_manager_proxy), blob_context); |
| - cache->InitBackend(); |
| + std::move(quota_manager_proxy), blob_context, 0); |
|
jkarlin
2016/11/11 18:24:57
nit: can you add /* cache size */ after the 0?
cmumford
2016/11/22 17:45:03
Done.
|
| + cache->SetObserver(cache_storage), cache->InitBackend(); |
| return base::WrapUnique(cache); |
| } |
| @@ -278,12 +279,13 @@ std::unique_ptr<CacheStorageCache> CacheStorageCache::CreatePersistentCache( |
| const base::FilePath& path, |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter, |
| scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy, |
| - base::WeakPtr<storage::BlobStorageContext> blob_context) { |
| - CacheStorageCache* cache = |
| - new CacheStorageCache(origin, cache_name, path, cache_storage, |
| - std::move(request_context_getter), |
| - std::move(quota_manager_proxy), blob_context); |
| - cache->InitBackend(); |
| + base::WeakPtr<storage::BlobStorageContext> blob_context, |
| + int64_t cache_size) { |
| + CacheStorageCache* cache = new CacheStorageCache( |
| + origin, cache_name, path, cache_storage, |
| + std::move(request_context_getter), std::move(quota_manager_proxy), |
| + blob_context, cache_size); |
| + cache->SetObserver(cache_storage), cache->InitBackend(); |
| return base::WrapUnique(cache); |
| } |
| @@ -497,6 +499,10 @@ void CacheStorageCache::GetSizeThenClose(const SizeCallback& callback) { |
| scheduler_->WrapCallbackToRunNext(callback)))); |
| } |
| +void CacheStorageCache::SetObserver(CacheStorageCacheObserver* observer) { |
| + cache_observer_ = observer; |
| +} |
| + |
| CacheStorageCache::~CacheStorageCache() { |
| quota_manager_proxy_->NotifyOriginNoLongerInUse(origin_); |
| } |
| @@ -508,7 +514,8 @@ CacheStorageCache::CacheStorageCache( |
| CacheStorage* cache_storage, |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter, |
| scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy, |
| - base::WeakPtr<storage::BlobStorageContext> blob_context) |
| + base::WeakPtr<storage::BlobStorageContext> blob_context, |
| + int64_t cache_size) |
| : origin_(origin), |
| cache_name_(cache_name), |
| path_(path), |
| @@ -518,7 +525,9 @@ CacheStorageCache::CacheStorageCache( |
| blob_storage_context_(blob_context), |
| scheduler_( |
| new CacheStorageScheduler(CacheStorageSchedulerClient::CLIENT_CACHE)), |
| + cache_size_(cache_size), |
| max_query_size_bytes_(kMaxQueryCacheResultBytes), |
| + cache_observer_(nullptr), |
| memory_only_(path.empty()), |
| weak_ptr_factory_(this) { |
| DCHECK(!origin_.is_empty()); |
| @@ -1184,12 +1193,19 @@ void CacheStorageCache::UpdateCacheSize() { |
| void CacheStorageCache::UpdateCacheSizeGotSize( |
| std::unique_ptr<CacheStorageCacheHandle> cache_handle, |
| int current_cache_size) { |
| + DCHECK_NE(current_cache_size, CacheStorage::kSizeUnknown); |
| int64_t old_cache_size = cache_size_; |
| cache_size_ = current_cache_size; |
| + int64_t size_delta = current_cache_size - old_cache_size; |
| + |
| quota_manager_proxy_->NotifyStorageModified( |
| storage::QuotaClient::kServiceWorkerCache, origin_, |
| - storage::kStorageTypeTemporary, current_cache_size - old_cache_size); |
| + storage::kStorageTypeTemporary, size_delta); |
| + |
| + if (cache_observer_) |
| + cache_observer_->CacheSizeSet( |
| + this, CacheStorageCacheObserver::Whence::DELTA, size_delta); |
| } |
| void CacheStorageCache::Delete(const CacheStorageBatchOperation& operation, |
| @@ -1374,6 +1390,11 @@ void CacheStorageCache::InitDidCreateBackend( |
| void CacheStorageCache::InitGotCacheSize(const base::Closure& callback, |
| CacheStorageError cache_create_error, |
| int cache_size) { |
| + if (cache_size_ != CacheStorage::kSizeUnknown) { |
| + // If this check fails, either there is a flaw in how Chrome remembers the |
| + // cache size, or the files in the cache have been modified. |
| + DCHECK_EQ(cache_size_, cache_size); |
|
jkarlin
2016/11/11 18:24:57
Hmm, how about DCHECK(cache_size == CacheStorage::
cmumford
2016/11/22 17:45:03
Done.
|
| + } |
| cache_size_ = cache_size; |
| initializing_ = false; |
| backend_state_ = (cache_create_error == CACHE_STORAGE_OK && backend_ && |
| @@ -1384,6 +1405,10 @@ void CacheStorageCache::InitGotCacheSize(const base::Closure& callback, |
| UMA_HISTOGRAM_ENUMERATION("ServiceWorkerCache.InitBackendResult", |
| cache_create_error, CACHE_STORAGE_ERROR_LAST + 1); |
| + if (cache_observer_) |
| + cache_observer_->CacheSizeSet(this, CacheStorageCacheObserver::Whence::SET, |
| + cache_size_); |
| + |
| callback.Run(); |
| } |