70 lines
2.9 KiB
C#
70 lines
2.9 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Configuration;
|
||
using Polly;
|
||
using Polly.Extensions.Http;
|
||
using Prism.Ioc;
|
||
using Prism.Modularity;
|
||
using System.Net.Http;
|
||
using YY.Admin.Core.Services;
|
||
using YY.Admin.Infrastructure.Hubs;
|
||
using YY.Admin.Infrastructure.Network;
|
||
using YY.Admin.Infrastructure.Storage;
|
||
using YY.Admin.Infrastructure.Sync;
|
||
|
||
namespace YY.Admin.Module;
|
||
|
||
public class SyncModule : IModule
|
||
{
|
||
public void RegisterTypes(IContainerRegistry containerRegistry)
|
||
{
|
||
containerRegistry.RegisterSingleton<TokenStore>();
|
||
containerRegistry.RegisterSingleton<HttpSyncClient>();
|
||
containerRegistry.RegisterSingleton<OutboxProcessor>();
|
||
containerRegistry.RegisterSingleton<IJeecgUserMirrorPullOutbox, JeecgUserMirrorPullOutbox>();
|
||
containerRegistry.RegisterSingleton<INetworkMonitor, NetworkMonitor>();
|
||
containerRegistry.RegisterSingleton<ISignalRService, StompWebSocketService>();
|
||
|
||
var serviceCollection = new ServiceCollection();
|
||
serviceCollection.AddHttpClient("JeecgApi", (sp, client) =>
|
||
{
|
||
var config = containerRegistry.GetContainer().Resolve<IConfiguration>();
|
||
var baseUrl = config.GetValue<string>("JeecgIntegration:BaseUrl")?.TrimEnd('/');
|
||
if (!string.IsNullOrWhiteSpace(baseUrl))
|
||
{
|
||
client.BaseAddress = new Uri(baseUrl);
|
||
}
|
||
|
||
var tokenStore = containerRegistry.GetContainer().Resolve<TokenStore>();
|
||
var token = tokenStore.GetTokenAsync(default).ConfigureAwait(false).GetAwaiter().GetResult();
|
||
if (!string.IsNullOrWhiteSpace(token))
|
||
{
|
||
client.DefaultRequestHeaders.Authorization = new System.Net.Http.Headers.AuthenticationHeaderValue("Bearer", token);
|
||
}
|
||
}).AddPolicyHandler(GetRetryPolicy());
|
||
|
||
var provider = serviceCollection.BuildServiceProvider();
|
||
var httpClientFactory = provider.GetRequiredService<IHttpClientFactory>();
|
||
containerRegistry.RegisterInstance(httpClientFactory);
|
||
}
|
||
|
||
public void OnInitialized(IContainerProvider containerProvider)
|
||
{
|
||
var networkMonitor = containerProvider.Resolve<INetworkMonitor>();
|
||
var outboxProcessor = containerProvider.Resolve<OutboxProcessor>();
|
||
var signalService = containerProvider.Resolve<ISignalRService>();
|
||
|
||
_ = networkMonitor.StartAsync(CancellationToken.None);
|
||
_ = outboxProcessor.StartConsumerAsync(CancellationToken.None);
|
||
// 用户镜像 + 设备指令：统一 STOMP（/ws/device），免密与设备 Token 模式均启动
|
||
_ = Task.Run(() => signalService.ConnectUnifiedDeviceChannelAsync(CancellationToken.None));
|
||
}
|
||
|
||
private static IAsyncPolicy<HttpResponseMessage> GetRetryPolicy()
|
||
{
|
||
return HttpPolicyExtensions
|
||
.HandleTransientHttpError()
|
||
.WaitAndRetryAsync(3, retryAttempt => TimeSpan.FromSeconds(Math.Pow(2, retryAttempt)));
|
||
}
|
||
|
||
}
|