Digital Twin Graph

.github/workflows/rust-ci.yml

@@ -64,6 +64,8 @@ jobs:
       - name: Cache Dependencies
         uses: Swatinem/rust-cache@v2
       - name: Build
-        run: cargo build
+        # Build the project with the `managed_subscribe` and `digital_twin_graph` features enabled.
+        run: cargo build --features "managed_subscribe,digital_twin_graph,digital_twin_registry"
       - name: Test
-        run: cargo test
+        # Test the project with the `managed_subscribe` and `digital_twin_graph` features enabled.      
+        run: cargo test --features "managed_subscribe,digital_twin_graph,digital_twin_registry"

Cargo.toml

@@ -16,6 +16,8 @@ members = [
 
   # extension
   "core/module/managed_subscribe",
+  "core/module/digital_twin_graph",
+  "core/module/digital_twin_registry",  
 
   # DTDL tools
   "dtdl-tools",
@@ -27,6 +29,7 @@ members = [
   "samples/common",
   "samples/protobuf_data_access",
   "samples/command",
+  "samples/graph",
   "samples/managed_subscribe",
   "samples/mixed",
   "samples/property",
@@ -67,6 +70,7 @@ strum = "0.26.1"
 strum_macros = "0.26.1"
 tokio = "1.29.1"
 tokio-console-subscriber = { version = "0.2.0", package = "console-subscriber" }
+tokio-retry = "0.3"
 tokio-stream = "0.1.14"
 tonic = "0.11.0"
 tonic-build = "0.11.0"

core/invehicle-digital-twin/Cargo.toml

@@ -19,6 +19,8 @@ http = { workspace = true }
 iref = { workspace = true }
 log = { workspace = true }
 common = { path = "../common" }
+digital_twin_graph = { path = "../module/digital_twin_graph", optional = true }
+digital_twin_registry = { path = "../module/digital_twin_registry", optional = true }
 managed_subscribe = { path = "../module/managed_subscribe", optional = true }
 parking_lot = { workspace = true }
 prost = { workspace = true }
@@ -38,5 +40,7 @@ yaml-rust = { workspace = true }
 tonic-build = { workspace = true }
 
 [features]
+digital_twin_graph = ["dep:digital_twin_graph"]
+digital_twin_registry = ["dep:digital_twin_registry"]
 managed_subscribe = ["dep:managed_subscribe"]
 tokio_console = ["dep:tokio-console-subscriber", "tokio/tracing"]

core/invehicle-digital-twin/src/main.rs

@@ -8,6 +8,12 @@
 #[cfg(feature = "managed_subscribe")]
 use managed_subscribe::managed_subscribe_module::ManagedSubscribeModule;
 
+#[cfg(feature = "digital_twin_graph")]
+use digital_twin_graph::digital_twin_graph_module::DigitalTwinGraphModule;
+
+#[cfg(feature = "digital_twin_registry")]
+use digital_twin_registry::digital_twin_registry_module::DigitalTwinRegistryModule;
+
 // End: Module references.
 
 #[allow(unused_imports)]
@@ -108,7 +114,7 @@ where
 
     #[cfg(feature = "managed_subscribe")]
     // (1) Adds the Managed Subscribe module to the service.
-    let server = {
+    let mut server = {
         // (2) Initialize the Managed Subscribe module, which implements GrpcModule.
         let managed_subscribe_module = ManagedSubscribeModule::new().await.map_err(|error| {
             error!("Unable to create Managed Subscribe module.");
@@ -129,6 +135,36 @@ where
         server.add_module(new_middleware, Box::new(managed_subscribe_module))
     };
 
+    #[cfg(feature = "digital_twin_graph")]
+    // (1) Adds the Managed Subscribe module to the service.
+    let mut server = {
+        // (2) Initialize the Digital Twin Graph module, which implements GrpcModule.
+        let digital_twin_graph_module = DigitalTwinGraphModule::new().await.map_err(|error| {
+            error!("Unable to create Digital Twin Graph module.");
+            error
+        })?;
+
+        info!("Initialized Digital Twin Graph module.");
+
+        // (3) Add the module with the updated middleware stack to the server.
+        server.add_module(server.middleware.clone(), Box::new(digital_twin_graph_module))
+    };
+
+    #[cfg(feature = "digital_twin_registry")]
+    // (1) Adds the Managed Subscribe module to the service.
+    let mut server = {
+        // (2) Initialize the Digital Twin Registry module, which implements GrpcModule.
+        let digital_twin_registry_module = DigitalTwinRegistryModule::new().await.map_err(|error| {
+            error!("Unable to create Digital Twin Registry module.");
+            error
+        })?;
+
+        info!("Initialized Digital Twin Registry module.");
+
+        // (3) Add the module with the updated middleware stack to the server.
+        server.add_module(server.middleware.clone(), Box::new(digital_twin_registry_module))
+    };
+
     // Construct the server.
     let builder = server.construct_server().add_service(base_service);
 

core/module/digital_twin_graph/Cargo.toml

@@ -0,0 +1,25 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+# SPDX-License-Identifier: MIT
+
+[package]
+name = "digital_twin_graph"
+version = "0.1.0"
+edition = "2021"
+license = "MIT"
+
+[dependencies]
+common = { path = "../../common" }
+core-protobuf-data-access = { path = "../../protobuf_data_access" }
+log = { workspace = true }
+serde = { workspace = true }
+serde_derive = { workspace = true }
+serde_json = { workspace = true }
+tokio = { workspace = true , features = ["full"] }
+tonic = { workspace = true }
+tower = { workspace = true }
+yaml-rust = { workspace = true }
+uuid = { workspace = true }
+
+[build-dependencies]
+tonic-build = { workspace = true }