SuZhouGuanHong/ShanMeiXinNengYuan.git

using SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using WIDESEAWCS_Core;
using WIDESEAWCS_Core.BaseServices;
using WIDESEAWCS_Core.Enums;
using WIDESEAWCS_Core.Helper;
using WIDESEAWCS_DTO.BasicInfo;
using WIDESEAWCS_QuartzJob.Models;
using WIDESEAWCS_QuartzJob.Repository;
 
namespace WIDESEAWCS_QuartzJob.Service
{
    /// <summary>
    /// 路由配置业务层
    /// </summary>
    public class RouterService : ServiceBase<Dt_Router, IRouterRepository>, IRouterService
    {
        private readonly IDeviceProtocolRepository _deviceProtocolRepository;
        private readonly IDeviceInfoRepository _deviceInfoRepository;
 
        /// <summary>
        /// 路由配置业务层
        /// </summary>
        /// <param name="BaseDal"></param>
        /// <param name="deviceProtocolRepository"></param>
        /// <param name="deviceInfoRepository"></param>
        public RouterService(IRouterRepository BaseDal, IDeviceProtocolRepository deviceProtocolRepository, IDeviceInfoRepository deviceInfoRepository) : base(BaseDal)
        {
            _deviceProtocolRepository = deviceProtocolRepository;
            _deviceInfoRepository = deviceInfoRepository;
        }
 
        /// <summary>
        /// 根据起点/当前位置、终点获取下一个子节点。
        /// </summary>
        /// <param name="startPosi">起点/当前位置。</param>
        /// <param name="endPosi">终点。</param>
        /// <returns>返回路由实体集合。</returns>
        public List<Dt_Router> QueryNextRoutes(string startPosi, string endPosi)
        {
            List<Dt_Router> routers = new List<Dt_Router>();
            try
            {
                // 一次性查询所有路由数据到内存
                List<Dt_Router> allRouters = BaseDal.QueryData(x => true);
                
                // 在内存中进行路径搜索
                routers = FindRoutesInMemory(startPosi, endPosi, allRouters, null);
                
                if (routers.Count == 0)
                {
                    throw new Exception($"该路径未配置或配置错误,请检查设备路由信息,起点:【{startPosi}】,终点:【{endPosi}】");
                }
            }
            catch (Exception ex)
            {
                // 记录错误信息
            }
            return routers;
        }
 
        /// <summary>
        /// 根据起点/当前位置、终点获取下一个子节点。
        /// </summary>
        /// <param name="startPosi">起点/当前位置。</param>
        /// <param name="endPosi">终点。</param>
        /// <param name="routeType">路由类型。</param>
        /// <returns>返回路由实体集合。</returns>
        public List<Dt_Router> QueryNextRoutes(string startPosi, string endPosi, int routeType)
        {
            List<Dt_Router> routers = new List<Dt_Router>();
            try
            {
                // 一次性查询指定类型的所有路由数据到内存
                List<Dt_Router> allRouters = BaseDal.QueryData(x => x.InOutType == routeType);
                
                // 在内存中进行路径搜索
                routers = FindRoutesInMemory(startPosi, endPosi, allRouters, routeType);
                
                if (routers.Count == 0)
                {
                    throw new Exception($"该路径未配置或配置错误,请检查设备路由信息,起点:【{startPosi}】,终点:【{endPosi}】");
                }
            }
            catch (Exception ex)
            {
                // 记录错误信息
            }
            return routers;
        }
 
