Task+ConcurrentQueue多執行緒程式設計

UP技術控發表於2022-02-03

　佇列（Queue）代表了一個先進先出的物件集合。當您需要對各項進行先進先出的訪問時，則使用佇列。當您在列表中新增一項，稱為入隊，當您從列表中移除一項時，稱為出隊。

　　ConcurrentQueue<T>佇列是一個高效的執行緒安全的佇列，是.Net Framework 4.0，System.Collections.Concurrent名稱空間下的一個資料結構。

Task是在ThreadPool的基礎上推出的，我們知道了ThreadPool的弊端：我們不能控制執行緒池中執行緒的執行順序，也不能獲取執行緒池內執行緒取消/異常/完成的通知。net4.0在ThreadPool的基礎上推出了Task，Task擁有執行緒池的優點，同時也解決了使用執行緒池不易控制的弊端。

下面我們來看下Task+ConcurrentQueue實現多執行緒程式設計

1、定義最大執行緒數，一般和本機的cpu 有關

    /// <summary>
        /// 執行緒總數
        /// </summary>
        private int threadNum = Convert.ToInt32(ConfigurationManager.AppSettings["ThreadNum"]);

ConcurrentQueue佇列

   /// <summary>
        /// 佇列
        /// </summary>
        private ConcurrentQueue<AssetRepayment> queues = new ConcurrentQueue<AssetRepayment>();

2、接下來我們把業務資料取出來，加到定義的 queues.Enqueue(l);

  var dt = DateTime.Now.Date;
            var list = AssetRepayService.GetRepayments().Where(o => o.AssetRepayStatus == AssetRepayStatus.NoSend && o.PlanRepaymentDate == dt && o.AssetRepayMode == AssetRepayMode.Withholding);
            int count = list.Count();
            if (count == 0)
            {
                LogHelper.WriteFatal("代扣充值-沒有可執行的資料");
                return;
            }
            totalCount = count;
            int allpage = count / 200 + (count % 200 == 0 ? 0 : 1);
            int page = 0;
            LogHelper.WriteFatal("代扣充值-可執行的資料:" + count + "條，頁數："+ allpage);
            

            do
            {
                LogHelper.WriteFatal("代扣充值-第:" + page + "頁");

                var ll = list.OrderBy(o=>o.Id).Skip(page++ * 200).Take(200).ToList();
                foreach (var l in ll)
                {
                    queues.Enqueue(l);
                }
            }
            while (page < allpage);

　　3、等資料全部載入到queues，我們接著下一步

  List<Task> tasks = new List<Task>();
            for (int i = 0; i < threadNum; i++)
            {
                var task = Task.Run(() =>
                {
                    Process();
                });
                tasks.Add(task);
            }
            var taskList = Task.Factory.ContinueWhenAll(tasks.ToArray(), (ts) =>
            {
            });
            taskList.Wait();

利用Task 處理資料

這裡需要注意的是

   var taskList = Task.Factory.ContinueWhenAll(tasks.ToArray(), (ts) =>
            {
            });
            taskList.Wait();

這個代表開始執行執行緒並且需要全部完成才會退出 ContinueWhenAll

4 接下去我們看下處理程式怎麼消費資料

從佇列取資料

  var currentIndex = Interlocked.Increment(ref index);
                    AssetRepayment repayId = null;
                    var isExit = queues.TryDequeue(out repayId);
                    if (!isExit)
                    {
                        break;
                    }

　　有資料的話往下走執行我們要的業務邏輯

    var service = context.GetService<IDeTransactionService>();
                        service.SubDeTransaction(repayId);

                        LogHelper.WriteFatal(string.Format("代扣充值 共{0}條 當前第{1}條", totalCount, currentIndex));

　　完整處理方法如下

  private void Process()
        {
            using (var context = new MefContext())
            {
                while (true)
                {
                    var currentIndex = Interlocked.Increment(ref index);
                    AssetRepayment repayId = null;
                    var isExit = queues.TryDequeue(out repayId);
                    if (!isExit)
                    {
                        break;
                    }
                    try
                    {
                        var service = context.GetService<IDeTransactionService>();
                        service.SubDeTransaction(repayId);

                        LogHelper.WriteFatal(string.Format("代扣充值 共{0}條 當前第{1}條", totalCount, currentIndex));
                    }
                    catch (Exception ex)
                    {
                        LogHelper.WriteError("代扣充值-", ex);
                    }
                }
            }
        }

到此為止，我們實現了 Task+ConcurrentQueue多執行緒程式設計。

完整程式碼塊

連結：https://pan.baidu.com/s/1jgpafTFssiVLmZhDe1CgYQ

提取碼：erib

Task+ConcurrentQueue多執行緒程式設計

相關文章