feat: Thor搭建

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/AiGateWay/HttpClientFactory.cs

@@ -0,0 +1,73 @@
+using System.Collections.Concurrent;
+
+namespace Yi.Framework.AiHub.Domain.AiGateWay;
+
+public static class HttpClientFactory
+{
+    /// <summary>
+    /// HttpClient池总数
+    /// </summary>
+    /// <returns></returns>
+    private static int _poolSize;
+
+    private static int PoolSize
+    {
+        get
+        {
+            if (_poolSize == 0)
+            {
+                // 获取环境变量
+                var poolSize = Environment.GetEnvironmentVariable("HttpClientPoolSize");
+                if (!string.IsNullOrEmpty(poolSize) && int.TryParse(poolSize, out var size))
+                {
+                    _poolSize = size;
+                }
+                else
+                {
+                    _poolSize = Environment.ProcessorCount;
+                }
+
+                if (_poolSize < 1)
+                {
+                    _poolSize = 2;
+                }
+            }
+
+            return _poolSize;
+        }
+    }
+
+    private static readonly ConcurrentDictionary<string, Lazy<List<HttpClient>>> HttpClientPool = new();
+
+    public static HttpClient GetHttpClient(string key)
+    {
+        return HttpClientPool.GetOrAdd(key, k => new Lazy<List<HttpClient>>(() =>
+        {
+            var clients = new List<HttpClient>(PoolSize);
+
+            for (var i = 0; i < PoolSize; i++)
+            {
+                clients.Add(new HttpClient(new SocketsHttpHandler
+                {
+                    PooledConnectionLifetime = TimeSpan.FromMinutes(30),
+                    PooledConnectionIdleTimeout = TimeSpan.FromMinutes(30),
+                    EnableMultipleHttp2Connections = true,
+                    // 连接超时5分钟
+                    ConnectTimeout = TimeSpan.FromMinutes(5),
+                    MaxAutomaticRedirections = 3,
+                    AllowAutoRedirect = true,
+                    Expect100ContinueTimeout = TimeSpan.FromMinutes(30),
+                })
+                {
+                    Timeout = TimeSpan.FromMinutes(30),
+                    DefaultRequestHeaders =
+                    {
+                        { "User-Agent", "yxai" },
+                    }
+                });
+            }
+
+            return clients;
+        })).Value[new Random().Next(0, PoolSize)];
+    }
+}