namespace NBC
{
    public class ParallelTaskCollection : TaskCollection
    {
        private int _currentIndex;

        /// <summary>
        ///     最大并行数量 (默认为9)
        /// </summary>
        public int ParallelNum = 9;

        protected override NTaskStatus RunTasksAndCheckIfDone()
        {
            var st = NTaskStatus.Running;

            if (CurRunTask.Count < ParallelNum && _currentIndex < RawList.Count)
            {
                var num = ParallelNum - CurRunTask.Count;
                for (var index = 0; index < num; index++)
                    if (_currentIndex < RawList.Count)
                    {
                        CurRunTask.Add(RawList[_currentIndex]);
                        _currentIndex += 1;
                    }
            }

            for (var index = 0; index < CurrentTask.Count; index++)
            {
                var element = CurrentTask[index];
                var childSt = element.Process();

                if (FailBreak && childSt == NTaskStatus.Fail)
                {
                    _errorMsg = element.ErrorMsg;
                    st = NTaskStatus.Fail;
                    break;
                }

                if (childSt == NTaskStatus.Success || childSt == NTaskStatus.Fail)
                {
                    CurrentTask.RemoveAt(index);
                    index--;
                    FinishList.Add(element);
                }
            }

            if (FinishList.Count >= RawList.Count) st = NTaskStatus.Success;

            // for (var index = 0; index < CurrentTask.Count; index++)
            // {
            //     var element = CurrentTask[index];
            //     var childSt = element.Process();
            //
            //     if (childSt >= Status.Success)
            //     {
            //         if (FailBreak && childSt == Status.Fail)
            //         {
            //             _errorMsg = element.ErrorMsg;
            //             st = Status.Fail;
            //         }
            //
            //         CurrentTask.RemoveAt(index);
            //         index--;
            //         FinishList.Add(element);
            //     }
            // }
            //
            // if (FinishList.Count >= RawList.Count)
            // {
            //     st = Status.Success;
            // }
            // else if (CurRunTask.Count < ParallelNum && _currentIndex < RawList.Count)
            // {
            //     var num = ParallelNum - CurRunTask.Count;
            //     for (var index = 0; index < num; index++)
            //     {
            //         if (_currentIndex < RawList.Count)
            //         {
            //             CurRunTask.Add(RawList[_currentIndex]);
            //             _currentIndex += 1;
            //         }
            //     }
            // }

            return st;
        }

        public override void Reset()
        {
            base.Reset();
            _currentIndex = 0;
        }

        public override void Stop()
        {
            base.Stop();
            _currentIndex = 0;
        }

        public override void Clear()
        {
            base.Clear();
            _currentIndex = 0;
        }
    }
}