        /// <summary>
        /// 在内存中查找从起点到终点的所有路由
        /// </summary>
        /// <param name="startPosi">起点位置</param>
        /// <param name="endPosi">终点位置</param>
        /// <param name="allRouters">所有路由数据</param>
        /// <param name="routeType">路由类型（可选）</param>
        /// <returns>符合条件的路由列表</returns>
        private List<Dt_Router> FindRoutesInMemory(string startPosi, string endPosi, List<Dt_Router> allRouters, int? routeType)
        {
            List<Dt_Router> result = new List<Dt_Router>();
            HashSet<int> addedIds = new HashSet<int>();
 
            // 构建索引：以 NextPosi 和 ChildPosi 为键的字典，加速查找
            var routersByNext = allRouters
                .GroupBy(r => r.NextPosi)
                .ToDictionary(g => g.Key, g => g.ToList());
            
            var routersByChild = allRouters
                .GroupBy(r => r.ChildPosi)
                .ToDictionary(g => g.Key, g => g.ToList());
 
            // 找到所有指向终点的路由（终点的父节点）
            List<Dt_Router> endRouters = new List<Dt_Router>();
            if (routersByNext.ContainsKey(endPosi))
                endRouters.AddRange(routersByNext[endPosi]);
            if (routersByChild.ContainsKey(endPosi))
                endRouters.AddRange(routersByChild[endPosi].Where(r => !endRouters.Any(e => e.Id == r.Id)));
 
            // 按 IsEnd 降序排序
            endRouters = endRouters.OrderByDescending(r => r.IsEnd).ToList();
 
            // 使用队列进行广度优先搜索
            Queue<(Dt_Router router, List<Dt_Router> path)> queue = new Queue<(Dt_Router, List<Dt_Router>)>();
            
            // 将所有终点路由加入队列
            foreach (var endRouter in endRouters)
            {
                queue.Enqueue((endRouter, new List<Dt_Router> { endRouter }));
            }
 
            // 广度优先搜索
            while (queue.Count > 0)
            {
                var (currentRouter, currentPath) = queue.Dequeue();
 
                // 如果当前路由的起点就是目标起点，找到完整路径
                if (currentRouter.StartPosi == startPosi)
                {
                    // 将路径中的所有路由添加到结果中（去重）
                    foreach (var router in currentPath)
                    {
                        if (!addedIds.Contains(router.Id))
                        {
                            result.Add(router);
                            addedIds.Add(router.Id);
                        }
                    }
                    continue;
                }
 
                // 查找当前路由起点的父路由
                List<Dt_Router> parentRouters = new List<Dt_Router>();
                if (routersByNext.ContainsKey(currentRouter.StartPosi))
                    parentRouters.AddRange(routersByNext[currentRouter.StartPosi]);
                if (routersByChild.ContainsKey(currentRouter.StartPosi))
                    parentRouters.AddRange(routersByChild[currentRouter.StartPosi].Where(r => !parentRouters.Any(p => p.Id == r.Id)));
 
                // 将父路由加入队列
                foreach (var parentRouter in parentRouters)
                {
                    // 避免循环引用
                    if (!currentPath.Any(p => p.Id == parentRouter.Id))
                    {
                        var newPath = new List<Dt_Router>(currentPath) { parentRouter };
                        queue.Enqueue((parentRouter, newPath));
                    }
                }
            }
 
            return result;
        }
 
        /// <summary>
        /// 根据起点获取下一个单个子节点路由
        /// </summary>
        /// <param name="startPosi">起点/当前位置</param>
        /// <returns>返回下一个路由节点，如果没有则返回null</returns>
        public Dt_Router QueryNextRoute(string startPosi)
        {
            try
            {
                // 查询从起点出发的所有路由
                List<Dt_Router> routes = BaseDal.QueryData(x => x.StartPosi == startPosi, 
                    new Dictionary<string, OrderByType> { { nameof(Dt_Router.IsEnd), OrderByType.Desc } });
                
                // 返回第一个路由
                return routes.FirstOrDefault();
            }
            catch (Exception ex)
            {
                // 记录错误信息
                return null;
            }
        }
 
        /// <summary>
        /// 根据起点和路由类型获取下一个单个子节点路由
        /// </summary>
        /// <param name="startPosi">起点/当前位置</param>
        /// <param name="routeType">路由类型</param>
        /// <returns>返回下一个路由节点，如果没有则返回null</returns>
        public Dt_Router QueryNextRoute(string startPosi, int routeType)
        {
            try
            {
                // 查询从起点出发的指定类型路由
                List<Dt_Router> routes = BaseDal.QueryData(x => x.StartPosi == startPosi && x.InOutType == routeType, 
                    new Dictionary<string, OrderByType> { { nameof(Dt_Router.IsEnd), OrderByType.Desc } });
                
                // 返回第一个路由
                return routes.FirstOrDefault();
            }
            catch (Exception ex)
            {
                // 记录错误信息
                return null;
            }
        }
 
