blob: 15a96050d0e0afd24c86e4be15992db1a32a0e2c [file] [log] [blame]
// Copyright 2022 The Fuchsia 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 SRC_DEVICES_BIN_DRIVER_MANAGER_NODE_REMOVAL_TRACKER_H_
#define SRC_DEVICES_BIN_DRIVER_MANAGER_NODE_REMOVAL_TRACKER_H_
#include <lib/async/cpp/task.h>
#include <lib/fit/function.h>
#include <unordered_map>
#include <unordered_set>
#include "src/devices/bin/driver_manager/node.h"
namespace driver_manager {
class NodeRemovalTracker {
public:
struct Node {
std::string name;
Collection collection;
NodeState state;
};
explicit NodeRemovalTracker(async_dispatcher_t* dispatcher) : dispatcher_(dispatcher) {}
NodeId RegisterNode(Node node);
void Notify(NodeId id, NodeState state);
void FinishEnumeration();
void set_pkg_callback(fit::callback<void()> callback);
void set_all_callback(fit::callback<void()> callback);
private:
void OnRemovalTimeout();
void CheckRemovalDone();
size_t remaining_pkg_node_count() const { return remaining_pkg_nodes_.size(); }
size_t remaining_node_count() const {
return remaining_pkg_nodes_.size() + remaining_non_pkg_nodes_.size();
}
bool fully_enumerated_ = false;
NodeId next_node_id_ = 0;
std::unordered_set<NodeId> remaining_pkg_nodes_;
std::unordered_set<NodeId> remaining_non_pkg_nodes_;
std::unordered_map<NodeId, Node> nodes_;
fit::callback<void()> pkg_callback_;
fit::callback<void()> all_callback_;
async_dispatcher_t* const dispatcher_;
// Task used to dump diagnostic data if node removal is hanging. The task is started when
// FinishEnumeration() is called and canceled when removal is complete.
async::TaskClosureMethod<NodeRemovalTracker, &NodeRemovalTracker::OnRemovalTimeout>
handle_timeout_task_{this};
};
} // namespace driver_manager
#endif // SRC_DEVICES_BIN_DRIVER_MANAGER_NODE_REMOVAL_TRACKER_H_