OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "webkit/quota/mock_storage_client.h" | 5 #include "webkit/quota/mock_storage_client.h" |
6 | 6 |
7 #include "base/atomic_sequence_num.h" | 7 #include "base/atomic_sequence_num.h" |
8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 9 #include "base/bind.h" |
9 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
10 #include "base/memory/singleton.h" | 11 #include "base/memory/singleton.h" |
11 #include "base/message_loop_proxy.h" | 12 #include "base/message_loop_proxy.h" |
12 #include "base/stl_util.h" | 13 #include "base/stl_util.h" |
13 #include "net/base/net_util.h" | 14 #include "net/base/net_util.h" |
14 #include "webkit/quota/quota_manager.h" | 15 #include "webkit/quota/quota_manager.h" |
15 | 16 |
16 using base::AtomicSequenceNumber; | 17 using base::AtomicSequenceNumber; |
17 | 18 |
18 namespace quota { | 19 namespace quota { |
(...skipping 22 matching lines...) Expand all Loading... |
41 }; | 42 }; |
42 | 43 |
43 } // anonymous namespace | 44 } // anonymous namespace |
44 | 45 |
45 MockStorageClient::MockStorageClient( | 46 MockStorageClient::MockStorageClient( |
46 QuotaManagerProxy* quota_manager_proxy, | 47 QuotaManagerProxy* quota_manager_proxy, |
47 const MockOriginData* mock_data, size_t mock_data_size) | 48 const MockOriginData* mock_data, size_t mock_data_size) |
48 : quota_manager_proxy_(quota_manager_proxy), | 49 : quota_manager_proxy_(quota_manager_proxy), |
49 id_(MockStorageClientIDSequencer::GetInstance()->NextMockID()), | 50 id_(MockStorageClientIDSequencer::GetInstance()->NextMockID()), |
50 mock_time_counter_(0), | 51 mock_time_counter_(0), |
51 runnable_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) { | 52 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) { |
52 for (size_t i = 0; i < mock_data_size; ++i) { | 53 for (size_t i = 0; i < mock_data_size; ++i) { |
53 origin_data_[make_pair(GURL(mock_data[i].origin), mock_data[i].type)] = | 54 origin_data_[make_pair(GURL(mock_data[i].origin), mock_data[i].type)] = |
54 mock_data[i].usage; | 55 mock_data[i].usage; |
55 } | 56 } |
56 } | 57 } |
57 | 58 |
58 MockStorageClient::~MockStorageClient() {} | 59 MockStorageClient::~MockStorageClient() {} |
59 | 60 |
60 void MockStorageClient::AddOriginAndNotify( | 61 void MockStorageClient::AddOriginAndNotify( |
61 const GURL& origin_url, StorageType type, int64 size) { | 62 const GURL& origin_url, StorageType type, int64 size) { |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
102 } | 103 } |
103 | 104 |
104 void MockStorageClient::OnQuotaManagerDestroyed() { | 105 void MockStorageClient::OnQuotaManagerDestroyed() { |
105 delete this; | 106 delete this; |
106 } | 107 } |
107 | 108 |
108 void MockStorageClient::GetOriginUsage(const GURL& origin_url, | 109 void MockStorageClient::GetOriginUsage(const GURL& origin_url, |
109 StorageType type, | 110 StorageType type, |
110 const GetUsageCallback& callback) { | 111 const GetUsageCallback& callback) { |
111 base::MessageLoopProxy::current()->PostTask( | 112 base::MessageLoopProxy::current()->PostTask( |
112 FROM_HERE, runnable_factory_.NewRunnableMethod( | 113 FROM_HERE, |
113 &MockStorageClient::RunGetOriginUsage, | 114 base::Bind(&MockStorageClient::RunGetOriginUsage, |
114 origin_url, type, callback)); | 115 weak_factory_.GetWeakPtr(), origin_url, type, callback)); |
115 } | 116 } |
116 | 117 |
117 void MockStorageClient::GetOriginsForType( | 118 void MockStorageClient::GetOriginsForType( |
118 StorageType type, const GetOriginsCallback& callback) { | 119 StorageType type, const GetOriginsCallback& callback) { |
119 base::MessageLoopProxy::current()->PostTask( | 120 base::MessageLoopProxy::current()->PostTask( |
120 FROM_HERE, runnable_factory_.NewRunnableMethod( | 121 FROM_HERE, |
121 &MockStorageClient::RunGetOriginsForType, | 122 base::Bind(&MockStorageClient::RunGetOriginsForType, |
122 type, callback)); | 123 weak_factory_.GetWeakPtr(), type, callback)); |
123 } | 124 } |
124 | 125 |
125 void MockStorageClient::GetOriginsForHost( | 126 void MockStorageClient::GetOriginsForHost( |
126 StorageType type, const std::string& host, | 127 StorageType type, const std::string& host, |
127 const GetOriginsCallback& callback) { | 128 const GetOriginsCallback& callback) { |
128 base::MessageLoopProxy::current()->PostTask( | 129 base::MessageLoopProxy::current()->PostTask( |
129 FROM_HERE, runnable_factory_.NewRunnableMethod( | 130 FROM_HERE, |
130 &MockStorageClient::RunGetOriginsForHost, | 131 base::Bind(&MockStorageClient::RunGetOriginsForHost, |
131 type, host, callback)); | 132 weak_factory_.GetWeakPtr(), type, host, callback)); |
132 } | 133 } |
133 | 134 |
134 void MockStorageClient::DeleteOriginData( | 135 void MockStorageClient::DeleteOriginData( |
135 const GURL& origin, StorageType type, | 136 const GURL& origin, StorageType type, |
136 const DeletionCallback& callback) { | 137 const DeletionCallback& callback) { |
137 base::MessageLoopProxy::current()->PostTask( | 138 base::MessageLoopProxy::current()->PostTask( |
138 FROM_HERE, runnable_factory_.NewRunnableMethod( | 139 FROM_HERE, |
139 &MockStorageClient::RunDeleteOriginData, | 140 base::Bind(&MockStorageClient::RunDeleteOriginData, |
140 origin, type, callback)); | 141 weak_factory_.GetWeakPtr(), origin, type, callback)); |
141 } | 142 } |
142 | 143 |
143 void MockStorageClient::RunGetOriginUsage( | 144 void MockStorageClient::RunGetOriginUsage( |
144 const GURL& origin_url, StorageType type, | 145 const GURL& origin_url, StorageType type, |
145 const GetUsageCallback& callback) { | 146 const GetUsageCallback& callback) { |
146 OriginDataMap::iterator find = origin_data_.find(make_pair(origin_url, type)); | 147 OriginDataMap::iterator find = origin_data_.find(make_pair(origin_url, type)); |
147 if (find == origin_data_.end()) { | 148 if (find == origin_data_.end()) { |
148 callback.Run(0); | 149 callback.Run(0); |
149 } else { | 150 } else { |
150 callback.Run(find->second); | 151 callback.Run(find->second); |
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
192 int64 delta = itr->second; | 193 int64 delta = itr->second; |
193 quota_manager_proxy_-> | 194 quota_manager_proxy_-> |
194 NotifyStorageModified(id(), origin_url, type, -delta); | 195 NotifyStorageModified(id(), origin_url, type, -delta); |
195 origin_data_.erase(itr); | 196 origin_data_.erase(itr); |
196 } | 197 } |
197 | 198 |
198 callback.Run(kQuotaStatusOk); | 199 callback.Run(kQuotaStatusOk); |
199 } | 200 } |
200 | 201 |
201 } // namespace quota | 202 } // namespace quota |
OLD | NEW |