using System.Collections.Concurrent;
|
|
namespace WIDESEA_Core.Core
|
{
|
/// <summary>
|
/// 轮询服务 - 线程安全的地址轮询选择
|
/// </summary>
|
public class RoundRobinService
|
{
|
/// <summary>
|
/// 轮询计数器 - key: 巷道前缀, value: 当前索引
|
/// </summary>
|
private readonly ConcurrentDictionary<string, int> _counters = new();
|
|
/// <summary>
|
/// 获取下一个地址（轮询）
|
/// </summary>
|
/// <param name="key">巷道前缀标识</param>
|
/// <param name="addresses">候选地址列表</param>
|
/// <returns>选中的目标地址</returns>
|
public string GetNextAddress(string key, List<string> addresses)
|
{
|
// 空列表检查
|
if (addresses == null || addresses.Count == 0)
|
return "10080";
|
|
// 单个地址，直接返回
|
if (addresses.Count == 1)
|
return addresses[0];
|
|
// 多个地址，使用轮询选择
|
// AddOrUpdate 是原子操作，线程安全
|
int index = _counters.AddOrUpdate(
|
key,
|
0, // 首次使用，从 0 开始
|
(k, oldValue) => (oldValue + 1) % addresses.Count // 轮询：递增后取模
|
);
|
|
return addresses[index];
|
}
|
}
|
}
|
