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)));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|