chrome/browser/ash/crosapi/resource_manager_ash.h

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_ASH_CROSAPI_RESOURCE_MANAGER_ASH_H_
#define CHROME_BROWSER_ASH_CROSAPI_RESOURCE_MANAGER_ASH_H_

#include <stdint.h>

#include "chromeos/ash/components/dbus/resourced/resourced_client.h"
#include "chromeos/crosapi/mojom/resource_manager.mojom.h"
#include "mojo/public/cpp/bindings/receiver_set.h"
#include "mojo/public/cpp/bindings/remote_set.h"

namespace crosapi {

// The ash-chrome implementation of the ResourceManager crosapi interface.
// This class must only be used from the main thread.
class ResourceManagerAsh : public mojom::ResourceManager,
                           public ash::ResourcedClient::Observer {
 public:
  ResourceManagerAsh();
  ResourceManagerAsh(const ResourceManagerAsh&) = delete;
  ResourceManagerAsh& operator=(const ResourceManagerAsh&) = delete;
  ~ResourceManagerAsh() override;

  void BindReceiver(mojo::PendingReceiver<mojom::ResourceManager> receiver);

  // ash::ResourcedClient::Observer:
  void OnMemoryPressure(ash::ResourcedClient::PressureLevel level,
                        uint64_t reclaim_target_kb) override;

  // crosapi::mojom::ResourceManager:
  void AddMemoryPressureObserver(
      mojo::PendingRemote<mojom::MemoryPressureObserver> observer) override;

 private:
  // Support any number of connections.
  mojo::ReceiverSet<mojom::ResourceManager> receivers_;

  // Support any number of memory pressure observers.
  mojo::RemoteSet<mojom::MemoryPressureObserver> observers_;
};

}  // namespace crosapi

#endif  // CHROME_BROWSER_ASH_CROSAPI_RESOURCE_MANAGER_ASH_H_