From 59779539e7e2522c379111dcd8aedbf750d3ddb6 Mon Sep 17 00:00:00 2001
From: sijchen <sijchen@cisco.com>
Date: Wed, 4 Nov 2015 10:29:08 -0800
Subject: [PATCH] add autolock in ThreadPoolTest to avoid possible conflict

---
 test/common/WelsThreadPoolTest.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/common/WelsThreadPoolTest.h b/test/common/WelsThreadPoolTest.h
index 955473de..62fff3d7 100644
--- a/test/common/WelsThreadPoolTest.h
+++ b/test/common/WelsThreadPoolTest.h
@@ -1,6 +1,7 @@
 #ifndef _WELS_THREAD_POOL_TEST_H_
 #define _WELS_THREAD_POOL_TEST_H_
 
+#include "WelsLock.h"
 #include "WelsThreadPool.h"
 
 using namespace WelsCommon;
@@ -14,12 +15,14 @@ class CThreadPoolTest : public IWelsThreadPoolSink {
   ~CThreadPoolTest() {}
 
   virtual int32_t OnTaskExecuted (IWelsTask* pTask) {
+    WelsCommon::CWelsAutoLock cAutoLock (m_cTaskCountLock);
     m_iTaskCount ++;
     //printf("Task execute over count is %d\n", m_iTaskCount);
     return cmResultSuccess;
   }
 
   virtual int32_t OnTaskCancelled (IWelsTask* pTask) {
+    WelsCommon::CWelsAutoLock cAutoLock (m_cTaskCountLock);
     m_iTaskCount ++;
     //printf("Task execute cancelled count is %d\n", m_iTaskCount);
     return cmResultSuccess;
@@ -31,6 +34,7 @@ class CThreadPoolTest : public IWelsThreadPoolSink {
 
  private:
   int32_t  m_iTaskCount;
+  WelsCommon::CWelsLock  m_cTaskCountLock;
 };