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
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_PLUS_ADDRESSES_PLUS_ADDRESS_JIT_ALLOCATOR_H_
#define COMPONENTS_PLUS_ADDRESSES_PLUS_ADDRESS_JIT_ALLOCATOR_H_
#include <optional>
#include "base/containers/flat_map.h"
#include "base/memory/raw_ref.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "components/plus_addresses/plus_address_allocator.h"
#include "components/plus_addresses/plus_address_types.h"
#include "url/origin.h"
namespace plus_addresses {
class PlusAddressHttpClient;
class PlusAddressService;
class PlusAddressJitAllocator : public PlusAddressAllocator {
public:
explicit PlusAddressJitAllocator(PlusAddressHttpClient* http_client);
~PlusAddressJitAllocator() override;
// PlusAddressAllocator:
void AllocatePlusAddress(const url::Origin& origin,
AllocationMode mode,
PlusAddressRequestCallback callback) override;
std::optional<PlusProfile> AllocatePlusAddressSynchronously(
const url::Origin& origin,
AllocationMode mode) override;
bool IsRefreshingSupported(const url::Origin& origin) const override;
void RemoveAllocatedPlusAddress(const PlusAddress& plus_address) override;
private:
// Checks the `profile_or_error` response before passing it on to `callback`.
// If it receives a 429 code, it disables refreshing until a cooldown period
// has expired.
void HandleRefreshResponse(PlusAddressRequestCallback callback,
const PlusProfileOrError& profile_or_error);
// Responsible for server communication. Owned by the `PlusAddressService` and
// outlives `this`.
const raw_ref<PlusAddressHttpClient> http_client_;
// Counts how many refresh attempts where made for an `Origin`. Serves to
// limit the number of refresh requests per session.
base::flat_map<url::Origin, int> refresh_attempts_;
// The last known time that the user reached the refresh limit.
base::TimeTicks time_refresh_limit_reached_;
base::WeakPtrFactory<PlusAddressJitAllocator> weak_ptr_factory_{this};
};
} // namespace plus_addresses
#endif // COMPONENTS_PLUS_ADDRESSES_PLUS_ADDRESS_JIT_ALLOCATOR_H_
|
