Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "components/data_use_measurement/core/data_use_recorder.h" | 5 #include "components/data_use_measurement/core/data_use_recorder.h" |
| 6 | |
|
RyanSturm
2016/11/29 20:33:00
Add this whitespace back.
Not at Google. Contact bengr
2016/11/30 00:21:16
Done.
| |
| 7 #include "net/url_request/url_request.h" | 6 #include "net/url_request/url_request.h" |
| 8 | 7 |
| 9 namespace data_use_measurement { | 8 namespace data_use_measurement { |
| 10 | 9 |
| 11 DataUseRecorder::DataUseRecorder() {} | 10 DataUseRecorder::DataUseRecorder() : main_url_request_(nullptr) {} |
| 12 | 11 |
| 13 DataUseRecorder::~DataUseRecorder() {} | 12 DataUseRecorder::~DataUseRecorder() {} |
| 14 | 13 |
| 15 bool DataUseRecorder::IsDataUseComplete() { | 14 bool DataUseRecorder::IsDataUseComplete() { |
| 16 return pending_url_requests_.empty() && pending_data_sources_.empty(); | 15 return pending_url_requests_.empty() && pending_data_sources_.empty(); |
| 17 } | 16 } |
| 18 | 17 |
| 19 void DataUseRecorder::OnBeforeUrlRequest(net::URLRequest* request) { | 18 void DataUseRecorder::AddPendingURLRequest(net::URLRequest* request) { |
| 20 pending_url_requests_.insert(request); | 19 pending_url_requests_.insert(request); |
| 21 } | 20 } |
| 22 | 21 |
| 23 void DataUseRecorder::OnUrlRequestDestroyed(net::URLRequest* request) { | 22 void DataUseRecorder::OnUrlRequestDestroyed(net::URLRequest* request) { |
| 24 pending_url_requests_.erase(request); | 23 pending_url_requests_.erase(request); |
| 25 } | 24 } |
| 26 | 25 |
| 26 void DataUseRecorder::RemoveAllPendingURLRequests() { | |
| 27 pending_url_requests_.clear(); | |
| 28 } | |
| 29 | |
| 30 void DataUseRecorder::OnBeforeUrlRequest(net::URLRequest* request) {} | |
| 31 | |
| 27 void DataUseRecorder::OnNetworkBytesReceived(net::URLRequest* request, | 32 void DataUseRecorder::OnNetworkBytesReceived(net::URLRequest* request, |
| 28 int64_t bytes_received) { | 33 int64_t bytes_received) { |
| 29 data_use_.total_bytes_received_ += bytes_received; | 34 data_use_.total_bytes_received_ += bytes_received; |
| 30 } | 35 } |
| 31 | 36 |
| 32 void DataUseRecorder::OnNetworkBytesSent(net::URLRequest* request, | 37 void DataUseRecorder::OnNetworkBytesSent(net::URLRequest* request, |
| 33 int64_t bytes_sent) { | 38 int64_t bytes_sent) { |
| 34 data_use_.total_bytes_sent_ += bytes_sent; | 39 data_use_.total_bytes_sent_ += bytes_sent; |
| 35 } | 40 } |
| 36 | 41 |
| 37 void DataUseRecorder::AddPendingDataSource(void* source) { | 42 void DataUseRecorder::AddPendingDataSource(void* source) { |
| 38 pending_data_sources_.insert(source); | 43 pending_data_sources_.insert(source); |
| 39 } | 44 } |
| 40 | 45 |
| 41 bool DataUseRecorder::HasPendingDataSource(void* source) { | 46 bool DataUseRecorder::HasPendingDataSource(void* source) { |
| 42 return pending_data_sources_.find(source) != pending_data_sources_.end(); | 47 return pending_data_sources_.find(source) != pending_data_sources_.end(); |
| 43 } | 48 } |
| 44 | 49 |
| 45 void DataUseRecorder::RemovePendingDataSource(void* source) { | 50 void DataUseRecorder::RemovePendingDataSource(void* source) { |
| 46 pending_data_sources_.erase(source); | 51 pending_data_sources_.erase(source); |
| 47 } | 52 } |
| 48 | 53 |
| 49 bool DataUseRecorder::HasPendingURLRequest(const net::URLRequest* request) { | 54 bool DataUseRecorder::HasPendingURLRequest(net::URLRequest* request) { |
| 50 return pending_url_requests_.find(request) != pending_url_requests_.end(); | 55 return pending_url_requests_.find(request) != pending_url_requests_.end(); |
| 51 } | 56 } |
| 52 | 57 |
| 58 void DataUseRecorder::MergeFrom(DataUseRecorder* other) { | |
| 59 data_use_.MergeFrom(other->data_use()); | |
| 60 } | |
| 61 | |
| 53 } // namespace data_use_measurement | 62 } // namespace data_use_measurement |
| OLD | NEW |