fuchsia_inspect_contrib/graph/
types.rs

1// Copyright 2024 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use fuchsia_inspect as inspect;
6use std::borrow::Cow;
7use std::marker::PhantomData;
8
9pub trait GraphObject {
10    type Id;
11    fn write_to_node(node: &inspect::Node, id: &Self::Id);
12}
13
14#[derive(Debug)]
15pub struct VertexMarker<T>(PhantomData<T>);
16#[derive(Debug)]
17pub struct EdgeMarker;
18
19impl GraphObject for EdgeMarker {
20    type Id = u64;
21
22    fn write_to_node(node: &inspect::Node, id: &Self::Id) {
23        node.record_uint("edge_id", *id);
24    }
25}
26
27impl<T: VertexId> GraphObject for VertexMarker<T> {
28    type Id = T;
29
30    fn write_to_node(node: &inspect::Node, id: &Self::Id) {
31        node.record_string("vertex_id", id.get_id().as_ref());
32    }
33}
34
35/// The ID of a vertex.
36pub trait VertexId {
37    /// Fetches the ID of a vertex, which must have a string representation.
38    fn get_id(&self) -> Cow<'_, str>;
39}
40
41impl<T: std::fmt::Display> VertexId for T {
42    fn get_id(&self) -> Cow<'_, str> {
43        Cow::Owned(format!("{self}"))
44    }
45}
46
47impl VertexId for str {
48    fn get_id(&self) -> Cow<'_, str> {
49        Cow::Borrowed(self)
50    }
51}