refactor: Embed object storage improvements

aneoconsulting · Dec 16, 2024 · d6e75c7 · d6e75c7
1 parent 5b1ddc5
commit d6e75c7
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/Adaptors/Embed/src/ObjectStorage.cs b/Adaptors/Embed/src/ObjectStorage.cs
@@ -17,7 +17,7 @@
 
 using System;
 using System.Collections.Generic;
-using System.Runtime.CompilerServices;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 
@@ -67,18 +67,17 @@ public Task<HealthCheckResult> Check(HealthCheckTag tag)
     await foreach (var val in valueChunks.WithCancellation(cancellationToken)
                                          .ConfigureAwait(false))
     {
-      array.AddRange(val.ToArray());
+      array.AddRange(val.Span);
     }
 
     return (array.ToArray(), array.Count);
   }
 
   /// <inheritdoc />
-  public async IAsyncEnumerable<byte[]> GetValuesAsync(byte[]                                     id,
-                                                       [EnumeratorCancellation] CancellationToken cancellationToken = default)
-  {
-    yield return id;
-  }
+  public IAsyncEnumerable<byte[]> GetValuesAsync(byte[]            id,
+                                                 CancellationToken cancellationToken = default)
+    => AsyncEnumerable.Repeat(id,
+                              1);
 
   /// <inheritdoc />
   public Task TryDeleteAsync(IEnumerable<byte[]> ids,