        /// <summary>
        /// 根据起点、终点方向和路由类型获取下一个单个子节点路由（智能选择朝向终点的路由）
        /// </summary>
        /// <param name="startPosi">起点/当前位置</param>
        /// <param name="endPosi">终点位置（用于方向判断）</param>
        /// <param name="routeType">路由类型</param>
        /// <returns>返回下一个路由节点，优先返回朝向终点的路由，如果没有则返回null</returns>
        public Dt_Router QueryNextRoute(string startPosi, string endPosi, int routeType)
        {
            try
            {
                // 查询从起点出发的指定类型路由
                List<Dt_Router> routes = BaseDal.QueryData(x => x.StartPosi == startPosi && x.InOutType == routeType, 
                    new Dictionary<string, OrderByType> { { nameof(Dt_Router.IsEnd), OrderByType.Desc } });
                
                if (routes.Count == 0)
                    return null;
 
                // 优先选择直接指向终点的路由
                Dt_Router directRoute = routes.FirstOrDefault(x => x.NextPosi == endPosi || x.ChildPosi == endPosi);
                if (directRoute != null)
                    return directRoute;
 
                // 如果没有直接路由，使用查找算法找到朝向终点的路由
                List<Dt_Router> allRouters = BaseDal.QueryData(x => x.InOutType == routeType);
                foreach (var route in routes)
                {
                    // 检查从这个路由的下一个位置是否能到达终点
                    var pathToEnd = FindRoutesInMemory(route.NextPosi, endPosi, allRouters, routeType);
                    if (pathToEnd.Count > 0)
                        return route;
                }
 
                // 如果都不能到达终点，返回第一个路由
                return routes.FirstOrDefault();
            }
            catch (Exception ex)
            {
                // 记录错误信息
                return null;
            }
        }
 
        /// <summary>
        /// 获取从起点到终点的完整路径（按顺序返回每个子节点路由）
        /// </summary>
        /// <param name="startPosi">起点位置</param>
        /// <param name="endPosi">终点位置</param>
        /// <param name="routeType">路由类型</param>
        /// <returns>返回有序的路由列表，如果找不到路径则返回空列表</returns>
        public List<Dt_Router> QueryRoutePath(string startPosi, string endPosi, int routeType)
        {
            List<Dt_Router> path = new List<Dt_Router>();
            string currentPosi = startPosi;
            HashSet<string> visitedPositions = new HashSet<string>();
            
            try
            {
                while (currentPosi != endPosi)
                {
                    if (visitedPositions.Contains(currentPosi))
                    {
                        break;
                    }
                    visitedPositions.Add(currentPosi);
 
                    Dt_Router nextRoute = QueryNextRoute(currentPosi, endPosi, routeType);
                    
                    if (nextRoute == null)
                    {
                        break;
                    }
                    
                    path.Add(nextRoute);
                    currentPosi = nextRoute.NextPosi;
                    
                    if (path.Count > 1000)
                    {
                        break;
                    }
                }
                
                if (currentPosi != endPosi)
                {
                    return new List<Dt_Router>();
                }
            }
            catch (Exception ex)
            {
                return new List<Dt_Router>();
            }
            
            return path;
        }
 
