chrome/browser/performance_manager/resource_coordinator_interface.h

// 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.

#ifndef CHROME_BROWSER_PERFORMANCE_MANAGER_RESOURCE_COORDINATOR_INTERFACE_H_
#define CHROME_BROWSER_PERFORMANCE_MANAGER_RESOURCE_COORDINATOR_INTERFACE_H_

#include <stdint.h>

#include "base/component_export.h"
#include "base/macros.h"
#include "chrome/browser/performance_manager/performance_manager.h"
#include "services/resource_coordinator/public/cpp/coordination_unit_id.h"
#include "services/resource_coordinator/public/mojom/coordination_unit_provider.mojom.h"
#include "services/resource_coordinator/public/mojom/service_constants.mojom.h"

namespace performance_manager {

template <class CoordinationUnitMojoPtr, class CoordinationUnitMojoRequest>
class ResourceCoordinatorInterface {
 public:
  ResourceCoordinatorInterface() = default;
  virtual ~ResourceCoordinatorInterface() = default;

  void AddBinding(CoordinationUnitMojoRequest request) {
    if (!service_)
      return;
    service_->AddBinding(std::move(request));
  }

  // Returns the ID. Note that this is meaningless for a singleton CU.
  resource_coordinator::CoordinationUnitID id() const { return cu_id_; }

  // Returns the remote endpoint interface.
  const CoordinationUnitMojoPtr& service() const { return service_; }

  // Expose the GetID function for testing.
  using CoordinationUnitMojo = typename CoordinationUnitMojoPtr::InterfaceType;
  using GetIDCallback = typename CoordinationUnitMojo::GetIDCallback;
  void GetID(GetIDCallback callback) {
    if (!service_)
      std::move(callback).Run(cu_id_);
    service_->GetID(std::move(callback));
  }

 protected:
  virtual void ConnectToService(
      resource_coordinator::mojom::CoordinationUnitProviderPtr& provider,
      const resource_coordinator::CoordinationUnitID& cu_id) = 0;

  void ConnectToService(PerformanceManager* performance_manager,
                        const resource_coordinator::CoordinationUnitID& cu_id) {
    if (!performance_manager)
      return;
    cu_id_ = cu_id;
    resource_coordinator::mojom::CoordinationUnitProviderPtr provider;
    performance_manager->BindInterface(mojo::MakeRequest(&provider));
    ConnectToService(provider, cu_id);
  }

  CoordinationUnitMojoPtr service_;
  resource_coordinator::CoordinationUnitID cu_id_;

 private:
  DISALLOW_COPY_AND_ASSIGN(ResourceCoordinatorInterface);
};

}  // namespace performance_manager

#endif  // CHROME_BROWSER_PERFORMANCE_MANAGER_RESOURCE_COORDINATOR_INTERFACE_H_