qworkerMsg.c 16.5 KB
Newer Older
D
dapan1121 已提交
1
#include "qworker.h"
S
common  
Shengliang Guan 已提交
2
#include "tcommon.h"
D
dapan1121 已提交
3 4 5 6
#include "executor.h"
#include "planner.h"
#include "query.h"
#include "qworkerInt.h"
D
dapan1121 已提交
7
#include "qworkerMsg.h"
D
dapan1121 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
#include "tmsg.h"
#include "tname.h"
#include "dataSinkMgt.h"


int32_t qwMallocFetchRsp(int32_t length, SRetrieveTableRsp **rsp) {
  int32_t msgSize = sizeof(SRetrieveTableRsp) + length;
  
  SRetrieveTableRsp *pRsp = (SRetrieveTableRsp *)rpcMallocCont(msgSize);
  if (NULL == pRsp) {
    qError("rpcMallocCont %d failed", msgSize);
    QW_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

  memset(pRsp, 0, sizeof(SRetrieveTableRsp));

  *rsp = pRsp;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
29 30


D
dapan1121 已提交
31
void qwBuildFetchRsp(void *msg, SOutputData *input, int32_t len, bool qComplete) {
D
dapan1121 已提交
32 33
  SRetrieveTableRsp *rsp = (SRetrieveTableRsp *)msg;
  
D
dapan1121 已提交
34
  rsp->useconds = htobe64(input->useconds);
D
dapan1121 已提交
35
  rsp->completed = qComplete;
D
dapan1121 已提交
36 37 38 39 40 41 42 43
  rsp->precision = input->precision;
  rsp->compressed = input->compressed;
  rsp->compLen = htonl(len);
  rsp->numOfRows = htonl(input->numOfRows);
}


void qwFreeFetchRsp(void *msg) {
D
dapan 已提交
44 45 46
  if (msg) {
    rpcFreeCont(msg);
  }
D
dapan1121 已提交
47 48
}

S
shm  
Shengliang Guan 已提交
49
int32_t qwBuildAndSendQueryRsp(SQWConnInfo *pConn, int32_t code) {
D
dapan1121 已提交
50 51 52 53 54
  SQueryTableRsp rsp = {.code = code};
  
  int32_t contLen = tSerializeSQueryTableRsp(NULL, 0, &rsp);
  void *msg = rpcMallocCont(contLen);
  tSerializeSQueryTableRsp(msg, contLen, &rsp);
D
dapan1121 已提交
55 56

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
57
    .msgType = TDMT_VND_QUERY_RSP,
D
dapan1121 已提交
58 59
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
60 61
    .pCont   = msg,
    .contLen = contLen,
D
dapan1121 已提交
62 63 64
    .code    = code,
  };

S
shm  
Shengliang Guan 已提交
65
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
66 67 68 69

  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
70
int32_t qwBuildAndSendReadyRsp(SQWConnInfo *pConn, int32_t code) {
D
dapan1121 已提交
71 72 73 74
  SResReadyRsp *pRsp = (SResReadyRsp *)rpcMallocCont(sizeof(SResReadyRsp));
  pRsp->code = code;

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
75
    .msgType = TDMT_VND_RES_READY_RSP,
D
dapan1121 已提交
76
    .handle  = pConn->handle,
D
dapan1121 已提交
77
    .ahandle = NULL,
D
dapan1121 已提交
78 79 80 81 82
    .pCont   = pRsp,
    .contLen = sizeof(*pRsp),
    .code    = code,
  };

S
shm  
Shengliang Guan 已提交
83
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
84 85 86 87

  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
88
int32_t qwBuildAndSendHbRsp(SQWConnInfo *pConn, SSchedulerHbRsp *pStatus, int32_t code) {
D
dapan1121 已提交
89 90 91
  int32_t contLen = tSerializeSSchedulerHbRsp(NULL, 0, pStatus);
  void *pRsp = rpcMallocCont(contLen);
  tSerializeSSchedulerHbRsp(pRsp, contLen, pStatus);
D
dapan1121 已提交
92 93

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
94
    .msgType = TDMT_VND_QUERY_HEARTBEAT_RSP,
D
dapan1121 已提交
95 96
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
97
    .pCont   = pRsp,
D
dapan1121 已提交
98 99
    .contLen = contLen,
    .code    = code,
D
dapan1121 已提交
100 101
  };

S
shm  
Shengliang Guan 已提交
102
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
103 104 105 106

  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
107
int32_t qwBuildAndSendFetchRsp(SQWConnInfo *pConn, SRetrieveTableRsp *pRsp, int32_t dataLength, int32_t code) {
D
dapan1121 已提交
108 109 110 111 112 113 114
  if (NULL == pRsp) {
    pRsp = (SRetrieveTableRsp *)rpcMallocCont(sizeof(SRetrieveTableRsp));
    memset(pRsp, 0, sizeof(SRetrieveTableRsp));
    dataLength = 0;
  }

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
115
    .msgType = TDMT_VND_FETCH_RSP,
D
dapan1121 已提交
116 117
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
118 119 120 121 122
    .pCont   = pRsp,
    .contLen = sizeof(*pRsp) + dataLength,
    .code    = code,
  };

S
shm  
Shengliang Guan 已提交
123
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
124 125 126 127

  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
128
int32_t qwBuildAndSendCancelRsp(SQWConnInfo *pConn, int32_t code) {
D
dapan1121 已提交
129 130 131 132
  STaskCancelRsp *pRsp = (STaskCancelRsp *)rpcMallocCont(sizeof(STaskCancelRsp));
  pRsp->code = code;

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
133
    .msgType = TDMT_VND_CANCEL_TASK_RSP,
D
dapan1121 已提交
134 135
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
136 137 138 139 140
    .pCont   = pRsp,
    .contLen = sizeof(*pRsp),
    .code    = code,
  };

S
shm  
Shengliang Guan 已提交
141
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
142 143 144
  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
145
int32_t qwBuildAndSendDropRsp(SQWConnInfo *pConn, int32_t code) {
D
dapan1121 已提交
146 147 148 149
  STaskDropRsp *pRsp = (STaskDropRsp *)rpcMallocCont(sizeof(STaskDropRsp));
  pRsp->code = code;

  SRpcMsg rpcRsp = {
D
dapan1121 已提交
150
    .msgType = TDMT_VND_DROP_TASK_RSP,
D
dapan1121 已提交
151 152
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
153 154 155 156 157
    .pCont   = pRsp,
    .contLen = sizeof(*pRsp),
    .code    = code,
  };

S
shm  
Shengliang Guan 已提交
158
  tmsgSendRsp(&rpcRsp);
D
dapan1121 已提交
159 160 161
  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
162
int32_t qwBuildAndSendShowRsp(SRpcMsg *pMsg, int32_t code) {
S
Shengliang Guan 已提交
163 164 165 166
  int32_t         numOfCols = 6;
  SVShowTablesRsp showRsp = {0};

  // showRsp.showId = 1;
wafwerar's avatar
wafwerar 已提交
167
  showRsp.tableMeta.pSchemas = taosMemoryCalloc(numOfCols, sizeof(SSchema));
S
Shengliang Guan 已提交
168 169 170 171
  if (showRsp.tableMeta.pSchemas == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }
D
dapan1121 已提交
172

173
  col_id_t  cols = 0;
S
Shengliang Guan 已提交
174
  SSchema *pSchema = showRsp.tableMeta.pSchemas;
D
dapan1121 已提交
175 176

  const SSchema *s = tGetTbnameColumnSchema();
S
Shengliang Guan 已提交
177
  *pSchema = createSchema(s->type, s->bytes, ++cols, "name");
D
dapan1121 已提交
178 179 180
  pSchema++;

  int32_t type = TSDB_DATA_TYPE_TIMESTAMP;
S
Shengliang Guan 已提交
181
  *pSchema = createSchema(type, tDataTypes[type].bytes, ++cols, "created");
D
dapan1121 已提交
182 183 184
  pSchema++;

  type = TSDB_DATA_TYPE_SMALLINT;
S
Shengliang Guan 已提交
185
  *pSchema = createSchema(type, tDataTypes[type].bytes, ++cols, "columns");
D
dapan1121 已提交
186 187
  pSchema++;

S
Shengliang Guan 已提交
188
  *pSchema = createSchema(s->type, s->bytes, ++cols, "stable");
D
dapan1121 已提交
189 190 191
  pSchema++;

  type = TSDB_DATA_TYPE_BIGINT;
S
Shengliang Guan 已提交
192
  *pSchema = createSchema(type, tDataTypes[type].bytes, ++cols, "uid");
D
dapan1121 已提交
193 194 195
  pSchema++;

  type = TSDB_DATA_TYPE_INT;
S
Shengliang Guan 已提交
196
  *pSchema = createSchema(type, tDataTypes[type].bytes, ++cols, "vgId");
D
dapan1121 已提交
197 198

  assert(cols == numOfCols);
S
Shengliang Guan 已提交
199 200 201 202 203
  showRsp.tableMeta.numOfColumns = cols;

  int32_t bufLen = tSerializeSShowRsp(NULL, 0, &showRsp);
  void   *pBuf = rpcMallocCont(bufLen);
  tSerializeSShowRsp(pBuf, bufLen, &showRsp);
D
dapan1121 已提交
204 205

  SRpcMsg rpcMsg = {
S
Shengliang Guan 已提交
206
      .handle = pMsg->handle,
D
dapan1121 已提交
207
      .ahandle = pMsg->ahandle,
S
Shengliang Guan 已提交
208 209 210
      .pCont = pBuf,
      .contLen = bufLen,
      .code = code,
D
dapan1121 已提交
211 212
  };

S
shm  
Shengliang Guan 已提交
213
  tmsgSendRsp(&rpcMsg);
D
dapan1121 已提交
214 215 216
  return TSDB_CODE_SUCCESS;
}

S
shm  
Shengliang Guan 已提交
217
int32_t qwBuildAndSendShowFetchRsp(SRpcMsg *pMsg, SVShowTablesFetchReq* pFetchReq) {
D
dapan1121 已提交
218 219 220 221 222 223 224 225 226 227 228 229
  SVShowTablesFetchRsp *pRsp = (SVShowTablesFetchRsp *)rpcMallocCont(sizeof(SVShowTablesFetchRsp));
  int32_t handle = htonl(pFetchReq->id);

  pRsp->numOfRows = 0;
  SRpcMsg rpcMsg = {
      .handle  = pMsg->handle,
      .ahandle = pMsg->ahandle,
      .pCont   = pRsp,
      .contLen = sizeof(*pRsp),
      .code    = 0,
  };

S
shm  
Shengliang Guan 已提交
230
  tmsgSendRsp(&rpcMsg);
D
dapan1121 已提交
231 232 233
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
234
int32_t qwBuildAndSendCQueryMsg(QW_FPARAMS_DEF, SQWConnInfo *pConn) {
D
dapan1121 已提交
235 236 237 238 239 240 241 242 243 244 245 246
  SQueryContinueReq * req = (SQueryContinueReq *)rpcMallocCont(sizeof(SQueryContinueReq));
  if (NULL == req) {
    QW_SCH_TASK_ELOG("rpcMallocCont %d failed", (int32_t)sizeof(SQueryContinueReq));
    QW_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

  req->header.vgId = mgmt->nodeId;
  req->sId = sId;
  req->queryId = qId;
  req->taskId = tId;

  SRpcMsg pNewMsg = {
D
dapan1121 已提交
247 248
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
D
dapan1121 已提交
249 250 251 252 253 254
    .msgType = TDMT_VND_QUERY_CONTINUE,
    .pCont   = req,
    .contLen = sizeof(SQueryContinueReq),
    .code    = 0,
  };

S
Shengliang Guan 已提交
255
  int32_t code = tmsgPutToQueue(&mgmt->msgCb, QUERY_QUEUE, &pNewMsg);
D
dapan1121 已提交
256
  if (TSDB_CODE_SUCCESS != code) {
H
Haojun Liao 已提交
257
    QW_SCH_TASK_ELOG("put query continue msg to queue failed, vgId:%d, code:%s", mgmt->nodeId, tstrerror(code));
D
dapan1121 已提交
258 259 260 261
    rpcFreeCont(req);
    QW_ERR_RET(code);
  }

D
dapan1121 已提交
262
  QW_SCH_TASK_DLOG("query continue msg put to queue, vgId:%d", mgmt->nodeId);
D
dapan1121 已提交
263 264 265 266

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
267 268 269 270 271 272 273 274

int32_t qwRegisterBrokenLinkArg(QW_FPARAMS_DEF, SQWConnInfo *pConn) {
  STaskDropReq * req = (STaskDropReq *)rpcMallocCont(sizeof(STaskDropReq));
  if (NULL == req) {
    QW_SCH_TASK_ELOG("rpcMallocCont %d failed", (int32_t)sizeof(STaskDropReq));
    QW_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }  

D
dapan1121 已提交
275 276 277 278 279
  req->header.vgId = htonl(mgmt->nodeId);
  req->sId = htobe64(sId);
  req->queryId = htobe64(qId);
  req->taskId = htobe64(tId);
  req->refId = htobe64(rId);
D
dapan1121 已提交
280 281 282 283 284 285 286 287 288 289
  
  SRpcMsg pMsg = {
    .handle  = pConn->handle,
    .ahandle = pConn->ahandle,
    .msgType = TDMT_VND_DROP_TASK,
    .pCont   = req,
    .contLen = sizeof(STaskDropReq),
    .code    = TSDB_CODE_RPC_NETWORK_UNAVAIL,
  };
  
S
Shengliang Guan 已提交
290
  tmsgRegisterBrokenLinkArg(&mgmt->msgCb, &pMsg);
D
dapan1121 已提交
291 292 293 294 295

  return TSDB_CODE_SUCCESS;
}


D
dapan1121 已提交
296 297 298 299 300 301 302 303 304 305
int32_t qWorkerProcessQueryMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  int32_t code = 0;
  SSubQueryMsg *msg = pMsg->pCont;
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
  
  if (NULL == msg || pMsg->contLen <= sizeof(*msg)) {
D
dapan1121 已提交
306
    QW_ELOG("invalid query msg, msg:%p, msgLen:%d", msg, pMsg->contLen);
D
dapan1121 已提交
307
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
D
dapan1121 已提交
308 309
  }

310
  msg->sId     = be64toh(msg->sId);
D
dapan1121 已提交
311
  msg->queryId = be64toh(msg->queryId);
312
  msg->taskId  = be64toh(msg->taskId);
D
dapan1121 已提交
313
  msg->refId   = be64toh(msg->refId);
314 315 316
  msg->phyLen  = ntohl(msg->phyLen);
  msg->sqlLen  = ntohl(msg->sqlLen);

D
dapan1121 已提交
317 318 319
  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
D
dapan1121 已提交
320
  int64_t  rId = msg->refId;
D
dapan1121 已提交
321

D
dapan1121 已提交
322 323 324
  SQWMsg qwMsg = {.node = node, .msg = msg->msg + msg->sqlLen, .msgLen = msg->phyLen};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
325

326
  char* sql = strndup(msg->msg, msg->sqlLen);
D
dapan1121 已提交
327
  QW_SCH_TASK_DLOG("processQuery start, node:%p, handle:%p, sql:%s", node, pMsg->handle, sql);
wafwerar's avatar
wafwerar 已提交
328
  taosMemoryFreeClear(sql);
329

D
dapan1121 已提交
330
  QW_ERR_RET(qwProcessQuery(QW_FPARAMS(), &qwMsg, msg->taskType));
331

D
dapan1121 已提交
332
  QW_SCH_TASK_DLOG("processQuery end, node:%p", node);
333 334

  return TSDB_CODE_SUCCESS;  
D
dapan1121 已提交
335 336
}

337
int32_t qWorkerProcessCQueryMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
D
dapan1121 已提交
338 339 340
  int32_t code = 0;
  int8_t status = 0;
  bool queryDone = false;
341
  SQueryContinueReq *msg = (SQueryContinueReq *)pMsg->pCont;
D
dapan1121 已提交
342 343
  bool needStop = false;
  SQWTaskCtx *handles = NULL;
D
dapan1121 已提交
344
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
D
dapan1121 已提交
345

D
dapan1121 已提交
346
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
D
dapan1121 已提交
347
    QW_ELOG("invalid cquery msg, msg:%p, msgLen:%d", msg, pMsg->contLen);
D
dapan1121 已提交
348
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
D
dapan1121 已提交
349 350
  }

351 352 353
  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
D
dapan1121 已提交
354
  int64_t rId = 0;
D
dapan1121 已提交
355

D
dapan1121 已提交
356 357 358
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
359

D
dapan1121 已提交
360
  QW_SCH_TASK_DLOG("processCQuery start, node:%p, handle:%p", node, pMsg->handle);
361 362 363

  QW_ERR_RET(qwProcessCQuery(QW_FPARAMS(), &qwMsg));

D
dapan1121 已提交
364
  QW_SCH_TASK_DLOG("processCQuery end, node:%p", node);
365 366

  return TSDB_CODE_SUCCESS;    
D
dapan1121 已提交
367 368 369 370 371 372 373
}

int32_t qWorkerProcessReadyMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg){
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

D
dapan1121 已提交
374
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
D
dapan1121 已提交
375 376
  SResReadyReq *msg = pMsg->pCont;
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
D
dapan1121 已提交
377
    QW_ELOG("invalid task ready msg, msg:%p, msgLen:%d", msg, pMsg->contLen);
D
dapan1121 已提交
378 379 380
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

D
dapan1121 已提交
381 382 383 384 385 386 387
  msg->sId = be64toh(msg->sId);
  msg->queryId = be64toh(msg->queryId);
  msg->taskId = be64toh(msg->taskId);

  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
D
dapan1121 已提交
388
  int64_t rId = 0;
D
dapan1121 已提交
389

D
dapan1121 已提交
390 391 392
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
393

D
dapan1121 已提交
394
  QW_SCH_TASK_DLOG("processReady start, node:%p, handle:%p", node, pMsg->handle);
395

D
dapan1121 已提交
396
  QW_ERR_RET(qwProcessReady(QW_FPARAMS(), &qwMsg));
397

D
dapan1121 已提交
398
  QW_SCH_TASK_DLOG("processReady end, node:%p", node);
D
dapan1121 已提交
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
  
  return TSDB_CODE_SUCCESS;
}

int32_t qWorkerProcessStatusMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  int32_t code = 0;
  SSchTasksStatusReq *msg = pMsg->pCont;
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
    qError("invalid task status msg");
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

D
dapan1121 已提交
415
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
D
dapan1121 已提交
416
  msg->sId = htobe64(msg->sId);
D
dapan1121 已提交
417
  uint64_t sId = msg->sId;
D
dapan1121 已提交
418 419 420

  SSchedulerStatusRsp *sStatus = NULL;
  
D
dapan1121 已提交
421
  //QW_ERR_JRET(qwGetSchTasksStatus(qWorkerMgmt, msg->sId, &sStatus));
D
dapan1121 已提交
422 423 424

_return:

D
dapan1121 已提交
425
  //QW_ERR_RET(qwBuildAndSendStatusRsp(pMsg, sStatus));
D
dapan1121 已提交
426 427 428 429 430 431 432 433 434 435 436 437 438

  return TSDB_CODE_SUCCESS;
}

int32_t qWorkerProcessFetchMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  SResFetchReq *msg = pMsg->pCont;
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
  
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
D
dapan1121 已提交
439
    QW_ELOG("invalid fetch msg, msg:%p, msgLen:%d", msg, pMsg->contLen);  
D
dapan1121 已提交
440 441 442
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

D
dapan1121 已提交
443 444 445
  msg->sId = be64toh(msg->sId);
  msg->queryId = be64toh(msg->queryId);
  msg->taskId = be64toh(msg->taskId);
D
dapan1121 已提交
446 447 448 449

  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
D
dapan1121 已提交
450
  int64_t rId = 0;
D
dapan1121 已提交
451

D
dapan1121 已提交
452 453 454
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
455

D
dapan1121 已提交
456
  QW_SCH_TASK_DLOG("processFetch start, node:%p, handle:%p", node, pMsg->handle);
457 458 459

  QW_ERR_RET(qwProcessFetch(QW_FPARAMS(), &qwMsg));

D
dapan1121 已提交
460
  QW_SCH_TASK_DLOG("processFetch end, node:%p", node);
461 462

  return TSDB_CODE_SUCCESS;  
D
dapan1121 已提交
463 464
}

