diff options
Diffstat (limited to 'chromium/google_apis/drive/request_sender.h')
-rw-r--r-- | chromium/google_apis/drive/request_sender.h | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/chromium/google_apis/drive/request_sender.h b/chromium/google_apis/drive/request_sender.h new file mode 100644 index 00000000000..dfc671c42bf --- /dev/null +++ b/chromium/google_apis/drive/request_sender.h @@ -0,0 +1,111 @@ +// 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. + +#ifndef GOOGLE_APIS_DRIVE_REQUEST_SENDER_H_ +#define GOOGLE_APIS_DRIVE_REQUEST_SENDER_H_ + +#include <set> +#include <string> + +#include "base/basictypes.h" +#include "base/callback_forward.h" +#include "base/memory/ref_counted.h" +#include "base/memory/scoped_ptr.h" +#include "base/memory/weak_ptr.h" +#include "base/threading/thread_checker.h" +#include "google_apis/drive/gdata_errorcode.h" + +namespace base { +class SequencedTaskRunner; +} + +namespace net { +class URLRequestContextGetter; +} + +namespace google_apis { + +class AuthenticatedRequestInterface; +class AuthServiceInterface; + +// Helper class that sends requests implementing +// AuthenticatedRequestInterface and handles retries and authentication. +class RequestSender { + public: + // |auth_service| is used for fetching OAuth tokens. It'll be owned by + // this RequestSender. + // + // |url_request_context_getter| is the context used to perform network + // requests from this RequestSender. + // + // |blocking_task_runner| is used for running blocking operation, e.g., + // parsing JSON response from the server. + // + // |custom_user_agent| will be used for the User-Agent header in HTTP + // requests issued through the request sender if the value is not empty. + RequestSender(AuthServiceInterface* auth_service, + net::URLRequestContextGetter* url_request_context_getter, + base::SequencedTaskRunner* blocking_task_runner, + const std::string& custom_user_agent); + ~RequestSender(); + + AuthServiceInterface* auth_service() { return auth_service_.get(); } + + net::URLRequestContextGetter* url_request_context_getter() const { + return url_request_context_getter_; + } + + base::SequencedTaskRunner* blocking_task_runner() const { + return blocking_task_runner_.get(); + } + + // Starts a request implementing the AuthenticatedRequestInterface + // interface, and makes the request retry upon authentication failures by + // calling back to RetryRequest. The |request| object is owned by this + // RequestSender. It will be deleted in RequestSender's destructor or + // in RequestFinished(). + // + // Returns a closure to cancel the request. The closure cancels the request + // if it is in-flight, and does nothing if it is already terminated. + base::Closure StartRequestWithRetry(AuthenticatedRequestInterface* request); + + // Notifies to this RequestSender that |request| has finished. + // TODO(kinaba): refactor the life time management and make this at private. + void RequestFinished(AuthenticatedRequestInterface* request); + + private: + // Called when the access token is fetched. + void OnAccessTokenFetched( + const base::WeakPtr<AuthenticatedRequestInterface>& request, + GDataErrorCode error, + const std::string& access_token); + + // Clears any authentication token and retries the request, which forces + // an authentication token refresh. + void RetryRequest(AuthenticatedRequestInterface* request); + + // Cancels the request. Used for implementing the returned closure of + // StartRequestWithRetry. + void CancelRequest( + const base::WeakPtr<AuthenticatedRequestInterface>& request); + + scoped_ptr<AuthServiceInterface> auth_service_; + scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_; + scoped_refptr<base::SequencedTaskRunner> blocking_task_runner_; + + std::set<AuthenticatedRequestInterface*> in_flight_requests_; + const std::string custom_user_agent_; + + base::ThreadChecker thread_checker_; + + // Note: This should remain the last member so it'll be destroyed and + // invalidate its weak pointers before any other members are destroyed. + base::WeakPtrFactory<RequestSender> weak_ptr_factory_; + + DISALLOW_COPY_AND_ASSIGN(RequestSender); +}; + +} // namespace google_apis + +#endif // GOOGLE_APIS_DRIVE_REQUEST_SENDER_H_ |