LMS.service/LMS.service/Configuration/RsaConfigurattions.cs

using LMS.Common.RSAKey;
using LMS.DAO;
using LMS.Repository.DB;
using LMS.Tools.Extensions;
using Microsoft.EntityFrameworkCore;

namespace LMS.service.Configuration
{
    public class RsaConfigurattions(IServiceProvider serviceProvider) : IHostedService
    {
        private readonly IServiceProvider _serviceProvider = serviceProvider;

        public async Task StartAsync(CancellationToken cancellationToken)
        {
            using var scope = _serviceProvider.CreateScope();
            var dbContext = scope.ServiceProvider.GetRequiredService<ApplicationDbContext>();
            using var transaction = await dbContext.Database.BeginTransactionAsync(cancellationToken);
            try
            {
                int count = await dbContext.RsaKeys.CountAsync(cancellationToken: cancellationToken);

                // 先删除无效的RSA数据
                List<RsaKeys> rsaKeys = await dbContext.RsaKeys.Where(x => x.UseCount > 50 || x.ExpirationTime < BeijingTimeExtension.GetBeijingTime()).ToListAsync(cancellationToken: cancellationToken);
                if (rsaKeys.Count > 0)
                {
                    dbContext.RsaKeys.RemoveRange(rsaKeys);
                }

                // 新增 保证数据库中有指定数量的RSA数据

                int AddCount = 20 - (count - rsaKeys.Count);
                for (int i = 0; i < AddCount; i++)
                {
                    RSAKeyGenerateModel rSAKeyGenerate = RsaKeyPairGenerator.InitRsaKey();

                    await dbContext.RsaKeys.AddAsync(new RsaKeys
                    {
                        PublicKey = rSAKeyGenerate.PublicKey,
                        EncryptedPrivateKey = rSAKeyGenerate.EncryptedPrivateKey,
                        EncryptionKey = rSAKeyGenerate.EncryptedKey,
                        EncryptionIV = rSAKeyGenerate.EncryptionIV,
                        UseCount = 0,
                        ExpirationTime = BeijingTimeExtension.GetBeijingTime().AddDays(10),
                        Id = Guid.NewGuid().ToString(),
                    }, cancellationToken);
                }

                await dbContext.SaveChangesAsync(cancellationToken);
                await transaction.CommitAsync(cancellationToken);
            }
            catch (Exception ex)
            {
                await transaction.RollbackAsync(cancellationToken);
                // 根据需要处理异常，例如终止应用程序
                Console.WriteLine(ex.Message);
                Environment.Exit(1);
            }
        }

        public Task StopAsync(CancellationToken cancellationToken)
        {
            return Task.CompletedTask;
        }
    }
}