S
Shengliang 已提交
465
int32_t qWorkerProcessFetchRsp(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
S
Shengliang Guan 已提交
466
  qProcessFetchRsp(NULL, pMsg, NULL);
S
Shengliang 已提交
467 468 469
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
470 471 472 473 474
int32_t qWorkerProcessCancelMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

D
dapan1121 已提交
475
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
D
dapan1121 已提交
476 477 478 479 480 481 482
  int32_t code = 0;
  STaskCancelReq *msg = pMsg->pCont;
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
    qError("invalid task cancel msg");  
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

D
dapan1121 已提交
483 484 485 486
  msg->sId = be64toh(msg->sId);
  msg->queryId = be64toh(msg->queryId);
  msg->taskId = be64toh(msg->taskId);
  msg->refId = be64toh(msg->refId);
D
dapan1121 已提交
487

D
dapan1121 已提交
488 489 490 491 492
  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
  int64_t  rId = msg->refId;

D
dapan1121 已提交
493 494 495 496
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;

D
dapan1121 已提交
497
  //QW_ERR_JRET(qwCancelTask(qWorkerMgmt, msg->sId, msg->queryId, msg->taskId));
D
dapan1121 已提交
498 499 500

_return:

S
shm  
Shengliang Guan 已提交
501
  QW_ERR_RET(qwBuildAndSendCancelRsp(&qwMsg.connInfo, code));
D
dapan1121 已提交
502
  QW_SCH_TASK_DLOG("cancel rsp send, handle:%p, code:%x - %s", qwMsg.connInfo.handle, code, tstrerror(code));
D
dapan1121 已提交
503 504 505 506 507 508 509 510 511 512 513 514 515 516

  return TSDB_CODE_SUCCESS;
}

