Implemented GetMachineReference for NaturalStateEngine

psibr · Jan 13, 2019 · b5f6931 · b5f6931
1 parent 3f5112b
commit b5f6931
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/REstate/Natural/NaturalStateEngine.cs b/src/REstate/Natural/NaturalStateEngine.cs
@@ -48,6 +48,8 @@ Task<INaturalStateMachine> GetMachineAsync(
             string machineId,
             CancellationToken cancellationToken = default);
 
+        INaturalStateMachine GetMachineReference(string machineId);
+
         Task<INaturalSchematic> GetSchematicAsync(
             string schematicName,
             CancellationToken cancellationToken = default);
@@ -169,6 +171,14 @@ public async Task<INaturalStateMachine> GetMachineAsync(
             return new NaturalStateMachine(machine);
         }
 
+
+        public INaturalStateMachine GetMachineReference(string machineId)
+        {
+            var machine = _stateEngine.GetMachineReference(machineId);
+
+            return new NaturalStateMachine(machine);
+        }
+
         public async Task<INaturalSchematic> GetSchematicAsync(
             string schematicName,
             CancellationToken cancellationToken = default)