/// <summary>
/// 取一个访问令牌
/// </summary>
/// <returns></returns>
public TokenResponse TakeToken()
{
if (dictUserToken.ContainsKey(this.UserName))
{
UserTokenInfo uti = dictUserToken[this.UserName];
//获取当前用户的读写锁
this.CurrentTokenLock = uti.TokenLock;
//this.CurrentTokenLock.EnterUpgradeableReadLock();
//Console.WriteLine("...EnterUpgradeableReadLock...,Thread ID:{0}",Thread.CurrentThread.ManagedThreadId);
this.OldToken = uti.Token;
//Thread.Sleep(1000);
Console.WriteLine("thread waite one.thread:{0}", Thread.CurrentThread.ManagedThreadId);
this.CurrentTokenLock.WaitOne(10000);
if (DateTime.Now.Subtract(uti.FirstUseTime).TotalSeconds >= uti.Token.expires_in - 2)
{
//等待所有使用此令牌的线程使用完成
Console.WriteLine("thread reset begin. thread:{0}", Thread.CurrentThread.ManagedThreadId);
this.CurrentTokenLock.Reset();
try
{
if (uti.UseCount > 0)
{
Thread.Sleep(2000);//睡2秒，等待之前的请求处理完
}
//防止线程重入，再次判断
if (DateTime.Now.Subtract(uti.FirstUseTime).TotalSeconds >= uti.Token.expires_in - 2)
{
//刷新令牌
OAuthClient oc = new OAuthClient();
var newToken = oc.RefreshToken(uti.Token);
uti.ResetToken(newToken);
this.TokenExctionMessage = oc.ExceptionMessage;
}
}
catch (Exception ex)
{
this.TokenExctionMessage = ex.Message;
return(null);
}
finally
{
Console.WriteLine("thread set. thread:{0}", Thread.CurrentThread.ManagedThreadId);
this.CurrentTokenLock.Set();
}
}
Console.WriteLine("thread continue. thread:{0}", Thread.CurrentThread.ManagedThreadId);
this.CurrentUserTokenInfo = uti;
uti.BeginUse();
this.CurrentTokenLock.Set();
return(uti.Token);
}
else
{
throw new Exception(this.UserName + " 还没有访问令牌。");
}
}
/// <summary>
/// 取一个访问令牌
/// </summary>
/// <returns>如果没有或者获取令牌失败，返回空</returns>
public TokenResponse TakeToken()
{
if (dictUserToken.ContainsKey(this.UserName))
{
UserTokenInfo uti = dictUserToken[this.UserName];
//获取当前用户的读写锁
//this.CurrentTokenLock = uti.TokenLock;
//this.CurrentTokenLock.EnterUpgradeableReadLock();
//Console.WriteLine("...EnterUpgradeableReadLock...,Thread ID:{0}",Thread.CurrentThread.ManagedThreadId);
this.OldToken = uti.Token;
//如果令牌超期，刷新令牌
if (DateTime.Now.Subtract(uti.FirstUseTime).TotalSeconds >= uti.Token.expires_in)
{
lock (uti.SyncObject)
{
//防止线程重入，再次判断
if (DateTime.Now.Subtract(uti.FirstUseTime).TotalSeconds >= uti.Token.expires_in)
{
//等待之前的用户使用完令牌
while (uti.UseCount > 0)
{
if (DateTime.Now.Subtract(uti.LastUseTime).TotalSeconds > 10)
{
//如果发出请求超过10秒使用计数还大于0，可以认为资源服务器响应缓慢，最终请求此资源可能会拒绝访问
this.TokenExctionMessage = "Resouce Server maybe Request TimeOut.";
OAuthClient.WriteErrorLog("00", "**警告** " + DateTime.Now.ToString() + "：用户" + this.UserName + " 最近一次使用当前令牌（"
+ uti.Token.AccessToken + "）已经超时（10秒）,使用次数：" + uti.UseCount + "。\r\n**下面将刷新令牌，但可能导致之前还未处理完的资源服务器访问被拒绝访问。");
break;
}
System.Threading.Thread.Sleep(100);
Console.WriteLine("----waite token Use Count is 0.--------");
}
//刷新令牌
try
{
OAuthClient oc = new OAuthClient();
var newToken = oc.RefreshToken(uti.Token);
if (newToken == null)
{
throw new Exception("Refresh Token Error:" + oc.ExceptionMessage);
}
else if (string.IsNullOrEmpty(newToken.AccessToken))
{
throw new Exception("Refresh Token Error:Empty AccessToken. Other Message:" + oc.ExceptionMessage);
}
uti.ResetToken(newToken);
this.TokenExctionMessage = oc.ExceptionMessage;
}
catch (Exception ex)
{
this.TokenExctionMessage = ex.Message;
return(null);
}
}
}//end lock
}
this.CurrentUserTokenInfo = uti;
uti.BeginUse();
//this.CurrentTokenLock.Set();
return(uti.Token);
}
else
{
//throw new Exception(this.UserName+" 还没有访问令牌。");
this.TokenExctionMessage = this.UserName + " 还没有访问令牌。";
return(null);
}
}