querymsg.c 17.9 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
Hongze Cheng 已提交
16
#include "tmsg.h"
D
dapan1121 已提交
17
#include "queryInt.h"
D
dapan1121 已提交
18
#include "query.h"
D
catalog  
dapan1121 已提交
19
#include "trpc.h"
20
#include "systable.h"
D
dapan 已提交
21

S
Shengliang Guan 已提交
22 23 24
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wformat-truncation"

D
dapan1121 已提交
25
int32_t (*queryBuildMsg[TDMT_MAX])(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallocFp)(int32_t)) = {0};
S
Shengliang Guan 已提交
26
int32_t (*queryProcessMsgRsp[TDMT_MAX])(void *output, char *msg, int32_t msgSize) = {0};
D
dapan 已提交
27

X
Xiaoyu Wang 已提交
28 29 30
int32_t queryBuildUseDbOutput(SUseDbOutput *pOut, SUseDbRsp *usedbRsp) {
  memcpy(pOut->db, usedbRsp->db, TSDB_DB_FNAME_LEN);
  pOut->dbId = usedbRsp->uid;
D
dapan1121 已提交
31

wafwerar's avatar
wafwerar 已提交
32
  pOut->dbVgroup = taosMemoryCalloc(1, sizeof(SDBVgInfo));
X
Xiaoyu Wang 已提交
33 34 35 36 37 38
  if (NULL == pOut->dbVgroup) {
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }

  pOut->dbVgroup->vgVersion = usedbRsp->vgVersion;
  pOut->dbVgroup->hashMethod = usedbRsp->hashMethod;
D
dapan1121 已提交
39

D
dapan1121 已提交
40 41
  qDebug("Got %d vgroup for db %s", usedbRsp->vgNum, usedbRsp->db);

D
dapan1121 已提交
42 43 44 45
  if (usedbRsp->vgNum <= 0) {
    return TSDB_CODE_SUCCESS;
  }
  
X
Xiaoyu Wang 已提交
46 47 48 49 50 51 52 53
  pOut->dbVgroup->vgHash =
      taosHashInit(usedbRsp->vgNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
  if (NULL == pOut->dbVgroup->vgHash) {
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }

  for (int32_t i = 0; i < usedbRsp->vgNum; ++i) {
    SVgroupInfo *pVgInfo = taosArrayGet(usedbRsp->pVgroupInfos, i);
D
dapan 已提交
54
    pOut->dbVgroup->numOfTable += pVgInfo->numOfTable;
D
dapan1121 已提交
55 56
    qDebug("the %dth vgroup, id %d, epNum %d, current %s port %d", i, pVgInfo->vgId, pVgInfo->epSet.numOfEps,
      pVgInfo->epSet.eps[pVgInfo->epSet.inUse].fqdn, pVgInfo->epSet.eps[pVgInfo->epSet.inUse].port);
X
Xiaoyu Wang 已提交
57 58 59 60 61 62 63 64
    if (0 != taosHashPut(pOut->dbVgroup->vgHash, &pVgInfo->vgId, sizeof(int32_t), pVgInfo, sizeof(SVgroupInfo))) {
      return TSDB_CODE_TSC_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
65
int32_t queryBuildTableMetaReqMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan1121 已提交
66
  SBuildTableInput *pInput = input;
D
dapan1121 已提交
67 68 69 70
  if (NULL == input || NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

S
Shengliang Guan 已提交
71 72
  STableInfoReq infoReq = {0};
  infoReq.header.vgId = pInput->vgId;
S
Shengliang Guan 已提交
73
  if (pInput->dbFName) {
S
Shengliang Guan 已提交
74
    tstrncpy(infoReq.dbFName, pInput->dbFName, TSDB_DB_FNAME_LEN);
D
dapan1121 已提交
75
  }
S
Shengliang Guan 已提交
76
  tstrncpy(infoReq.tbName, pInput->tbName, TSDB_TABLE_NAME_LEN);
D
dapan1121 已提交
77

S
Shengliang Guan 已提交
78
  int32_t bufLen = tSerializeSTableInfoReq(NULL, 0, &infoReq);
D
dapan1121 已提交
79
  void   *pBuf = (*mallcFp)(bufLen);
S
Shengliang Guan 已提交
80 81 82 83
  tSerializeSTableInfoReq(pBuf, bufLen, &infoReq);

  *msg = pBuf;
  *msgLen = bufLen;
D
dapan1121 已提交
84 85 86 87

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
88
int32_t queryBuildUseDbMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
S
Shengliang Guan 已提交
89 90
  SBuildUseDBInput *pInput = input;
  if (NULL == pInput || NULL == msg || NULL == msgLen) {
D
dapan1121 已提交
91 92 93
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

S
Shengliang Guan 已提交
94
  SUseDbReq usedbReq = {0};
S
Shengliang Guan 已提交
95
  strncpy(usedbReq.db, pInput->db, sizeof(usedbReq.db));
S
Shengliang Guan 已提交
96
  usedbReq.db[sizeof(usedbReq.db) - 1] = 0;
S
Shengliang Guan 已提交
97
  usedbReq.vgVersion = pInput->vgVersion;
X
Xiaoyu Wang 已提交
98
  usedbReq.dbId = pInput->dbId;
D
dapan 已提交
99
  usedbReq.numOfTable = pInput->numOfTable;
D
dapan1121 已提交
100

S
Shengliang Guan 已提交
101
  int32_t bufLen = tSerializeSUseDbReq(NULL, 0, &usedbReq);
D
dapan1121 已提交
102
  void   *pBuf = (*mallcFp)(bufLen);
S
Shengliang Guan 已提交
103
  tSerializeSUseDbReq(pBuf, bufLen, &usedbReq);
D
dapan1121 已提交
104

S
Shengliang Guan 已提交
105 106
  *msg = pBuf;
  *msgLen = bufLen;
D
dapan1121 已提交
107

S
Shengliang Guan 已提交
108
  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
109 110
}

D
dapan1121 已提交
111
int32_t queryBuildQnodeListMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan1121 已提交
112 113 114 115 116 117 118 119
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SQnodeListReq qnodeListReq = {0};
  qnodeListReq.rowNum = -1;

  int32_t bufLen = tSerializeSQnodeListReq(NULL, 0, &qnodeListReq);
D
dapan1121 已提交
120
  void   *pBuf = (*mallcFp)(bufLen);
D
dapan1121 已提交
121 122 123 124 125 126 127 128
  tSerializeSQnodeListReq(pBuf, bufLen, &qnodeListReq);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
129
int32_t queryBuildGetDBCfgMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan1121 已提交
130 131 132 133 134 135 136 137
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SDbCfgReq dbCfgReq = {0};
  strcpy(dbCfgReq.db, input);

  int32_t bufLen = tSerializeSDbCfgReq(NULL, 0, &dbCfgReq);
D
dapan1121 已提交
138
  void   *pBuf = (*mallcFp)(bufLen);
D
dapan1121 已提交
139 140 141 142 143 144 145
  tSerializeSDbCfgReq(pBuf, bufLen, &dbCfgReq);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
146

D
dapan1121 已提交
147
int32_t queryBuildGetIndexMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan1121 已提交
148 149 150 151 152 153 154 155
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SUserIndexReq indexReq = {0};
  strcpy(indexReq.indexFName, input);

  int32_t bufLen = tSerializeSUserIndexReq(NULL, 0, &indexReq);
D
dapan1121 已提交
156
  void   *pBuf = (*mallcFp)(bufLen);
D
dapan1121 已提交
157 158 159 160 161 162 163 164
  tSerializeSUserIndexReq(pBuf, bufLen, &indexReq);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
165
int32_t queryBuildRetrieveFuncMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan1121 已提交
166 167 168 169 170 171
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SRetrieveFuncReq funcReq = {0};
  funcReq.numOfFuncs = 1;
D
dapan1121 已提交
172
  funcReq.ignoreCodeComment = true;
D
dapan1121 已提交
173 174 175 176
  funcReq.pFuncNames = taosArrayInit(1, strlen(input) + 1);
  taosArrayPush(funcReq.pFuncNames, input);

  int32_t bufLen = tSerializeSRetrieveFuncReq(NULL, 0, &funcReq);
D
dapan1121 已提交
177
  void   *pBuf = (*mallcFp)(bufLen);
D
dapan1121 已提交
178 179 180 181 182 183 184 185 186
  tSerializeSRetrieveFuncReq(pBuf, bufLen, &funcReq);

  taosArrayDestroy(funcReq.pFuncNames);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
187

D
dapan1121 已提交
188
int32_t queryBuildGetUserAuthMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
D
dapan 已提交
189 190 191 192 193 194 195 196
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SGetUserAuthReq req = {0};
  strncpy(req.user, input, sizeof(req.user));

  int32_t bufLen = tSerializeSGetUserAuthReq(NULL, 0, &req);
D
dapan1121 已提交
197
  void   *pBuf = (*mallcFp)(bufLen);
D
dapan 已提交
198 199 200 201 202 203 204 205
  tSerializeSGetUserAuthReq(pBuf, bufLen, &req);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
int32_t queryBuildGetTbIndexMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  STableIndexReq indexReq = {0};
  strcpy(indexReq.tbFName, input);

  int32_t bufLen = tSerializeSTableIndexReq(NULL, 0, &indexReq);
  void   *pBuf = (*mallcFp)(bufLen);
  tSerializeSTableIndexReq(pBuf, bufLen, &indexReq);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
int32_t queryBuildGetTbCfgMsg(void *input, char **msg, int32_t msgSize, int32_t *msgLen, void*(*mallcFp)(int32_t)) {
  if (NULL == msg || NULL == msgLen) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  SBuildTableInput *pInput = input;
  STableCfgReq cfgReq = {0};
  cfgReq.header.vgId = pInput->vgId;
  strcpy(cfgReq.dbFName, pInput->dbFName);
  strcpy(cfgReq.tbName, pInput->tbName);

  int32_t bufLen = tSerializeSTableCfgReq(NULL, 0, &cfgReq);
  void   *pBuf = (*mallcFp)(bufLen);
  tSerializeSTableCfgReq(pBuf, bufLen, &cfgReq);

  *msg = pBuf;
  *msgLen = bufLen;

  return TSDB_CODE_SUCCESS;
}

D
dapan 已提交
245

S
Shengliang Guan 已提交
246
int32_t queryProcessUseDBRsp(void *output, char *msg, int32_t msgSize) {
S
Shengliang Guan 已提交
247 248 249 250
  SUseDbOutput *pOut = output;
  SUseDbRsp     usedbRsp = {0};
  int32_t       code = -1;

D
dapan1121 已提交
251
  if (NULL == output || NULL == msg || msgSize <= 0) {
S
Shengliang Guan 已提交
252 253
    code = TSDB_CODE_TSC_INVALID_INPUT;
    goto PROCESS_USEDB_OVER;
D
dapan1121 已提交
254 255
  }

S
Shengliang Guan 已提交
256 257
  if (tDeserializeSUseDbRsp(msg, msgSize, &usedbRsp) != 0) {
    qError("invalid use db rsp msg, msgSize:%d", msgSize);
S
Shengliang Guan 已提交
258 259
    code = TSDB_CODE_INVALID_MSG;
    goto PROCESS_USEDB_OVER;
D
dapan1121 已提交
260 261
  }

S
Shengliang Guan 已提交
262 263
  if (usedbRsp.vgNum < 0) {
    qError("invalid db[%s] vgroup number[%d]", usedbRsp.db, usedbRsp.vgNum);
S
Shengliang Guan 已提交
264 265
    code = TSDB_CODE_TSC_INVALID_VALUE;
    goto PROCESS_USEDB_OVER;
D
dapan1121 已提交
266 267
  }

X
Xiaoyu Wang 已提交
268
  code = queryBuildUseDbOutput(pOut, &usedbRsp);
269

S
Shengliang Guan 已提交
270
PROCESS_USEDB_OVER:
X
Xiaoyu Wang 已提交
271

S
Shengliang Guan 已提交
272 273 274
  if (code != 0) {
    if (pOut) {
      if (pOut->dbVgroup) taosHashCleanup(pOut->dbVgroup->vgHash);
wafwerar's avatar
wafwerar 已提交
275
      taosMemoryFreeClear(pOut->dbVgroup);
S
Shengliang Guan 已提交
276 277
    }
    qError("failed to process usedb rsp since %s", terrstr());
D
dapan1121 已提交
278
  }
S
Shengliang Guan 已提交
279

S
Shengliang Guan 已提交
280
  tFreeSUsedbRsp(&usedbRsp);
D
dapan1121 已提交
281 282 283
  return code;
}

S
Shengliang Guan 已提交
284
static int32_t queryConvertTableMetaMsg(STableMetaRsp *pMetaMsg) {
D
dapan1121 已提交
285 286 287 288 289 290 291 292 293 294
  if (pMetaMsg->numOfTags < 0 || pMetaMsg->numOfTags > TSDB_MAX_TAGS) {
    qError("invalid numOfTags[%d] in table meta rsp msg", pMetaMsg->numOfTags);
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

  if (pMetaMsg->numOfColumns > TSDB_MAX_COLUMNS || pMetaMsg->numOfColumns <= 0) {
    qError("invalid numOfColumns[%d] in table meta rsp msg", pMetaMsg->numOfColumns);
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

S
Shengliang Guan 已提交
295
  if (pMetaMsg->tableType != TSDB_SUPER_TABLE && pMetaMsg->tableType != TSDB_CHILD_TABLE &&
X
Xiaoyu Wang 已提交
296
      pMetaMsg->tableType != TSDB_NORMAL_TABLE && pMetaMsg->tableType != TSDB_SYSTEM_TABLE) {
D
dapan1121 已提交
297 298 299 300 301 302 303 304 305 306 307 308 309 310
    qError("invalid tableType[%d] in table meta rsp msg", pMetaMsg->tableType);
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

  if (pMetaMsg->sversion < 0) {
    qError("invalid sversion[%d] in table meta rsp msg", pMetaMsg->sversion);
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

  if (pMetaMsg->tversion < 0) {
    qError("invalid tversion[%d] in table meta rsp msg", pMetaMsg->tversion);
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

S
Shengliang Guan 已提交
311
  if (pMetaMsg->pSchemas[0].colId != PRIMARYKEY_TIMESTAMP_COL_ID) {
312
    qError("invalid colId[%" PRIi16 "] for the first column in table meta rsp msg", pMetaMsg->pSchemas[0].colId);
D
dapan1121 已提交
313 314 315 316 317 318
    return TSDB_CODE_TSC_INVALID_VALUE;
  }

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
319
int32_t queryCreateTableMetaFromMsg(STableMetaRsp *msg, bool isStb, STableMeta **pMeta) {
D
dapan1121 已提交
320 321
  int32_t total = msg->numOfColumns + msg->numOfTags;
  int32_t metaSize = sizeof(STableMeta) + sizeof(SSchema) * total;
S
Shengliang Guan 已提交
322

wafwerar's avatar
wafwerar 已提交
323
  STableMeta *pTableMeta = taosMemoryCalloc(1, metaSize);
D
dapan1121 已提交
324 325 326 327
  if (NULL == pTableMeta) {
    qError("calloc size[%d] failed", metaSize);
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }
D
dapan1121 已提交
328

D
dapan1121 已提交
329 330 331
  pTableMeta->vgId = isStb ? 0 : msg->vgId;
  pTableMeta->tableType = isStb ? TSDB_SUPER_TABLE : msg->tableType;
  pTableMeta->uid = isStb ? msg->suid : msg->tuid;
D
dapan1121 已提交
332 333 334 335 336 337 338 339
  pTableMeta->suid = msg->suid;
  pTableMeta->sversion = msg->sversion;
  pTableMeta->tversion = msg->tversion;

  pTableMeta->tableInfo.numOfTags = msg->numOfTags;
  pTableMeta->tableInfo.precision = msg->precision;
  pTableMeta->tableInfo.numOfColumns = msg->numOfColumns;

S
Shengliang Guan 已提交
340
  memcpy(pTableMeta->schema, msg->pSchemas, sizeof(SSchema) * total);
D
dapan1121 已提交
341

S
Shengliang Guan 已提交
342
  for (int32_t i = 0; i < msg->numOfColumns; ++i) {
D
dapan1121 已提交
343 344 345
    pTableMeta->tableInfo.rowSize += pTableMeta->schema[i].bytes;
  }

D
dapan1121 已提交
346 347 348 349 350 351
  qDebug("table %s uid %" PRIx64 " meta returned, type %d vgId %d db %s stb %s suid %" PRIx64 " sver %d tver %d" PRIx64 
    " tagNum %d colNum %d precision %d rowSize %d", 
    msg->tbName, pTableMeta->uid, pTableMeta->tableType, pTableMeta->vgId, msg->dbFName, msg->stbName, pTableMeta->suid,
    pTableMeta->sversion, pTableMeta->tversion, pTableMeta->tableInfo.numOfTags, pTableMeta->tableInfo.numOfColumns,
    pTableMeta->tableInfo.precision, pTableMeta->tableInfo.rowSize);

D
dapan1121 已提交
352 353 354 355
  *pMeta = pTableMeta;
  return TSDB_CODE_SUCCESS;
}

S
Shengliang Guan 已提交
356
int32_t queryProcessTableMetaRsp(void *output, char *msg, int32_t msgSize) {
D
dapan1121 已提交
357
  int32_t       code = 0;
S
Shengliang Guan 已提交
358 359 360 361 362 363
  STableMetaRsp metaRsp = {0};

  if (NULL == output || NULL == msg || msgSize <= 0) {
    code = TSDB_CODE_TSC_INVALID_INPUT;
    goto PROCESS_META_OVER;
  }
D
dapan1121 已提交
364

S
Shengliang Guan 已提交
365 366 367 368 369 370
  if (tDeserializeSTableMetaRsp(msg, msgSize, &metaRsp) != 0) {
    code = TSDB_CODE_INVALID_MSG;
    goto PROCESS_META_OVER;
  }

  code = queryConvertTableMetaMsg(&metaRsp);
D
dapan1121 已提交
371
  if (code != TSDB_CODE_SUCCESS) {
S
Shengliang Guan 已提交
372
    goto PROCESS_META_OVER;
D
dapan1121 已提交
373 374
  }

D
dapan1121 已提交
375
  if (0 != strcmp(metaRsp.dbFName, TSDB_INFORMATION_SCHEMA_DB) && !tIsValidSchema(metaRsp.pSchemas, metaRsp.numOfColumns, metaRsp.numOfTags)) {
S
Shengliang Guan 已提交
376 377
    code = TSDB_CODE_TSC_INVALID_VALUE;
    goto PROCESS_META_OVER;
D
dapan1121 已提交
378 379
  }

S
Shengliang Guan 已提交
380 381 382
  STableMetaOutput *pOut = output;
  strcpy(pOut->dbFName, metaRsp.dbFName);
  pOut->dbId = metaRsp.dbId;
D
dapan1121 已提交
383

S
Shengliang Guan 已提交
384
  if (metaRsp.tableType == TSDB_CHILD_TABLE) {
D
dapan1121 已提交
385
    SET_META_TYPE_BOTH_TABLE(pOut->metaType);
D
dapan1121 已提交
386

S
Shengliang Guan 已提交
387 388
    strcpy(pOut->ctbName, metaRsp.tbName);
    strcpy(pOut->tbName, metaRsp.stbName);
D
dapan1121 已提交
389

S
Shengliang Guan 已提交
390 391 392 393 394 395
    pOut->ctbMeta.vgId = metaRsp.vgId;
    pOut->ctbMeta.tableType = metaRsp.tableType;
    pOut->ctbMeta.uid = metaRsp.tuid;
    pOut->ctbMeta.suid = metaRsp.suid;

    code = queryCreateTableMetaFromMsg(&metaRsp, true, &pOut->tbMeta);
D
dapan1121 已提交
396
  } else {
D
dapan1121 已提交
397
    SET_META_TYPE_TABLE(pOut->metaType);
S
Shengliang Guan 已提交
398 399 400 401 402 403
    strcpy(pOut->tbName, metaRsp.tbName);
    code = queryCreateTableMetaFromMsg(&metaRsp, (metaRsp.tableType == TSDB_SUPER_TABLE), &pOut->tbMeta);
  }

PROCESS_META_OVER:
  if (code != 0) {
D
dapan1121 已提交
404
    qError("failed to process table meta rsp since %s", tstrerror(code));
D
dapan1121 已提交
405
  }
S
Shengliang Guan 已提交
406 407

  tFreeSTableMetaRsp(&metaRsp);
D
dapan1121 已提交
408 409 410
  return code;
}

D
dapan1121 已提交
411 412 413

int32_t queryProcessQnodeListRsp(void *output, char *msg, int32_t msgSize) {
  SQnodeListRsp out = {0};
D
dapan 已提交
414
  int32_t       code = 0;
D
dapan1121 已提交
415 416 417

  if (NULL == output || NULL == msg || msgSize <= 0) {
    code = TSDB_CODE_TSC_INVALID_INPUT;
D
dapan 已提交
418
    return code;
D
dapan1121 已提交
419 420
  }

D
dapan1121 已提交
421
  out.qnodeList = (SArray *)output;
D
dapan1121 已提交
422 423 424
  if (tDeserializeSQnodeListRsp(msg, msgSize, &out) != 0) {
    qError("invalid qnode list rsp msg, msgSize:%d", msgSize);
    code = TSDB_CODE_INVALID_MSG;
D
dapan 已提交
425
    return code;
D
dapan1121 已提交
426 427 428 429 430
  }

  return code;
}

D
dapan1121 已提交
431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
int32_t queryProcessGetDbCfgRsp(void *output, char *msg, int32_t msgSize) {
  SDbCfgRsp out = {0};

  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  if (tDeserializeSDbCfgRsp(msg, msgSize, &out) != 0) {
    qError("tDeserializeSDbCfgRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }

  memcpy(output, &out, sizeof(out));

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463
int32_t queryProcessGetIndexRsp(void *output, char *msg, int32_t msgSize) {
  SUserIndexRsp out = {0};

  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  if (tDeserializeSUserIndexRsp(msg, msgSize, &out) != 0) {
    qError("tDeserializeSUserIndexRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }

  memcpy(output, &out, sizeof(out));

  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
464

D
dapan1121 已提交
465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489
int32_t queryProcessRetrieveFuncRsp(void *output, char *msg, int32_t msgSize) {
  SRetrieveFuncRsp out = {0};

  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  if (tDeserializeSRetrieveFuncRsp(msg, msgSize, &out) != 0) {
    qError("tDeserializeSRetrieveFuncRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }

  if (1 != out.numOfFuncs) {
    qError("invalid func num returned, numOfFuncs:%d", out.numOfFuncs);
    return TSDB_CODE_INVALID_MSG;
  }

  SFuncInfo * funcInfo = taosArrayGet(out.pFuncInfos, 0);

  memcpy(output, funcInfo, sizeof(*funcInfo));
  taosArrayDestroy(out.pFuncInfos);

  return TSDB_CODE_SUCCESS;
}

D
dapan 已提交
490 491 492 493 494 495 496 497 498 499 500 501 502
int32_t queryProcessGetUserAuthRsp(void *output, char *msg, int32_t msgSize) {
  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  if (tDeserializeSGetUserAuthRsp(msg, msgSize, (SGetUserAuthRsp *)output) != 0) {
    qError("tDeserializeSGetUserAuthRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
503 504 505 506
int32_t queryProcessGetTbIndexRsp(void *output, char *msg, int32_t msgSize) {
  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }
D
dapan 已提交
507

D
dapan1121 已提交
508
  STableIndexRsp *out = (STableIndexRsp*)output;
D
dapan1121 已提交
509
  if (tDeserializeSTableIndexRsp(msg, msgSize, out) != 0) {
D
dapan1121 已提交
510 511 512
    qError("tDeserializeSTableIndexRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }
D
dapan 已提交
513
  
D
dapan1121 已提交
514 515 516
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
int32_t queryProcessGetTbCfgRsp(void *output, char *msg, int32_t msgSize) {
  if (NULL == output || NULL == msg || msgSize <= 0) {
    return TSDB_CODE_TSC_INVALID_INPUT;
  }

  STableCfgRsp *out = taosMemoryCalloc(1, sizeof(STableCfgRsp));
  if (tDeserializeSTableCfgRsp(msg, msgSize, out) != 0) {
    qError("tDeserializeSTableCfgRsp failed, msgSize:%d", msgSize);
    return TSDB_CODE_INVALID_MSG;
  }

  *(STableCfgRsp**)output = out;
  
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
533 534 535 536 537 538 539 540 541 542
void initQueryModuleMsgHandle() {
  queryBuildMsg[TMSG_INDEX(TDMT_VND_TABLE_META)]       = queryBuildTableMetaReqMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_TABLE_META)]       = queryBuildTableMetaReqMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_USE_DB)]           = queryBuildUseDbMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_QNODE_LIST)]       = queryBuildQnodeListMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_GET_DB_CFG)]       = queryBuildGetDBCfgMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_GET_INDEX)]        = queryBuildGetIndexMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_RETRIEVE_FUNC)]    = queryBuildRetrieveFuncMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_GET_USER_AUTH)]    = queryBuildGetUserAuthMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_GET_TABLE_INDEX)]  = queryBuildGetTbIndexMsg;
D
dapan1121 已提交
543 544
  queryBuildMsg[TMSG_INDEX(TDMT_VND_TABLE_CFG)]        = queryBuildGetTbCfgMsg;
  queryBuildMsg[TMSG_INDEX(TDMT_MND_TABLE_CFG)]        = queryBuildGetTbCfgMsg;
D
dapan1121 已提交
545 546 547 548 549 550 551 552 553 554

  queryProcessMsgRsp[TMSG_INDEX(TDMT_VND_TABLE_META)]      = queryProcessTableMetaRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_TABLE_META)]      = queryProcessTableMetaRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_USE_DB)]          = queryProcessUseDBRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_QNODE_LIST)]      = queryProcessQnodeListRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_GET_DB_CFG)]      = queryProcessGetDbCfgRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_GET_INDEX)]       = queryProcessGetIndexRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_RETRIEVE_FUNC)]   = queryProcessRetrieveFuncRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_GET_USER_AUTH)]   = queryProcessGetUserAuthRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_GET_TABLE_INDEX)] = queryProcessGetTbIndexRsp;
D
dapan1121 已提交
555 556
  queryProcessMsgRsp[TMSG_INDEX(TDMT_VND_TABLE_CFG)]       = queryProcessGetTbCfgRsp;
  queryProcessMsgRsp[TMSG_INDEX(TDMT_MND_TABLE_CFG)]       = queryProcessGetTbCfgRsp;
D
dapan 已提交
557 558
}

D
dapan1121 已提交
559
#pragma GCC diagnostic pop