up

src/Concelier/__Libraries/StellaOps.Concelier.Storage.Postgres/SourceStateAdapter.cs

@@ -4,14 +4,15 @@ using System.Collections.Generic;
 using MongoDB.Bson;
 using StellaOps.Concelier.Storage.Postgres.Models;
 using StellaOps.Concelier.Storage.Postgres.Repositories;
+using Contracts = StellaOps.Concelier.Storage.Contracts;
 using MongoContracts = StellaOps.Concelier.Storage.Mongo;
 
 namespace StellaOps.Concelier.Storage.Postgres;
 
 /// <summary>
-/// Adapter that satisfies the legacy source state contract using PostgreSQL storage.
+/// Adapter that satisfies the legacy source state contract using PostgreSQL storage and provides a Postgres-native cursor contract.
 /// </summary>
-public sealed class PostgresSourceStateAdapter : MongoContracts.ISourceStateRepository
+public sealed class PostgresSourceStateAdapter : MongoContracts.ISourceStateRepository, Contracts.ISourceStateStore
 {
     private readonly ISourceRepository _sourceRepository;
     private readonly Repositories.ISourceStateRepository _stateRepository;
@@ -134,6 +135,18 @@ public sealed class PostgresSourceStateAdapter : MongoContracts.ISourceStateRepo
         _ = await _stateRepository.UpsertAsync(entity, cancellationToken).ConfigureAwait(false);
     }
 
+    async Task<Contracts.SourceCursorState?> Contracts.ISourceStateStore.TryGetAsync(string sourceName, CancellationToken cancellationToken)
+        => (await TryGetAsync(sourceName, cancellationToken).ConfigureAwait(false))?.ToStorageCursorState();
+
+    Task Contracts.ISourceStateStore.UpdateCursorAsync(string sourceName, JsonDocument cursor, DateTimeOffset completedAt, CancellationToken cancellationToken)
+        => UpdateCursorAsync(sourceName, cursor.ToBsonDocument(), completedAt, cancellationToken);
+
+    Task Contracts.ISourceStateStore.MarkFailureAsync(string sourceName, DateTimeOffset now, TimeSpan backoff, string reason, CancellationToken cancellationToken)
+        => MarkFailureAsync(sourceName, now, backoff, reason, cancellationToken);
+
+    Task Contracts.ISourceStateStore.UpsertAsync(Contracts.SourceCursorState record, CancellationToken cancellationToken)
+        => UpsertAsync(record.ToMongoSourceStateRecord(), cancellationToken);
+
     private async Task<SourceEntity> EnsureSourceAsync(string sourceName, CancellationToken cancellationToken)
     {
         var existing = await _sourceRepository.GetByKeyAsync(sourceName, cancellationToken).ConfigureAwait(false);