1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "content/browser/feature_observer.h"
6
7 #include "content/public/browser/content_browser_client.h"
8 #include "content/public/browser/feature_observer_client.h"
9 #include "content/public/common/content_client.h"
10
11 namespace content {
12
FeatureObserver(FeatureObserverClient * client,GlobalFrameRoutingId id)13 FeatureObserver::FeatureObserver(FeatureObserverClient* client,
14 GlobalFrameRoutingId id)
15 : client_(client), id_(id) {
16 DCHECK(client_);
17
18 for (size_t i = 0;
19 i <= static_cast<size_t>(blink::mojom::ObservedFeatureType::kMaxValue);
20 ++i) {
21 features_by_type_[i].set_disconnect_handler(base::BindRepeating(
22 [](mojo::ReceiverSet<blink::mojom::ObservedFeature>* set,
23 FeatureObserverClient* client, GlobalFrameRoutingId id,
24 blink::mojom::ObservedFeatureType type) {
25 if (!set->empty())
26 return;
27
28 // Notify if this is the last receiver.
29 client->OnStopUsing(id, type);
30 },
31 base::Unretained(&features_by_type_[i]), client_, id_,
32 static_cast<blink::mojom::ObservedFeatureType>(i)));
33 }
34 }
35
36 FeatureObserver::~FeatureObserver() = default;
37
GetFeatureObserver(mojo::PendingReceiver<blink::mojom::FeatureObserver> receiver)38 void FeatureObserver::GetFeatureObserver(
39 mojo::PendingReceiver<blink::mojom::FeatureObserver> receiver) {
40 observers_.Add(this, std::move(receiver));
41 }
42
Register(mojo::PendingReceiver<blink::mojom::ObservedFeature> feature,blink::mojom::ObservedFeatureType type)43 void FeatureObserver::Register(
44 mojo::PendingReceiver<blink::mojom::ObservedFeature> feature,
45 blink::mojom::ObservedFeatureType type) {
46 DCHECK(client_);
47
48 auto& set = features_by_type_[static_cast<int>(type)];
49
50 // Notify if this is the first receiver.
51 if (set.empty())
52 client_->OnStartUsing(id_, type);
53
54 set.Add(nullptr, std::move(feature));
55 }
56
57 } // namespace content
58