        /// <summary>
        /// 根据设备编号获取对应的路由点位编号(输送线站台编号)信息
        /// </summary>
        /// <param name="deviceCode">设备编号</param>
        /// <returns>返回路由点位编号(输送线站台编号)集合</returns>
        public List<string> QueryAllPositions(string deviceCode)
        {
            // 创建一个字符串列表，用于存储所有位置
            List<string> positions = new List<string>();
            try
            {
                // 查询所有进入路由器的位置
                List<string> inRouterPositions = BaseDal.QueryData(x => x.ChildPosiDeviceCode == deviceCode && x.InOutType == RouterInOutType.In.ObjToInt()).GroupBy(x => x.StartPosi).Select(x => x.Key).ToList();
 
                // 查询所有离开路由器的位置
                List<string> outRouterPositions = BaseDal.QueryData(x => x.ChildPosiDeviceCode == deviceCode && x.InOutType == RouterInOutType.Out.ObjToInt()).GroupBy(x => x.ChildPosi).Select(x => x.Key).ToList();
 
                // 将进入和离开路由器的位置添加到列表中
                positions.AddRange(inRouterPositions);
                positions.AddRange(outRouterPositions);
                // 返回去重后的位置列表
                return positions.GroupBy(x => x).Select(x => x.Key).ToList();
            }
            catch
            {
 
            }
            // 返回位置列表
            return positions;
        }
        /// <summary>
        /// 获取路由表中所有完整的路由信息(前端调用展示数据)。
        /// </summary>
        /// <returns>匿名对象集合。</returns>
        public List<object> GetAllWholeRouters()
        {
            List<object> data = new List<object>();
            // 查询所有路由
            List<Dt_Router> allRouters = BaseDal.QueryData(x => true);
            // 查询所有结束的路由，并按Id排序
            List<Dt_Router> dt_Routers = allRouters.Where(x => x.IsEnd).OrderBy(x => x.Id).ToList();
 
            // 遍历所有结束的路由
            foreach (var item in dt_Routers)
            {
                // 获取当前路由的子路由
                string routes = $"{item.NextPosi},";
                // 获取当前路由的父路由
                string str = GetPreviousRoutes(item.StartPosi, allRouters, item.InOutType);
                // 如果父路由不为空
                if (!string.IsNullOrEmpty(str))
                {
                    // 去掉最后一个逗号
                    if (str.EndsWith(","))
                        str = str.Substring(0, str.Length - 1);
                    // 将父路由添加到子路由中
                    routes += str;
                }
                // 如果当前路由是入口
                if (item.InOutType == RouterInOutType.In.ObjToInt())
                {
                    // 将子路由反转并添加到data中
                    List<string> itemRouters = routes.Split(",").Reverse().ToList();
                    object obj = new { type = RouterInOutType.In, routes = itemRouters };
                    data.Add(obj);
                }
                // 如果当前路由是出口
                else
                {
                    // 将子路由反转并添加到data中
                    List<string> itemRouters = routes.Split(",").Reverse().ToList();
                    object obj = new { type = RouterInOutType.Out, routes = itemRouters };
                    data.Add(obj);
                }
            }
 
            // 返回data
            return data;
        }
 
        private string GetPreviousRoutes(string startPosi, List<Dt_Router> allRouters, int routerType)
        {
            // 定义一个空字符串routers
            string routers = string.Empty;
            // 判断startPosi是否为空
            if (!string.IsNullOrEmpty(startPosi))
            {
                // 判断routers是否以逗号结尾
                if (!routers.EndsWith(","))
                    // 如果不是，则将startPosi添加到routers中，并在后面加上逗号
                    routers += $"{startPosi},";
                else
                    // 如果是，则将startPosi添加到routers中
                    routers += $"{startPosi}";
            }
            // 从allRouters中筛选出NextPosi等于startPosi且InOutType等于routerType的元素，并转换为List
            List<Dt_Router> preRouters = allRouters.Where(x => x.NextPosi == startPosi && x.InOutType == routerType).ToList();
            // 遍历preRouters中的每个元素
            foreach (var item in preRouters)
            {
                // 调用GetPreviousRoutes方法，传入item.StartPosi、allRouters和routerType，并将返回值赋给str
                string str = GetPreviousRoutes(item.StartPosi, allRouters, routerType);
                // 判断str是否为空
                if (!string.IsNullOrEmpty(str))
                {
                    // 判断routers是否以逗号结尾
                    if (routers.EndsWith(","))
                        // 如果是，则将str添加到routers中
                        routers += $"{str}";
                    else
                        // 如果不是，则将str添加到routers中，并在后面加上逗号
                        routers += $"{str},";
                }
            }
            // 返回routers
            return routers;
        }
 
