JiShe.CollectBus/src/JiShe.CollectBus.Application/Plugins/TcpMonitor.cs

using System;
using System.Runtime.CompilerServices;
using System.Threading.Tasks;
using DeviceDetectorNET.Parser.Device;
using DotNetCore.CAP;
using JiShe.CollectBus.Ammeters;
using JiShe.CollectBus.Common.Enums;
using JiShe.CollectBus.Common.Extensions;
using JiShe.CollectBus.Enums;
using JiShe.CollectBus.Interceptors;
using JiShe.CollectBus.IotSystems.Devices;
using JiShe.CollectBus.IotSystems.MessageReceiveds;
using JiShe.CollectBus.Protocol.Contracts;
using MassTransit;
using Microsoft.Extensions.Logging;
using TouchSocket.Core;
using TouchSocket.Sockets;
using Volo.Abp.Caching;
using Volo.Abp.DependencyInjection;
using Volo.Abp.Domain.Entities;
using Volo.Abp.Domain.Repositories;

namespace JiShe.CollectBus.Plugins
{
    public partial class TcpMonitor : PluginBase, ITransientDependency, ITcpReceivedPlugin, ITcpConnectingPlugin, ITcpConnectedPlugin, ITcpClosedPlugin
    {
        private readonly IPublishEndpoint _producerBus;
        private readonly ILogger<TcpMonitor> _logger;
        private readonly IRepository<Device, Guid> _deviceRepository;
        private readonly IDistributedCache<AmmeterInfo> _ammeterInfoCache;

        /// <summary>
        /// 
        /// </summary>
        /// <param name="producerBus"></param>
        /// <param name="logger"></param>
        /// <param name="deviceRepository"></param>
        /// <param name="ammeterInfoCache"></param>
        public TcpMonitor(IPublishEndpoint producerBus,
            ILogger<TcpMonitor> logger,
            IRepository<Device, Guid> deviceRepository,
            IDistributedCache<AmmeterInfo> ammeterInfoCache)
        {
            _producerBus = producerBus;
            _logger = logger;
            _deviceRepository = deviceRepository;
            _ammeterInfoCache = ammeterInfoCache;
        }

        public async Task OnTcpReceived(ITcpSession client, ReceivedDataEventArgs e)
        {
            var messageHexString = Convert.ToHexString(e.ByteBlock.Span);
            var hexStringList = messageHexString.StringToPairs();
            var aFn = (int?)hexStringList.GetAnalyzeValue(CommandChunkEnum.AFN);
            var fn = (int?)hexStringList.GetAnalyzeValue(CommandChunkEnum.FN);
            var aTuple = (Tuple<string, int>)hexStringList.GetAnalyzeValue(CommandChunkEnum.A);
            if (aFn.HasValue && fn.HasValue && aTuple != null && !string.IsNullOrWhiteSpace(aTuple.Item1))
            {
                var tcpSessionClient = (ITcpSessionClient)client;

                if ((AFN)aFn == AFN.链路接口检测)
                {
                    switch (fn)
                    {
                        case 1:
                            await OnTcpLoginReceived(tcpSessionClient, messageHexString, aTuple.Item1);
                            break;
                        case 3:
                            //心跳帧有两种情况：
                            //1. 集中器先有登录帧，再有心跳帧
                            //2. 集中器没有登录帧，只有心跳帧
                            await OnTcpHeartbeatReceived(tcpSessionClient, messageHexString, aTuple.Item1);
                            break;
                        default:
                            _logger.LogError($"指令初步解析失败，指令内容：{messageHexString}");
                            break;
                    }
                }
                else
                {
                    await OnTcpNormalReceived(tcpSessionClient, messageHexString, aTuple.Item1);
                }
            }
            else
            {
                _logger.LogError($"指令初步解析失败，指令内容：{messageHexString}");
            }

            await e.InvokeNext();
        }

        //[GeneratorPlugin(typeof(ITcpConnectingPlugin))]
        public async Task OnTcpConnecting(ITcpSession client, ConnectingEventArgs e)
        {
            var tcpSessionClient = (ITcpSessionClient)client;

            _logger.LogInformation($"[TCP] ID:{tcpSessionClient.Id} IP:{client.GetIPPort()}正在连接中...");
            await e.InvokeNext();
        }

        //[GeneratorPlugin(typeof(ITcpConnectedPlugin))]
        public async Task OnTcpConnected(ITcpSession client, ConnectedEventArgs e)
        {
            var tcpSessionClient = (ITcpSessionClient)client;


            _logger.LogInformation($"[TCP] ID:{tcpSessionClient.Id} IP:{client.GetIPPort()}已连接");
            await e.InvokeNext();
        }

