// Licensed to StellaOps under the BUSL-1.1 license.
using Npgsql;

namespace StellaOps.ReachGraph.Persistence;

public sealed partial class PostgresReachGraphRepository
{
    private static async Task SetTenantContextAsync(
        NpgsqlConnection connection,
        string tenantId,
        CancellationToken cancellationToken)
    {
        await using var command = connection.CreateCommand();
        command.CommandText = "SELECT set_config('app.tenant_id', @TenantId, false);";
        command.Parameters.AddWithValue("TenantId", tenantId);
        await command.ExecuteNonQueryAsync(cancellationToken).ConfigureAwait(false);
    }
}