FISLib_Old/Vinno.vCloud.Common.FIS/ConnectionCheckerV2.cs

using System;
using System.Threading;
using System.Threading.Tasks;
using Vinno.IUS.Common.Log;
using Vinno.vCloud.Common.FIS.Helper;
using WingInterfaceLibrary.DTO.ServerInfo;
using WingInterfaceLibrary.Interface;

namespace Vinno.vCloud.Common.FIS
{
    internal class ConnectionCheckerV2
    {
        private readonly IVinnoServerService _vinnoServerService;

        private readonly ManualResetEvent _waitEvent = new ManualResetEvent(false);

        private bool _defaultStatus;

        /// <summary>
        ///get or set connection check cycle
        /// </summary>
        internal int ConnectionCheckCycle { get; set; }

        public event EventHandler Offlined;

        public ConnectionCheckerV2(IVinnoServerService vinnoServerService, int connectionCheckCycle)
        {
            _vinnoServerService = vinnoServerService;
            ConnectionCheckCycle = connectionCheckCycle;
        }

        /// <summary>
        /// Start the Heartrate to keep the session available.
        /// </summary>
        public void Start(bool defaultStatus)
        {
            _defaultStatus = defaultStatus;
            Task.Run(() =>
            {
                while (!_waitEvent.WaitOne(TimeSpan.FromSeconds(ConnectionCheckCycle)))
                {
                    DoCheck();
                }
            });
        }

        /// <summary>
        /// Stop the Heartrate
        /// </summary>
        public void Stop()
        {
            _waitEvent.Set();
        }

        private void DoCheck()
        {
            try
            {
                if (_defaultStatus)
                {
                    if (!Check())
                    {
                        OnOfflined();
                    }
                }
                else
                {
                    OnOfflined();
                }
            }
            catch (Exception ex)
            {
                Logger.WriteLineError($"ConnectionCheckerV2 do check failed:{ex}");
            }
        }

        public bool Check()
        {
            try
            {
                EchoResult result = JsonRpcHelper.Echo(_vinnoServerService);
                if (result == null || result.Code != 0)
                {
                    return false;
                }
                else
                {
                    return true;
                }
            }
            catch (Exception ex)
            {
                Logger.WriteLineError($"ConnectionCheckerV2 check error:{ex}");
                return false;
            }
        }

        private void OnOfflined()
        {
            Offlined?.Invoke(this, EventArgs.Empty);
        }
    }
}