-
Notifications
You must be signed in to change notification settings - Fork 6.6k
/
request_throttler.cc
94 lines (80 loc) · 3.34 KB
/
request_throttler.cc
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/feed/core/v2/request_throttler.h"
#include <vector>
#include "base/logging.h"
#include "components/feed/core/v2/config.h"
#include "components/feed/core/v2/prefs.h"
#include "components/prefs/pref_service.h"
namespace feed {
namespace {
// Returns the maximum number of requests per day for this request type.
// -1 indicates there is no limit.
int GetMaxRequestsPerDay(NetworkRequestType request_type) {
const Config& config = GetFeedConfig();
switch (request_type) {
case NetworkRequestType::kFeedQuery:
case NetworkRequestType::kWebFeedListContents:
case NetworkRequestType::kQueryInteractiveFeed:
case NetworkRequestType::kQueryBackgroundFeed:
return config.max_feed_query_requests_per_day;
case NetworkRequestType::kUploadActions:
return config.max_action_upload_requests_per_day;
case NetworkRequestType::kNextPage:
case NetworkRequestType::kQueryNextPage:
return config.max_next_page_requests_per_day;
case NetworkRequestType::kListWebFeeds:
return config.max_list_web_feeds_requests_per_day;
case NetworkRequestType::kListRecommendedWebFeeds:
return config.max_list_recommended_web_feeds_requests_per_day;
case NetworkRequestType::kUnfollowWebFeed:
case NetworkRequestType::kFollowWebFeed:
case NetworkRequestType::kSingleWebFeedListContents:
case NetworkRequestType::kQueryWebFeed:
return -1;
}
}
int DaysSinceOrigin(const base::Time& time_value) {
// |LocalMidnight()| DCHECKs on some platforms if |time_value| is too small
// (like zero). So if time is before the unix epoch, return 0.
return time_value < base::Time::UnixEpoch()
? 0
: time_value.LocalMidnight().since_origin().InDays();
}
} // namespace
RequestThrottler::RequestThrottler(PrefService* pref_service)
: pref_service_(pref_service) {
DCHECK(pref_service);
}
bool RequestThrottler::RequestQuota(NetworkRequestType request_type) {
ResetCountersIfDayChanged();
const int max_requests_per_day = GetMaxRequestsPerDay(request_type);
if (max_requests_per_day == -1)
return true;
// Fetch request counts from prefs. There's an entry for each request type.
// We may need to resize the list.
std::vector<int> request_counts =
feed::prefs::GetThrottlerRequestCounts(*pref_service_);
const size_t request_count_index = static_cast<size_t>(request_type);
if (request_counts.size() <= request_count_index)
request_counts.resize(request_count_index + 1);
int& requests_already_made = request_counts[request_count_index];
if (requests_already_made >= max_requests_per_day)
return false;
requests_already_made++;
feed::prefs::SetThrottlerRequestCounts(request_counts, *pref_service_);
return true;
}
void RequestThrottler::ResetCountersIfDayChanged() {
// Grant new quota on local midnight to spread out when clients that start
// making un-throttled requests to server.
const base::Time now = base::Time::Now();
const bool day_changed =
DaysSinceOrigin(feed::prefs::GetLastRequestTime(*pref_service_)) !=
DaysSinceOrigin(now);
feed::prefs::SetLastRequestTime(now, *pref_service_);
if (day_changed)
feed::prefs::SetThrottlerRequestCounts({}, *pref_service_);
}
} // namespace feed