        //[GeneratorPlugin(typeof(ITcpClosedPlugin))]//ITcpSessionClient
        public async Task OnTcpClosed(ITcpSession client, ClosedEventArgs e)
        {

            var tcpSessionClient = (ITcpSessionClient)client;
            var entity = await _deviceRepository.FindAsync(a => a.ClientId == tcpSessionClient.Id);
            if (entity != null)
            {
                entity.UpdateByOnClosed();
                await _deviceRepository.UpdateAsync(entity);
            }
            else
            {
                _logger.LogWarning($"[TCP] ID:{tcpSessionClient.Id} IP:{client.GetIPPort()}已关闭连接,但采集程序检索失败");
            }

            await e.InvokeNext();
        }

        /// <summary>
        /// 登录帧处理
        /// </summary>
        /// <param name="client"></param>
        /// <param name="messageHexString"></param>
        /// <param name="deviceNo">集中器编号</param>
        /// <returns></returns>
        private async Task OnTcpLoginReceived(ITcpSessionClient client, string messageHexString, string deviceNo)
        {
            string oldClientId = $"{client.Id}";

            await client.ResetIdAsync(deviceNo);

            var entity = await _deviceRepository.FindAsync(a => a.Number == deviceNo);
            if (entity == null)
            {
                await _deviceRepository.InsertAsync(new Device(deviceNo, oldClientId, DateTime.Now, DateTime.Now, DeviceStatus.Online));
            }
            else
            {
                entity.UpdateByLoginAndHeartbeat(oldClientId);
                await _deviceRepository.UpdateAsync(entity);
            }

            var messageReceivedLoginEvent = new MessageReceivedLogin
            {
                ClientId = deviceNo,
                ClientIp = client.IP,
                ClientPort = client.Port,
                MessageHexString = messageHexString,
                DeviceNo = deviceNo,
                MessageId = NewId.NextGuid().ToString()
            };
            //await _producerBus.PublishAsync(ProtocolConst.SubscriberReceivedLoginEventName, messageReceivedLoginEvent);

            await _producerBus.Publish( messageReceivedLoginEvent);
        }

        private async Task OnTcpHeartbeatReceived(ITcpSessionClient client, string messageHexString, string deviceNo)
        {
            string clientId = deviceNo;
            string oldClientId = $"{client.Id}";

            var entity = await _deviceRepository.FindAsync(a => a.Number == deviceNo);
            if (entity == null) //没有登录帧的设备，只有心跳帧
            {
                await client.ResetIdAsync(clientId);
                await _deviceRepository.InsertAsync(new Device(deviceNo, oldClientId, DateTime.Now, DateTime.Now, DeviceStatus.Online));
            }
            else
            {
                if (clientId != oldClientId)
                {
                    entity.UpdateByLoginAndHeartbeat(oldClientId);
                }
                else
                {
                    entity.UpdateByLoginAndHeartbeat();
                }
                 
                await _deviceRepository.UpdateAsync(entity);
            }

            var messageReceivedHeartbeatEvent = new MessageReceivedHeartbeat
            {
                ClientId = clientId,
                ClientIp = client.IP,
                ClientPort = client.Port,
                MessageHexString = messageHexString,
                DeviceNo = deviceNo,
                MessageId = NewId.NextGuid().ToString()
            };
            //await _producerBus.PublishAsync(ProtocolConst.SubscriberReceivedHeartbeatEventName, messageReceivedHeartbeatEvent);
            await _producerBus.Publish(messageReceivedHeartbeatEvent);
        }

        private async Task OnTcpNormalReceived(ITcpSessionClient client, string messageHexString, string deviceNo)
        {
            await _producerBus.Publish(new MessageReceived
            {
                ClientId = client.Id,
                ClientIp = client.IP,
                ClientPort = client.Port,
                MessageHexString = messageHexString,
                DeviceNo = deviceNo,
                MessageId = NewId.NextGuid().ToString()
            });

            //await _producerBus.PublishAsync(ProtocolConst.SubscriberReceivedEventName, new MessageReceived
            //{
            //    ClientId = client.Id,
            //    ClientIp = client.IP,
            //    ClientPort = client.Port,
            //    MessageHexString = messageHexString,
            //    DeviceNo = deviceNo,
            //    MessageId = NewId.NextGuid().ToString()
            //});
        }
    }
}