CyberX8/Framework/Common/Utilities/PeriodicJob.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using Aitex.Core.RT.Log;

namespace Aitex.Core.Util
{
    public class PeriodicJob
    {
        Thread _thread;
        int _interval;
        Func<bool> _func;
        
        CancellationTokenSource _cancelFlag = new CancellationTokenSource();        //for quit thread 
        ManualResetEvent _waitFlag = new ManualResetEvent(true);                    //for pause purpose
        ManualResetEvent _sleepFlag = new ManualResetEvent(false);                //for sleep time

        object _locker = new object();

        #region 属性
        /// <summary>
        /// 间隔时间
        /// </summary>
        public int Interval { get { return _interval; } }
        #endregion

        public PeriodicJob(int interval, Func<bool> func, string name, bool isStartNow=false, bool isBackground=true)
        {
            _thread = new Thread(new ParameterizedThreadStart(ThreadFunction));
            _thread.Name = name;
            _thread.IsBackground = isBackground;

            LOG.WriteLog(eEvent.EV_SYSTEM_CONFIG, "System", $"Create PeriodJob {name}");
            _interval = interval;
            _func = func;

            if (isStartNow)
                Start();
        }

        public void Start()
        {
            //lock (_locker)
            {
                if (_thread == null)
                    return;

                _waitFlag.Set();

                if (_thread.IsAlive)
                    return;

                _thread.Start(this);
            }
        }

        public void Pause()
        {
            //lock (_locker)
            {
                _waitFlag.Reset();
            }
        }

        public void Stop(bool showLog=true)
        {
            try
            {
                //lock (_locker)
                {
                    _sleepFlag.Set(); //do not sleep

                    _waitFlag.Set();    //do not pause

                    _cancelFlag.Cancel();   //quit

                    if (_thread == null)
                        return;

                    if (_thread.ThreadState != ThreadState.Suspended)
                    {
                        try
                        {
                            _thread.Abort();
                        }
                        catch (Exception ex)
                        {
                            if (showLog)
                            {
                                LOG.WriteExeption("Thread aborted exception", ex);
                            }
                        }
                    }

                    _thread = null;
                }
            }
            catch (Exception ex)
            {
                LOG.WriteExeption("Thread stop exception", ex);
            }
        }

        void ThreadFunction(object param)
        {
            PeriodicJob t = (PeriodicJob)param;
            t.Run();
        }


        public void ChangeInterval(int msInterval)
        {
            _interval = msInterval;
        }

        void Run()
        {
            while (!_cancelFlag.IsCancellationRequested)
            {
                try
                {
                    _waitFlag.WaitOne();
                    DateTime now = DateTime.Now;

                    if (!_func())
                        break;
                    int elapseTime = (int)DateTime.Now.Subtract(now).TotalMilliseconds;
                    _sleepFlag.WaitOne(Math.Max(_interval - elapseTime, 30));
                }
                catch (Exception ex)
                {
                    LOG.WriteExeption(ex);
                }
            }
            
        }
    }
}