|
1 /* |
|
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
3 * |
|
4 * Copyright 2011 Olaf Wintermann. All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions are met: |
|
8 * |
|
9 * 1. Redistributions of source code must retain the above copyright |
|
10 * notice, this list of conditions and the following disclaimer. |
|
11 * |
|
12 * 2. Redistributions in binary form must reproduce the above copyright |
|
13 * notice, this list of conditions and the following disclaimer in the |
|
14 * documentation and/or other materials provided with the distribution. |
|
15 * |
|
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
26 * POSSIBILITY OF SUCH DAMAGE. |
|
27 */ |
|
28 |
|
29 #include <stdio.h> |
|
30 #include <stdlib.h> |
|
31 |
|
32 #include "../ucx/map.h" |
|
33 |
|
34 #include "threadpools.h" |
|
35 |
|
36 |
|
37 UcxMap *thread_pool_map = NULL; |
|
38 int numthrpools = 0; |
|
39 |
|
40 threadpool_t *default_thread_pool = NULL; |
|
41 |
|
42 threadpool_t *last_thrpool_c = NULL; |
|
43 |
|
44 int create_threadpool(sstr_t name, ThreadPoolConfig *cfg) { |
|
45 if(thread_pool_map == NULL) { |
|
46 thread_pool_map = ucx_map_new(16); |
|
47 } |
|
48 |
|
49 threadpool_t *pool = ucx_map_sstr_get(thread_pool_map, name); |
|
50 if(pool) { |
|
51 /* TODO: reconfig thread pool */ |
|
52 return 0; |
|
53 } else { |
|
54 threadpool_t *tp = threadpool_new(cfg->min_threads); |
|
55 |
|
56 int ret = ucx_map_sstr_put(thread_pool_map, name, tp); |
|
57 |
|
58 if(ret == 0) { |
|
59 numthrpools++; |
|
60 last_thrpool_c = tp; |
|
61 } |
|
62 |
|
63 return ret; |
|
64 } |
|
65 } |
|
66 |
|
67 int check_thread_pool_cfg() { |
|
68 if(numthrpools > 0 ) { |
|
69 if(default_thread_pool) { |
|
70 return 0; |
|
71 } else { |
|
72 default_thread_pool = last_thrpool_c; |
|
73 return 0; |
|
74 } |
|
75 } |
|
76 |
|
77 ThreadPoolConfig cfg; |
|
78 cfg.min_threads = 4; |
|
79 cfg.max_threads = 8; |
|
80 cfg.queue_size = 64; |
|
81 cfg.stack_size = 262144; |
|
82 |
|
83 return create_threadpool(sstr("default"), &cfg); |
|
84 } |
|
85 |
|
86 threadpool_t* get_default_threadpool() { |
|
87 return default_thread_pool; |
|
88 } |
|
89 |