1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef GOOGLE_APIS_DRIVE_FILES_LIST_REQUEST_RUNNER_H_
#define GOOGLE_APIS_DRIVE_FILES_LIST_REQUEST_RUNNER_H_
#include <memory>
#include <string>
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "google_apis/drive/drive_api_requests.h"
#include "google_apis/drive/drive_api_url_generator.h"
#include "google_apis/drive/drive_common_callbacks.h"
namespace google_apis {
class RequestSender;
// Runs file list requests (the FileListRequest class) with a backoff retry
// logic in case of the DRIVE_RESPONSE_TOO_LARGE error code.
class FilesListRequestRunner {
public:
FilesListRequestRunner(
RequestSender* request_sender,
const google_apis::DriveApiUrlGenerator& url_generator);
// Creates a FilesListRequest instance and starts the request with a backoff
// retry in case of DRIVE_RESPONSE_TOO_LARGE error code.
CancelCallbackOnce CreateAndStartWithSizeBackoff(
int max_results,
FilesListCorpora corpora,
const std::string& team_drive_id,
const std::string& q,
const std::string& fields,
FileListCallback callback);
FilesListRequestRunner(const FilesListRequestRunner&) = delete;
FilesListRequestRunner& operator=(const FilesListRequestRunner&) = delete;
~FilesListRequestRunner();
void SetRequestCompletedCallbackForTesting(base::OnceClosure callback);
private:
// Called when the cancelling callback returned by
// CreateAndStartWithSizeBackoff is invoked. Once called cancels the current
// request.
void OnCancel(CancelCallbackOnce* cancel_callback);
// Called when a single request is completed with either a success or an
// error. In case of DRIVE_RESPONSE_TOO_LARGE it will retry the request with
// half of the requests.
void OnCompleted(int max_results,
FilesListCorpora corpora,
const std::string& team_drive_id,
const std::string& q,
const std::string& fields,
FileListCallback callback,
CancelCallbackOnce* cancel_callback,
ApiErrorCode error,
std::unique_ptr<FileList> entry);
raw_ptr<RequestSender> request_sender_; // Not owned.
const google_apis::DriveApiUrlGenerator url_generator_; // Not owned.
base::OnceClosure request_completed_callback_for_testing_;
// 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<FilesListRequestRunner> weak_ptr_factory_{this};
};
} // namespace google_apis
#endif // GOOGLE_APIS_DRIVE_FILES_LIST_REQUEST_RUNNER_H_
|
