[Common] Add sink to IWelsTask
Showing the changes to test/common/WelsThreadPoolTest.cpp (1 of 10 files). Show all
Unified · Split
| @@ -16,7 +16,7 @@ class CSimpleTask : public IWelsTask { · 15 unmodified lines | |||
| 16 | public: | 16 | public: |
| 17 | static uint32_t id; | 17 | static uint32_t id; |
| 18 | 18 | ||
| 19 | CSimpleTask() { | 19 | CSimpleTask (WelsCommon::IWelsTaskSink* pSink) : IWelsTask (pSink) { |
| 20 | m_uiID = id ++; | 20 | m_uiID = id ++; |
| 21 | } | 21 | } |
| 22 | 22 | ||
| @@ -38,18 +38,25 @@ uint32_t CSimpleTask::id = 0; · 15 unmodified lines | |||
| 38 | 38 | ||
| 39 | 39 | ||
| 40 | TEST (CThreadPoolTest, CThreadPoolTest) { | 40 | TEST (CThreadPoolTest, CThreadPoolTest) { |
| 41 | CSimpleTask tasks[TEST_TASK_NUM]; | ||
| 42 | CThreadPoolTest cThreadPoolTest; | 41 | CThreadPoolTest cThreadPoolTest; |
| 42 | CSimpleTask* aTasks[TEST_TASK_NUM]; | ||
| 43 | CWelsThreadPool cThreadPool (&cThreadPoolTest); | 43 | CWelsThreadPool cThreadPool (&cThreadPoolTest); |
| 44 | 44 | ||
| 45 | int32_t i; | 45 | int32_t i; |
| 46 | for (i = 0; i < TEST_TASK_NUM; i++) { | ||
| 47 | aTasks[i] = new CSimpleTask (&cThreadPoolTest); | ||
| 48 | } | ||
| 46 | 49 | ||
| 47 | for (i = 0; i < TEST_TASK_NUM; i++) { | 50 | for (i = 0; i < TEST_TASK_NUM; i++) { |
| 48 | cThreadPool.QueueTask (&tasks[i]); | 51 | cThreadPool.QueueTask (aTasks[i]); |
| 49 | } | 52 | } |
| 50 | 53 | ||
| 51 | while (cThreadPoolTest.GetTaskCount() < TEST_TASK_NUM) { | 54 | while (cThreadPoolTest.GetTaskCount() < TEST_TASK_NUM) { |
| 52 | WelsSleep (1); | 55 | WelsSleep (1); |
| 53 | } | 56 | } |
| 57 | |||
| 58 | for (i = 0; i < TEST_TASK_NUM; i++) { | ||
| 59 | delete aTasks[i]; | ||
| 60 | } | ||
| 54 | } | 61 | } |
| 55 | 62 | ||