int32_t qWorkerProcessDropMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  int32_t code = 0;
  STaskDropReq *msg = pMsg->pCont;
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
  
  if (NULL == msg || pMsg->contLen < sizeof(*msg)) {
D
dapan1121 已提交
517
    QW_ELOG("invalid task drop msg, msg:%p, msgLen:%d", msg, pMsg->contLen);
D
dapan1121 已提交
518 519 520
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

D
dapan1121 已提交
521 522 523
  msg->sId = be64toh(msg->sId);
  msg->queryId = be64toh(msg->queryId);
  msg->taskId = be64toh(msg->taskId);
D
dapan1121 已提交
524
  msg->refId = be64toh(msg->refId);
D
dapan1121 已提交
525 526 527 528

  uint64_t sId = msg->sId;
  uint64_t qId = msg->queryId;
  uint64_t tId = msg->taskId;
D
dapan1121 已提交
529
  int64_t  rId = msg->refId;
D
dapan1121 已提交
530

D
dapan1121 已提交
531 532 533
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
534

D
dapan1121 已提交
535 536 537
  if (TSDB_CODE_RPC_NETWORK_UNAVAIL == pMsg->code) {
    QW_SCH_TASK_DLOG("receive drop task due to network broken, error:%s", tstrerror(pMsg->code));    
  }
D
dapan1121 已提交
538

D
dapan1121 已提交
539
  QW_SCH_TASK_DLOG("processDrop start, node:%p, handle:%p", node, pMsg->handle);
540 541 542

  QW_ERR_RET(qwProcessDrop(QW_FPARAMS(), &qwMsg));

D
dapan1121 已提交
543
  QW_SCH_TASK_DLOG("processDrop end, node:%p", node);
544 545

  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
546 547
}

