summaryrefslogtreecommitdiff
path: root/chromium/services/resource_coordinator/resource_coordinator_service.cc
blob: 855e1310ab72f680569baa441ff346b560a61604 (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
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
95
96
97
98
99
100
101
// Copyright 2017 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 "services/resource_coordinator/resource_coordinator_service.h"

#include <utility>

#include "base/memory/ptr_util.h"
#include "services/resource_coordinator/memory_instrumentation/coordinator_impl.h"
#if !defined(TOOLKIT_QT)
#include "services/resource_coordinator/observers/metrics_collector.h"
#endif
#include "services/resource_coordinator/observers/tab_signal_generator_impl.h"
#include "services/resource_coordinator/service_callbacks_impl.h"
#include "services/resource_coordinator/tracing/agent_registry.h"
#include "services/resource_coordinator/tracing/coordinator.h"
#include "services/service_manager/public/cpp/service_context.h"

namespace resource_coordinator {

std::unique_ptr<service_manager::Service> ResourceCoordinatorService::Create() {
  auto resource_coordinator_service =
      base::MakeUnique<ResourceCoordinatorService>();

  return resource_coordinator_service;
}

ResourceCoordinatorService::ResourceCoordinatorService()
    : weak_factory_(this) {}

ResourceCoordinatorService::~ResourceCoordinatorService() {
  ref_factory_.reset();
}

void ResourceCoordinatorService::OnStart() {
  ref_factory_.reset(new service_manager::ServiceContextRefFactory(
      base::Bind(&service_manager::ServiceContext::RequestQuit,
                 base::Unretained(context()))));

  registry_.AddInterface(base::Bind(ServiceCallbacksImpl::Create,
                                    base::Unretained(ref_factory_.get()),
                                    base::Unretained(this)));

  registry_.AddInterface(
      base::Bind(&CoordinationUnitIntrospectorImpl::BindToInterface,
                 base::Unretained(&introspector_)));

  // Register new |CoordinationUnitGraphObserver| implementations here.
  auto tab_signal_generator_impl = base::MakeUnique<TabSignalGeneratorImpl>();
  registry_.AddInterface(
      base::Bind(&TabSignalGeneratorImpl::BindToInterface,
                 base::Unretained(tab_signal_generator_impl.get())));
  coordination_unit_manager_.RegisterObserver(
      std::move(tab_signal_generator_impl));

#if !defined(TOOLKIT_QT)
  coordination_unit_manager_.RegisterObserver(
      base::MakeUnique<MetricsCollector>());
#endif

  coordination_unit_manager_.OnStart(&registry_, ref_factory_.get());

  // TODO(chiniforooshan): The abstract class Coordinator in the
  // public/cpp/memory_instrumentation directory should not be needed anymore.
  // We should be able to delete that and rename
  // memory_instrumentation::CoordinatorImpl to
  // memory_instrumentation::Coordinator.
  memory_instrumentation_coordinator_ =
      base::MakeUnique<memory_instrumentation::CoordinatorImpl>(
          context()->connector());
  registry_.AddInterface(base::BindRepeating(
      &memory_instrumentation::CoordinatorImpl::BindCoordinatorRequest,
      base::Unretained(memory_instrumentation_coordinator_.get())));

  tracing_agent_registry_ = base::MakeUnique<tracing::AgentRegistry>();
  registry_.AddInterface(
      base::BindRepeating(&tracing::AgentRegistry::BindAgentRegistryRequest,
                          base::Unretained(tracing_agent_registry_.get())));

  tracing_coordinator_ = base::MakeUnique<tracing::Coordinator>();
  registry_.AddInterface(
      base::BindRepeating(&tracing::Coordinator::BindCoordinatorRequest,
                          base::Unretained(tracing_coordinator_.get())));
}

void ResourceCoordinatorService::OnBindInterface(
    const service_manager::BindSourceInfo& source_info,
    const std::string& interface_name,
    mojo::ScopedMessagePipeHandle interface_pipe) {
  registry_.BindInterface(interface_name, std::move(interface_pipe),
                          source_info);
}

void ResourceCoordinatorService::SetUkmRecorder(
    std::unique_ptr<ukm::MojoUkmRecorder> ukm_recorder) {
  ukm_recorder_ = std::move(ukm_recorder);
  coordination_unit_manager_.set_ukm_recorder(ukm_recorder_.get());
}

}  // namespace resource_coordinator