예제 #1
0
        /// <summary>
        /// Then main running part of main thread.
        /// <para></para>
        /// Main thread creates thread to run tasks, and manage the threads' count.
        /// </summary>
        private void Main()
        {
            while (toState != TaskPoolState.STOP)
            {
                run.WaitOne();
                int count = 0;
                while (true)
                {
                    locker.EnterReadLock();
                    count = running.Count;
                    locker.ExitReadLock();

                    if (count >= maxThreadCount || toState != TaskPoolState.RUN)
                    {
                        break;
                    }

                    if (!AddThread())
                    {
                        break;
                    }
                }

                locker.EnterWriteLock();
                if (count == 0)
                {
                    currentState = TaskPoolState.PAUSE;
                }
                else
                {
                    currentState = TaskPoolState.RUN;
                }
                locker.ExitWriteLock();
            }
        }
예제 #2
0
        /// <summary>
        /// Try to pause task box until no running tasks.
        /// </summary>
        /// <param name="waitInterval"></param>
        /// <returns></returns>
        public bool Pause(int waitInterval = 1000)
        {
            /// set toState to pause.
            locker.EnterWriteLock();
            if (toState == TaskPoolState.RUN)
            {
                toState = TaskPoolState.PAUSE;
            }
            locker.ExitWriteLock();

            TaskPoolState state;

            while (true)
            {
                locker.EnterReadLock();
                state = currentState;
                locker.ExitReadLock();
                if (state != TaskPoolState.RUN)
                {
                    return(true);
                }

                Thread.Sleep(waitInterval);
            }
        }
예제 #3
0
 /// <summary>
 /// Resume the task pool to fetch new tasks and run.
 /// </summary>
 public void Resume()
 {
     locker.EnterWriteLock();
     toState = TaskPoolState.RUN;
     locker.ExitWriteLock();
     run.Set();
 }
예제 #4
0
        private TaskBox(int maxThreadCount, bool isPreRunOne, bool isPostRunOne)
        {
            this.maxThreadCount = maxThreadCount;
            this.isPreRunOne    = isPreRunOne;
            this.isPostRunOne   = isPostRunOne;

            remains      = new Queue <IPoolTask>();
            running      = new Dictionary <string, IPoolTask>();
            toState      = TaskPoolState.RUN;
            currentState = TaskPoolState.PAUSE;

            new Thread(Main).Start();
        }
예제 #5
0
        public static void Start()
        {
            if (_state == TaskPoolState.Unstarted)
            {
                _state = TaskPoolState.Starting;
                Init();

                foreach (var queue in _pool)
                {
                    new Thread(queue.Value.Dispatcher.Start).Start();
                }
                _state = TaskPoolState.Running;
            }
        }
예제 #6
0
        /// <summary>
        /// Constructor of TaskPool.
        /// </summary>
        /// <param name="taskGenerator"></param>
        /// <param name="maxThreadCount"></param>
        private TaskPool(IPoolTaskGenerator <IPoolTask> taskGenerator, int maxThreadCount)
        {
            toState             = TaskPoolState.RUN;
            currentState        = TaskPoolState.PAUSE;
            this.maxThreadCount = maxThreadCount;

            /// Start a thread to fetch <see cref="IPoolTask"/>.
            taskFetch = new TaskFetchThread(taskGenerator);
            Task thread = new Task(taskFetch.Run);

            thread.Start();

            mainThread = new Thread(Main);
            mainThread.Start();
        }
예제 #7
0
        public static void Stop()
        {
            if (_state == TaskPoolState.Running)
            {
                _state = TaskPoolState.Stopping;
                foreach (var kvp in _pool)
                {
                    kvp.Value.Dispatcher.Stop();
                }

                while (_pool.Any(kvp => !kvp.Value.Dispatcher.IsIdle))
                {
                    Thread.Sleep(1000);
                }

                foreach (var kvp in _pool)
                {
                    kvp.Value.Dispose();
                }

                _pool.Clear();
                _state = TaskPoolState.Unstarted;
            }
        }
