Chromium Code Reviews| Index: components/data_use_measurement/core/data_use_measurement.cc |
| diff --git a/components/data_use_measurement/core/data_use_measurement.cc b/components/data_use_measurement/core/data_use_measurement.cc |
| index 5ded37e7fe909e0d0e6929b772d9cec655a7afa9..b0ff81ddce705482693c492276114b1e4488c224 100644 |
| --- a/components/data_use_measurement/core/data_use_measurement.cc |
| +++ b/components/data_use_measurement/core/data_use_measurement.cc |
| @@ -10,6 +10,7 @@ |
| #include "build/build_config.h" |
| #include "components/data_use_measurement/core/data_use_user_data.h" |
| #include "components/data_use_measurement/core/url_request_classifier.h" |
| +#include "components/domain_reliability/uploader.h" |
| #include "net/base/network_change_notifier.h" |
| #include "net/base/upload_data_stream.h" |
| #include "net/http/http_response_headers.h" |
| @@ -92,6 +93,14 @@ void DataUseMeasurement::OnBeforeURLRequest(net::URLRequest* request) { |
| data_use_user_data = new DataUseUserData( |
| DataUseUserData::ServiceName::NOT_TAGGED, CurrentAppState()); |
| request->SetUserData(DataUseUserData::kUserDataKey, data_use_user_data); |
| + } else if (domain_reliability::DomainReliabilityUploader:: |
|
RyanSturm
2016/12/01 01:45:20
If domain_reliability isn't adding DataUseUserData
|
| + OriginatedFromDomainReliability(*request)) { |
| + // Detect if the request originated from DomainReliability. |
| + // DataUseUserData::AttachToFetcher() cannot be called from domain |
| + // reliability, since it sets userdata on URLFetcher for its purposes. |
| + data_use_user_data = new DataUseUserData( |
| + DataUseUserData::ServiceName::DOMAIN_RELIABILITY, CurrentAppState()); |
| + request->SetUserData(DataUseUserData::kUserDataKey, data_use_user_data); |
| } |
| } |