PathServiceInitializer.cs 3.68 KB
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Rcs.Application.Services;
using Rcs.Infrastructure.PathFinding.Services;

namespace Rcs.Infrastructure.PathFinding
{
    /// <summary>
    /// 寻路服务初始化器 - 应用启动时自动初始化合并映射和逆向边关系
    /// @author zzy
    /// </summary>
    public class PathServiceInitializer : IHostedService
    {
        private readonly ILogger<PathServiceInitializer> _logger;
        private readonly IMapMappingService _mapMappingService;
        private readonly IUnifiedTrafficControlService _unifiedTrafficControlService;
        private CancellationTokenSource? _startupInitCts;
        private Task? _startupInitTask;

        public PathServiceInitializer(
            ILogger<PathServiceInitializer> logger,
            IMapMappingService mapMappingService,
            IUnifiedTrafficControlService unifiedTrafficControlService)
        {
            _logger = logger;
            _mapMappingService = mapMappingService;
            _unifiedTrafficControlService = unifiedTrafficControlService;
        }

        public Task StartAsync(CancellationToken cancellationToken)
        {
            if (_startupInitTask != null)
            {
                return Task.CompletedTask;
            }

            _startupInitCts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
            _startupInitTask = Task.Run(() => RunInitializationAsync(_startupInitCts.Token), CancellationToken.None);

            _logger.LogInformation("寻路服务初始化已转后台执行,不阻塞应用启动");
            return Task.CompletedTask;
        }

        public async Task StopAsync(CancellationToken cancellationToken)
        {
            _startupInitCts?.Cancel();

            if (_startupInitTask != null)
            {
                try
                {
                    await _startupInitTask.WaitAsync(cancellationToken);
                }
                catch (OperationCanceledException)
                {
                    // 应用正在停止,允许取消等待。
                }
            }

            _startupInitCts?.Dispose();
            _startupInitCts = null;
            _startupInitTask = null;
            _logger.LogInformation("寻路服务已停止");
        }

        private async Task RunInitializationAsync(CancellationToken cancellationToken)
        {
            _logger.LogInformation("开始初始化寻路服务...");

            try
            {
                // 1. 初始化地图合并映射(从Redis加载所有地图,按MapCode分组合并)
                var mergedTopologies = await _mapMappingService.InitializeAllMappingsAsync();

                if (mergedTopologies.Count == 0)
                {
                    _logger.LogWarning("未找到任何地图数据,寻路服务初始化完成但无可用地图");
                    return;
                }

                // 2. 从合并映射初始化逆向边关系(用于双向互斥检测)
                await _unifiedTrafficControlService.InitializeFromMergedMappingsAsync(mergedTopologies);

                _logger.LogInformation("寻路服务初始化完成: {MapCodeCount} 个MapCode分组已加载", mergedTopologies.Count);
            }
            catch (OperationCanceledException)
            {
                _logger.LogInformation("寻路服务初始化已取消");
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "寻路服务初始化失败");
                // 不抛出异常,允许应用继续启动,后续可以手动重试初始化
            }
        }
    }
}