예제 #8
0
        /// <summary>
        /// Add a thread to run tasks.
        /// </summary>
        /// <returns></returns>
        private bool AddThread()
        {
            int       count = 0;
            IPoolTask task  = FetchTask();

            if (task == null)
            {
                return(false);
            }

            Task thread = null;

            TaskPoolState m_toState;

            thread = new Task(() =>
            {
                while (true)
                {
                    if (task != null)
                    {
                        locker.EnterReadLock();
                        m_toState = toState;
                        locker.ExitReadLock();

                        if (m_toState == TaskPoolState.RUN)
                        {
                            task.Run();
                            RemoveRunning(task);
                            task = null;

                            /// When finished a task, check <see cref="toState"/>.
                            locker.EnterReadLock();
                            m_toState = toState;
                            locker.ExitReadLock();
                            if (m_toState == TaskPoolState.RUN)
                            {
                                task = FetchTask();
                                if (task != null)
                                {
                                    AddRunning(task, thread);
                                }
                            }
                        }
                        else
                        {
                            task.Cancel();
                            RemoveRunning(task, false);
                            task = null;
                        }
                    }

                    // When the intent is not to run, then cancel current task,
                    if (task == null)
                    {
                        runningLocker.EnterReadLock();
                        count = running.Count;
                        runningLocker.ExitReadLock();
                        if (count == 0)
                        {
                            locker.EnterWriteLock();
                            currentState = TaskPoolState.PAUSE;
                            locker.ExitWriteLock();
                        }
                        break;
                    }
                }
            });
            AddRunning(task, thread);
            thread.Start();
            return(true);
        }
예제 #9
0
 public void Stop()
 {
     Abort();
     WakeUpAll();
     State = TaskPoolState.Stopped;
 }
예제 #10
0
 public void Start()
 {
     State = TaskPoolState.Started;
     foreach (var executor in _executors)
     {
         executor.Run();
     }
 }
예제 #11
0
        /// <summary>
        /// Create a thread, and fetch a task to run in it.
        /// </summary>
        /// <returns></returns>
        private bool AddThread()
        {
            if (toState != TaskPoolState.RUN)
            {
                return(false);
            }

            IPoolTask task = taskFetch.FetchNewTask();

            if (task == null)
            {
                return(false);
            }

            Task thread = null;

            thread = new Task(() =>
            {
                while (true)
                {
                    task.Run();

                    /// When current task is finished, remove it from <see cref="running"/>.
                    locker.EnterWriteLock();
                    running.Remove(task.Key);
                    locker.ExitWriteLock();

                    /// If pool will run, then fetch a new task and add it to <see cref="running"/>
                    if (toState == TaskPoolState.RUN)
                    {
                        task = taskFetch.FetchNewTask();
                        if (task != null)
                        {
                            locker.EnterWriteLock();
                            task.AttachThread(thread);
                            running.Add(task.Key, task);
                            locker.ExitWriteLock();
                        }
                    }
                    else
                    {
                        task = null;
                    }

                    /// When this thread are running to end, get count of running tasks.
                    /// If count equals 0, it means there is no running task.
                    /// Then set current state to <see cref="TaskPoolState.PAUSE"/>
                    if (task == null)
                    {
                        int count = 0;
                        locker.EnterUpgradeableReadLock();
                        count = running.Count;
                        if (count == 0)
                        {
                            locker.EnterWriteLock();
                            currentState = TaskPoolState.PAUSE;
                            locker.ExitWriteLock();
                        }
                        locker.ExitUpgradeableReadLock();
                        break;
                    }
                }
            });
            task.AttachThread(thread);

            locker.EnterWriteLock();
            running.Add(task.Key, task);
            locker.ExitWriteLock();
            thread.Start();
            return(true);
        }