using Microsoft.Extensions.Caching.Distributed;
|
using Microsoft.Extensions.Caching.Memory;
|
using Newtonsoft.Json;
|
using SqlSugar;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
using WIDESEAWCS_Core.Helper;
|
|
namespace WIDESEAWCS_Core.Caches
|
{
|
public class MemoryCacheService : ICacheService
|
{
|
protected IMemoryCache _cache;
|
public MemoryCacheService(IMemoryCache cache)
|
{
|
_cache = cache;
|
|
}
|
|
public bool Add(string key, string value, int expireSeconds = -1, bool isSliding = false)
|
{
|
return AddObject(key, value, expireSeconds, isSliding);
|
}
|
|
public bool AddObject(string key, object value, int expireSeconds = -1, bool isSliding = false)
|
{
|
if (expireSeconds > 0)
|
{
|
_cache.Set(key,
|
value,
|
new MemoryCacheEntryOptions()
|
.SetSlidingExpiration(new TimeSpan(0, 0, expireSeconds))
|
);
|
}
|
else
|
{
|
_cache.Set(key, value);
|
}
|
|
return true;
|
}
|
|
public void AddOrUpdate(string key, string value, int expireSeconds = -1, bool isSliding = false)
|
{
|
if (!string.IsNullOrEmpty(Get(key)))
|
{
|
Remove(key);
|
Add(key, value, expireSeconds, isSliding);
|
}
|
else
|
{
|
Add(key, value, expireSeconds, isSliding);
|
}
|
}
|
|
public void AddOrUpdate(string key, object value, int expireSeconds = -1, bool isSliding = false)
|
{
|
if (!string.IsNullOrEmpty(Get(key)))
|
{
|
Remove(key);
|
Add(key, value.Serialize(), expireSeconds, isSliding);
|
}
|
else
|
{
|
Add(key, value.Serialize(), expireSeconds, isSliding);
|
}
|
}
|
|
public void Dispose()
|
{
|
if (_cache != null)
|
_cache.Dispose();
|
GC.SuppressFinalize(this);
|
}
|
|
public bool Exists(string key)
|
{
|
if (key == null)
|
{
|
throw new ArgumentNullException(nameof(key));
|
}
|
return _cache.Get(key) != null;
|
}
|
|
public T? Get<T>(string key) where T : class
|
{
|
if (key == null)
|
{
|
throw new ArgumentNullException(nameof(key));
|
}
|
object? value = _cache.Get(key);
|
if (value == null)
|
{
|
throw new ArgumentNullException(nameof(key));
|
}
|
return value as T;
|
}
|
|
public string? Get(string key)
|
{
|
try
|
{
|
return _cache.Get(key)?.ToString();
|
}
|
catch
|
{
|
return string.Empty;
|
}
|
}
|
|
public object? Get(Type type, string key)
|
{
|
try
|
{
|
object res = _cache.Get(key);
|
return res == null ? default : JsonConvert.DeserializeObject(res?.ToString() ?? "{}", type);
|
}
|
catch
|
{
|
return string.Empty;
|
}
|
}
|
|
public bool Remove(string key)
|
{
|
if (key == null)
|
{
|
throw new ArgumentNullException(nameof(key));
|
}
|
_cache.Remove(key);
|
|
return !Exists(key);
|
}
|
|
public void Remove(IEnumerable<string> keys)
|
{
|
if (keys == null)
|
{
|
throw new ArgumentNullException(nameof(keys));
|
}
|
|
keys.ToList().ForEach(item => _cache.Remove(item));
|
}
|
|
public bool TryAdd(string key, string value, int expireSeconds = -1)
|
{
|
if (Exists(key)) return false;
|
return Add(key, value, expireSeconds);
|
}
|
|
public bool TryAdd<T>(string key, T value, int expireSeconds = -1) where T : class
|
{
|
if (Exists(key)) return false;
|
return AddObject(key, value, expireSeconds);
|
}
|
|
public bool TryGetValue(string key, out string? value)
|
{
|
value = _cache.Get(key)?.ToString();
|
return value != null;
|
}
|
|
public bool TryGetValue<T>(string key, out T? value) where T : class
|
{
|
if (_cache.TryGetValue(key, out object? obj) && obj != null)
|
{
|
value = obj as T ?? JsonConvert.DeserializeObject<T>(obj.ToString() ?? "");
|
return value != null;
|
}
|
value = default;
|
return false;
|
}
|
|
public bool TryRemove(string key, out string? value)
|
{
|
value = _cache.Get(key)?.ToString();
|
if (value == null) return false;
|
_cache.Remove(key);
|
return true;
|
}
|
|
public bool TryUpdate(string key, string newValue, int expireSeconds = -1)
|
{
|
if (!Exists(key)) return false;
|
Remove(key);
|
Add(key, newValue, expireSeconds);
|
return true;
|
}
|
|
public string GetOrAdd(string key, string value, int expireSeconds = -1)
|
{
|
var existing = _cache.Get(key)?.ToString();
|
if (existing != null) return existing;
|
Add(key, value, expireSeconds);
|
return value;
|
}
|
|
public string GetOrAdd(string key, Func<string, string> valueFactory, int expireSeconds = -1)
|
{
|
var existing = _cache.Get(key)?.ToString();
|
if (existing != null) return existing;
|
var value = valueFactory(key);
|
Add(key, value, expireSeconds);
|
return value;
|
}
|
|
public T GetOrAdd<T>(string key, Func<string, T> valueFactory, int expireSeconds = -1) where T : class
|
{
|
if (_cache.TryGetValue(key, out object? obj) && obj != null)
|
{
|
if (obj is T t) return t;
|
var deserialized = JsonConvert.DeserializeObject<T>(obj.ToString() ?? "");
|
if (deserialized != null) return deserialized;
|
}
|
var value = valueFactory(key);
|
AddObject(key, value, expireSeconds);
|
return value;
|
}
|
}
|
}
|
