Initial commit: Add project files
This commit is contained in:
+103
@@ -0,0 +1,103 @@
|
||||
using System.Text.Json;
|
||||
using Dapper;
|
||||
using KArtSell.BuildingBlocks.Data;
|
||||
using KArtSell.BuildingBlocks.Hashing;
|
||||
using KArtSell.BuildingBlocks.Reliability;
|
||||
using KArtSell.Modules.ModelOperations.Application;
|
||||
|
||||
namespace KArtSell.Modules.ModelOperations.Infrastructure;
|
||||
|
||||
public sealed class DapperModelOperationRequestRepository(
|
||||
IDbConnectionFactory connectionFactory,
|
||||
IOutboxWriter outboxWriter) : IModelOperationRequestRepository
|
||||
{
|
||||
private const string InsertSql = """
|
||||
insert into evaluation.model_operation_request
|
||||
(request_id, schedule_id, operation_code, scope_key, automation_mode, idempotency_key,
|
||||
dataset_id, data_hash, model_version, config_version, code_sha, contract_version,
|
||||
lifecycle_state, correlation_id, status, requested_at)
|
||||
values
|
||||
(@RequestId, @ScheduleId, @OperationCode, @ScopeKey, @AutomationMode, @IdempotencyKey,
|
||||
@DatasetId, @DataHash, @ModelVersion, @ConfigVersion, @CodeSha, @ContractVersion,
|
||||
@LifecycleState, @CorrelationId, 'REQUESTED', @RequestedAt)
|
||||
on conflict (idempotency_key) do nothing;
|
||||
""";
|
||||
|
||||
private const string InsertStatusEventSql = """
|
||||
insert into evaluation.model_operation_status_event
|
||||
(event_id, request_id, from_status, to_status, reason_code, payload_hash,
|
||||
actor_type, correlation_id, occurred_at)
|
||||
values
|
||||
(@EventId, @RequestId, null, 'REQUESTED', null, @PayloadHash,
|
||||
'SYSTEM', @CorrelationId, @OccurredAt);
|
||||
""";
|
||||
|
||||
public async Task<bool> AddAsync(ModelOperationRequest request, CancellationToken cancellationToken)
|
||||
{
|
||||
await using var connection = await connectionFactory.OpenAsync(cancellationToken);
|
||||
await using var transaction = await connection.BeginTransactionAsync(cancellationToken);
|
||||
|
||||
var affected = await connection.ExecuteAsync(new CommandDefinition(
|
||||
InsertSql,
|
||||
new
|
||||
{
|
||||
request.RequestId,
|
||||
request.ScheduleId,
|
||||
request.OperationCode,
|
||||
request.ScopeKey,
|
||||
request.AutomationMode,
|
||||
request.IdempotencyKey,
|
||||
request.Context.VersionSet.DatasetId,
|
||||
request.Context.VersionSet.DataHash,
|
||||
request.Context.VersionSet.ModelVersion,
|
||||
request.Context.VersionSet.ConfigVersion,
|
||||
request.Context.VersionSet.CodeSha,
|
||||
request.Context.VersionSet.ContractVersion,
|
||||
request.Context.LifecycleState,
|
||||
request.CorrelationId,
|
||||
request.RequestedAt
|
||||
},
|
||||
transaction,
|
||||
cancellationToken: cancellationToken));
|
||||
|
||||
if (affected == 1)
|
||||
{
|
||||
var payload = JsonSerializer.Serialize(new
|
||||
{
|
||||
requestId = request.RequestId,
|
||||
request.OperationCode,
|
||||
request.ScopeKey,
|
||||
request.AutomationMode,
|
||||
versionSet = request.Context.VersionSet,
|
||||
boundary = "NO_AUTO_MODEL_MUTATION"
|
||||
});
|
||||
var payloadHash = ContentHasher.Sha256(payload);
|
||||
|
||||
await connection.ExecuteAsync(new CommandDefinition(
|
||||
InsertStatusEventSql,
|
||||
new
|
||||
{
|
||||
EventId = Guid.NewGuid(),
|
||||
request.RequestId,
|
||||
PayloadHash = payloadHash,
|
||||
request.CorrelationId,
|
||||
OccurredAt = request.RequestedAt
|
||||
},
|
||||
transaction,
|
||||
cancellationToken: cancellationToken));
|
||||
|
||||
var message = new OutboxMessage(
|
||||
request.RequestId,
|
||||
"ModelOperationRequested",
|
||||
1,
|
||||
payload,
|
||||
request.CorrelationId,
|
||||
request.RequestedAt,
|
||||
payloadHash);
|
||||
await outboxWriter.AddAsync(connection, transaction, message, cancellationToken);
|
||||
}
|
||||
|
||||
await transaction.CommitAsync(cancellationToken);
|
||||
return affected == 1;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user