OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ | 5 #ifndef CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ |
6 #define CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ | 6 #define CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
(...skipping 14 matching lines...) Expand all Loading... |
25 | 25 |
26 // Registers a stream, and sets its URL. | 26 // Registers a stream, and sets its URL. |
27 void RegisterStream(scoped_refptr<Stream> stream); | 27 void RegisterStream(scoped_refptr<Stream> stream); |
28 | 28 |
29 // Clones a stream. Returns true on success, or false if |src_url| doesn't | 29 // Clones a stream. Returns true on success, or false if |src_url| doesn't |
30 // exist. | 30 // exist. |
31 bool CloneStream(const GURL& url, const GURL& src_url); | 31 bool CloneStream(const GURL& url, const GURL& src_url); |
32 | 32 |
33 void UnregisterStream(const GURL& url); | 33 void UnregisterStream(const GURL& url); |
34 | 34 |
| 35 // Called by Stream instances to request increase of memory usage. If the |
| 36 // total memory usage for this registry is going to exceed the limit, |
| 37 // returns false. Otherwise, updates |total_memory_usage_| and returns true. |
| 38 // |
| 39 // |current_size| is the up-to-date size of ByteStream of the Stream instance |
| 40 // and |increase| must be the amount of data going to be added to the Stream |
| 41 // instance. |
| 42 bool UpdateMemoryUsage(const GURL& url, size_t current_size, size_t increase); |
| 43 |
35 // Gets the stream associated with |url|. Returns NULL if there is no such | 44 // Gets the stream associated with |url|. Returns NULL if there is no such |
36 // stream. | 45 // stream. |
37 scoped_refptr<Stream> GetStream(const GURL& url); | 46 scoped_refptr<Stream> GetStream(const GURL& url); |
38 | 47 |
| 48 void set_max_memory_usage_for_testing(size_t size) { |
| 49 max_memory_usage_ = size; |
| 50 } |
| 51 |
39 private: | 52 private: |
40 typedef std::map<GURL, scoped_refptr<Stream> > StreamMap; | 53 typedef std::map<GURL, scoped_refptr<Stream> > StreamMap; |
41 | 54 |
42 StreamMap streams_; | 55 StreamMap streams_; |
43 | 56 |
| 57 size_t total_memory_usage_; |
| 58 |
| 59 // Maximum amount of memory allowed to use for Stream instances registered |
| 60 // with this registry. |
| 61 size_t max_memory_usage_; |
| 62 |
44 DISALLOW_COPY_AND_ASSIGN(StreamRegistry); | 63 DISALLOW_COPY_AND_ASSIGN(StreamRegistry); |
45 }; | 64 }; |
46 | 65 |
47 } // namespace content | 66 } // namespace content |
48 | 67 |
49 #endif // CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ | 68 #endif // CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_ |
50 | |
51 | |
OLD | NEW |