D
dapan1121 已提交
548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568
int32_t qWorkerProcessHbMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  int32_t code = 0;
  SSchedulerHbReq req = {0};
  SQWorkerMgmt *mgmt = (SQWorkerMgmt *)qWorkerMgmt;
  
  if (NULL == pMsg->pCont) {
    QW_ELOG("invalid hb msg, msg:%p, msgLen:%d", pMsg->pCont, pMsg->contLen);
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }  

  if (tDeserializeSSchedulerHbReq(pMsg->pCont, pMsg->contLen, &req)) {
    QW_ELOG("invalid hb msg, msg:%p, msgLen:%d", pMsg->pCont, pMsg->contLen);
    tFreeSSchedulerHbReq(&req);
    QW_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  uint64_t sId = req.sId;
D
dapan1121 已提交
569 570 571
  SQWMsg qwMsg = {.node = node, .msg = NULL, .msgLen = 0};
  qwMsg.connInfo.handle = pMsg->handle;
  qwMsg.connInfo.ahandle = pMsg->ahandle;
D
dapan1121 已提交
572

D
dapan1121 已提交
573
  QW_SCH_DLOG("processHb start, node:%p, handle:%p", node, pMsg->handle);
D
dapan1121 已提交
574 575 576 577 578 579 580 581

  QW_ERR_RET(qwProcessHb(mgmt, &qwMsg, &req));

  QW_SCH_DLOG("processHb end, node:%p", node);

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
582 583 584 585 586 587 588
int32_t qWorkerProcessShowMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  int32_t code = 0;
  SVShowTablesReq *pReq = pMsg->pCont;
S
shm  
Shengliang Guan 已提交
589
  QW_RET(qwBuildAndSendShowRsp(pMsg, code));
D
dapan1121 已提交
590 591 592 593 594 595 596 597
}

int32_t qWorkerProcessShowFetchMsg(void *node, void *qWorkerMgmt, SRpcMsg *pMsg) {
  if (NULL == node || NULL == qWorkerMgmt || NULL == pMsg) {
    return TSDB_CODE_QRY_INVALID_INPUT;
  }

  SVShowTablesFetchReq *pFetchReq = pMsg->pCont;
S
shm  
Shengliang Guan 已提交
598
  QW_RET(qwBuildAndSendShowFetchRsp(pMsg, pFetchReq));
D
dapan1121 已提交
599
}