ThreadPool/example.cpp at master · shenxuebing/ThreadPool · GitHub - github.com

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <iostream>
#include <vector>
#include <chrono>

#include "ThreadPool.h"

int main()
{

    ThreadPool pool(4);
    std::vector< std::future<int> > results;

    for(int i = 0; i < 8; ++i) {
        results.emplace_back(
            pool.enqueue([i] {
                std::cout << "hello " << i << std::endl;
                std::this_thread::sleep_for(std::chrono::seconds(10));
                std::cout << "world11111 " << i << std::endl;
                return i*i;
            })
        );
    }

	/*for (auto&& result : results)
		std::cout << result.get() << ' ';
	std::cout << std::endl;*/

    std::vector< std::future<int> > results2;
	std::vector<int> cores = { 0, 1 };
	ThreadPool pool2(4, cores, ThreadPool::Priority::HIGH);
	for (int i = 0; i < 8; ++i)
	{
		results2.emplace_back(
			pool2.enqueue([i]
				{
					std::cout << "hello " << i << std::endl;
					std::this_thread::sleep_for(std::chrono::seconds(1));
					std::cout << "world " << i << std::endl;
					return i * i;
				})
		);
	}

	for (auto&& result : results2)
		std::cout << result.get() << ' ';
	std::cout << std::endl;

	ThreadPool pool3(std::thread::hardware_concurrency(),
		{},
		ThreadPool::Priority::REALTIME);
	std::vector< std::future<int> > results3;

	for (int i = 0; i < 8; ++i)
	{
		results3.emplace_back(
			pool3.enqueue([i]
				{
					std::cout << "hello " << i << std::endl;
					std::this_thread::sleep_for(std::chrono::seconds(1));
					std::cout << "world " << i << std::endl;
					return i * i;
				})
		);
	}

	for (auto&& result : results3)
		std::cout << result.get() << ' ';
	std::cout << std::endl;


#if defined(_WIN32)
	ThreadPool pool4(4, { 0,1 }, 10);  // Windows -2~15
#elif defined(__linux__)
	ThreadPool pool4(4, { 0,1 }, 50);  // Linux 1~99
#endif
	std::vector< std::future<int> > results4;
    for (int i = 0; i < 8; ++i)
    {
        results4.emplace_back(
            pool4.enqueue([i]
                {
                    std::cout << "hello " << i << std::endl;
                    std::this_thread::sleep_for(std::chrono::seconds(1));
                    std::cout << "world " << i << std::endl;
                    return i * i;
                })
        );
    }
    for (auto&& result : results4)
        std::cout << result.get() << ' ';


	pool.drain();  // wait for all tasks to complete

    return 0;
}