89 lines
2.5 KiB
C#
89 lines
2.5 KiB
C#
// ***********************************************************************
|
|
// Assembly : Helper
|
|
// Author : yubaolee
|
|
// Created : 12-16-2016
|
|
//
|
|
// Last Modified By : yubaolee
|
|
// Last Modified On : 12-21-2016
|
|
// 使用微软默认带超时的Cache
|
|
// File: CacheContext.cs
|
|
// ***********************************************************************
|
|
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using Microsoft.Extensions.Caching.Memory;
|
|
|
|
namespace Infrastructure.Cache
|
|
{
|
|
public class CacheContext : ICacheContext
|
|
{
|
|
private IMemoryCache _objCache;
|
|
private static readonly ConcurrentDictionary<string, ConcurrentQueue<string>> queueDic = new ConcurrentDictionary<string, ConcurrentQueue<string>>();
|
|
public CacheContext(IMemoryCache objCache)
|
|
{
|
|
_objCache = objCache;
|
|
}
|
|
|
|
public override T Get<T>(string key)
|
|
{
|
|
return _objCache.Get<T>(key);
|
|
}
|
|
|
|
public override bool Set<T>(string key, T t, DateTime expire)
|
|
{
|
|
var obj = Get<T>(key);
|
|
if (obj != null)
|
|
{
|
|
Remove(key);
|
|
}
|
|
|
|
_objCache.Set(key, t, new MemoryCacheEntryOptions()
|
|
.SetAbsoluteExpiration(expire)); //绝对过期时间
|
|
|
|
return true;
|
|
}
|
|
|
|
public override bool Remove(string key)
|
|
{
|
|
_objCache.Remove(key);
|
|
return true;
|
|
}
|
|
/// <summary>
|
|
/// 添加并发锁
|
|
/// </summary>
|
|
/// <param name="id">值</param>
|
|
/// <param name="func">执行方法</param>
|
|
/// <param name="key">键</param>
|
|
/// <returns></returns>
|
|
public override async Task<T> Lock<T>(string id, Func<string, Task<T>> func, string key = "hc")
|
|
{
|
|
var queue = queueDic.GetOrAdd(id, new ConcurrentQueue<string>());
|
|
var token = Guid.NewGuid().ToString();
|
|
queue.Enqueue(token);
|
|
queue.TryPeek(out string newToken);
|
|
if (token != newToken)
|
|
{
|
|
while (true)
|
|
{
|
|
Thread.Sleep(10);
|
|
queue.TryPeek(out newToken);
|
|
if (token == newToken)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
try
|
|
{
|
|
return await func(id);
|
|
}
|
|
finally
|
|
{
|
|
queue.TryDequeue(out string _);
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|