win32/reals/real_threadpool.h

// Copyright (c) Microsoft. All rights reserved.

#ifndef REAL_THREADPOOL_H
#define REAL_THREADPOOL_H

#include "macro_utils/macro_utils.h"

#include "c_pal/thandle.h"

#define R2(X) REGISTER_GLOBAL_MOCK_HOOK(X, real_##X);

#define REGISTER_THREADPOOL_GLOBAL_MOCK_HOOK()          \
    MU_FOR_EACH_1(R2,                                   \
        threadpool_create, \
        threadpool_create_work_item, \
        threadpool_schedule_work, \
        threadpool_schedule_work_item, \
        threadpool_timer_start, \
        threadpool_timer_restart, \
        threadpool_timer_cancel \
    ) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_MOVE(THREADPOOL), THANDLE_MOVE(real_THREADPOOL)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE(THREADPOOL), THANDLE_INITIALIZE(real_THREADPOOL)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE_MOVE(THREADPOOL), THANDLE_INITIALIZE_MOVE(real_THREADPOOL)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_ASSIGN(THREADPOOL), THANDLE_ASSIGN(real_THREADPOOL)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_MOVE(THREADPOOL_WORK_ITEM), THANDLE_MOVE(real_THREADPOOL_WORK_ITEM)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE(THREADPOOL_WORK_ITEM), THANDLE_INITIALIZE(real_THREADPOOL_WORK_ITEM)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE_MOVE(THREADPOOL_WORK_ITEM), THANDLE_INITIALIZE_MOVE(real_THREADPOOL_WORK_ITEM)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_ASSIGN(THREADPOOL_WORK_ITEM), THANDLE_ASSIGN(real_THREADPOOL_WORK_ITEM)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_MOVE(THREADPOOL_TIMER), THANDLE_MOVE(real_THREADPOOL_TIMER)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE(THREADPOOL_TIMER), THANDLE_INITIALIZE(real_THREADPOOL_TIMER)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_INITIALIZE_MOVE(THREADPOOL_TIMER), THANDLE_INITIALIZE_MOVE(real_THREADPOOL_TIMER)) \
    REGISTER_GLOBAL_MOCK_HOOK(THANDLE_ASSIGN(THREADPOOL_TIMER), THANDLE_ASSIGN(real_THREADPOOL_TIMER)) \

#ifdef __cplusplus
extern "C" {
#endif

    typedef struct THREADPOOL_TAG real_THREADPOOL;
    THANDLE_TYPE_DECLARE(real_THREADPOOL);
    typedef struct THREADPOOL_WORK_ITEM_TAG real_THREADPOOL_WORK_ITEM;
    THANDLE_TYPE_DECLARE(real_THREADPOOL_WORK_ITEM);
    typedef struct THREADPOOL_TIMER_TAG real_THREADPOOL_TIMER;
    THANDLE_TYPE_DECLARE(real_THREADPOOL_TIMER);

    THANDLE(THREADPOOL) real_threadpool_create(EXECUTION_ENGINE_HANDLE execution_engine);

    THANDLE(THREADPOOL_WORK_ITEM) real_threadpool_create_work_item(THANDLE(THREADPOOL) threadpool, THREADPOOL_WORK_FUNCTION work_function, void* work_function_context);

    int real_threadpool_schedule_work_item(THANDLE(THREADPOOL) threadpool, THANDLE(THREADPOOL_WORK_ITEM) threadpool_work_item);

    int real_threadpool_schedule_work(THANDLE(THREADPOOL) threadpool, THREADPOOL_WORK_FUNCTION work_function, void* work_function_context);

    THANDLE(THREADPOOL_TIMER) real_threadpool_timer_start(THANDLE(THREADPOOL) threadpool, uint32_t start_delay_ms, uint32_t timer_period_ms, THREADPOOL_WORK_FUNCTION work_function, void* work_function_context);

    int real_threadpool_timer_restart(THANDLE(THREADPOOL_TIMER) timer, uint32_t start_delay_ms, uint32_t timer_period_ms);

    void real_threadpool_timer_cancel(THANDLE(THREADPOOL_TIMER) timer);

#ifdef __cplusplus
}
#endif

#endif //REAL_THREADPOOL_H