TeamInklings/Octobot
TeamInklings/Octobot
1
0
Fork 1
mirror of https://github.com/TeamOctolings/Octobot.git synced 2025-04-20 00:43:36 +03:00
Code Issues Wiki Activity

Save important data about scheduled events to disk

Browse source 
Signed-off-by: Octol1ttle <l1ttleofficial@outlook.com>
This commit is contained in:
Octol1ttle 2023-06-02 17:08:55 +05:00
parent e116263a9d
commit 0538cbb164
Signed by: Octol1ttle
GPG key ID: B77C34313AEE1FFF
3 changed files with 43 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
Data/GuildData.cs
View file

@ -1,11 +1,22 @@
using System.Globalization;
namespace Boyfriend.Data; namespace Boyfriend.Data;
public class GuildData { public class GuildData {
public readonly GuildConfiguration Configuration; public readonly GuildConfiguration Configuration;
public readonly string ConfigurationPath; public readonly string ConfigurationPath;
public GuildData(GuildConfiguration configuration, string configurationPath) { public readonly Dictionary<ulong, ScheduledEventData> ScheduledEvents;
public readonly string ScheduledEventsPath;
public GuildData(
GuildConfiguration configuration, string configurationPath,
Dictionary<ulong, ScheduledEventData> scheduledEvents, string scheduledEventsPath) {
Configuration = configuration; Configuration = configuration;
ConfigurationPath = configurationPath; ConfigurationPath = configurationPath;
ScheduledEvents = scheduledEvents;
ScheduledEventsPath = scheduledEventsPath;
} }
public CultureInfo Culture => Configuration.Culture;
} }

12
Data/ScheduledEventData.cs Normal file
View file

@ -0,0 +1,12 @@
using Remora.Discord.API.Abstractions.Objects;
namespace Boyfriend.Data;
public class ScheduledEventData {
public DateTimeOffset? ActualStartTime;
public GuildScheduledEventStatus Status;
public ScheduledEventData(GuildScheduledEventStatus status) {
Status = status;
}
}

25
Data/Services/GuildDataService.cs
View file

@ -14,14 +14,17 @@ public class GuildDataService : IHostedService {
public async Task StopAsync(CancellationToken ct) { public async Task StopAsync(CancellationToken ct) {
var tasks = new List<Task>(); var tasks = new List<Task>();
foreach (var data in _datas.Values) { foreach (var data in _datas.Values) {
await using var stream = File.OpenWrite(data.ConfigurationPath); await using var configStream = File.OpenWrite(data.ConfigurationPath);
tasks.Add(JsonSerializer.SerializeAsync(stream, data.Configuration, cancellationToken: ct)); tasks.Add(JsonSerializer.SerializeAsync(configStream, data.Configuration, cancellationToken: ct));
await using var eventsStream = File.OpenWrite(data.ScheduledEventsPath);
tasks.Add(JsonSerializer.SerializeAsync(eventsStream, data.ScheduledEvents, cancellationToken: ct));
} }
await Task.WhenAll(tasks); await Task.WhenAll(tasks);
} }
private async Task<GuildData> GetData(Snowflake guildId, CancellationToken ct = default) { public async Task<GuildData> GetData(Snowflake guildId, CancellationToken ct = default) {
return _datas.TryGetValue(guildId, out var data) ? data : await InitializeData(guildId, ct); return _datas.TryGetValue(guildId, out var data) ? data : await InitializeData(guildId, ct);
} }
@ -29,16 +32,26 @@ public class GuildDataService : IHostedService {
var idString = $"{guildId}"; var idString = $"{guildId}";
var memberDataDir = $"{guildId}/MemberData"; var memberDataDir = $"{guildId}/MemberData";
var configurationPath = $"{guildId}/Configuration.json"; var configurationPath = $"{guildId}/Configuration.json";
var scheduledEventsPath = $"{guildId}/ScheduledEvents.json";
if (!Directory.Exists(idString)) Directory.CreateDirectory(idString); if (!Directory.Exists(idString)) Directory.CreateDirectory(idString);
if (!Directory.Exists(memberDataDir)) Directory.CreateDirectory(memberDataDir); if (!Directory.Exists(memberDataDir)) Directory.CreateDirectory(memberDataDir);
if (!File.Exists(configurationPath)) await File.WriteAllTextAsync(configurationPath, "{}", ct); if (!File.Exists(configurationPath)) await File.WriteAllTextAsync(configurationPath, "{}", ct);
if (!File.Exists(scheduledEventsPath)) await File.WriteAllTextAsync(scheduledEventsPath, "{}", ct);
await using var stream = File.OpenRead(configurationPath); await using var configurationStream = File.OpenRead(configurationPath);
var configuration var configuration
= JsonSerializer.DeserializeAsync<GuildConfiguration>( = JsonSerializer.DeserializeAsync<GuildConfiguration>(
stream, cancellationToken: ct); configurationStream, cancellationToken: ct);
var data = new GuildData(await configuration ?? new GuildConfiguration(), configurationPath); await using var eventsStream = File.OpenRead(scheduledEventsPath);
var events
= JsonSerializer.DeserializeAsync<Dictionary<ulong, ScheduledEventData>>(
eventsStream, cancellationToken: ct);
var data = new GuildData(
await configuration ?? new GuildConfiguration(), configurationPath,
await events ?? new Dictionary<ulong, ScheduledEventData>(),
scheduledEventsPath);
_datas.Add(guildId, data); _datas.Add(guildId, data);
return data; return data;
} }