up

src/Signals/StellaOps.Signals/Persistence/IProcSnapshotRepository.cs

@@ -0,0 +1,42 @@
+using StellaOps.Signals.Models;
+
+namespace StellaOps.Signals.Persistence;
+
+/// <summary>
+/// Repository for persisting and querying proc snapshot documents.
+/// </summary>
+public interface IProcSnapshotRepository
+{
+    /// <summary>
+    /// Upsert a proc snapshot document.
+    /// </summary>
+    Task<ProcSnapshotDocument> UpsertAsync(ProcSnapshotDocument document, CancellationToken cancellationToken);
+
+    /// <summary>
+    /// Get a proc snapshot by ID.
+    /// </summary>
+    Task<ProcSnapshotDocument?> GetByIdAsync(string id, CancellationToken cancellationToken);
+
+    /// <summary>
+    /// Get all proc snapshots for a specific image digest.
+    /// </summary>
+    Task<IReadOnlyList<ProcSnapshotDocument>> GetByImageDigestAsync(
+        string tenant,
+        string imageDigest,
+        int limit = 100,
+        CancellationToken cancellationToken = default);
+
+    /// <summary>
+    /// Get the most recent proc snapshot for an image digest and runtime type.
+    /// </summary>
+    Task<ProcSnapshotDocument?> GetLatestAsync(
+        string tenant,
+        string imageDigest,
+        string runtimeType,
+        CancellationToken cancellationToken);
+
+    /// <summary>
+    /// Delete expired proc snapshots.
+    /// </summary>
+    Task<int> DeleteExpiredAsync(CancellationToken cancellationToken);
+}

src/Signals/StellaOps.Signals/Persistence/InMemoryProcSnapshotRepository.cs

@@ -0,0 +1,95 @@
+using System.Collections.Concurrent;
+using StellaOps.Signals.Models;
+
+namespace StellaOps.Signals.Persistence;
+
+/// <summary>
+/// In-memory implementation of <see cref="IProcSnapshotRepository"/> for testing and development.
+/// </summary>
+public sealed class InMemoryProcSnapshotRepository : IProcSnapshotRepository
+{
+    private readonly ConcurrentDictionary<string, ProcSnapshotDocument> _documents = new(StringComparer.OrdinalIgnoreCase);
+    private readonly TimeProvider _timeProvider;
+
+    public InMemoryProcSnapshotRepository(TimeProvider timeProvider)
+    {
+        _timeProvider = timeProvider ?? throw new ArgumentNullException(nameof(timeProvider));
+    }
+
+    public Task<ProcSnapshotDocument> UpsertAsync(ProcSnapshotDocument document, CancellationToken cancellationToken)
+    {
+        ArgumentNullException.ThrowIfNull(document);
+
+        _documents[document.Id] = document;
+        return Task.FromResult(document);
+    }
+
+    public Task<ProcSnapshotDocument?> GetByIdAsync(string id, CancellationToken cancellationToken)
+    {
+        ArgumentException.ThrowIfNullOrWhiteSpace(id);
+
+        _documents.TryGetValue(id, out var document);
+        return Task.FromResult(document);
+    }
+
+    public Task<IReadOnlyList<ProcSnapshotDocument>> GetByImageDigestAsync(
+        string tenant,
+        string imageDigest,
+        int limit = 100,
+        CancellationToken cancellationToken = default)
+    {
+        ArgumentException.ThrowIfNullOrWhiteSpace(tenant);
+        ArgumentException.ThrowIfNullOrWhiteSpace(imageDigest);
+
+        var normalizedDigest = imageDigest.ToLowerInvariant();
+        var results = _documents.Values
+            .Where(d => string.Equals(d.Tenant, tenant, StringComparison.OrdinalIgnoreCase) &&
+                        string.Equals(d.ImageDigest, normalizedDigest, StringComparison.OrdinalIgnoreCase))
+            .OrderByDescending(d => d.CapturedAt)
+            .Take(limit)
+            .ToList();
+
+        return Task.FromResult<IReadOnlyList<ProcSnapshotDocument>>(results);
+    }
+
+    public Task<ProcSnapshotDocument?> GetLatestAsync(
+        string tenant,
+        string imageDigest,
+        string runtimeType,
+        CancellationToken cancellationToken)
+    {
+        ArgumentException.ThrowIfNullOrWhiteSpace(tenant);
+        ArgumentException.ThrowIfNullOrWhiteSpace(imageDigest);
+        ArgumentException.ThrowIfNullOrWhiteSpace(runtimeType);
+
+        var normalizedDigest = imageDigest.ToLowerInvariant();
+        var result = _documents.Values
+            .Where(d => string.Equals(d.Tenant, tenant, StringComparison.OrdinalIgnoreCase) &&
+                        string.Equals(d.ImageDigest, normalizedDigest, StringComparison.OrdinalIgnoreCase) &&
+                        string.Equals(d.RuntimeType, runtimeType, StringComparison.OrdinalIgnoreCase))
+            .OrderByDescending(d => d.CapturedAt)
+            .FirstOrDefault();
+
+        return Task.FromResult(result);
+    }
+
+    public Task<int> DeleteExpiredAsync(CancellationToken cancellationToken)
+    {
+        var now = _timeProvider.GetUtcNow();
+        var expiredIds = _documents
+            .Where(kv => kv.Value.ExpiresAt.HasValue && kv.Value.ExpiresAt.Value < now)
+            .Select(kv => kv.Key)
+            .ToList();
+
+        var deletedCount = 0;
+        foreach (var id in expiredIds)
+        {
+            if (_documents.TryRemove(id, out _))
+            {
+                deletedCount++;
+            }
+        }
+
+        return Task.FromResult(deletedCount);
+    }
+}