120 lines
3.3 KiB
C
120 lines
3.3 KiB
C
/*
|
|
* 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 _TD_SCHEDULER_H_
|
|
#define _TD_SCHEDULER_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include "catalog.h"
|
|
#include "planner.h"
|
|
|
|
extern tsem_t schdRspSem;
|
|
|
|
typedef struct SQueryProfileSummary {
|
|
int64_t startTs; // Object created and added into the message queue
|
|
int64_t endTs; // the timestamp when the task is completed
|
|
int64_t cputime; // total cpu cost, not execute elapsed time
|
|
|
|
int64_t loadRemoteDataDuration; // remote io time
|
|
int64_t loadNativeDataDuration; // native disk io time
|
|
|
|
uint64_t loadNativeData; // blocks + SMA + header files
|
|
uint64_t loadRemoteData; // remote data acquired by exchange operator.
|
|
|
|
uint64_t waitDuration; // the time to waiting to be scheduled in queue does matter, so we need to record it
|
|
int64_t addQTs; // the time to be added into the message queue, used to calculate the waiting duration in queue.
|
|
|
|
uint64_t totalRows;
|
|
uint64_t loadRows;
|
|
uint32_t totalBlocks;
|
|
uint32_t loadBlocks;
|
|
uint32_t loadBlockAgg;
|
|
uint32_t skipBlocks;
|
|
uint64_t resultSize; // generated result size in Kb.
|
|
} SQueryProfileSummary;
|
|
|
|
typedef struct STaskInfo {
|
|
SQueryNodeAddr addr;
|
|
SSubQueryMsg* msg;
|
|
} STaskInfo;
|
|
|
|
typedef struct SSchdFetchParam {
|
|
void** pData;
|
|
int32_t* code;
|
|
} SSchdFetchParam;
|
|
|
|
typedef void (*schedulerExecFp)(SExecResult* pResult, void* param, int32_t code);
|
|
typedef void (*schedulerFetchFp)(void* pResult, void* param, int32_t code);
|
|
typedef bool (*schedulerChkKillFp)(void* param);
|
|
|
|
typedef struct SSchedulerReq {
|
|
bool syncReq;
|
|
bool localReq;
|
|
SRequestConnInfo* pConn;
|
|
SArray* pNodeList;
|
|
SQueryPlan* pDag;
|
|
int64_t allocatorRefId;
|
|
const char* sql;
|
|
int64_t startTs;
|
|
schedulerExecFp execFp;
|
|
schedulerFetchFp fetchFp;
|
|
void* cbParam;
|
|
schedulerChkKillFp chkKillFp;
|
|
void* chkKillParam;
|
|
SExecResult* pExecRes;
|
|
void** pFetchRes;
|
|
} SSchedulerReq;
|
|
|
|
int32_t schedulerInit(void);
|
|
|
|
int32_t schedulerExecJob(SSchedulerReq* pReq, int64_t* pJob);
|
|
|
|
int32_t schedulerFetchRows(int64_t jobId, SSchedulerReq* pReq);
|
|
|
|
void schedulerFetchRowsA(int64_t job, schedulerFetchFp fp, void* param);
|
|
|
|
int32_t schedulerGetTasksStatus(int64_t job, SArray* pSub);
|
|
|
|
void schedulerStopQueryHb(void* pTrans);
|
|
|
|
int32_t schedulerUpdatePolicy(int32_t policy);
|
|
int32_t schedulerEnableReSchedule(bool enableResche);
|
|
|
|
/**
|
|
* Cancel query job
|
|
* @param pJob
|
|
* @return
|
|
*/
|
|
// int32_t scheduleCancelJob(void *pJob);
|
|
|
|
/**
|
|
* Free the query job
|
|
* @param pJob
|
|
*/
|
|
void schedulerFreeJob(int64_t* job, int32_t errCode);
|
|
|
|
void schedulerDestroy(void);
|
|
|
|
void schdExecCallback(SExecResult* pResult, void* param, int32_t code);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_SCHEDULER_H_*/
|