diff options
author | vnugent <public@vaughnnugent.com> | 2024-02-02 21:14:19 -0500 |
---|---|---|
committer | vnugent <public@vaughnnugent.com> | 2024-02-02 21:14:19 -0500 |
commit | 52645b724834e669788a45edb9d135f243432540 (patch) | |
tree | 56a5ea56fdf339f00934b3e4cd88212ea3608b29 /back-end/src/Cache/SearchResultCache.cs | |
parent | dd5e5164262c5ff7ecf6db1003d41d81e6364c09 (diff) |
download file formats, button tweaks, more cache infra
Diffstat (limited to 'back-end/src/Cache/SearchResultCache.cs')
-rw-r--r-- | back-end/src/Cache/SearchResultCache.cs | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/back-end/src/Cache/SearchResultCache.cs b/back-end/src/Cache/SearchResultCache.cs new file mode 100644 index 0000000..c7a263a --- /dev/null +++ b/back-end/src/Cache/SearchResultCache.cs @@ -0,0 +1,111 @@ +// Copyright (C) 2024 Vaughn Nugent +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as +// published by the Free Software Foundation, either version 3 of the +// License, or (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program. If not, see <https://www.gnu.org/licenses/>. + +using System; +using System.Threading; +using System.Threading.Tasks; + +using VNLib.Net.Http; +using VNLib.Data.Caching; +using VNLib.Plugins; +using VNLib.Plugins.Extensions.Loading; +using VNLib.Plugins.Extensions.VNCache; +using VNLib.Plugins.Extensions.VNCache.DataModel; + +namespace SimpleBookmark.Cache +{ + [ConfigurationName("search_cache", Required = false)] + internal sealed class SearchResultCache + { + private readonly IEntityCache<ResultCacheEntry>? _cache; + + /// <summary> + /// Gets a value that indicates if the configuration enabled result caching + /// </summary> + public bool IsEnabled => _cache != null; + + public SearchResultCache(PluginBase plugin) : this(plugin, null) + { } + + public SearchResultCache(PluginBase plugin, IConfigScope? config) + { + string? cachePrefix = config?.GetRequiredProperty("cachePrefix", p => p.GetString()!); + bool isEnabled = config?.GetRequiredProperty("enabled", p => p.GetBoolean()) ?? true; + + if (!isEnabled) + { + return; + } + + IGlobalCacheProvider? cache = plugin.GetDefaultGlobalCache(); + if (cache != null) + { + if (cachePrefix != null) + { + _cache = cache.GetPrefixedCache(cachePrefix) + .CreateEntityCache<ResultCacheEntry>( + new MemPackCacheSerializer(null), + new MemPackCacheSerializer(null) + ); + } + else + { + //non-prefixed cache + _cache = cache.CreateEntityCache<ResultCacheEntry>( + new MemPackCacheSerializer(null), + new MemPackCacheSerializer(null) + ); + } + } + } + + public async Task<IMemoryResponseReader?> GetCachedResultAsync(string[] keys, CancellationToken cancellation) + { + ResultCacheEntry? entry = await _cache!.GetAsync($"{keys}", cancellation); + return entry is null ? null : new ResultResponseReader(entry); + } + + public Task StoreResultAsync(Memory<byte> data, string[] keys, CancellationToken cancellation) + { + //Init new entry + ResultCacheEntry entry = new() + { + Payload = data, + Created = DateTime.UtcNow + }; + + return _cache!.UpsertAsync($"{keys}", entry, cancellation); + } + + public Task DeleteEntry(string[] keys, CancellationToken cancellation) => _cache!.RemoveAsync($"{keys}", cancellation); + + private sealed class ResultResponseReader(ResultCacheEntry entry) : IMemoryResponseReader + { + private int _position; + + ///<inheritdoc/> + public int Remaining => entry.Payload.Length - _position; + + ///<inheritdoc/> + public void Advance(int written) => _position += written; + + ///<inheritdoc/> + public void Close() => entry.Dispose(); + + ///<inheritdoc/> + public ReadOnlyMemory<byte> GetMemory() => entry.Payload.Slice(_position); + } + } +} |