thread_pool.h

Go to the documentation of this file.

#pragma once

 
/* Storage capacities in elements; text buffers include the terminating NUL. */
#define THREAD_POOL_NAME_BUFFER_SIZE 64
 
#include <stdint.h>
#include <stdbool.h>
#include "common.h"
#include <libplatform/mutex.h>
#include <libplatform/cond.h>
#include <libplatform/thread.h>
 
// Forward declaration
typedef struct thread_pool thread_pool_t;
typedef struct thread_pool_entry thread_pool_entry_t;

typedef struct thread_pool_work_entry {
  void *(*work_func)(void *);          
  void *work_arg;                      
  struct thread_pool_work_entry *next; 
} thread_pool_work_entry_t;

struct thread_pool_entry {
  platform_thread_t thread;               
  int stop_id;                             
  void *(*thread_func)(void *);            
  void *thread_arg;                        
  char name[THREAD_POOL_NAME_BUFFER_SIZE]; 
  struct thread_pool_entry *next;          
};

struct thread_pool {
  char name[THREAD_POOL_NAME_BUFFER_SIZE]; 
  thread_pool_entry_t *threads;            
  mutex_t threads_mutex;                   
  size_t thread_count;                     
 
  // Work queue mode fields (only used if num_workers > 0)
  bool is_work_queue_mode;              
  size_t num_workers;                   
  thread_pool_work_entry_t *work_queue; 
  mutex_t work_queue_mutex;             
  cond_t work_available;                
  bool shutdown_requested;              
};

thread_pool_t *thread_pool_create(const char *pool_name);

thread_pool_t *thread_pool_create_with_workers(const char *pool_name, size_t num_workers);

void thread_pool_destroy(thread_pool_t *pool);

asciichat_error_t thread_pool_queue_work(const char *name, thread_pool_t *pool, void *(*work_func)(void *),
                                         void *work_arg);

asciichat_error_t thread_pool_spawn(thread_pool_t *pool, void *(*thread_func)(void *), void *thread_arg, int stop_id,
                                    const char *thread_name);

asciichat_error_t thread_pool_stop_all(thread_pool_t *pool);

size_t thread_pool_get_count(const thread_pool_t *pool);

bool thread_pool_has_threads(const thread_pool_t *pool);

asciichat_error_t thread_pool_interrupt_all(thread_pool_t *pool, int sig);