Network traffic annotation added to drive/base_requests.

components/drive/service/drive_api_service.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/drive/service/drive_api_service.h"
 
 #include <stddef.h>
 
 #include <string>
 #include <vector>
@@ skipping 236 matching lines @@
   batch_request_->Commit();
   batch_request_.reset();
 }
 
 DriveAPIService::DriveAPIService(
     OAuth2TokenService* oauth2_token_service,
     net::URLRequestContextGetter* url_request_context_getter,
     base::SequencedTaskRunner* blocking_task_runner,
     const GURL& base_url,
     const GURL& base_thumbnail_url,
-    const std::string& custom_user_agent)
+    const std::string& custom_user_agent,
+    const net::NetworkTrafficAnnotationTag& traffic_annotation)
     : oauth2_token_service_(oauth2_token_service),
       url_request_context_getter_(url_request_context_getter),
       blocking_task_runner_(blocking_task_runner),
-      url_generator_(base_url, base_thumbnail_url,
+      url_generator_(base_url,
+                     base_thumbnail_url,
                      google_apis::GetTeamDrivesIntegrationSwitch()),
-      custom_user_agent_(custom_user_agent) {
-}
+      custom_user_agent_(custom_user_agent),
+      traffic_annotation_(traffic_annotation) {}
 
 DriveAPIService::~DriveAPIService() {
   DCHECK(thread_checker_.CalledOnValidThread());
   if (sender_.get())
     sender_->auth_service()->RemoveObserver(this);
 }
 
 void DriveAPIService::Initialize(const std::string& account_id) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
   std::vector<std::string> scopes;
   scopes.push_back(kDriveScope);
   scopes.push_back(kDriveAppsReadonlyScope);
   scopes.push_back(kDriveAppsScope);
 
   // Note: The following scope is used to support GetShareUrl on Drive API v2.
   // Unfortunately, there is no support on Drive API v2, so we need to fall back
   // to GData WAPI for the GetShareUrl.
   scopes.push_back(kDocsListScope);
 
   sender_.reset(new RequestSender(
-      new google_apis::AuthService(oauth2_token_service_,
-                                   account_id,
-                                   url_request_context_getter_.get(),
-                                   scopes),
-      url_request_context_getter_.get(),
-      blocking_task_runner_.get(),
-      custom_user_agent_));
+      new google_apis::AuthService(oauth2_token_service_, account_id,
+                                   url_request_context_getter_.get(), scopes),
+      url_request_context_getter_.get(), blocking_task_runner_.get(),
+      custom_user_agent_, traffic_annotation_));
   sender_->auth_service()->AddObserver(this);
 
   files_list_request_runner_.reset(
       new FilesListRequestRunner(sender_.get(), url_generator_));
 }
 
 void DriveAPIService::AddObserver(DriveServiceObserver* observer) {
   observers_.AddObserver(observer);
 }
 
@@ skipping 575 matching lines @@
   // RequestSender before the request is committed because the request has a
   // reference to RequestSender and we should ensure to delete the request when
   // the sender is deleted. Resolve the circulating dependency and fix it.
   const google_apis::CancelCallback callback =
       sender_->StartRequestWithAuthRetry(std::move(request));
   return base::MakeUnique<BatchRequestConfigurator>(
       weak_ref, sender_->blocking_task_runner(), url_generator_, callback);
 }
 
 }  // namespace drive