        /// <summary>
        /// 添加完整路由信息(前端调用配置路由信息)。
        /// </summary>
        /// <param name="routersAddDTOs">设备路由配置添加DTO</param>
        /// <param name="routerType">路由类型</param>
        /// <returns>返回处理结果</returns>
        public WebResponseContent AddRouters(List<RoutersAddDTO> routersAddDTOs, int routerType)
        {
            WebResponseContent content = new WebResponseContent();
            try
            {
                // 检查子位置编号是否重复
                if (routersAddDTOs.GroupBy(x => x.ChildPositionCode).Where(x => !string.IsNullOrEmpty(x.Key)).Select(x => x.Count()).Any(x => x > 1))
                {
                    return content = WebResponseContent.Instance.Error("子位置编号重复");
                }
 
                // 检查根位置编号是否重复
                if (routersAddDTOs.GroupBy(x => x.PositionCode).Select(x => x.Count()).Any(x => x > 1))
                {
                    return content = WebResponseContent.Instance.Error("根位置编号重复");
                }
                // 查询设备信息
                List<dynamic> deviceCode = _deviceInfoRepository.QueryTabs<Dt_DeviceInfo, Dt_DeviceProtocol, dynamic>((a, b) => new object[] { JoinType.Inner, a.Id == b.DeviceId }, (a, b) => new { b.DeviceChildCode, a.DeviceCode }, (a, b) => true, x => true).Distinct().ToList();
 
                List<Dt_Router> routers = new List<Dt_Router>();
                // 遍历routersAddDTOs，生成Dt_Router对象
                for (int i = 0; i < routersAddDTOs.Count - 1; i++)
                {
                    dynamic obj = deviceCode.FirstOrDefault(x => x.DeviceChildCode == routersAddDTOs[i + 1].PositionCode || x.DeviceChildCode == routersAddDTOs[i + 1].ChildPositionCode);
                    Dt_Router router = new Dt_Router()
                    {
                        ChildPosi = routersAddDTOs[i + 1].PositionCode,
                        ChildPosiDeviceCode = obj.DeviceCode,
                        Depth = 1,
                        InOutType = routerType,
                        NextPosi = routersAddDTOs[i + 1].PositionCode,
                        SrmColumn = string.IsNullOrEmpty(routersAddDTOs[i].SCColumn) ? int.TryParse(routersAddDTOs[i + 1].SCColumn, out int col) ? col : null : int.TryParse(routersAddDTOs[i].SCColumn, out int col2) ? col2 : null,
                        SrmLayer = string.IsNullOrEmpty(routersAddDTOs[i].SCLayer) ? int.TryParse(routersAddDTOs[i + 1].SCLayer, out int lay) ? lay : null : int.TryParse(routersAddDTOs[i].SCLayer, out int lay2) ? lay2 : null,
                        SrmRow = string.IsNullOrEmpty(routersAddDTOs[i].SCRow) ? int.TryParse(routersAddDTOs[i + 1].SCRow, out int row) ? row : null : int.TryParse(routersAddDTOs[i].SCRow, out int row2) ? row2 : null,
                        StartPosi = routersAddDTOs[i].PositionCode,
                        IsEnd = false
                    };
                    // 如果是最后一个元素，设置终点位置编号
                    if (i == routersAddDTOs.Count - 2)
                    {
                        if (routerType == (int)RouterInOutType.Out)
                            router.ChildPosi = routersAddDTOs[i + 1].ChildPositionCode;
                        router.IsEnd = true;
                    }
                    routers.Add(router);
                }
                // 检查起点位置编号与子位置编号是否相同
                if (routers.Any(x => x.StartPosi == x.ChildPosi))
                {
                    return content = WebResponseContent.Instance.Error("输入数据起点位置编号与子位置编号相同");
                }
                // 检查起点位置编号与终点位置编号是否相同
                if (routers.Any(x => x.StartPosi == x.NextPosi))
                {
                    return content = WebResponseContent.Instance.Error("输入数据起点位置编号与终点位置编号相同");
                }
                // 查询数据库中已有的路由信息
                List<Dt_Router> dt_Routers = BaseDal.QueryData(x => x.InOutType == routerType);
 
                // 移除重复的路由信息
                dt_Routers.ForEach(x =>
                {
                    var t = routers.FirstOrDefault(v => v.StartPosi == x.StartPosi && v.NextPosi == x.NextPosi);
                    if (t != null)
                    {
                        routers.Remove(t);
                    }
                    var r = routers.FirstOrDefault(v => v.StartPosi == x.StartPosi && v.ChildPosi == x.ChildPosi);
                    if (r != null)
                    {
                        routers.Remove(r);
                    }
                });
 
                // 添加新的路由信息
                BaseDal.AddData(routers);
                content = WebResponseContent.Instance.OK();
            }
            catch (Exception ex)
            {
                content = WebResponseContent.Instance.Error(ex.Message);
            }
            return content;
        }
    }
}