tsched.h 2.3 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef TDENGINE_TSCHED_H
#define TDENGINE_TSCHED_H

#ifdef __cplusplus
extern "C" {
#endif

J
Jeff Tao 已提交
23 24
typedef struct SSchedMsg {
  void (*fp)(struct SSchedMsg *);
H
hzcheng 已提交
25
  void (*tfp)(void *, void *);
S
slguan 已提交
26
  void *msg;
H
hzcheng 已提交
27 28 29 30
  void *ahandle;
  void *thandle;
} SSchedMsg;

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
/**
 * Create a thread-safe ring-buffer based task queue and return the instance. A thread
 * pool will be created to consume the messages in the queue.
 * @param capacity the queue capacity
 * @param numOfThreads the number of threads for the thread pool
 * @param label the label of the queue
 * @return the created queue scheduler
 */
void *taosInitScheduler(int capacity, int numOfThreads, const char *label);

/**
 * Create a thread-safe ring-buffer based task queue and return the instance.
 * Same as taosInitScheduler, and it also print the queue status every 1 minite.
 * @param capacity the queue capacity
 * @param numOfThreads the number of threads for the thread pool
 * @param label the label of the queue
 * @param tmrCtrl the timer controller, tmr_ctrl_t*
 * @return the created queue scheduler
 */
void *taosInitSchedulerWithInfo(int capacity, int numOfThreads, const char *label, void *tmrCtrl);

/**
 * Clean up the queue scheduler instance and free the memory.
 * @param queueScheduler the queue scheduler to free
 */
void taosCleanUpScheduler(void *queueScheduler);

/**
 * Schedule a new task to run, the task is described by pMsg.
 * The function may be blocked if no thread is available to execute the task.
 * That may happen when all threads are busy.
 * @param queueScheduler the queue scheduler instance
 * @param pMsg the message for the task
 */
void taosScheduleTask(void *queueScheduler, SSchedMsg *pMsg);
H
hzcheng 已提交
66 67 68 69 70 71

#ifdef __cplusplus
}
#endif

#endif  // TDENGINE_TSCHED_H