RPCS3/archived-discord-bot
1
0
Fork 0
mirror of https://github.com/RPCS3/discord-bot.git synced 2026-01-31 01:25:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix MemoryCacheExtensions and its tests

Browse Source
This commit is contained in:
13xforever
2025-03-26 12:01:47 +05:00
parent 78a87aa02d
commit c4aa6a48f5
2 changed files with 13 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
CompatBot/Utils/Extensions/MemoryCacheExtensions.cs
View File

@@ -6,48 +6,15 @@ namespace CompatBot.Utils;
internal static class MemoryCacheExtensions
{
private static readonly object throaway = new object();
private static readonly object throaway = new();
public static List<T> GetCacheKeys<T>(this MemoryCache memoryCache)
{
// idk why it requires access before it populates the internal state
memoryCache.TryGetValue("str", out _);
memoryCache.TryGetValue(throaway, out _);
// get the internal state object
var stateField = memoryCache.GetType()
.GetFields(BindingFlags.NonPublic | BindingFlags.Instance)
.FirstOrDefault(fi => fi.Name == "_coherentState");
var coherentState = stateField?.GetValue(memoryCache);
if (coherentState is null)
{
Config.Log.Error($"Looks like {nameof(MemoryCache)} internals have changed in {nameof(coherentState)}");
return [];
}
// get the actual underlying key-value object
var stringField = coherentState.GetType()
.GetFields(BindingFlags.NonPublic | BindingFlags.Instance)
.FirstOrDefault(fi => fi.Name == "_stringEntries");
var nonStringField = coherentState.GetType()
.GetFields(BindingFlags.NonPublic | BindingFlags.Instance)
.FirstOrDefault(fi => fi.Name == "_nonStringEntries");
if (stringField is null || nonStringField is null)
{
Config.Log.Error($"Looks like {nameof(MemoryCache)} internals have changed in {nameof(stringField)}");
return [];
}
// read the keys
var value = typeof(T) == typeof(string)
? (IDictionary?)stringField.GetValue(coherentState)
: (IDictionary?)nonStringField.GetValue(coherentState);
return value?.Keys.OfType<T>().ToList() ?? [];
}
=> memoryCache.Keys.OfType<T>().ToList();
public static Dictionary<TKey, ICacheEntry?> GetCacheEntries<TKey>(this MemoryCache memoryCache)
where TKey: notnull
{
//memoryCache.TryGetValue();
var stateField = memoryCache.GetType()
.GetFields(BindingFlags.NonPublic | BindingFlags.Instance)
.FirstOrDefault(fi => fi.Name == "_coherentState");
@@ -58,13 +25,16 @@ internal static class MemoryCacheExtensions
return new();
}
var field = coherentState.GetType()
var entriesName = "_nonStringEntries";
if (typeof(TKey) == typeof(string))
entriesName = "_stringEntries";
var entriesField = coherentState.GetType() // MemoryCache.CoherentState
.GetFields(BindingFlags.NonPublic | BindingFlags.Instance)
.FirstOrDefault(fi => fi.Name == "_entries");
var cacheEntries = (IDictionary?)field?.GetValue(coherentState);
.FirstOrDefault(fi => fi.Name == entriesName);
var cacheEntries = (IDictionary?)entriesField?.GetValue(coherentState);
if (cacheEntries is null)
{
Config.Log.Error($"Looks like {nameof(MemoryCache)} internals have changed in {cacheEntries}");
Config.Log.Error($"Looks like {nameof(MemoryCache)} internals have changed in {nameof(coherentState)} cache entries");
return new(0);
}

6
Tests/MemoryCacheExtensionTests.cs
View File

@@ -32,11 +32,11 @@ public class MemoryCacheExtensionTests
public void GetCacheEntriesTest()
{
var cache = new MemoryCache(new MemoryCacheOptions { ExpirationScanFrequency = TimeSpan.FromHours(1) });
Assert.That(cache.GetCacheKeys<int>(), Is.Empty);
Assert.That(cache.GetCacheEntries<int>(), Is.Empty);
cache.Set(13, "val13");
cache.Set("bob", 69);
Assert.That(cache.GetCacheKeys<int>(), Has.Count.EqualTo(1));
Assert.That(cache.GetCacheKeys<string>(), Has.Count.EqualTo(1));
Assert.That(cache.GetCacheEntries<int>(), Has.Count.EqualTo(1));
Assert.That(cache.GetCacheEntries<string>(), Has.Count.EqualTo(1));
}
}