summaryrefslogtreecommitdiff
path: root/chromium/content/browser/feature_observer.cc
blob: d5fd35b5b2368d897909095893d7054fd8f55d94 (plain)
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
// Copyright 2020 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.

#include "content/browser/feature_observer.h"

#include "content/public/browser/content_browser_client.h"
#include "content/public/browser/feature_observer_client.h"
#include "content/public/common/content_client.h"

namespace content {

FeatureObserver::FeatureObserver(FeatureObserverClient* client,
                                 GlobalRenderFrameHostId id)
    : client_(client), id_(id) {
  DCHECK(client_);

  for (size_t i = 0;
       i <= static_cast<size_t>(blink::mojom::ObservedFeatureType::kMaxValue);
       ++i) {
    features_by_type_[i].set_disconnect_handler(base::BindRepeating(
        [](mojo::ReceiverSet<blink::mojom::ObservedFeature>* set,
           FeatureObserverClient* client, GlobalRenderFrameHostId id,
           blink::mojom::ObservedFeatureType type) {
          if (!set->empty())
            return;

          // Notify if this is the last receiver.
          client->OnStopUsing(id, type);
        },
        base::Unretained(&features_by_type_[i]), client_, id_,
        static_cast<blink::mojom::ObservedFeatureType>(i)));
  }
}

FeatureObserver::~FeatureObserver() = default;

void FeatureObserver::GetFeatureObserver(
    mojo::PendingReceiver<blink::mojom::FeatureObserver> receiver) {
  observers_.Add(this, std::move(receiver));
}

void FeatureObserver::Register(
    mojo::PendingReceiver<blink::mojom::ObservedFeature> feature,
    blink::mojom::ObservedFeatureType type) {
  DCHECK(client_);

  auto& set = features_by_type_[static_cast<int>(type)];

  // Notify if this is the first receiver.
  if (set.empty())
    client_->OnStartUsing(id_, type);

  set.Add(nullptr, std::move(feature));
}

}  // namespace content