jinko.yikatong/Src/Dlhis.Basic/Device/Services/Implements/DeviceConnectService.cs

using System.Net;
using System.Net.Sockets;
using Dlhis.Basic.Device.Request;
using Dlhis.Basic.Device.Response;
using Dlhis.Entity.Basic.Entity;
using Dlhis.Entity.Device.Entities;
using Dlhis.Entity.Device.Enums;
using Kurisu.AspNetCore.Utils.Extensions;
using Kurisu.Core.CustomClass;
using Kurisu.SqlSugar.Services;
using Mapster;

namespace Dlhis.Basic.Device.Services.Implements;

public class DeviceConnectService : IDeviceConnectService
{
    private readonly IDbContext _dbContext;

    public DeviceConnectService(IDbContext dbContext)
    {
        _dbContext = dbContext;
    }


    public async Task<string> ResigterAsync(DeviceResigterRequest request)
    {
        var businessPlace = await _dbContext.Queryable<BusinessPlaceEntity>().Where(x => x.CustomCode == request.BusinessCustomCode).FirstAsync();
        businessPlace.ThrowIfNull("营业点不存在");

        var device = await GetEnableDeviceAsync(request.UniqueNo, false, request.DeployPlace, businessPlace.TenantId);
        var registry = await _dbContext.Queryable<DeviceRegistryEntity>().Where(x => x.DeviceUniqueNo == request.UniqueNo).FirstAsync();

        if (registry == null)
        {
            registry = new DeviceRegistryEntity()
            {
                DeviceUniqueNo = request.UniqueNo,
                LastHeartbeatTime = DateTime.Now,
                RegisterTime = DateTime.Now,
                BusinessPlaceCode = businessPlace?.Code,
                TenantId = device.TenantId,
                CreatedBy = 1,
                CreateTime = DateTime.Now,
                ModifiedBy = 1,
                ModifiedTime = DateTime.Now,
            };

            await _dbContext.InsertAsync(registry);
            return "设备注册成功";
        }

        return "请勿重复注册设备";
    }

    public async Task<List<DeviceNotSetBusinessPlaceResponse>> GetNotSetBusinessPlaceDevicesAsync()
    {
        var devices = await _dbContext.Queryable<DeviceEntity>().Where(x => x.Enable).ToListAsync();
        devices = devices.Where(x => x.Type == DeviceTypeEnumType.Meal).ToList();

        return devices.Adapt<List<DeviceNotSetBusinessPlaceResponse>>();
    }

    public async Task SetResigterBusinessPlaceAsync(DeviceSetResigterBusinessPlaceRequest request)
    {
        var device = await GetEnableDeviceAsync(request.UniqueNo);
        (device.Type != DeviceTypeEnumType.Meal).ThrowIfTrue("非就餐类型设备,无法设置");

        var registry = await _dbContext.Queryable<DeviceRegistryEntity>().Where(x => x.DeviceUniqueNo == request.UniqueNo).FirstAsync();
        registry.ThrowIfNull($"{request.UniqueNo}的设备注册信息不存在,无法设置营业点");

        if (request.BusinessPlaceCode.HasValue)
        {
            var businessPlace = await _dbContext.Queryable<BusinessPlaceEntity>().Where(x => x.Code == request.BusinessPlaceCode).FirstAsync();
            businessPlace.ThrowIfNull("营业点不存在,无法设置到设备注册信息中"); 
        }

        //更新营业点信息
        registry.BusinessPlaceCode = request.BusinessPlaceCode;

        await _dbContext.UpdateAsync(registry);
    }


    public async Task HeartbeatAsync(DeviceHeartbeatRequest request)
    {
        var registry = await _dbContext.Queryable<DeviceRegistryEntity>().Where(x => x.DeviceUniqueNo == request.UniqueNo).FirstAsync();
        registry.ThrowIfNull($"注册信息不存在,无法发起心跳");

        registry.LastHeartbeatTime = DateTime.Now;
        await _dbContext.UpdateAsync(registry);
    }


    public async Task<bool> TryConnectAsync(DeviceConnectionTryRequest request)
    {
        try
        {
            var ipInfo = request.ConnectionInfo.ToObject<IPPortDto>();

            //超时时间
            var CancellationTokenSource = new CancellationTokenSource(300);
            await new TcpClient().ConnectAsync(IPAddress.Parse(ipInfo.IP), ipInfo.Port, CancellationTokenSource.Token);

            return true;
        }
        catch (Exception)
        {
            return false;
        }
    }


    //******************************************************** help methods **********************************************************

    /// <summary>
    /// 获取启用的设备
    /// </summary>
    /// <param name="uniqueNo"></param>
    /// <param name="checkExists"></param>
    /// <param name="deployPlace"></param>
    /// <param name="tenantId"></param>
    /// <returns></returns>
    private async Task<DeviceEntity> GetEnableDeviceAsync(string uniqueNo, bool checkExists = true, string deployPlace = "", string tenantId = "")
    {
        var device = await _dbContext.Queryable<DeviceEntity>()
       .Where(x => x.UniqueNo == uniqueNo)
       .FirstAsync();

        if (checkExists)
        {
            device.ThrowIfNull($"未知的设备唯一编码{uniqueNo}");
            device.Enable.ThrowIfFalse("设备尚未启用");
        }
        else
        {
            if (device == null)
            {
                device = new DeviceEntity()
                {
                    Code = Guid.NewGuid(),
                    ConnectionInfo = string.Empty,
                    TemplateCode = null,
                    UniqueNo = uniqueNo,
                    TenantId = tenantId,
                    Supplier = string.Empty,
                    Funcs = new List<DeviceFuncEnumType>(),
                    DeployPlace = deployPlace,
                    HeartbeatInterval = 0,
                    DeployDirection = string.Empty,
                    Enable = true,
                    DeployPosition = string.Empty,
                    CreatedBy = 1,
                    CreateTime = DateTime.Now,
                    ModifiedBy = 1,
                    ModifiedTime = DateTime.Now,
                };

                await _dbContext.InsertAsync(device);
            }
        }

        return device;
    }
}