143 lines
5.0 KiB
C#
143 lines
5.0 KiB
C#
// using System.Collections.Concurrent;
|
|
// using System.Text.Json;
|
|
// using BTCPayApp.Core.Attempt2;
|
|
// using BTCPayApp.Core.Contracts;
|
|
// using BTCPayApp.Core.Data;
|
|
// using BTCPayApp.VSS;
|
|
// using BTCPayServer.Lightning;
|
|
// using Microsoft.EntityFrameworkCore;
|
|
// using Microsoft.EntityFrameworkCore.Diagnostics;
|
|
// using Microsoft.Extensions.Logging;
|
|
// using VSSProto;
|
|
//
|
|
// namespace BTCPayApp.Core;
|
|
//
|
|
//
|
|
//
|
|
// public class VSSMapperInterceptor : SaveChangesInterceptor
|
|
// {
|
|
//
|
|
// public VSSMapperInterceptor(BTCPayConnectionManager btcPayConnectionManager, ILogger<VSSMapperInterceptor> logger)
|
|
// {
|
|
// }
|
|
//
|
|
// private ConcurrentDictionary<EventId, object> PendingEvents = new ConcurrentDictionary<EventId, object>();
|
|
// public override ValueTask<int> SavedChangesAsync(SaveChangesCompletedEventData eventData, int result,
|
|
// CancellationToken cancellationToken = new CancellationToken())
|
|
// {
|
|
// return base.SavedChangesAsync(eventData, result, cancellationToken);
|
|
// }
|
|
//
|
|
// private IVSSAPI api;
|
|
// public override ValueTask<InterceptionResult<int>> SavingChangesAsync(DbContextEventData eventData, InterceptionResult<int> result,
|
|
// CancellationToken cancellationToken = new CancellationToken())
|
|
// {
|
|
// foreach (var entry in eventData.Context.ChangeTracker.Entries())
|
|
// {
|
|
//
|
|
// if (entry.Entity is LightningPayment lightningPayment)
|
|
// {
|
|
// if (entry.State == EntityState.Deleted)
|
|
// {
|
|
//
|
|
// api.DeleteObjectAsync(new DeleteObjectRequest
|
|
// {
|
|
// KeyValue = new KeyValue()
|
|
// {
|
|
//
|
|
// }
|
|
// Key = $"LightningPayment/{lightningPayment.Id}"
|
|
// });
|
|
// }
|
|
// }
|
|
// if (entry.Entity is Channel channel)
|
|
// {
|
|
//
|
|
// }
|
|
// if (entry.Entity is Setting setting)
|
|
// {
|
|
//
|
|
// }
|
|
// }
|
|
//
|
|
// return base.SavingChangesAsync(eventData, result, cancellationToken);
|
|
// }
|
|
//
|
|
// public override Task SaveChangesCanceledAsync(DbContextEventData eventData,
|
|
// CancellationToken cancellationToken = new CancellationToken())
|
|
// {
|
|
// PendingEvents.Remove(eventData.EventId, out _);
|
|
// return base.SaveChangesCanceledAsync(eventData, cancellationToken);
|
|
// }
|
|
//
|
|
// public override Task SaveChangesFailedAsync(DbContextErrorEventData eventData,
|
|
// CancellationToken cancellationToken = new CancellationToken())
|
|
// {
|
|
// PendingEvents.Remove(eventData.EventId, out _);
|
|
// return base.SaveChangesFailedAsync(eventData, cancellationToken);
|
|
// }
|
|
//
|
|
//
|
|
// }
|
|
//
|
|
|
|
|
|
using System.Text.Json;
|
|
using AsyncKeyedLock;
|
|
using BTCPayApp.Core.Contracts;
|
|
using BTCPayApp.Core.Data;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
public class DatabaseConfigProvider: IConfigProvider
|
|
{
|
|
private readonly IDbContextFactory<AppDbContext> _dbContextFactory;
|
|
private readonly ILogger<DatabaseConfigProvider> _logger;
|
|
private AsyncKeyedLocker<string> _lock = new();
|
|
|
|
public DatabaseConfigProvider(IDbContextFactory<AppDbContext> dbContextFactory, ILogger<DatabaseConfigProvider> logger)
|
|
{
|
|
_dbContextFactory = dbContextFactory;
|
|
_logger = logger;
|
|
}
|
|
|
|
public async Task<T?> Get<T>(string key)
|
|
{
|
|
await using var dbContext = await _dbContextFactory.CreateDbContextAsync();
|
|
var config = await dbContext.Settings.FindAsync(key);
|
|
if (typeof(T) == typeof(byte[]))
|
|
return (T?) (config?.Value as object);
|
|
return config is null ? default : JsonSerializer.Deserialize<T>(config.Value);
|
|
}
|
|
|
|
public async Task Set<T>(string key, T? value, bool backup)
|
|
{
|
|
using var releaser = await _lock.LockAsync(key);
|
|
_logger.LogDebug("Setting {key} to {value} {backup}", key, value, backup? "backup": "no backup");
|
|
await using var dbContext = await _dbContextFactory.CreateDbContextAsync();
|
|
if (value is null)
|
|
{
|
|
try
|
|
{
|
|
dbContext.Settings.Remove(new Setting {Key = key});
|
|
await dbContext.SaveChangesAsync();
|
|
}
|
|
catch (DbUpdateException)
|
|
{
|
|
}
|
|
return;
|
|
}
|
|
|
|
var newValue = typeof(T) == typeof(byte[])? value as byte[]:JsonSerializer.SerializeToUtf8Bytes(value);
|
|
var setting = new Setting {Key = key, Value = newValue, Backup = backup};
|
|
await dbContext.Upsert(setting, CancellationToken.None);
|
|
|
|
}
|
|
|
|
public async Task<IEnumerable<string>> List(string prefix)
|
|
{
|
|
await using var dbContext = await _dbContextFactory.CreateDbContextAsync();
|
|
return await dbContext.Settings.Where(s => s.Key.StartsWith(prefix)).Select(s => s.Key).ToListAsync();
|
|
}
|
|
}
|