-
-
Notifications
You must be signed in to change notification settings - Fork 124
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
2950cbb
commit 9ffa58b
Showing
17 changed files
with
275 additions
and
53 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -10,4 +10,4 @@ | |
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,158 @@ | ||
| using System.Threading.Channels; | ||
| using CTFServer.Repositories; | ||
| using CTFServer.Utils; | ||
| using Microsoft.Extensions.Caching.Distributed; | ||
|
|
||
| namespace CTFServer.Services; | ||
|
|
||
| /// <summary> | ||
| /// 缓存更新请求 | ||
| /// </summary> | ||
| public class CacheRequest | ||
| { | ||
| public string Key { get; set; } = string.Empty; | ||
| public string[] Params { get; set; } = Array.Empty<string>(); | ||
| public DistributedCacheEntryOptions? Options { get; set; } | ||
|
|
||
| public CacheRequest(string key, DistributedCacheEntryOptions? options = null, params string[] _params) | ||
| { | ||
| Key = key; | ||
| Params = _params; | ||
| Options = options; | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// 缓存请求处理接口 | ||
| /// </summary> | ||
| public interface ICacheRequestHandler | ||
| { | ||
| public string? CacheKey(CacheRequest request); | ||
| public Task<byte[]> Handler(AsyncServiceScope scope, CacheRequest request, CancellationToken token = default); | ||
| } | ||
|
|
||
| public class CacheMaker : IHostedService | ||
| { | ||
| private readonly ILogger<CacheMaker> logger; | ||
| private readonly IDistributedCache cache; | ||
| private readonly ChannelReader<CacheRequest> channelReader; | ||
| private readonly IServiceScopeFactory serviceScopeFactory; | ||
| private CancellationTokenSource TokenSource { get; set; } = new CancellationTokenSource(); | ||
| private Dictionary<string, ICacheRequestHandler> cacheHandlers = new(); | ||
|
|
||
| public CacheMaker( | ||
| ILogger<CacheMaker> _logger, | ||
| IDistributedCache _cache, | ||
| ChannelReader<CacheRequest> _channelReader, | ||
| IServiceScopeFactory _serviceScopeFactory) | ||
| { | ||
| logger = _logger; | ||
| cache = _cache; | ||
| channelReader = _channelReader; | ||
| serviceScopeFactory = _serviceScopeFactory; | ||
| } | ||
|
|
||
| public void AddCacheRequestHandler<T>(string key) where T : ICacheRequestHandler, new() | ||
| => cacheHandlers.Add(key, new T()); | ||
|
|
||
| private async Task Maker(CancellationToken token = default) | ||
| { | ||
| logger.SystemLog($"缓存更新线程已启动", TaskStatus.Pending, LogLevel.Debug); | ||
|
|
||
| try | ||
| { | ||
| await foreach (var item in channelReader.ReadAllAsync(token)) | ||
| { | ||
| if (!cacheHandlers.ContainsKey(item.Key)) | ||
| { | ||
| logger.SystemLog($"缓存更新线程未找到匹配的请求：{item.Key}", TaskStatus.NotFound, LogLevel.Warning); | ||
| continue; | ||
| } | ||
|
|
||
| var handler = cacheHandlers[item.Key]; | ||
| var key = handler.CacheKey(item); | ||
|
|
||
| if (key is null) | ||
| { | ||
| logger.SystemLog($"无效的缓存更新请求：{item.Key}", TaskStatus.NotFound, LogLevel.Warning); | ||
| continue; | ||
| } | ||
|
|
||
| var updateLock = $"_CacheUpdateLock_{key}"; | ||
|
|
||
| logger.SystemLog($"缓存更新线程开始处理更新请求：{key}", TaskStatus.Pending, LogLevel.Debug); | ||
|
|
||
| if (await cache.GetAsync(updateLock, token) is not null) | ||
| { | ||
| // only one GZCTF instance will never encounter this problem | ||
| logger.SystemLog($"缓存更新线程已锁定：{key}", TaskStatus.Pending, LogLevel.Debug); | ||
| continue; | ||
| } | ||
|
|
||
| await using var scope = serviceScopeFactory.CreateAsyncScope(); | ||
|
|
||
| try | ||
| { | ||
| await cache.SetAsync(updateLock, Array.Empty<byte>(), new DistributedCacheEntryOptions | ||
| { | ||
| AbsoluteExpirationRelativeToNow = TimeSpan.FromSeconds(60) | ||
| }, token); | ||
|
|
||
| var bytes = await handler.Handler(scope, item, token); | ||
|
|
||
| if (bytes is not null && bytes.Length > 0) | ||
| { | ||
| await cache.SetAsync(key, bytes, item.Options ?? new DistributedCacheEntryOptions(), token); | ||
| logger.SystemLog($"缓存已更新：{key} @ {bytes.Length} bytes", TaskStatus.Success, LogLevel.Debug); | ||
| } | ||
| else | ||
| { | ||
| logger.SystemLog($"缓存生成失败：{key}", TaskStatus.Failed, LogLevel.Warning); | ||
| } | ||
| } | ||
| catch (Exception e) | ||
| { | ||
| logger.SystemLog($"缓存更新线程更新失败：{key} @ {e.Message}", TaskStatus.Failed, LogLevel.Error); | ||
| } | ||
| finally | ||
| { | ||
| await cache.RemoveAsync(updateLock, token); | ||
| } | ||
|
|
||
| token.ThrowIfCancellationRequested(); | ||
| } | ||
| } | ||
| catch (OperationCanceledException) | ||
| { | ||
| logger.SystemLog($"任务取消，缓存更新线程将退出", TaskStatus.Exit, LogLevel.Debug); | ||
| } | ||
| finally | ||
| { | ||
| logger.SystemLog($"缓存更新线程已退出", TaskStatus.Exit, LogLevel.Debug); | ||
| } | ||
| } | ||
|
|
||
| public Task StartAsync(CancellationToken token) | ||
| { | ||
| TokenSource = new CancellationTokenSource(); | ||
|
|
||
| #region Add Handlers | ||
|
|
||
| AddCacheRequestHandler<ScoreboardCacheHandler>(CacheKey.ScoreBoardBase); | ||
|
|
||
| #endregion | ||
|
|
||
| _ = Maker(TokenSource.Token); | ||
|
|
||
| return Task.CompletedTask; | ||
| } | ||
|
|
||
| public Task StopAsync(CancellationToken token) | ||
| { | ||
| TokenSource.Cancel(); | ||
|
|
||
| logger.SystemLog("缓存更新已停用", TaskStatus.Success, LogLevel.Debug); | ||
|
|
||
| return Task.CompletedTask; | ||
| } | ||
| } |
Oops, something went wrong.