c# - 使用 Async/Await 时，如何在请求属性更改时更新所有请求？

Question

我正在使用 async/await 每秒向使用其 API 的服务发出多个请求。我遇到的问题是当我需要刷新 token 时(它每小时都会过期)。 token 过期后，我从服务返回 401 未授权错误。那是我刷新 token 并再次重试失败请求的时候。 token 刷新正常，但我发现即使刷新 token 后，许多后续请求仍使用旧 token 发送。以下是此功能中使用的方法。想知道是否有什么突出的原因是这种意外行为的罪魁祸首。

public void Process(id)
{
    var tasks = items.Select(async item =>
    {
        var response = await SendRequestAsync(() => CreateRequest(item.Url));
        //do something with response
        await Process(item.subId); //recursive call to process sub items.
    }).ToList();

    if (tasks.Any())
        await Task.WhenAll(tasks);      

}

public HttpRequestMessage CreateRequest(string url)
{
    var request = new HttpRequestMessage(HttpMethod.Get, url);
    request.Headers.Add("Authorization", "Bearer " + AppSettings.AccessToken); 
    return request;
}

public async Task<HttpResponseMessage> SendRequestAsync(Func<HttpRequestMessage> funcReq)
{   
    var response = await ExecuteRequestAsync(funcReq());

    while (response.StatusCode == HttpStatusCode.Unauthorized)
    {
        await RefreshTokenAsync();
        return await ExecuteRequestAsync(funcReq());  //assuming func ensures that CreateRequest is called each time, so I'll always have a new request with the updated token.
    }  

    return response;            
}

private async Task<HttpResponseMessage> ExecuteRequestAsync(HttpRequestMessage request)
{
    var client = new HttpClient();
    var response = await client.SendAsync(request);
    return response;    
}

public async Task RefreshTokenAsync()
{
    await semaphoreSlim.WaitAsync();
    try
    {      
        if ((DateTime.Now - refreshTime).TotalMinutes < 60) //tokens last for an hour, so after the refresh is made by the first request that failed, subsequent requests should have the latest token.
            return; 

        Token newToken = GetNewToken();
        AppSettings.AccessToken = newToken.AccessToken //AppSettings is a singleton wrapper class for app.cofig app settings
        refreshTime = DateTime.Now
    }
    finally
    {
        semaphoreSlim.Release();
    }
}

Answer 1

这不是答案。只是我没有在哪里发布评论中讨论的代码。

Prabhu，我认为这样的事情应该有效，以便在获得 401 之前更新 token 。只有当您可以假设 token 过期的频率时，这才有效。

public HttpRequestMessage CreateRequest(string url)
{
    var request = new HttpRequestMessage(HttpMethod.Get, url);
    request.Headers.Add("Authorization", "Bearer " + GetUpToDateAccessToken()); 
    return request;
}

private Token GetUpToDateAccessToken()
{
    _readWriteLockSlim.EnterReadLock();
    try
    {      
        return _latestToken;
    }
    finally
    {
        _readWriteLockSlim.ExitReadLock();
    }  
}

可以使用计时器每 60 分钟更新一次 token 。同步是通过读写锁完成的。这将是计时器的节拍处理程序(您可以使用 System.Timers.Timer)。

private void UpdateToken()
{  
  _readWriteLockSlim.EnterWriteLock();
  try 
  {
     if ((DateTime.Now - refreshTime).TotalMinutes >= 60) 
     {
         Token newToken = GetNewToken();
         _latestToken = newToken.AccessToken;
          refreshTime = DateTime.Now;
     }
  }
  finally
  {
    _readWriteLockSlim.ExitWriteLock();
  }
}

如您所述，如果不能保证 60 分钟的有效期，那么这将无法按预期工作。也许您可以每 5 分钟左右重新生成一次 token ，以确保您不会使用无效 token 发出请求。

最后，要处理 401，因为它们仍然会发生，您可以将 SendRequestAsync 中的 while 循环修改为:

if (response.StatusCode == HttpStatusCode.Unauthorized)
{
   UpdateToken();
   return await ExecuteRequestAsync(funcReq());
}