From 900bb5e845c3ad44defbd427cae3d44a4a43321f Mon Sep 17 00:00:00 2001 From: ivarlovlie Date: Sat, 25 Feb 2023 13:15:44 +0100 Subject: feat: Initial commit --- code/api/src/Jobs/AccessTokenCleanupJob.cs | 20 ++++++++++++++++++++ code/api/src/Jobs/JobRegister.cs | 23 +++++++++++++++++++++++ code/api/src/Jobs/VaultTokenRenewalJob.cs | 24 ++++++++++++++++++++++++ 3 files changed, 67 insertions(+) create mode 100644 code/api/src/Jobs/AccessTokenCleanupJob.cs create mode 100644 code/api/src/Jobs/JobRegister.cs create mode 100644 code/api/src/Jobs/VaultTokenRenewalJob.cs (limited to 'code/api/src/Jobs') diff --git a/code/api/src/Jobs/AccessTokenCleanupJob.cs b/code/api/src/Jobs/AccessTokenCleanupJob.cs new file mode 100644 index 0000000..20b450c --- /dev/null +++ b/code/api/src/Jobs/AccessTokenCleanupJob.cs @@ -0,0 +1,20 @@ +namespace IOL.GreatOffice.Api.Jobs; + +public class AccessTokenCleanupJob : IJob +{ + private readonly ILogger _logger; + private readonly MainAppDatabase _context; + + public AccessTokenCleanupJob(ILogger logger, MainAppDatabase context) { + _logger = logger; + _context = context; + } + + public Task Execute(IJobExecutionContext context) { + var staleTokens = _context.AccessTokens.Where(c => c.ExpiryDate < AppDateTime.UtcNow).ToList(); + if (staleTokens.IsNullOrEmpty()) return Task.CompletedTask; + _logger.LogInformation("Removing {0} stale tokens", staleTokens.Count()); + _context.AccessTokens.RemoveRange(staleTokens); + return Task.CompletedTask; + } +} diff --git a/code/api/src/Jobs/JobRegister.cs b/code/api/src/Jobs/JobRegister.cs new file mode 100644 index 0000000..1da7d5b --- /dev/null +++ b/code/api/src/Jobs/JobRegister.cs @@ -0,0 +1,23 @@ +namespace IOL.GreatOffice.Api.Jobs; + +public static class JobRegister +{ + private static readonly JobKey AccessTokenCleanupKey = new("AccessTokenCleanupKey"); + private static readonly JobKey VaultTokenRenewalKey = new("VaultTokenRenewalKey"); + + public static IServiceCollectionQuartzConfigurator RegisterJobs(this IServiceCollectionQuartzConfigurator configurator) { + configurator.AddJob(AccessTokenCleanupKey); + configurator.AddJob(VaultTokenRenewalKey); + configurator.AddTrigger(options => { + options.ForJob(AccessTokenCleanupKey) + .WithIdentity(AccessTokenCleanupKey.Name + "-trigger") + .WithCronSchedule("0 0 0/1 ? * * *"); + }); + configurator.AddTrigger(options => { + options.ForJob(VaultTokenRenewalKey) + .WithIdentity(VaultTokenRenewalKey.Name + "-trigger") + .WithCronSchedule("0 0 0/1 ? * * *"); + }); + return configurator; + } +} \ No newline at end of file diff --git a/code/api/src/Jobs/VaultTokenRenewalJob.cs b/code/api/src/Jobs/VaultTokenRenewalJob.cs new file mode 100644 index 0000000..1768629 --- /dev/null +++ b/code/api/src/Jobs/VaultTokenRenewalJob.cs @@ -0,0 +1,24 @@ +namespace IOL.GreatOffice.Api.Jobs; + +public class VaultTokenRenewalJob : IJob +{ + private readonly ILogger _logger; + private readonly VaultService _vaultService; + + public VaultTokenRenewalJob(ILogger logger, VaultService vaultService) { + _logger = logger; + _vaultService = vaultService; + } + + public async Task Execute(IJobExecutionContext context) { + _logger.LogInformation("Starting vault token renewal"); + var renew = await _vaultService.RenewTokenAsync(); + if (renew == default) { + _logger.LogCritical("Renewal did not succeed"); + return; + } + + var token = await _vaultService.LookupTokenAsync(); + _logger.LogInformation("Token was renewed, new expire time {expires}", token.Data.ExpireTime); + } +} \ No newline at end of file -- cgit v1.3