SecMPS/api_sqlsugar/Warehouse/Services/device_manager/Partial/gateway_nodesService.cs

/*
 *所有关于gateway_nodes类的业务代码应在此处编写
*可使用repository.调用常用方法，获取EF/Dapper等信息
*如果需要事务请使用repository.DbContextBeginTransaction
*也可使用DBServerProvider.手动获取数据库相关信息
*用户信息、权限、角色等使用UserContext.Current操作
*gateway_nodesService对增、删、改查、导入、导出、审核业务代码扩展参照ServiceFunFilter
*/
using VolPro.Core.BaseProvider;
using VolPro.Core.Extensions.AutofacManager;
using VolPro.Entity.DomainModels;
using System.Linq;
using VolPro.Core.Utilities;
using System.Linq.Expressions;
using VolPro.Core.Extensions;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.AspNetCore.Http;
using Warehouse.IRepositories;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using System.Text.Json;

namespace Warehouse.Services
{
    public partial class gateway_nodesService
    {
        private readonly IHttpContextAccessor _httpContextAccessor;
        private readonly Igateway_nodesRepository _repository;//访问数据库

        [ActivatorUtilitiesConstructor]
        public gateway_nodesService(
            Igateway_nodesRepository dbRepository,
            IHttpContextAccessor httpContextAccessor
            )
        : base(dbRepository)
        {
            _httpContextAccessor = httpContextAccessor;
            _repository = dbRepository;
            //多租户会用到这init代码，其他情况可以不用
            //base.Init(dbRepository);
        }

        /// <summary>
        /// 网关注册（Upsert）。
        /// NodeCode 匹配则更新适配器类型/地址/在线状态并返回已有 NodeId，
        /// NodeCode 不匹配且 Token 验证通过则插入新记录。
        /// </summary>
        public async Task<gateway_nodes> RegisterNodeAsync(string nodeCode, string token, string adapterTypes, string baseUrl)
        {
            var existing = await _repository.FindAsIQueryable<gateway_nodes>()
                .FirstOrDefaultAsync(x => x.NodeCode == nodeCode);

            gateway_nodes entity;
            if (existing != null)
            {
                // 已存在：验证Token，更新网关上报字段
                if (existing.NodeToken != token)
                    throw new UnauthorizedAccessException("NodeToken 不匹配");

                existing.AdapterTypes = adapterTypes;
                existing.BaseUrl = baseUrl;
                existing.IsOnline = "在线";
                existing.LastHeartbeat = DateTime.Now;
                _repository.DbContext.Updateable(existing).ExecuteCommand();
                entity = existing;
            }
            else
            {
                // 新节点：直接插入
                entity = new gateway_nodes
                {
                    NodeCode = nodeCode,
                    NodeName = nodeCode,
                    NodeToken = token,
                    AdapterTypes = adapterTypes,
                    BaseUrl = baseUrl,
                    IsOnline = "在线",
                    Enable = "启用",
                    LastHeartbeat = DateTime.Now,
                    CreateDate = DateTime.Now
                };
                _repository.DbContext.Insertable(entity).ExecuteCommand();
            }
            return entity;
        }

        /// <summary>
        /// 心跳更新。更新 LastHeartbeat 并标记在线。
        /// </summary>
        public async Task UpdateHeartbeatAsync(string nodeCode, string token)
        {
            var entity = _repository.FindAsIQueryable<gateway_nodes>()
                .FirstOrDefaultAsync(x => x.NodeCode == nodeCode && x.NodeToken == token);
            if (entity == null)
                throw new UnauthorizedAccessException("认证失败：NodeCode 或 Token 无效");

            entity.IsOnline = "在线";
            entity.LastHeartbeat = DateTime.Now;
            _repository.DbContext.Updateable(entity).ExecuteCommand();
        }

        /// <summary>
        /// 设备数据同步。按照字段分治原则写入 base_device：
        /// 首次入库写全量，后续仅更新网关字段（IsOnline/ExtraData/ParentDeviceId等）。
        /// parentSourceId 解析为 ParentDeviceId。
        /// </summary>
        public async Task<(int added, int updated)> SyncDevicesAsync(int gatewayNodeId, List<SyncDeviceItem> devices)
        {
            var db = _repository.DbContext;

            // 批量查询已有设备映射表（用于 parentSourceId → ParentDeviceId 解析）
            var adapterCodes = devices.Select(d => d.AdapterCode).Distinct().ToList();
            var existingIds = db.Queryable<base_device>()
                .Where(x => x.NodeId == gatewayNodeId && adapterCodes.Contains(x.AdapterCode))
                .ToList()
                .ToDictionary(x => (x.AdapterCode, x.SourceId), x => x.DeviceId);

            int added = 0, updated = 0;
            foreach (var d in devices)
            {
                var key = (d.AdapterCode, d.SourceId);
                existingIds.TryGetValue(key, out var existingId);
                bool isNew = existingId == 0;

                // 解析 parentSourceId → ParentDeviceId
                int? parentDeviceId = null;
                if (!string.IsNullOrEmpty(d.ParentSourceId))
                {
                    existingIds.TryGetValue((d.AdapterCode, d.ParentSourceId), out var pid);
                    if (pid > 0) parentDeviceId = pid;
                }

                if (isNew)
                {
                    // 首次入库写全量
                    var entity = new base_device
                    {
                        DeviceName = d.Name ?? $"DEV_{d.SourceId}",
                        AdapterCode = d.AdapterCode,
                        SourceId = d.SourceId,
                        DeviceCategory = d.Category,
                        DeviceGroup = d.Group,
                        NodeId = gatewayNodeId,
                        IsParent = d.IsParent ? "是" : "否",
                        ParentDeviceId = parentDeviceId,
                        IsOnline = d.IsOnline ? "在线" : "离线",
                        IpAddress = d.IpAddress,
                        Port = d.Port,
                        ExtraData = d.ExtraDataJson,
                        Enable = "启用",
                        LastSyncTime = DateTime.Now,
                        CreateDate = DateTime.Now
                    };
                    db.Insertable(entity).ExecuteCommand();
                    added++;
                }
                else
                {
                    // 已有记录：仅更新网关字段
                    var entity = db.Queryable<base_device>().InSingle(existingId);
                    if (entity != null)
                    {
                        entity.IsOnline = d.IsOnline ? "在线" : "离线";
                        entity.IsParent = d.IsParent ? "是" : "否";
                        entity.ParentDeviceId = parentDeviceId ?? entity.ParentDeviceId;
                        entity.IpAddress = d.IpAddress;
                        entity.Port = d.Port;
                        entity.ExtraData = d.ExtraDataJson ?? entity.ExtraData;
                        entity.LastSyncTime = DateTime.Now;
                        db.Updateable(entity).ExecuteCommand();
                        updated++;
                    }
                }
            }
            return (added, updated);
        }
    }

    /// <summary>网关同步设备条目（A3 接口接收的数据模型）</summary>
    public class SyncDeviceItem
    {
        public string AdapterCode { get; set; } = "";
        public string SourceId { get; set; } = "";
        public string? Name { get; set; }
        public string? Category { get; set; }
        public string? Group { get; set; }
        public bool IsParent { get; set; }
        public string? ParentSourceId { get; set; }
        public bool IsOnline { get; set; }
        public string? IpAddress { get; set; }
        public int? Port { get; set; }
        public string? ExtraDataJson { get; set; }
    }
}