Files
Umbraco-CMS/src/Umbraco.Core/Cache/DictionaryAppCache.cs
2022-01-21 11:43:58 +01:00

112 lines
3.4 KiB
C#

using System;
using System.Collections;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Text.RegularExpressions;
using Umbraco.Extensions;
namespace Umbraco.Cms.Core.Cache
{
/// <summary>
/// Implements <see cref="IAppCache"/> on top of a concurrent dictionary.
/// </summary>
public class DictionaryAppCache : IRequestCache
{
/// <summary>
/// Gets the internal items dictionary, for tests only!
/// </summary>
private readonly ConcurrentDictionary<string, object?> _items = new ConcurrentDictionary<string, object?>();
public int Count => _items.Count;
/// <inheritdoc />
public bool IsAvailable => true;
/// <inheritdoc />
public virtual object? Get(string key)
{
return _items.TryGetValue(key, out var value) ? value : null;
}
/// <inheritdoc />
public virtual object? Get(string key, Func<object?> factory)
{
return _items.GetOrAdd(key, _ => factory());
}
public bool Set(string key, object? value) => _items.TryAdd(key, value);
public bool Remove(string key) => _items.TryRemove(key, out _);
/// <inheritdoc />
public virtual IEnumerable<object?> SearchByKey(string keyStartsWith)
{
var items = new List<object?>();
foreach (var (key, value) in _items)
if (key.InvariantStartsWith(keyStartsWith))
items.Add(value);
return items;
}
/// <inheritdoc />
public IEnumerable<object?> SearchByRegex(string regex)
{
var compiled = new Regex(regex, RegexOptions.Compiled);
var items = new List<object?>();
foreach (var (key, value) in _items)
if (compiled.IsMatch(key))
items.Add(value);
return items;
}
/// <inheritdoc />
public virtual void Clear()
{
_items.Clear();
}
/// <inheritdoc />
public virtual void Clear(string key)
{
_items.TryRemove(key, out _);
}
/// <inheritdoc />
public virtual void ClearOfType(Type type)
{
_items.RemoveAll(kvp => kvp.Value != null && kvp.Value.GetType() == type);
}
/// <inheritdoc />
public virtual void ClearOfType<T>()
{
var typeOfT = typeof(T);
ClearOfType(typeOfT);
}
/// <inheritdoc />
public virtual void ClearOfType<T>(Func<string, T, bool> predicate)
{
var typeOfT = typeof(T);
_items.RemoveAll(kvp => kvp.Value != null && kvp.Value.GetType() == typeOfT && predicate(kvp.Key, (T)kvp.Value));
}
/// <inheritdoc />
public virtual void ClearByKey(string keyStartsWith)
{
_items.RemoveAll(kvp => kvp.Key.InvariantStartsWith(keyStartsWith));
}
/// <inheritdoc />
public virtual void ClearByRegex(string regex)
{
var compiled = new Regex(regex, RegexOptions.Compiled);
_items.RemoveAll(kvp => compiled.IsMatch(kvp.Key));
}
public IEnumerator<KeyValuePair<string, object?>> GetEnumerator() => _items.GetEnumerator();
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
}
}