breeze/workflow
1
0
Fork 0
mirror of https://github.com/sogou/workflow.git synced 2026-02-08 01:33:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add thrdpool_decrease().

Browse Source
This commit is contained in:
Xie Han
2024-05-26 03:18:32 +08:00
committed by xiehan
parent fec3f6ced8
commit 30a78d30b3
2 changed files with 39 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/kernel/thrdpool.c
View File

@@ -41,13 +41,31 @@ struct __thrdpool_task_entry
static pthread_t __zero_tid;
static void __thrdpool_exit_routine(void *context)
{
thrdpool_t *pool = (thrdpool_t *)context;
pthread_t tid;
/* One thread joins another. Don't need to keep all thread IDs. */
pthread_mutex_lock(&pool->mutex);
tid = pool->tid;
pool->tid = pthread_self();
if (--pool->nthreads == 0 && pool->terminate)
pthread_cond_signal(pool->terminate);
pthread_mutex_unlock(&pool->mutex);
if (!pthread_equal(tid, __zero_tid))
pthread_join(tid, NULL);
pthread_exit(NULL);
}
static void *__thrdpool_routine(void *arg)
{
thrdpool_t *pool = (thrdpool_t *)arg;
struct __thrdpool_task_entry *entry;
void (*task_routine)(void *);
void *task_context;
pthread_t tid;
pthread_setspecific(pool->key, pool);
while (!pool->terminate)
@@ -69,17 +87,7 @@ static void *__thrdpool_routine(void *arg)
}
}
/* One thread joins another. Don't need to keep all thread IDs. */
pthread_mutex_lock(&pool->mutex);
tid = pool->tid;
pool->tid = pthread_self();
if (--pool->nthreads == 0)
pthread_cond_signal(pool->terminate);
pthread_mutex_unlock(&pool->mutex);
if (!pthread_equal(tid, __zero_tid))
pthread_join(tid, NULL);
__thrdpool_exit_routine(pool);
return NULL;
}
@@ -227,6 +235,23 @@ int thrdpool_increase(thrdpool_t *pool)
return -1;
}
int thrdpool_decrease(thrdpool_t *pool)
{
void *buf = malloc(sizeof (struct __thrdpool_task_entry));
struct __thrdpool_task_entry *entry;
if (buf)
{
entry = (struct __thrdpool_task_entry *)buf;
entry->task.routine = __thrdpool_exit_routine;
entry->task.context = pool;
msgqueue_put_head(entry, pool->msgqueue);
return 0;
}
return -1;
}
inline int thrdpool_in_pool(thrdpool_t *pool);
int thrdpool_in_pool(thrdpool_t *pool)
@@ -247,7 +272,7 @@ void thrdpool_destroy(void (*pending)(const struct thrdpool_task *),
if (!entry)
break;
if (pending)
if (pending && entry->task.routine != __thrdpool_exit_routine)
pending(&entry->task);
free(entry);

1
src/kernel/thrdpool.h
View File

@@ -49,6 +49,7 @@ extern "C"
thrdpool_t *thrdpool_create(size_t nthreads, size_t stacksize);
int thrdpool_schedule(const struct thrdpool_task *task, thrdpool_t *pool);
int thrdpool_increase(thrdpool_t *pool);
int thrdpool_decrease(thrdpool_t *pool);
int thrdpool_in_pool(thrdpool_t *pool);
void thrdpool_destroy(void (*pending)(const struct thrdpool_task *),
thrdpool_t *pool);