Redid stores. Added user store. Added Channel Manager, to manage data from a single channel.
This commit is contained in:
@@ -1,223 +1,102 @@
|
||||
using System.Collections.Immutable;
|
||||
using System.Text;
|
||||
using HermesSocketLibrary.db;
|
||||
using HermesSocketServer.Models;
|
||||
using HermesSocketServer.Validators;
|
||||
|
||||
namespace HermesSocketServer.Store
|
||||
{
|
||||
public class VoiceStore : IStore<string, string>
|
||||
public class VoiceStore : GroupSaveStore<string, Voice>
|
||||
{
|
||||
private readonly VoiceIdValidator _idValidator;
|
||||
private readonly VoiceNameValidator _nameValidator;
|
||||
private readonly Database _database;
|
||||
private readonly IValidator _voiceIdValidator;
|
||||
private readonly IValidator _voiceNameValidator;
|
||||
private readonly Serilog.ILogger _logger;
|
||||
private readonly IDictionary<string, string> _voices;
|
||||
private readonly IList<string> _added;
|
||||
private readonly IList<string> _modified;
|
||||
private readonly IList<string> _deleted;
|
||||
private readonly object _lock;
|
||||
|
||||
public DateTime PreviousSave;
|
||||
private readonly GroupSaveSqlGenerator<Voice> _generator;
|
||||
|
||||
|
||||
public VoiceStore(Database database, VoiceIdValidator voiceIdValidator, VoiceNameValidator voiceNameValidator, Serilog.ILogger logger)
|
||||
public VoiceStore(VoiceIdValidator voiceIdValidator, VoiceNameValidator voiceNameValidator, Database database, Serilog.ILogger logger) : base(logger)
|
||||
{
|
||||
_idValidator = voiceIdValidator;
|
||||
_nameValidator = voiceNameValidator;
|
||||
_database = database;
|
||||
_voiceIdValidator = voiceIdValidator;
|
||||
_voiceNameValidator = voiceNameValidator;
|
||||
_logger = logger;
|
||||
_voices = new Dictionary<string, string>();
|
||||
_added = new List<string>();
|
||||
_modified = new List<string>();
|
||||
_deleted = new List<string>();
|
||||
_lock = new object();
|
||||
|
||||
PreviousSave = DateTime.UtcNow;
|
||||
var ctp = new Dictionary<string, string>
|
||||
{
|
||||
{ "id", "Id" },
|
||||
{ "name", "Name" }
|
||||
};
|
||||
_generator = new GroupSaveSqlGenerator<Voice>(ctp);
|
||||
}
|
||||
|
||||
public string? Get(string key)
|
||||
{
|
||||
if (_voices.TryGetValue(key, out var voice))
|
||||
return voice;
|
||||
return null;
|
||||
}
|
||||
|
||||
public IDictionary<string, string> Get()
|
||||
{
|
||||
return _voices.ToImmutableDictionary();
|
||||
}
|
||||
|
||||
public async Task Load()
|
||||
public override async Task Load()
|
||||
{
|
||||
string sql = "SELECT id, name FROM \"TtsVoice\";";
|
||||
await _database.Execute(sql, new Dictionary<string, object>(), (reader) =>
|
||||
{
|
||||
var id = reader.GetString(0);
|
||||
var name = reader.GetString(1);
|
||||
_voices.Add(id, name);
|
||||
});
|
||||
_logger.Information($"Loaded {_voices.Count} TTS voices from database.");
|
||||
}
|
||||
|
||||
public void Remove(string? key)
|
||||
{
|
||||
if (key == null)
|
||||
return;
|
||||
|
||||
lock (_lock)
|
||||
{
|
||||
if (_voices.ContainsKey(key))
|
||||
string id = reader.GetString(0);
|
||||
_store.Add(id, new Voice()
|
||||
{
|
||||
_voices.Remove(key);
|
||||
if (!_added.Remove(key))
|
||||
{
|
||||
_modified.Remove(key);
|
||||
if (!_deleted.Contains(key))
|
||||
_deleted.Add(key);
|
||||
}
|
||||
}
|
||||
}
|
||||
Id = id,
|
||||
Name = reader.GetString(1),
|
||||
});
|
||||
});
|
||||
_logger.Information($"Loaded {_store.Count} TTS voices from database.");
|
||||
}
|
||||
|
||||
public async Task<bool> Save()
|
||||
public override void OnInitialAdd(string key, Voice value)
|
||||
{
|
||||
var changes = false;
|
||||
var sb = new StringBuilder();
|
||||
var sql = "";
|
||||
_idValidator.Check(value.Id);
|
||||
_nameValidator.Check(value.Name);
|
||||
}
|
||||
|
||||
public override void OnInitialModify(string key, Voice value)
|
||||
{
|
||||
_nameValidator.Check(value.Name);
|
||||
}
|
||||
|
||||
public override void OnInitialRemove(string key)
|
||||
{
|
||||
}
|
||||
|
||||
public override async Task<bool> Save()
|
||||
{
|
||||
int count = 0;
|
||||
string sql = string.Empty;
|
||||
|
||||
if (_added.Any())
|
||||
{
|
||||
int count = _added.Count;
|
||||
sb.Append("INSERT INTO \"TtsVoice\" (id, name) VALUES ");
|
||||
lock (_lock)
|
||||
{
|
||||
foreach (var voiceId in _added)
|
||||
{
|
||||
string voice = _voices[voiceId];
|
||||
sb.Append("('")
|
||||
.Append(voiceId)
|
||||
.Append("','")
|
||||
.Append(voice)
|
||||
.Append("'),");
|
||||
}
|
||||
sb.Remove(sb.Length - 1, 1)
|
||||
.Append(';');
|
||||
|
||||
sql = sb.ToString();
|
||||
sb.Clear();
|
||||
count = _added.Count;
|
||||
sql = _generator.GenerateInsertSql("TtsVoice", _added.Select(a => _store[a]), ["id", "name"]);
|
||||
_added.Clear();
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
_logger.Debug($"About to save {count} voices to database.");
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.Error(ex, "Failed to save TTS voices on database: " + sql);
|
||||
}
|
||||
changes = true;
|
||||
_logger.Debug($"ADD {count} " + sql);
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
|
||||
if (_modified.Any())
|
||||
{
|
||||
int count = _modified.Count;
|
||||
sb.Append("UPDATE \"TtsVoice\" as t SET name = c.name FROM (VALUES ");
|
||||
lock (_lock)
|
||||
{
|
||||
foreach (var voiceId in _modified)
|
||||
{
|
||||
string voice = _voices[voiceId];
|
||||
sb.Append("('")
|
||||
.Append(voiceId)
|
||||
.Append("','")
|
||||
.Append(voice)
|
||||
.Append("'),");
|
||||
}
|
||||
sb.Remove(sb.Length - 1, 1)
|
||||
.Append(") AS c(id, name) WHERE id = c.id;");
|
||||
|
||||
|
||||
sql = sb.ToString();
|
||||
sb.Clear();
|
||||
count = _modified.Count;
|
||||
sql = _generator.GenerateUpdateSql("TtsVoice", _modified.Select(m => _store[m]), ["id"], ["name"]);
|
||||
_modified.Clear();
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
_logger.Debug($"About to update {count} voices on the database.");
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.Error(ex, "Failed to modify TTS voices on database: " + sql);
|
||||
}
|
||||
changes = true;
|
||||
_logger.Debug($"MOD {count} " + sql);
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
|
||||
if (_deleted.Any())
|
||||
{
|
||||
int count = _deleted.Count;
|
||||
sb.Append("DELETE FROM \"TtsVoice\" WHERE id IN (");
|
||||
lock (_lock)
|
||||
{
|
||||
foreach (var voiceId in _deleted)
|
||||
{
|
||||
sb.Append("'")
|
||||
.Append(voiceId)
|
||||
.Append("',");
|
||||
}
|
||||
sb.Remove(sb.Length - 1, 1)
|
||||
.Append(");");
|
||||
|
||||
|
||||
sql = sb.ToString();
|
||||
sb.Clear();
|
||||
count = _deleted.Count;
|
||||
sql = _generator.GenerateDeleteSql("TtsVoice", _deleted, ["id"]);
|
||||
_deleted.Clear();
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
_logger.Debug($"About to delete {count} voices from the database.");
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.Error(ex, "Failed to modify TTS voices on database: " + sql);
|
||||
}
|
||||
changes = true;
|
||||
_logger.Debug($"DEL {count} " + sql);
|
||||
await _database.ExecuteScalar(sql);
|
||||
}
|
||||
return changes;
|
||||
}
|
||||
|
||||
public bool Set(string? key, string? value)
|
||||
{
|
||||
if (key == null || value == null)
|
||||
return false;
|
||||
_voiceNameValidator.Check(value);
|
||||
|
||||
lock (_lock)
|
||||
{
|
||||
if (_voices.TryGetValue(key, out var voice))
|
||||
{
|
||||
|
||||
if (voice != value)
|
||||
{
|
||||
_voices[key] = value;
|
||||
if (!_added.Contains(key) && !_modified.Contains(key))
|
||||
_modified.Add(key);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
_voiceIdValidator.Check(key);
|
||||
_voices.Add(key, value);
|
||||
if (!_deleted.Remove(key) && !_added.Contains(key))
|
||||
_added.Add(key);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user