tscServer.c 86.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/>.
 */

S
slguan 已提交
16
#include "os.h"
H
hzcheng 已提交
17 18 19 20
#include "tcache.h"
#include "trpc.h"
#include "tscProfile.h"
#include "tscSecondaryMerge.h"
H
hjxilinx 已提交
21
#include "tscSubquery.h"
H
hzcheng 已提交
22 23 24 25 26 27 28
#include "tscUtil.h"
#include "tschemautil.h"
#include "tsclient.h"
#include "tsocket.h"
#include "ttime.h"
#include "ttimer.h"
#include "tutil.h"
S
slguan 已提交
29
#include "tscLog.h"
陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
30
#include "qsqltype.h"
H
hzcheng 已提交
31 32 33

#define TSC_MGMT_VNODE 999

S
slguan 已提交
34
SRpcIpSet  tscMgmtIpSet;
S
slguan 已提交
35 36
SRpcIpSet  tscDnodeIpSet;

37 38
int (*tscBuildMsg[TSDB_SQL_MAX])(SSqlObj *pSql, SSqlInfo *pInfo) = {0};

H
hzcheng 已提交
39 40 41
int (*tscProcessMsgRsp[TSDB_SQL_MAX])(SSqlObj *pSql);
void tscProcessActivityTimer(void *handle, void *tmrId);
int tscKeepConn[TSDB_SQL_MAX] = {0};
42

B
Bomin Zhang 已提交
43
TSKEY tscGetSubscriptionProgress(void* sub, int64_t uid, TSKEY dflt);
44 45
void tscUpdateSubscriptionProgress(void* sub, int64_t uid, TSKEY ts);
void tscSaveSubscriptionProgress(void* sub);
H
hzcheng 已提交
46

S
slguan 已提交
47
static int32_t minMsgSize() { return tsRpcHeadSize + 100; }
H
hzcheng 已提交
48

49
static void tscSetDnodeIpList(SSqlObj* pSql, SCMVgroupInfo* pVgroupInfo) {
50 51
  SRpcIpSet* pIpList = &pSql->ipList;
  
52
  pIpList->numOfIps = pVgroupInfo->numOfIps;
53 54
  pIpList->inUse    = 0;
  
55 56 57
  for(int32_t i = 0; i < pVgroupInfo->numOfIps; ++i) {
    strcpy(pIpList->fqdn[i], pVgroupInfo->ipAddr[i].fqdn);
    pIpList->port[i] = pVgroupInfo->ipAddr[i].port;
58 59 60
  }
}

S
slguan 已提交
61
void tscPrintMgmtIp() {
S
slguan 已提交
62 63
  if (tscMgmtIpSet.numOfIps <= 0) {
    tscError("invalid mgmt IP list:%d", tscMgmtIpSet.numOfIps);
S
slguan 已提交
64
  } else {
S
slguan 已提交
65
    for (int i = 0; i < tscMgmtIpSet.numOfIps; ++i) {
J
jtao1735 已提交
66
      tscTrace("mgmt index:%d %s:%d", i, tscMgmtIpSet.fqdn[i], tscMgmtIpSet.port[i]);
S
slguan 已提交
67
    }
S
slguan 已提交
68 69 70
  }
}

71
void tscSetMgmtIpList(SRpcIpSet *pIpList) {
S
slguan 已提交
72 73 74
  tscMgmtIpSet.numOfIps = pIpList->numOfIps;
  tscMgmtIpSet.inUse = pIpList->inUse;
  for (int32_t i = 0; i < tscMgmtIpSet.numOfIps; ++i) {
J
jtao1735 已提交
75
    tscMgmtIpSet.port[i] = htons(pIpList->port[i]);
S
slguan 已提交
76 77 78
  }
}

S
slguan 已提交
79
void tscUpdateIpSet(void *ahandle, SRpcIpSet *pIpSet) {
S
slguan 已提交
80
  tscMgmtIpSet = *pIpSet;
S
slguan 已提交
81 82 83 84
  tscTrace("mgmt IP list is changed for ufp is called, numOfIps:%d inUse:%d", tscMgmtIpSet.numOfIps, tscMgmtIpSet.inUse);
  for (int32_t i = 0; i < tscMgmtIpSet.numOfIps; ++i) {
    tscTrace("index:%d fqdn:%s port:%d", i, tscMgmtIpSet.fqdn[i], tscMgmtIpSet.port[i]);
  }
S
slguan 已提交
85 86
}

H
hjxilinx 已提交
87 88 89 90 91 92 93
/*
 * For each management node, try twice at least in case of poor network situation.
 * If the client start to connect to a non-management node from the client, and the first retry may fail due to
 * the poor network quality. And then, the second retry get the response with redirection command.
 * The retry will not be executed since only *two* retry is allowed in case of single management node in the cluster.
 * Therefore, we need to multiply the retry times by factor of 2 to fix this problem.
 */
94
UNUSED_FUNC
H
hjxilinx 已提交
95 96
static int32_t tscGetMgmtConnMaxRetryTimes() {
  int32_t factor = 2;
S
slguan 已提交
97
  return tscMgmtIpSet.numOfIps * factor;
H
hjxilinx 已提交
98 99
}

H
hzcheng 已提交
100 101 102 103 104 105 106 107 108 109 110 111
void tscProcessHeartBeatRsp(void *param, TAOS_RES *tres, int code) {
  STscObj *pObj = (STscObj *)param;
  if (pObj == NULL) return;
  if (pObj != pObj->signature) {
    tscError("heart beat msg, pObj:%p, signature:%p invalid", pObj, pObj->signature);
    return;
  }

  SSqlObj *pSql = pObj->pHb;
  SSqlRes *pRes = &pSql->res;

  if (code == 0) {
112
    SCMHeartBeatRsp *pRsp = (SCMHeartBeatRsp *)pRes->pRsp;
S
slguan 已提交
113
    SRpcIpSet *      pIpList = &pRsp->ipList;
114 115
    if (pIpList->numOfIps > 0) 
      tscSetMgmtIpList(pIpList);
S
slguan 已提交
116

H
hzcheng 已提交
117 118 119
    if (pRsp->killConnection) {
      tscKillConnection(pObj);
    } else {
S
slguan 已提交
120 121
      if (pRsp->queryId) tscKillQuery(pObj, htonl(pRsp->queryId));
      if (pRsp->streamId) tscKillStream(pObj, htonl(pRsp->streamId));
H
hzcheng 已提交
122 123
    }
  } else {
H
hjxilinx 已提交
124
    tscTrace("heart beat failed, code:%s", tstrerror(code));
H
hzcheng 已提交
125 126 127 128 129 130 131 132 133 134 135 136 137
  }

  taosTmrReset(tscProcessActivityTimer, tsShellActivityTimer * 500, pObj, tscTmr, &pObj->pTimer);
}

void tscProcessActivityTimer(void *handle, void *tmrId) {
  STscObj *pObj = (STscObj *)handle;

  if (pObj == NULL) return;
  if (pObj->signature != pObj) return;
  if (pObj->pTimer != tmrId) return;

  if (pObj->pHb == NULL) {
S
slguan 已提交
138 139 140
    SSqlObj *pSql = (SSqlObj *)calloc(1, sizeof(SSqlObj));
    if (NULL == pSql) return;

H
hzcheng 已提交
141
    pSql->fp = tscProcessHeartBeatRsp;
H
hjxilinx 已提交
142
    
143 144 145 146
    SQueryInfo *pQueryInfo = NULL;
    tscGetQueryInfoDetailSafely(&pSql->cmd, 0, &pQueryInfo);
    pQueryInfo->command = TSDB_SQL_HB;
    
147
    pSql->cmd.command = TSDB_SQL_HB;
S
slguan 已提交
148 149 150 151 152
    if (TSDB_CODE_SUCCESS != tscAllocPayload(&(pSql->cmd), TSDB_DEFAULT_PAYLOAD_SIZE)) {
      tfree(pSql);
      return;
    }

153
    pSql->cmd.command = TSDB_SQL_HB;
H
hzcheng 已提交
154 155 156 157
    pSql->param = pObj;
    pSql->pTscObj = pObj;
    pSql->signature = pSql;
    pObj->pHb = pSql;
158
    tscAddSubqueryInfo(&pObj->pHb->cmd);
159

S
slguan 已提交
160
    tscTrace("%p pHb is allocated, pObj:%p", pObj->pHb, pObj);
H
hzcheng 已提交
161 162 163
  }

  if (tscShouldFreeHeatBeat(pObj->pHb)) {
S
slguan 已提交
164
    tscTrace("%p free HB object and release connection", pObj);
H
hzcheng 已提交
165 166 167 168 169 170 171 172 173
    tscFreeSqlObj(pObj->pHb);
    tscCloseTscObj(pObj);
    return;
  }

  tscProcessSql(pObj->pHb);
}

int tscSendMsgToServer(SSqlObj *pSql) {
174
  STscObj* pObj = pSql->pTscObj;
H
hjxilinx 已提交
175 176 177
  SSqlCmd* pCmd = &pSql->cmd;
  
  char *pMsg = rpcMallocCont(pCmd->payloadLen);
S
slguan 已提交
178
  if (NULL == pMsg) {
S
slguan 已提交
179 180
    tscError("%p msg:%s malloc fail", pSql, taosMsg[pSql->cmd.msgType]);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
H
hzcheng 已提交
181 182
  }

S
slguan 已提交
183
  if (pSql->cmd.command < TSDB_SQL_MGMT) {
S
slguan 已提交
184
    memcpy(pMsg, pSql->cmd.payload + tsRpcHeadSize, pSql->cmd.payloadLen);
J
jtao1735 已提交
185 186 187 188 189
  } else {
    pSql->ipList = tscMgmtIpSet;
    memcpy(pMsg, pSql->cmd.payload, pSql->cmd.payloadLen);
  }

陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
190
  // tscTrace("%p msg:%s is sent to server", pSql, taosMsg[pSql->cmd.msgType]);
191

J
jtao1735 已提交
192
  SRpcMsg rpcMsg = {
193 194 195 196
      .msgType = pSql->cmd.msgType,
      .pCont   = pMsg,
      .contLen = pSql->cmd.payloadLen,
      .handle  = pSql,
H
hjxilinx 已提交
197
      .code    = 0
J
jtao1735 已提交
198
  };
199
  rpcSendRequest(pObj->pDnodeConn, &pSql->ipList, &rpcMsg);
H
hzcheng 已提交
200

S
slguan 已提交
201
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
202 203
}

J
jtao1735 已提交
204
void tscProcessMsgFromServer(SRpcMsg *rpcMsg, SRpcIpSet *pIpSet) {
205
  SSqlObj *pSql = (SSqlObj *)rpcMsg->handle;
206 207 208 209 210
  if (pSql == NULL) {
    tscError("%p sql is already released", pSql->signature);
    return;
  }
  if (pSql->signature != pSql) {
H
hzcheng 已提交
211
    tscError("%p sql is already released, signature:%p", pSql, pSql->signature);
S
slguan 已提交
212
    return;
H
hzcheng 已提交
213 214
  }

S
slguan 已提交
215 216 217
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;
  STscObj *pObj = pSql->pTscObj;
陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
218
  // tscTrace("%p msg:%s is received from server", pSql, taosMsg[rpcMsg->msgType]);
H
hzcheng 已提交
219

220
  if (pObj->signature != pObj) {
S
slguan 已提交
221 222
    tscTrace("%p sql is already released or DB connection is closed, freed:%d pObj:%p signature:%p", pSql, pSql->freed,
             pObj, pObj->signature);
223 224 225
    if (pSql != pObj->pSql) {
      tscFreeSqlObj(pSql);
    }
226
    rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
227
    return;
H
hzcheng 已提交
228 229
  }

J
jtao1735 已提交
230 231 232 233 234 235
  if (pCmd->command < TSDB_SQL_MGMT) {
    if (pIpSet) pSql->ipList = *pIpSet;
  } else {
    if (pIpSet) tscMgmtIpSet = *pIpSet;
  }

236 237
  if (rpcMsg->pCont == NULL) {
    rpcMsg->code = TSDB_CODE_NETWORK_UNAVAIL;
S
slguan 已提交
238
  } else {
239
    STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
240 241
    if (rpcMsg->code == TSDB_CODE_NOT_ACTIVE_TABLE || rpcMsg->code == TSDB_CODE_INVALID_TABLE_ID ||
        rpcMsg->code == TSDB_CODE_INVALID_VNODE_ID || rpcMsg->code == TSDB_CODE_NOT_ACTIVE_VNODE ||
S
slguan 已提交
242
        rpcMsg->code == TSDB_CODE_NETWORK_UNAVAIL || rpcMsg->code == TSDB_CODE_NOT_ACTIVE_TABLE ||
243
        rpcMsg->code == TSDB_CODE_TABLE_ID_MISMATCH) {
S
slguan 已提交
244 245 246 247 248 249 250 251 252 253 254
      /*
       * not_active_table: 1. the virtual node may fail to create table, since the procedure of create table is asynchronized,
       *                   the virtual node may have not create table till now, so try again by using the new metermeta.
       *                   2. this requested table may have been removed by other client, so we need to renew the
       *                   metermeta here.
       *
       * not_active_vnode: current vnode is move to other node due to node balance procedure or virtual node have been
       *                   removed. So, renew metermeta and try again.
       * not_active_session: db has been move to other node, the vnode does not exist on this dnode anymore.
       */
      if (pCmd->command == TSDB_SQL_CONNECT) {
255 256
        rpcMsg->code = TSDB_CODE_NETWORK_UNAVAIL;
        rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
257 258
        return;
      } else if (pCmd->command == TSDB_SQL_HB) {
259 260
        rpcMsg->code = TSDB_CODE_NOT_READY;
        rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
261 262
        return;
      } else {
263
        tscWarn("%p it shall renew table meta, code:%s, retry:%d", pSql, tstrerror(rpcMsg->code), ++pSql->retry);
264
        
265
        pSql->res.code = rpcMsg->code;  // keep the previous error code
266 267 268 269 270 271 272 273
        if (pSql->retry > pSql->maxRetry) {
          tscError("%p max retry %d reached, give up", pSql, pSql->maxRetry);
        } else {
          rpcMsg->code = tscRenewMeterMeta(pSql, pTableMetaInfo->name);
          if (pTableMetaInfo->pTableMeta) {
            tscSendMsgToServer(pSql);
          }
  
274
          rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
275 276
          return;
        }
H
hzcheng 已提交
277 278
      }
    }
S
slguan 已提交
279
  }
280
  
H
hzcheng 已提交
281
  pRes->rspLen = 0;
282
  
H
hzcheng 已提交
283
  if (pRes->code != TSDB_CODE_QUERY_CANCELLED) {
284
    pRes->code = (rpcMsg->code != TSDB_CODE_SUCCESS) ? rpcMsg->code : TSDB_CODE_NETWORK_UNAVAIL;
H
hzcheng 已提交
285
  } else {
H
hjxilinx 已提交
286
    tscTrace("%p query is cancelled, code:%d", pSql, tstrerror(pRes->code));
H
hzcheng 已提交
287 288
  }

S
slguan 已提交
289 290 291 292 293
  if (pRes->code == TSDB_CODE_SUCCESS) {
    tscTrace("%p reset retry counter to be 0 due to success rsp, old:%d", pSql, pSql->retry);
    pSql->retry = 0;
  }

S
slguan 已提交
294
  if (pRes->code != TSDB_CODE_QUERY_CANCELLED) {
295
    assert(rpcMsg->msgType == pCmd->msgType + 1);
296
    pRes->code    = rpcMsg->code;
297
    pRes->rspType = rpcMsg->msgType;
298
    pRes->rspLen  = rpcMsg->contLen;
H
hzcheng 已提交
299

300
    if (pRes->rspLen > 0 && rpcMsg->pCont) {
301 302 303 304 305
      char *tmp = (char *)realloc(pRes->pRsp, pRes->rspLen);
      if (tmp == NULL) {
        pRes->code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      } else {
        pRes->pRsp = tmp;
306
        memcpy(pRes->pRsp, rpcMsg->pCont, pRes->rspLen);
S
slguan 已提交
307
      }
308 309
    } else {
      pRes->pRsp = NULL;
S
slguan 已提交
310 311
    }

H
hzcheng 已提交
312 313 314 315
    /*
     * There is not response callback function for submit response.
     * The actual inserted number of points is the first number.
     */
316
    if (rpcMsg->msgType == TSDB_MSG_TYPE_SUBMIT_RSP && pRes->pRsp != NULL) {
317
      SShellSubmitRspMsg *pMsg = (SShellSubmitRspMsg*)pRes->pRsp;
S
slguan 已提交
318 319 320 321 322 323 324
      pMsg->code = htonl(pMsg->code);
      pMsg->numOfRows = htonl(pMsg->numOfRows);
      pMsg->affectedRows = htonl(pMsg->affectedRows);
      pMsg->failedRows = htonl(pMsg->failedRows);
      pMsg->numOfFailedBlocks = htonl(pMsg->numOfFailedBlocks);

      pRes->numOfRows += pMsg->affectedRows;
陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
325 326
      tscTrace("%p SQL cmd:%s, code:%s inserted rows:%d rspLen:%d", pSql, sqlCmd[pCmd->command], 
          tstrerror(pRes->code), pMsg->affectedRows, pRes->rspLen);
S
slguan 已提交
327
    } else {
陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
328
      tscTrace("%p SQL cmd:%s, code:%s rspLen:%d", pSql, sqlCmd[pCmd->command], tstrerror(pRes->code), pRes->rspLen);
H
hzcheng 已提交
329 330
    }
  }
331
  
332 333
  if (pRes->code == TSDB_CODE_SUCCESS && tscProcessMsgRsp[pCmd->command])
    rpcMsg->code = (*tscProcessMsgRsp[pCmd->command])(pSql);
H
hjxilinx 已提交
334
  
335
  if (rpcMsg->code != TSDB_CODE_ACTION_IN_PROGRESS) {
336
    void *taosres = tscKeepConn[pCmd->command] ? pSql : NULL;
H
[td-32]  
hjxilinx 已提交
337
    rpcMsg->code = pRes->code ? pRes->code : pRes->numOfRows;
338
    
339
    tscTrace("%p SQL result:%s res:%p", pSql, tstrerror(pRes->code), pSql);
H
hzcheng 已提交
340

341 342
    /*
     * Whether to free sqlObj or not should be decided before call the user defined function, since this SqlObj
H
hjxilinx 已提交
343 344
     * may be freed in UDF, and reused by other threads before tscShouldBeFreed called, in which case
     * tscShouldBeFreed checks an object which is actually allocated by other threads.
345 346
     *
     * If this block of memory is re-allocated for an insert thread, in which tscKeepConn[command] equals to 0,
H
hjxilinx 已提交
347
     * the tscShouldBeFreed will success and tscFreeSqlObj free it immediately.
348
     */
H
hjxilinx 已提交
349
    bool shouldFree = tscShouldBeFreed(pSql);
H
hjxilinx 已提交
350
    (*pSql->fp)(pSql->param, taosres, rpcMsg->code);
H
hzcheng 已提交
351

352
    if (shouldFree) {
353
      tscTrace("%p sqlObj is automatically freed", pSql);
sangshuduo's avatar
sangshuduo 已提交
354
      tscFreeSqlObj(pSql);
H
hzcheng 已提交
355 356 357
    }
  }

358
  rpcFreeCont(rpcMsg->pCont);
H
hzcheng 已提交
359 360
}

S
slguan 已提交
361 362 363
int doProcessSql(SSqlObj *pSql) {
  SSqlCmd *pCmd = &pSql->cmd;
  SSqlRes *pRes = &pSql->res;
364 365
  int32_t code = TSDB_CODE_SUCCESS;
  
H
hjxilinx 已提交
366 367 368 369 370 371 372
  if (pCmd->command == TSDB_SQL_SELECT ||
      pCmd->command == TSDB_SQL_FETCH ||
      pCmd->command == TSDB_SQL_RETRIEVE ||
      pCmd->command == TSDB_SQL_INSERT ||
      pCmd->command == TSDB_SQL_CONNECT ||
      pCmd->command == TSDB_SQL_HB ||
      pCmd->command == TSDB_SQL_META ||
H
hjxilinx 已提交
373
      pCmd->command == TSDB_SQL_STABLEVGROUP) {
374 375 376 377 378 379
    pRes->code = tscBuildMsg[pCmd->command](pSql, NULL);
  }
  
  if (pRes->code != TSDB_CODE_SUCCESS) {
    tscQueueAsyncRes(pSql);
    return pRes->code;
S
slguan 已提交
380
  }
381

382
  code = tscSendMsgToServer(pSql);
383
  if (code != TSDB_CODE_SUCCESS) {
S
slguan 已提交
384
    pRes->code = code;
H
hjxilinx 已提交
385
    tscQueueAsyncRes(pSql);
S
slguan 已提交
386
  }
H
hjxilinx 已提交
387 388
  
  return TSDB_CODE_SUCCESS;
S
slguan 已提交
389 390 391
}

int tscProcessSql(SSqlObj *pSql) {
392 393
  char *   name = NULL;
  SSqlCmd *pCmd = &pSql->cmd;
394 395
  
  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
396
  STableMetaInfo *pTableMetaInfo = NULL;
397
  uint16_t        type = 0;
398

399
  if (pQueryInfo != NULL) {
H
hjxilinx 已提交
400
    pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
401 402
    if (pTableMetaInfo != NULL) {
      name = pTableMetaInfo->name;
403
    }
404

405
    type = pQueryInfo->type;
406
  
H
hjxilinx 已提交
407
    // while numOfTables equals to 0, it must be Heartbeat
H
hjxilinx 已提交
408
    assert((pQueryInfo->numOfTables == 0 && pQueryInfo->command == TSDB_SQL_HB) || pQueryInfo->numOfTables > 0);
S
slguan 已提交
409
  }
410

陶建辉(Jeff)'s avatar
陶建辉(Jeff) 已提交
411
  tscTrace("%p SQL cmd:%s will be processed, name:%s, type:%d", pSql, sqlCmd[pCmd->command], name, type);
H
hjxilinx 已提交
412
  if (pCmd->command < TSDB_SQL_MGMT) { // the pTableMetaInfo cannot be NULL
H
hjxilinx 已提交
413
    if (pTableMetaInfo == NULL) {
H
hjxilinx 已提交
414 415 416
      pSql->res.code = TSDB_CODE_OTHERS;
      return pSql->res.code;
    }
H
hjxilinx 已提交
417 418
  } else if (pCmd->command < TSDB_SQL_LOCAL) {
    pSql->ipList = tscMgmtIpSet; //?
H
hzcheng 已提交
419 420 421
  } else {  // local handler
    return (*tscProcessMsgRsp[pCmd->command])(pSql);
  }
422
  
S
slguan 已提交
423 424
  return doProcessSql(pSql);
}
H
hzcheng 已提交
425

H
hjxilinx 已提交
426
void tscKillSTableQuery(SSqlObj *pSql) {
427 428 429
  SSqlCmd* pCmd = &pSql->cmd;
  
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
430
  if (!tscIsTwoStageSTableQuery(pQueryInfo, 0)) {
H
hzcheng 已提交
431 432 433 434 435 436
    return;
  }

  for (int i = 0; i < pSql->numOfSubs; ++i) {
    SSqlObj *pSub = pSql->pSubs[i];

S
slguan 已提交
437
    if (pSub == NULL) {
H
hzcheng 已提交
438 439
      continue;
    }
S
slguan 已提交
440

H
hzcheng 已提交
441 442 443 444 445
    /*
     * here, we cannot set the command = TSDB_SQL_KILL_QUERY. Otherwise, it may cause
     * sub-queries not correctly released and master sql object of metric query reaches an abnormal state.
     */
    pSql->pSubs[i]->res.code = TSDB_CODE_QUERY_CANCELLED;
S
slguan 已提交
446
    //taosStopRpcConn(pSql->pSubs[i]->thandle);
H
hzcheng 已提交
447 448 449 450 451 452 453 454 455 456 457
  }

  /*
   * 1. if the subqueries are not launched or partially launched, we need to waiting the launched
   * query return to successfully free allocated resources.
   * 2. if no any subqueries are launched yet, which means the metric query only in parse sql stage,
   * set the res.code, and return.
   */
  const int64_t MAX_WAITING_TIME = 10000;  // 10 Sec.
  int64_t       stime = taosGetTimestampMs();

H
hjxilinx 已提交
458
  while (pCmd->command != TSDB_SQL_RETRIEVE_LOCALMERGE && pCmd->command != TSDB_SQL_RETRIEVE_EMPTY_RESULT) {
H
hzcheng 已提交
459 460 461 462 463 464
    taosMsleep(100);
    if (taosGetTimestampMs() - stime > MAX_WAITING_TIME) {
      break;
    }
  }

H
hjxilinx 已提交
465
  tscTrace("%p super table query cancelled", pSql);
H
hzcheng 已提交
466 467
}

J
jtao1735 已提交
468
int tscBuildFetchMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
469 470 471 472 473
  char *pMsg, *pStart;

  pStart = pSql->cmd.payload + tsRpcHeadSize;
  pMsg = pStart;

474
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg *)pMsg;
S
slguan 已提交
475
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
S
slguan 已提交
476 477
  pMsg += sizeof(pSql->res.qhandle);

478
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
S
slguan 已提交
479
  pRetrieveMsg->free = htons(pQueryInfo->type);
480
  pMsg += sizeof(pQueryInfo->type);
H
hzcheng 已提交
481

482
  // todo valid the vgroupId at the client side
H
hjxilinx 已提交
483 484
  STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
  
weixin_48148422's avatar
weixin_48148422 已提交
485
  if (UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo)) {
H
hjxilinx 已提交
486
    int32_t vgIndex = pTableMetaInfo->vgroupIndex;
487
    
H
hjxilinx 已提交
488 489
    SVgroupsInfo* pVgroupInfo = pTableMetaInfo->vgroupList;
    pRetrieveMsg->header.vgId = htonl(pVgroupInfo->vgroups[vgIndex].vgId);
490
  } else {
H
hjxilinx 已提交
491
    STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
492
    pRetrieveMsg->header.vgId = htonl(pTableMeta->vgroupInfo.vgId);
493 494
  }
  
495 496 497 498
  pMsg += sizeof(SRetrieveTableMsg);
  
  pRetrieveMsg->header.contLen = htonl(pSql->cmd.payloadLen);
  
J
jtao1735 已提交
499
  pSql->cmd.msgType = TSDB_MSG_TYPE_FETCH;
500
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
501 502
}

503
int tscBuildSubmitMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hjxilinx 已提交
504
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
H
hjxilinx 已提交
505
  STableMeta* pTableMeta = tscGetMetaInfo(pQueryInfo, 0)->pTableMeta;
H
hjxilinx 已提交
506
  
507 508 509 510
  char* pMsg = pSql->cmd.payload + tsRpcHeadSize;
  
  // NOTE: shell message size should not include SMsgDesc
  int32_t size = pSql->cmd.payloadLen - sizeof(SMsgDesc);
H
hjxilinx 已提交
511
  
512
  SMsgDesc* pMsgDesc = (SMsgDesc*) pMsg;
513 514
  
  pMsgDesc->numOfVnodes = htonl(1);       //todo set the right number of vnodes
515
  pMsg += sizeof(SMsgDesc);
H
hjxilinx 已提交
516
  
517
  SSubmitMsg *pShellMsg = (SSubmitMsg *)pMsg;
H
hjxilinx 已提交
518
  int32_t vgId = pTableMeta->vgroupInfo.vgId;
519
  
H
hjxilinx 已提交
520
  pShellMsg->header.vgId = htonl(vgId);
521
  pShellMsg->header.contLen = htonl(size);
522
  pShellMsg->length = pShellMsg->header.contLen;
H
hjxilinx 已提交
523
  
524
  pShellMsg->numOfBlocks = htonl(pSql->cmd.numOfTablesInSubmit);  // number of meters to be inserted
H
hzcheng 已提交
525

H
hjxilinx 已提交
526
  // pSql->cmd.payloadLen is set during copying data into payload
S
slguan 已提交
527
  pSql->cmd.msgType = TSDB_MSG_TYPE_SUBMIT;
528
  tscSetDnodeIpList(pSql, &pTableMeta->vgroupInfo);
529
  
S
slguan 已提交
530
  tscTrace("%p build submit msg, vgId:%d numOfVgroup:%d numberOfIP:%d", pSql, vgId, htonl(pMsgDesc->numOfVnodes), pSql->ipList.numOfIps);
531
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
532 533 534
}

/*
535
 * for table query, simply return the size <= 1k
H
hzcheng 已提交
536
 */
537
static int32_t tscEstimateQueryMsgSize(SSqlCmd *pCmd, int32_t clauseIndex) {
H
hzcheng 已提交
538
  const static int32_t MIN_QUERY_MSG_PKT_SIZE = TSDB_MAX_BYTES_PER_ROW * 5;
539
  SQueryInfo *         pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
H
hzcheng 已提交
540

541
  int32_t srcColListSize = taosArrayGetSize(pQueryInfo->colList) * sizeof(SColumnInfo);
H
hjxilinx 已提交
542 543 544 545
  
  size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
  int32_t exprSize = sizeof(SSqlFuncMsg) * numOfExprs;
  
546
  return MIN_QUERY_MSG_PKT_SIZE + minMsgSize() + sizeof(SQueryTableMsg) + srcColListSize + exprSize + 4096;
H
hzcheng 已提交
547 548
}

549
static char *doSerializeTableInfo(SQueryTableMsg* pQueryMsg, SSqlObj *pSql, char *pMsg) {
550
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, pSql->cmd.clauseIndex, 0);
B
Bomin Zhang 已提交
551
  TSKEY dfltKey = htobe64(pQueryMsg->window.skey);
552

H
hjxilinx 已提交
553
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
weixin_48148422's avatar
weixin_48148422 已提交
554
  if (UTIL_TABLE_IS_NORMAL_TABLE(pTableMetaInfo) || pTableMetaInfo->pVgroupTables == NULL) {
555 556
    
    SCMVgroupInfo* pVgroupInfo = NULL;
weixin_48148422's avatar
weixin_48148422 已提交
557
    if (UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo)) {
558 559
      int32_t index = pTableMetaInfo->vgroupIndex;
      assert(index >= 0);
H
hjxilinx 已提交
560
  
561 562
      pVgroupInfo = &pTableMetaInfo->vgroupList->vgroups[index];
      tscTrace("%p query on stable, vgIndex:%d, numOfVgroups:%d", pSql, index, pTableMetaInfo->vgroupList->numOfVgroups);
H
hjxilinx 已提交
563 564
    } else {
      pVgroupInfo = &pTableMeta->vgroupInfo;
565
    }
weixin_48148422's avatar
weixin_48148422 已提交
566

567 568
    tscSetDnodeIpList(pSql, pVgroupInfo);
    pQueryMsg->head.vgId = htonl(pVgroupInfo->vgId);
weixin_48148422's avatar
weixin_48148422 已提交
569

570 571 572
    STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
    pTableIdInfo->tid = htonl(pTableMeta->sid);
    pTableIdInfo->uid = htobe64(pTableMeta->uid);
B
Bomin Zhang 已提交
573
    pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pTableMeta->uid, dfltKey));
weixin_48148422's avatar
weixin_48148422 已提交
574

575 576 577 578 579 580
    pQueryMsg->numOfTables = htonl(1);  // set the number of tables
    pMsg += sizeof(STableIdInfo);
  } else {
    int32_t index = pTableMetaInfo->vgroupIndex;
    int32_t numOfVgroups = taosArrayGetSize(pTableMetaInfo->pVgroupTables);
    assert(index >= 0 && index < numOfVgroups);
weixin_48148422's avatar
weixin_48148422 已提交
581

582
    tscTrace("%p query on stable, vgIndex:%d, numOfVgroups:%d", pSql, index, numOfVgroups);
weixin_48148422's avatar
weixin_48148422 已提交
583

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
    SVgroupTableInfo* pTableIdList = taosArrayGet(pTableMetaInfo->pVgroupTables, index);
    
    // set the vgroup info
    tscSetDnodeIpList(pSql, &pTableIdList->vgInfo);
    pQueryMsg->head.vgId = htonl(pTableIdList->vgInfo.vgId);
    
    int32_t numOfTables = taosArrayGetSize(pTableIdList->itemList);
    pQueryMsg->numOfTables = htonl(numOfTables);  // set the number of tables
  
    // serialize each table id info
    for(int32_t i = 0; i < numOfTables; ++i) {
      STableIdInfo* pItem = taosArrayGet(pTableIdList->itemList, i);
      
      STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
      pTableIdInfo->tid = htonl(pItem->tid);
      pTableIdInfo->uid = htobe64(pItem->uid);
B
Bomin Zhang 已提交
600
      pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pItem->uid, dfltKey));
601 602 603 604
      pMsg += sizeof(STableIdInfo);
    }
  }
  
H
Haojun Liao 已提交
605 606
  tscTrace("%p vgId:%d, query on table:%s, tid:%d, uid:%" PRIu64, pSql, htonl(pQueryMsg->head.vgId), pTableMetaInfo->name,
      pTableMeta->sid, pTableMeta->uid);
H
hjxilinx 已提交
607
  
608 609 610
  return pMsg;
}

611
int tscBuildQueryMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
612 613
  SSqlCmd *pCmd = &pSql->cmd;

614
  int32_t size = tscEstimateQueryMsgSize(pCmd, pCmd->clauseIndex);
H
hzcheng 已提交
615

S
slguan 已提交
616 617 618 619
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for query msg", pSql);
    return -1;
  }
620
  
621
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
622
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
623
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
624
  
H
hjxilinx 已提交
625
  if (taosArrayGetSize(pQueryInfo->colList) <= 0 && !tscQueryTags(pQueryInfo)) {
626 627 628
    tscError("%p illegal value of numOfCols in query msg: %d", pSql, tscGetNumOfColumns(pTableMeta));
    return -1;
  }
629 630 631 632 633 634 635 636 637 638
  
  if (pQueryInfo->intervalTime < 0) {
    tscError("%p illegal value of aggregation time interval in query msg: %ld", pSql, pQueryInfo->intervalTime);
    return -1;
  }
  
  if (pQueryInfo->groupbyExpr.numOfGroupCols < 0) {
    tscError("%p illegal value of numOfGroupCols in query msg: %d", pSql, pQueryInfo->groupbyExpr.numOfGroupCols);
    return -1;
  }
639 640

  char *pStart = pCmd->payload + tsRpcHeadSize;
H
hzcheng 已提交
641

S
slguan 已提交
642
  SQueryTableMsg *pQueryMsg = (SQueryTableMsg *)pStart;
H
hzcheng 已提交
643

644
  int32_t numOfTags = taosArrayGetSize(pTableMetaInfo->tagColList);
645
  
646
  if (pQueryInfo->order.order == TSDB_ORDER_ASC) {
H
hjxilinx 已提交
647 648
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.skey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.ekey);
H
hzcheng 已提交
649
  } else {
H
hjxilinx 已提交
650 651
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.ekey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.skey);
H
hzcheng 已提交
652 653
  }

654 655
  pQueryMsg->order          = htons(pQueryInfo->order.order);
  pQueryMsg->orderColId     = htons(pQueryInfo->order.orderColId);
656
  pQueryMsg->fillType    = htons(pQueryInfo->fillType);
657 658
  pQueryMsg->limit          = htobe64(pQueryInfo->limit.limit);
  pQueryMsg->offset         = htobe64(pQueryInfo->limit.offset);
659
  pQueryMsg->numOfCols      = htons(taosArrayGetSize(pQueryInfo->colList));
660 661
  pQueryMsg->intervalTime   = htobe64(pQueryInfo->intervalTime);
  pQueryMsg->slidingTime    = htobe64(pQueryInfo->slidingTime);
H
hjxilinx 已提交
662
  pQueryMsg->slidingTimeUnit = pQueryInfo->slidingTimeUnit;
663
  pQueryMsg->numOfGroupCols = htons(pQueryInfo->groupbyExpr.numOfGroupCols);
664
  pQueryMsg->numOfTags      = htonl(numOfTags);
weixin_48148422's avatar
weixin_48148422 已提交
665
  pQueryMsg->tagNameRelType = htons(pQueryInfo->tagCond.relType);
666
  pQueryMsg->queryType      = htons(pQueryInfo->type);
H
hjxilinx 已提交
667 668 669
  
  size_t numOfOutput = tscSqlExprNumOfExprs(pQueryInfo);
  pQueryMsg->numOfOutput = htons(numOfOutput);
H
hjxilinx 已提交
670 671
  if (numOfOutput < 0) {
    tscError("%p illegal value of number of output columns in query msg: %d", pSql, numOfOutput);
H
hzcheng 已提交
672 673 674 675
    return -1;
  }

  // set column list ids
676 677
  size_t numOfCols = taosArrayGetSize(pQueryInfo->colList);
  char *pMsg = (char *)(pQueryMsg->colList) + numOfCols * sizeof(SColumnInfo);
H
hjxilinx 已提交
678
  SSchema *pSchema = tscGetTableSchema(pTableMeta);
679
  
680 681 682
  for (int32_t i = 0; i < numOfCols; ++i) {
    SColumn *pCol = taosArrayGetP(pQueryInfo->colList, i);
    SSchema *pColSchema = &pSchema[pCol->colIndex.columnIndex];
H
hzcheng 已提交
683

684
    if (pCol->colIndex.columnIndex >= tscGetNumOfColumns(pTableMeta) || pColSchema->type < TSDB_DATA_TYPE_BOOL ||
685 686 687 688 689
        pColSchema->type > TSDB_DATA_TYPE_NCHAR) {
      tscError("%p sid:%d uid:%" PRIu64" id:%s, column index out of range, numOfColumns:%d, index:%d, column name:%s",
          pSql, pTableMeta->sid, pTableMeta->uid, pTableMetaInfo->name, tscGetNumOfColumns(pTableMeta), pCol->colIndex,
               pColSchema->name);

690
      return TSDB_CODE_INVALID_SQL;
691
    }
H
hzcheng 已提交
692 693 694

    pQueryMsg->colList[i].colId = htons(pColSchema->colId);
    pQueryMsg->colList[i].bytes = htons(pColSchema->bytes);
695
    pQueryMsg->colList[i].type  = htons(pColSchema->type);
S
slguan 已提交
696
    pQueryMsg->colList[i].numOfFilters = htons(pCol->numOfFilters);
H
hzcheng 已提交
697

S
slguan 已提交
698 699 700
    // append the filter information after the basic column information
    for (int32_t f = 0; f < pCol->numOfFilters; ++f) {
      SColumnFilterInfo *pColFilter = &pCol->filterInfo[f];
H
hzcheng 已提交
701

S
slguan 已提交
702
      SColumnFilterInfo *pFilterMsg = (SColumnFilterInfo *)pMsg;
703
      pFilterMsg->filterstr = htons(pColFilter->filterstr);
S
slguan 已提交
704 705

      pMsg += sizeof(SColumnFilterInfo);
H
hzcheng 已提交
706

707
      if (pColFilter->filterstr) {
S
slguan 已提交
708 709 710 711 712 713 714 715 716 717
        pFilterMsg->len = htobe64(pColFilter->len);
        memcpy(pMsg, (void *)pColFilter->pz, pColFilter->len + 1);
        pMsg += (pColFilter->len + 1);  // append the additional filter binary info
      } else {
        pFilterMsg->lowerBndi = htobe64(pColFilter->lowerBndi);
        pFilterMsg->upperBndi = htobe64(pColFilter->upperBndi);
      }

      pFilterMsg->lowerRelOptr = htons(pColFilter->lowerRelOptr);
      pFilterMsg->upperRelOptr = htons(pColFilter->upperRelOptr);
H
hzcheng 已提交
718

S
slguan 已提交
719 720 721 722 723
      if (pColFilter->lowerRelOptr == TSDB_RELATION_INVALID && pColFilter->upperRelOptr == TSDB_RELATION_INVALID) {
        tscError("invalid filter info");
        return -1;
      }
    }
H
hzcheng 已提交
724 725
  }

H
hjxilinx 已提交
726
  SSqlFuncMsg *pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hjxilinx 已提交
727
  for (int32_t i = 0; i < tscSqlExprNumOfExprs(pQueryInfo); ++i) {
728
    SSqlExpr *pExpr = tscSqlExprGet(pQueryInfo, i);
H
hzcheng 已提交
729

H
hjxilinx 已提交
730
    if (!tscValidateColumnId(pTableMetaInfo, pExpr->colInfo.colId)) {
731
      /* column id is not valid according to the cached table meta, the table meta is expired */
H
hzcheng 已提交
732 733 734 735
      tscError("%p table schema is not matched with parsed sql", pSql);
      return -1;
    }

736 737 738
    pSqlFuncExpr->colInfo.colId    = htons(pExpr->colInfo.colId);
    pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
    pSqlFuncExpr->colInfo.flag     = htons(pExpr->colInfo.flag);
H
hzcheng 已提交
739

740
    pSqlFuncExpr->functionId  = htons(pExpr->functionId);
H
hzcheng 已提交
741
    pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
H
hjxilinx 已提交
742
    pMsg += sizeof(SSqlFuncMsg);
H
hzcheng 已提交
743 744 745 746 747 748 749

    for (int32_t j = 0; j < pExpr->numOfParams; ++j) {
      pSqlFuncExpr->arg[j].argType = htons((uint16_t)pExpr->param[j].nType);
      pSqlFuncExpr->arg[j].argBytes = htons(pExpr->param[j].nLen);

      if (pExpr->param[j].nType == TSDB_DATA_TYPE_BINARY) {
        memcpy(pMsg, pExpr->param[j].pz, pExpr->param[j].nLen);
750
        pMsg += pExpr->param[j].nLen;
H
hzcheng 已提交
751 752 753 754 755
      } else {
        pSqlFuncExpr->arg[j].argValue.i64 = htobe64(pExpr->param[j].i64Key);
      }
    }

H
hjxilinx 已提交
756
    pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hzcheng 已提交
757
  }
758
  
759
  // serialize the table info (sid, uid, tags)
760 761
  pMsg = doSerializeTableInfo(pQueryMsg, pSql, pMsg);
  
762
  SSqlGroupbyExpr *pGroupbyExpr = &pQueryInfo->groupbyExpr;
763
  if (pGroupbyExpr->numOfGroupCols > 0) {
S
slguan 已提交
764
    pQueryMsg->orderByIdx = htons(pGroupbyExpr->orderIndex);
H
hzcheng 已提交
765 766
    pQueryMsg->orderType = htons(pGroupbyExpr->orderType);

S
slguan 已提交
767
    for (int32_t j = 0; j < pGroupbyExpr->numOfGroupCols; ++j) {
768 769
      SColIndex* pCol = taosArrayGet(pGroupbyExpr->columnInfo, j);
  
S
slguan 已提交
770 771 772
      *((int16_t *)pMsg) = pCol->colId;
      pMsg += sizeof(pCol->colId);

773 774
      *((int16_t *)pMsg) += pCol->colIndex;
      pMsg += sizeof(pCol->colIndex);
S
slguan 已提交
775 776 777

      *((int16_t *)pMsg) += pCol->flag;
      pMsg += sizeof(pCol->flag);
H
hjxilinx 已提交
778 779 780
      
      memcpy(pMsg, pCol->name, tListLen(pCol->name));
      pMsg += tListLen(pCol->name);
S
slguan 已提交
781 782 783
    }
  }

784
  if (pQueryInfo->fillType != TSDB_FILL_NONE) {
H
hjxilinx 已提交
785
    for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
786 787
      *((int64_t *)pMsg) = htobe64(pQueryInfo->fillVal[i]);
      pMsg += sizeof(pQueryInfo->fillVal[0]);
S
slguan 已提交
788 789
    }
  }
790 791 792 793 794 795 796 797 798
  
  if (numOfTags != 0) {
    int32_t numOfColumns = tscGetNumOfColumns(pTableMeta);
    int32_t numOfTagColumns = tscGetNumOfTags(pTableMeta);
    int32_t total = numOfTagColumns + numOfColumns;
    
    pSchema = tscGetTableTagSchema(pTableMeta);
    
    for (int32_t i = 0; i < numOfTags; ++i) {
H
hjxilinx 已提交
799
      SColumn *pCol = taosArrayGetP(pTableMetaInfo->tagColList, i);
800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
      SSchema *pColSchema = &pSchema[pCol->colIndex.columnIndex];

      if ((pCol->colIndex.columnIndex >= numOfTagColumns || pCol->colIndex.columnIndex < -1) ||
          (pColSchema->type < TSDB_DATA_TYPE_BOOL || pColSchema->type > TSDB_DATA_TYPE_NCHAR)) {
        tscError("%p sid:%d uid:%" PRIu64 " id:%s, tag index out of range, totalCols:%d, numOfTags:%d, index:%d, column name:%s",
                 pSql, pTableMeta->sid, pTableMeta->uid, pTableMetaInfo->name, total, numOfTagColumns,
                 pCol->colIndex, pColSchema->name);

        return TSDB_CODE_INVALID_SQL;
      }
  
      SColumnInfo* pTagCol = (SColumnInfo*) pMsg;
  
      pTagCol->colId = htons(pColSchema->colId);
      pTagCol->bytes = htons(pColSchema->bytes);
      pTagCol->type  = htons(pColSchema->type);
      pTagCol->numOfFilters = 0;
      
      pMsg += sizeof(SColumnInfo);
    }
  }
S
slguan 已提交
821 822 823 824 825 826

  // compressed ts block
  pQueryMsg->tsOffset = htonl(pMsg - pStart);
  int32_t tsLen = 0;
  int32_t numOfBlocks = 0;

827
  if (pQueryInfo->tsBuf != NULL) {
H
hjxilinx 已提交
828
    STSVnodeBlockInfo *pBlockInfo = tsBufGetVnodeBlockInfo(pQueryInfo->tsBuf, pTableMetaInfo->vgroupIndex);
829
    assert(QUERY_IS_JOIN_QUERY(pQueryInfo->type) && pBlockInfo != NULL);  // this query should not be sent
S
slguan 已提交
830 831

    // todo refactor
832 833
    fseek(pQueryInfo->tsBuf->f, pBlockInfo->offset, SEEK_SET);
    fread(pMsg, pBlockInfo->compLen, 1, pQueryInfo->tsBuf->f);
S
slguan 已提交
834 835 836 837

    pMsg += pBlockInfo->compLen;
    tsLen = pBlockInfo->compLen;
    numOfBlocks = pBlockInfo->numOfBlocks;
H
hzcheng 已提交
838 839
  }

S
slguan 已提交
840 841
  pQueryMsg->tsLen = htonl(tsLen);
  pQueryMsg->tsNumOfBlocks = htonl(numOfBlocks);
842 843
  if (pQueryInfo->tsBuf != NULL) {
    pQueryMsg->tsOrder = htonl(pQueryInfo->tsBuf->tsOrder);
H
hzcheng 已提交
844 845
  }

H
hjxilinx 已提交
846
  // serialize tag column query condition
847
  if (pQueryInfo->tagCond.pCond != NULL && taosArrayGetSize(pQueryInfo->tagCond.pCond) > 0) {
848 849
    STagCond* pTagCond = &pQueryInfo->tagCond;
    
850
    SCond *pCond = tsGetSTableQueryCond(pTagCond, pTableMeta->uid);
851
    if (pCond != NULL && pCond->cond != NULL) {
852 853
      pQueryMsg->tagCondLen = htons(pCond->len);
      memcpy(pMsg, pCond->cond, pCond->len);
854
      
855
      pMsg += pCond->len;
856 857 858
    }
  }
  
weixin_48148422's avatar
weixin_48148422 已提交
859 860 861 862 863 864 865 866
  if (pQueryInfo->tagCond.tbnameCond.cond == NULL) {
    *pMsg = 0;
    pMsg++;
  } else {
    strcpy(pMsg, pQueryInfo->tagCond.tbnameCond.cond);
    pMsg += strlen(pQueryInfo->tagCond.tbnameCond.cond) + 1;
  }

867
  int32_t msgLen = pMsg - pStart;
H
hzcheng 已提交
868 869 870

  tscTrace("%p msg built success,len:%d bytes", pSql, msgLen);
  pCmd->payloadLen = msgLen;
S
slguan 已提交
871
  pSql->cmd.msgType = TSDB_MSG_TYPE_QUERY;
H
hjxilinx 已提交
872
  
873
  pQueryMsg->head.contLen = htonl(msgLen);
H
hzcheng 已提交
874
  assert(msgLen + minMsgSize() <= size);
875 876

  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
877 878
}

879 880
int32_t tscBuildCreateDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
881
  pCmd->payloadLen = sizeof(SCMCreateDbMsg);
S
slguan 已提交
882
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DB;
H
hzcheng 已提交
883

884
  SCMCreateDbMsg *pCreateDbMsg = (SCMCreateDbMsg*)pCmd->payload;
885

886
  assert(pCmd->numOfClause == 1);
887
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
888
  strncpy(pCreateDbMsg->db, pTableMetaInfo->name, tListLen(pCreateDbMsg->db));
H
hzcheng 已提交
889

890
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
891 892
}

893 894
int32_t tscBuildCreateDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
895
  pCmd->payloadLen = sizeof(SCMCreateDnodeMsg);
S
slguan 已提交
896 897 898 899
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
900

901
  SCMCreateDnodeMsg *pCreate = (SCMCreateDnodeMsg *)pCmd->payload;
J
jtao1735 已提交
902 903
  strncpy(pCreate->ep, pInfo->pDCLInfo->a[0].z, pInfo->pDCLInfo->a[0].n);
  
S
slguan 已提交
904
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DNODE;
H
hzcheng 已提交
905

906
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
907 908
}

909 910
int32_t tscBuildAcctMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
911
  pCmd->payloadLen = sizeof(SCMCreateAcctMsg);
S
slguan 已提交
912 913 914 915
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
916

917
  SCMCreateAcctMsg *pAlterMsg = (SCMCreateAcctMsg *)pCmd->payload;
H
hzcheng 已提交
918

919 920
  SSQLToken *pName = &pInfo->pDCLInfo->user.user;
  SSQLToken *pPwd = &pInfo->pDCLInfo->user.passwd;
H
hzcheng 已提交
921

922 923
  strncpy(pAlterMsg->user, pName->z, pName->n);
  strncpy(pAlterMsg->pass, pPwd->z, pPwd->n);
S
slguan 已提交
924

925
  SCreateAcctSQL *pAcctOpt = &pInfo->pDCLInfo->acctOpt;
H
hzcheng 已提交
926

927 928 929 930 931 932 933 934
  pAlterMsg->cfg.maxUsers = htonl(pAcctOpt->maxUsers);
  pAlterMsg->cfg.maxDbs = htonl(pAcctOpt->maxDbs);
  pAlterMsg->cfg.maxTimeSeries = htonl(pAcctOpt->maxTimeSeries);
  pAlterMsg->cfg.maxStreams = htonl(pAcctOpt->maxStreams);
  pAlterMsg->cfg.maxPointsPerSecond = htonl(pAcctOpt->maxPointsPerSecond);
  pAlterMsg->cfg.maxStorage = htobe64(pAcctOpt->maxStorage);
  pAlterMsg->cfg.maxQueryTime = htobe64(pAcctOpt->maxQueryTime);
  pAlterMsg->cfg.maxConnections = htonl(pAcctOpt->maxConnections);
H
hzcheng 已提交
935

936 937 938 939 940 941 942 943 944 945 946 947 948
  if (pAcctOpt->stat.n == 0) {
    pAlterMsg->cfg.accessState = -1;
  } else {
    if (pAcctOpt->stat.z[0] == 'r' && pAcctOpt->stat.n == 1) {
      pAlterMsg->cfg.accessState = TSDB_VN_READ_ACCCESS;
    } else if (pAcctOpt->stat.z[0] == 'w' && pAcctOpt->stat.n == 1) {
      pAlterMsg->cfg.accessState = TSDB_VN_WRITE_ACCCESS;
    } else if (strncmp(pAcctOpt->stat.z, "all", 3) == 0 && pAcctOpt->stat.n == 3) {
      pAlterMsg->cfg.accessState = TSDB_VN_ALL_ACCCESS;
    } else if (strncmp(pAcctOpt->stat.z, "no", 2) == 0 && pAcctOpt->stat.n == 2) {
      pAlterMsg->cfg.accessState = 0;
    }
  }
H
hzcheng 已提交
949

S
slguan 已提交
950
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_ACCT;
951
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
952 953
}

954 955
int32_t tscBuildUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
956
  pCmd->payloadLen = sizeof(SCMCreateUserMsg);
S
slguan 已提交
957

S
slguan 已提交
958 959 960 961 962
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

963
  SCMCreateUserMsg *pAlterMsg = (SCMCreateUserMsg*)pCmd->payload;
H
hzcheng 已提交
964

965 966 967
  SUserInfo *pUser = &pInfo->pDCLInfo->user;
  strncpy(pAlterMsg->user, pUser->user.z, pUser->user.n);
  pAlterMsg->flag = pUser->type;
H
hzcheng 已提交
968

969 970 971 972
  if (pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
    pAlterMsg->privilege = (char)pCmd->count;
  } else if (pUser->type == TSDB_ALTER_USER_PASSWD) {
    strncpy(pAlterMsg->pass, pUser->passwd.z, pUser->passwd.n);
H
hjxilinx 已提交
973 974
  } else { // create user password info
    strncpy(pAlterMsg->pass, pUser->passwd.z, pUser->passwd.n);
975
  }
H
hzcheng 已提交
976

977
  if (pUser->type == TSDB_ALTER_USER_PASSWD || pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
S
slguan 已提交
978
    pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_USER;
979
  } else {
S
slguan 已提交
980
    pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_USER;
981
  }
H
hzcheng 已提交
982

983
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
984 985
}

986 987
int32_t tscBuildCfgDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
988
  pCmd->payloadLen = sizeof(SCMCfgDnodeMsg);
guanshengliang's avatar
guanshengliang 已提交
989
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONFIG_DNODE;
990 991
  return TSDB_CODE_SUCCESS;
}
S
slguan 已提交
992

993 994
int32_t tscBuildDropDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
995
  pCmd->payloadLen = sizeof(SCMDropDbMsg);
H
hzcheng 已提交
996

S
slguan 已提交
997 998 999 1000 1001
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

1002
  SCMDropDbMsg *pDropDbMsg = (SCMDropDbMsg*)pCmd->payload;
H
hzcheng 已提交
1003

1004
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1005
  strncpy(pDropDbMsg->db, pTableMetaInfo->name, tListLen(pDropDbMsg->db));
1006
  pDropDbMsg->ignoreNotExists = pInfo->pDCLInfo->existsCheck ? 1 : 0;
H
hzcheng 已提交
1007

S
slguan 已提交
1008
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DB;
1009
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1010 1011
}

1012 1013
int32_t tscBuildDropTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1014
  pCmd->payloadLen = sizeof(SCMDropTableMsg);
H
hzcheng 已提交
1015

S
slguan 已提交
1016 1017 1018 1019
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
1020

1021
  SCMDropTableMsg *pDropTableMsg = (SCMDropTableMsg*)pCmd->payload;
1022
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1023
  strcpy(pDropTableMsg->tableId, pTableMetaInfo->name);
S
slguan 已提交
1024
  pDropTableMsg->igNotExists = pInfo->pDCLInfo->existsCheck ? 1 : 0;
H
hzcheng 已提交
1025

S
slguan 已提交
1026
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_TABLE;
1027
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1028 1029
}

1030
int32_t tscBuildDropDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1031
  SSqlCmd *pCmd = &pSql->cmd;
1032
  pCmd->payloadLen = sizeof(SCMDropDnodeMsg);
S
slguan 已提交
1033 1034 1035 1036
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
1037

1038
  SCMDropDnodeMsg *pDrop = (SCMDropDnodeMsg *)pCmd->payload;
1039
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
J
jtao1735 已提交
1040
  strcpy(pDrop->ep, pTableMetaInfo->name);
S
slguan 已提交
1041
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DNODE;
H
hzcheng 已提交
1042

1043
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1044 1045
}

S
[TD-16]  
slguan 已提交
1046
int32_t tscBuildDropUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1047
  SSqlCmd *pCmd = &pSql->cmd;
1048
  pCmd->payloadLen = sizeof(SCMDropUserMsg);
S
slguan 已提交
1049
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_USER;
H
hzcheng 已提交
1050

S
slguan 已提交
1051 1052 1053 1054
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
1055

1056
  SCMDropUserMsg *pDropMsg = (SCMDropUserMsg*)pCmd->payload;
1057
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1058
  strcpy(pDropMsg->user, pTableMetaInfo->name);
H
hzcheng 已提交
1059

1060
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1061 1062
}

S
[TD-16]  
slguan 已提交
1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079
int32_t tscBuildDropAcctMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
  pCmd->payloadLen = sizeof(SCMDropUserMsg);
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_ACCT;

  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

  SCMDropUserMsg *pDropMsg = (SCMDropUserMsg*)pCmd->payload;
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
  strcpy(pDropMsg->user, pTableMetaInfo->name);

  return TSDB_CODE_SUCCESS;
}

1080 1081
int32_t tscBuildUseDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1082
  pCmd->payloadLen = sizeof(SCMUseDbMsg);
H
hzcheng 已提交
1083

S
slguan 已提交
1084 1085 1086 1087
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
1088

1089
  SCMUseDbMsg *pUseDbMsg = (SCMUseDbMsg*)pCmd->payload;
1090
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1091
  strcpy(pUseDbMsg->db, pTableMetaInfo->name);
S
slguan 已提交
1092
  pCmd->msgType = TSDB_MSG_TYPE_CM_USE_DB;
H
hzcheng 已提交
1093

1094
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1095 1096
}

1097
int32_t tscBuildShowMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1098
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1099
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1100
  pCmd->msgType = TSDB_MSG_TYPE_CM_SHOW;
1101
  pCmd->payloadLen = sizeof(SCMShowMsg) + 100;
H
hzcheng 已提交
1102

S
slguan 已提交
1103 1104 1105
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
S
slguan 已提交
1106
  }
H
hzcheng 已提交
1107

1108
  SCMShowMsg *pShowMsg = (SCMShowMsg*)pCmd->payload;
S
slguan 已提交
1109

1110
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1111
  size_t nameLen = strlen(pTableMetaInfo->name);
S
slguan 已提交
1112
  if (nameLen > 0) {
H
hjxilinx 已提交
1113
    strcpy(pShowMsg->db, pTableMetaInfo->name);  // prefix is set here
H
hzcheng 已提交
1114
  } else {
S
slguan 已提交
1115
    strcpy(pShowMsg->db, pObj->db);
H
hzcheng 已提交
1116 1117
  }

1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
  SShowInfo *pShowInfo = &pInfo->pDCLInfo->showOpt;
  pShowMsg->type = pShowInfo->showType;

  if (pShowInfo->showType != TSDB_MGMT_TABLE_VNODES) {
    SSQLToken *pPattern = &pShowInfo->pattern;
    if (pPattern->type > 0) {  // only show tables support wildcard query
      strncpy(pShowMsg->payload, pPattern->z, pPattern->n);
      pShowMsg->payloadLen = htons(pPattern->n);
    }
  } else {
    SSQLToken *pIpAddr = &pShowInfo->prefix;
    assert(pIpAddr->n > 0 && pIpAddr->type > 0);
H
hzcheng 已提交
1130

1131 1132 1133 1134
    strncpy(pShowMsg->payload, pIpAddr->z, pIpAddr->n);
    pShowMsg->payloadLen = htons(pIpAddr->n);
  }

1135
  pCmd->payloadLen = sizeof(SCMShowMsg) + pShowMsg->payloadLen;
1136
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1137 1138
}

1139
int32_t tscBuildKillMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1140
  SSqlCmd *pCmd = &pSql->cmd;
1141
  pCmd->payloadLen = sizeof(SCMKillQueryMsg);
H
hzcheng 已提交
1142

S
slguan 已提交
1143 1144 1145 1146
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
1147

1148
  SCMKillQueryMsg *pKill = (SCMKillQueryMsg*)pCmd->payload;
1149 1150 1151
  strncpy(pKill->queryId, pInfo->pDCLInfo->ip.z, pInfo->pDCLInfo->ip.n);
  switch (pCmd->command) {
    case TSDB_SQL_KILL_QUERY:
S
slguan 已提交
1152
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_QUERY;
1153 1154
      break;
    case TSDB_SQL_KILL_CONNECTION:
S
slguan 已提交
1155
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_CONN;
1156 1157
      break;
    case TSDB_SQL_KILL_STREAM:
S
slguan 已提交
1158
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_STREAM;
1159 1160 1161
      break;
  }
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1162 1163
}

1164
int tscEstimateCreateTableMsgLength(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1165 1166
  SSqlCmd *pCmd = &(pSql->cmd);

guanshengliang's avatar
guanshengliang 已提交
1167
  int32_t size = minMsgSize() + sizeof(SCMCreateTableMsg);
H
hzcheng 已提交
1168

1169
  SCreateTableSQL *pCreateTableInfo = pInfo->pCreateTableInfo;
1170
  if (pCreateTableInfo->type == TSQL_CREATE_TABLE_FROM_STABLE) {
H
hzcheng 已提交
1171 1172
    size += sizeof(STagData);
  } else {
S
slguan 已提交
1173
    size += sizeof(SSchema) * (pCmd->numOfCols + pCmd->count);
H
hzcheng 已提交
1174
  }
1175

1176 1177 1178
  if (pCreateTableInfo->pSelect != NULL) {
    size += (pCreateTableInfo->pSelect->selectToken.n + 1);
  }
H
hzcheng 已提交
1179 1180 1181 1182

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1183
int tscBuildCreateTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1184
  int              msgLen = 0;
S
slguan 已提交
1185
  SSchema *        pSchema;
H
hzcheng 已提交
1186
  int              size = 0;
1187 1188 1189
  SSqlCmd *pCmd = &pSql->cmd;

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
1190
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1191 1192

  // Reallocate the payload size
1193
  size = tscEstimateCreateTableMsgLength(pSql, pInfo);
S
slguan 已提交
1194 1195
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for create table msg", pSql);
1196
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
S
slguan 已提交
1197
  }
H
hzcheng 已提交
1198 1199


1200
  SCMCreateTableMsg *pCreateTableMsg = (SCMCreateTableMsg *)pCmd->payload;
H
hjxilinx 已提交
1201
  strcpy(pCreateTableMsg->tableId, pTableMetaInfo->name);
1202 1203

  // use dbinfo from table id without modifying current db info
H
hjxilinx 已提交
1204
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pCreateTableMsg->db);
H
hzcheng 已提交
1205

1206 1207 1208
  SCreateTableSQL *pCreateTable = pInfo->pCreateTableInfo;

  pCreateTableMsg->igExists = pCreateTable->existCheck ? 1 : 0;
H
hzcheng 已提交
1209 1210 1211 1212
  pCreateTableMsg->numOfColumns = htons(pCmd->numOfCols);
  pCreateTableMsg->numOfTags = htons(pCmd->count);

  pCreateTableMsg->sqlLen = 0;
S
slguan 已提交
1213
  char *pMsg = (char *)pCreateTableMsg->schema;
H
hzcheng 已提交
1214

1215 1216
  int8_t type = pInfo->pCreateTableInfo->type;
  if (type == TSQL_CREATE_TABLE_FROM_STABLE) {  // create by using super table, tags value
1217 1218 1219 1220 1221 1222 1223
    STagData* pTag = &pInfo->pCreateTableInfo->usingInfo.tagdata;
    *(int32_t*)pMsg = htonl(pTag->dataLen);
    pMsg += sizeof(int32_t);
    memcpy(pMsg, pTag->name, sizeof(pTag->name));
    pMsg += sizeof(pTag->name);
    memcpy(pMsg, pTag->data, pTag->dataLen);
    pMsg += pTag->dataLen;
1224
  } else {  // create (super) table
1225
    pSchema = (SSchema *)pCreateTableMsg->schema;
1226

H
hzcheng 已提交
1227
    for (int i = 0; i < pCmd->numOfCols + pCmd->count; ++i) {
H
hjxilinx 已提交
1228
      TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1229 1230 1231 1232

      pSchema->type = pField->type;
      strcpy(pSchema->name, pField->name);
      pSchema->bytes = htons(pField->bytes);
1233

H
hzcheng 已提交
1234 1235 1236 1237
      pSchema++;
    }

    pMsg = (char *)pSchema;
1238 1239
    if (type == TSQL_CREATE_STREAM) {  // check if it is a stream sql
      SQuerySQL *pQuerySql = pInfo->pCreateTableInfo->pSelect;
H
hzcheng 已提交
1240

1241 1242 1243
      strncpy(pMsg, pQuerySql->selectToken.z, pQuerySql->selectToken.n + 1);
      pCreateTableMsg->sqlLen = htons(pQuerySql->selectToken.n + 1);
      pMsg += pQuerySql->selectToken.n + 1;
H
hzcheng 已提交
1244 1245 1246
    }
  }

H
hjxilinx 已提交
1247
  tscFieldInfoClear(&pQueryInfo->fieldsInfo);
H
hzcheng 已提交
1248

S
slguan 已提交
1249
  msgLen = pMsg - (char*)pCreateTableMsg;
S
slguan 已提交
1250
  pCreateTableMsg->contLen = htonl(msgLen);
H
hzcheng 已提交
1251
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1252
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_TABLE;
H
hzcheng 已提交
1253 1254

  assert(msgLen + minMsgSize() <= size);
1255
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1256 1257 1258
}

int tscEstimateAlterTableMsgLength(SSqlCmd *pCmd) {
1259
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
guanshengliang's avatar
guanshengliang 已提交
1260
  return minMsgSize() + sizeof(SCMAlterTableMsg) + sizeof(SSchema) * tscNumOfFields(pQueryInfo) +
H
hzcheng 已提交
1261 1262 1263
         TSDB_EXTRA_PAYLOAD_SIZE;
}

1264
int tscBuildAlterTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1265
  SCMAlterTableMsg *pAlterTableMsg;
1266
  char *          pMsg;
H
hzcheng 已提交
1267 1268 1269
  int             msgLen = 0;
  int             size = 0;

1270 1271 1272
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);

H
hjxilinx 已提交
1273
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1274 1275

  size = tscEstimateAlterTableMsgLength(pCmd);
S
slguan 已提交
1276 1277 1278 1279
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for alter table msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1280

1281
  pAlterTableMsg = (SCMAlterTableMsg *)pCmd->payload;
H
hzcheng 已提交
1282

H
hjxilinx 已提交
1283
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pAlterTableMsg->db);
H
hzcheng 已提交
1284

1285 1286
  SAlterTableSQL *pAlterInfo = pInfo->pAlterInfo;

H
hjxilinx 已提交
1287
  strcpy(pAlterTableMsg->tableId, pTableMetaInfo->name);
1288
  pAlterTableMsg->type = htons(pAlterInfo->type);
1289

1290
  pAlterTableMsg->numOfCols = htons(tscNumOfFields(pQueryInfo));
S
slguan 已提交
1291
  SSchema *pSchema = pAlterTableMsg->schema;
S
slguan 已提交
1292
  for (int i = 0; i < pAlterTableMsg->numOfCols; ++i) {
H
hjxilinx 已提交
1293
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1294 1295 1296 1297 1298 1299 1300 1301

    pSchema->type = pField->type;
    strcpy(pSchema->name, pField->name);
    pSchema->bytes = htons(pField->bytes);
    pSchema++;
  }

  pMsg = (char *)pSchema;
1302 1303 1304
  pAlterTableMsg->tagValLen = htonl(pAlterInfo->tagData.dataLen);
  memcpy(pMsg, pAlterInfo->tagData.data, pAlterInfo->tagData.dataLen);
  pMsg += pAlterInfo->tagData.dataLen;
H
hzcheng 已提交
1305

S
slguan 已提交
1306
  msgLen = pMsg - (char*)pAlterTableMsg;
H
hzcheng 已提交
1307
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1308
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_TABLE;
H
hzcheng 已提交
1309 1310

  assert(msgLen + minMsgSize() <= size);
1311

1312
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1313 1314
}

1315
int tscAlterDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1316
  SSqlCmd *pCmd = &pSql->cmd;
1317
  pCmd->payloadLen = sizeof(SCMAlterDbMsg);
S
slguan 已提交
1318
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_DB;
H
hzcheng 已提交
1319

1320
  SCMAlterDbMsg *pAlterDbMsg = (SCMAlterDbMsg*)pCmd->payload;
1321
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1322
  strcpy(pAlterDbMsg->db, pTableMetaInfo->name);
H
hzcheng 已提交
1323

1324
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1325 1326
}

1327
int tscBuildRetrieveFromMgmtMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1328
  SSqlCmd *pCmd = &pSql->cmd;
J
jtao1735 已提交
1329
  pCmd->msgType = TSDB_MSG_TYPE_CM_RETRIEVE;
S
slguan 已提交
1330
  pCmd->payloadLen = sizeof(SRetrieveTableMsg);
S
slguan 已提交
1331

S
slguan 已提交
1332 1333 1334
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
H
hzcheng 已提交
1335
  }
S
slguan 已提交
1336

S
slguan 已提交
1337 1338 1339 1340
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg*)pCmd->payload;
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
  pRetrieveMsg->free = htons(pQueryInfo->type);
H
hzcheng 已提交
1341

1342
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1343 1344
}

1345
static int tscSetResultPointer(SQueryInfo *pQueryInfo, SSqlRes *pRes) {
1346
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
H
hzcheng 已提交
1347 1348 1349
    return pRes->code;
  }

H
hjxilinx 已提交
1350
  for (int i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
H
hjxilinx 已提交
1351
    int16_t offset = tscFieldInfoGetOffset(pQueryInfo, i);
H
hjxilinx 已提交
1352
    pRes->tsrow[i] = ((char*) pRes->data + offset * pRes->numOfRows);
H
hzcheng 已提交
1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366
  }

  return 0;
}

/*
 * this function can only be called once.
 * by using pRes->rspType to denote its status
 *
 * if pRes->rspType is 1, no more result
 */
static int tscLocalResultCommonBuilder(SSqlObj *pSql, int32_t numOfRes) {
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;
1367

1368
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
1369

H
hzcheng 已提交
1370 1371 1372 1373 1374 1375 1376
  pRes->code = TSDB_CODE_SUCCESS;

  if (pRes->rspType == 0) {
    pRes->numOfRows = numOfRes;
    pRes->row = 0;
    pRes->rspType = 1;

1377
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1378
  } else {
S
slguan 已提交
1379
    tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
  }

  uint8_t code = pSql->res.code;
  if (pSql->fp) {
    if (code == TSDB_CODE_SUCCESS) {
      (*pSql->fp)(pSql->param, pSql, pSql->res.numOfRows);
    } else {
      tscQueueAsyncRes(pSql);
    }
  }

  return code;
}

int tscProcessDescribeTableRsp(SSqlObj *pSql) {
S
slguan 已提交
1395
  SSqlCmd *       pCmd = &pSql->cmd;
1396
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hzcheng 已提交
1397

H
hjxilinx 已提交
1398
  STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
H
hjxilinx 已提交
1399 1400
  
  int32_t numOfRes = tinfo.numOfColumns + tinfo.numOfTags;
H
hzcheng 已提交
1401 1402 1403 1404
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

int tscProcessTagRetrieveRsp(SSqlObj *pSql) {
1405
//  SSqlCmd *pCmd = &pSql->cmd;
1406

1407 1408
//  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
//  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
S
slguan 已提交
1409 1410

  int32_t numOfRes = 0;
H
hjxilinx 已提交
1411
#if 0
1412
  if (tscSqlExprGet(pQueryInfo, 0)->functionId == TSDB_FUNC_TAGPRJ) {
H
hjxilinx 已提交
1413
    numOfRes = pTableMetaInfo->pMetricMeta->numOfTables;
H
hzcheng 已提交
1414 1415 1416
  } else {
    numOfRes = 1;  // for count function, there is only one output.
  }
H
hjxilinx 已提交
1417 1418 1419
  
#endif

H
hzcheng 已提交
1420 1421 1422
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

H
hjxilinx 已提交
1423
int tscProcessRetrieveLocalMergeRsp(SSqlObj *pSql) {
H
hzcheng 已提交
1424 1425 1426
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

1427
  pRes->code = tscDoLocalMerge(pSql);
H
hjxilinx 已提交
1428
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hzcheng 已提交
1429 1430

  if (pRes->code == TSDB_CODE_SUCCESS && pRes->numOfRows > 0) {
1431
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1432 1433 1434
  }

  pRes->row = 0;
1435
  pRes->completed = (pRes->numOfRows == 0);
H
hzcheng 已提交
1436

1437
  int32_t code = pRes->code;
H
hjxilinx 已提交
1438 1439 1440 1441
  if (pRes->code == TSDB_CODE_SUCCESS) {
    (*pSql->fp)(pSql->param, pSql, pRes->numOfRows);
  } else {
    tscQueueAsyncRes(pSql);
H
hzcheng 已提交
1442 1443 1444 1445 1446
  }

  return code;
}

S
slguan 已提交
1447
int tscProcessEmptyResultRsp(SSqlObj *pSql) { return tscLocalResultCommonBuilder(pSql, 0); }
H
hzcheng 已提交
1448

1449
int tscBuildConnectMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1450
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1451
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1452
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONNECT;
1453
  pCmd->payloadLen = sizeof(SCMConnectMsg);
H
hzcheng 已提交
1454

S
slguan 已提交
1455 1456 1457 1458 1459
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

1460
  SCMConnectMsg *pConnect = (SCMConnectMsg*)pCmd->payload;
H
hzcheng 已提交
1461 1462 1463 1464 1465

  char *db;  // ugly code to move the space
  db = strstr(pObj->db, TS_PATH_DELIMITER);
  db = (db == NULL) ? pObj->db : db + 1;
  strcpy(pConnect->db, db);
S
slguan 已提交
1466
  strcpy(pConnect->clientVersion, version);
S
slguan 已提交
1467
  strcpy(pConnect->msgVersion, "");
H
hzcheng 已提交
1468

1469
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1470 1471
}

H
hjxilinx 已提交
1472
int tscBuildTableMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1473
  SCMTableInfoMsg *pInfoMsg;
S
slguan 已提交
1474
  char *         pMsg;
H
hzcheng 已提交
1475 1476
  int            msgLen = 0;

B
Bomin Zhang 已提交
1477 1478 1479 1480
  char *tmpData = NULL;
  uint32_t len = pSql->cmd.payloadLen;
  if (len > 0) {
    tmpData = calloc(1, len);
1481 1482 1483 1484
    if (NULL == tmpData) {
      return TSDB_CODE_CLI_OUT_OF_MEMORY;
    }

H
hzcheng 已提交
1485
    // STagData is in binary format, strncpy is not available
B
Bomin Zhang 已提交
1486
    memcpy(tmpData, pSql->cmd.payload, len);
H
hzcheng 已提交
1487 1488
  }

1489 1490 1491
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

H
hjxilinx 已提交
1492
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
S
slguan 已提交
1493

1494
  pInfoMsg = (SCMTableInfoMsg *)pCmd->payload;
H
hjxilinx 已提交
1495
  strcpy(pInfoMsg->tableId, pTableMetaInfo->name);
H
hjxilinx 已提交
1496
  pInfoMsg->createFlag = htons(pSql->cmd.autoCreated ? 1 : 0);
S
slguan 已提交
1497

1498
  pMsg = (char*)pInfoMsg + sizeof(SCMTableInfoMsg);
H
hzcheng 已提交
1499

B
Bomin Zhang 已提交
1500 1501 1502
  if (pSql->cmd.autoCreated && len > 0) {
    memcpy(pInfoMsg->tags, tmpData, len);
    pMsg += len;
H
hzcheng 已提交
1503 1504
  }

H
hjxilinx 已提交
1505
  pCmd->payloadLen = pMsg - (char*)pInfoMsg;;
S
slguan 已提交
1506
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLE_META;
H
hzcheng 已提交
1507 1508 1509 1510

  tfree(tmpData);

  assert(msgLen + minMsgSize() <= pCmd->allocSize);
1511
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1512 1513
}

S
slguan 已提交
1514
/**
1515
 *  multi table meta req pkg format:
1516
 *  | SMgmtHead | SCMMultiTableInfoMsg | tableId0 | tableId1 | tableId2 | ......
S
slguan 已提交
1517 1518
 *      no used         4B
 **/
1519
int tscBuildMultiMeterMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
guanshengliang's avatar
guanshengliang 已提交
1520
#if 0
S
slguan 已提交
1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
  SSqlCmd *pCmd = &pSql->cmd;

  // copy payload content to temp buff
  char *tmpData = 0;
  if (pCmd->payloadLen > 0) {
    tmpData = calloc(1, pCmd->payloadLen + 1);
    if (NULL == tmpData) return -1;
    memcpy(tmpData, pCmd->payload, pCmd->payloadLen);
  }

  // fill head info
  SMgmtHead *pMgmt = (SMgmtHead *)(pCmd->payload + tsRpcHeadSize);
S
slguan 已提交
1533
  memset(pMgmt->db, 0, TSDB_TABLE_ID_LEN);  // server don't need the db
S
slguan 已提交
1534

1535
  SCMMultiTableInfoMsg *pInfoMsg = (SCMMultiTableInfoMsg *)(pCmd->payload + tsRpcHeadSize + sizeof(SMgmtHead));
S
slguan 已提交
1536
  pInfoMsg->numOfTables = htonl((int32_t)pCmd->count);
S
slguan 已提交
1537 1538

  if (pCmd->payloadLen > 0) {
S
slguan 已提交
1539
    memcpy(pInfoMsg->tableIds, tmpData, pCmd->payloadLen);
S
slguan 已提交
1540 1541 1542 1543
  }

  tfree(tmpData);

1544
  pCmd->payloadLen += sizeof(SMgmtHead) + sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1545
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLES_META;
S
slguan 已提交
1546 1547 1548

  assert(pCmd->payloadLen + minMsgSize() <= pCmd->allocSize);

S
slguan 已提交
1549
  tscTrace("%p build load multi-metermeta msg completed, numOfTables:%d, msg size:%d", pSql, pCmd->count,
S
slguan 已提交
1550 1551 1552
           pCmd->payloadLen);

  return pCmd->payloadLen;
guanshengliang's avatar
guanshengliang 已提交
1553 1554
#endif
  return 0;  
S
slguan 已提交
1555 1556
}

H
hjxilinx 已提交
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582
//static UNUSED_FUNC int32_t tscEstimateMetricMetaMsgSize(SSqlCmd *pCmd) {
////  const int32_t defaultSize =
////      minMsgSize() + sizeof(SSuperTableMetaMsg) + sizeof(SMgmtHead) + sizeof(int16_t) * TSDB_MAX_TAGS;
////  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
////
////  int32_t n = 0;
////  size_t size = taosArrayGetSize(pQueryInfo->tagCond.pCond);
////  for (int32_t i = 0; i < size; ++i) {
////    assert(0);
//////    n += strlen(pQueryInfo->tagCond.cond[i].cond);
////  }
////
////  int32_t tagLen = n * TSDB_NCHAR_SIZE;
////  if (pQueryInfo->tagCond.tbnameCond.cond != NULL) {
////    tagLen += strlen(pQueryInfo->tagCond.tbnameCond.cond) * TSDB_NCHAR_SIZE;
////  }
////
////  int32_t joinCondLen = (TSDB_TABLE_ID_LEN + sizeof(int16_t)) * 2;
////  int32_t elemSize = sizeof(SSuperTableMetaElemMsg) * pQueryInfo->numOfTables;
////
////  int32_t colSize = pQueryInfo->groupbyExpr.numOfGroupCols*sizeof(SColIndex);
////
////  int32_t len = tagLen + joinCondLen + elemSize + colSize + defaultSize;
////
////  return MAX(len, TSDB_DEFAULT_PAYLOAD_SIZE);
//}
H
hzcheng 已提交
1583

H
hjxilinx 已提交
1584 1585 1586
int tscBuildSTableVgroupMsg(SSqlObj *pSql, SSqlInfo *pInfo) {

#if 0
S
slguan 已提交
1587
  SSuperTableMetaMsg *pMetaMsg;
H
hzcheng 已提交
1588 1589
  char *          pMsg, *pStart;
  int             msgLen = 0;
S
slguan 已提交
1590
  int             tableIndex = 0;
H
hzcheng 已提交
1591

1592 1593 1594
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

1595
  STagCond *pTagCond = &pQueryInfo->tagCond;
S
slguan 已提交
1596

H
hjxilinx 已提交
1597
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
H
hzcheng 已提交
1598 1599

  int32_t size = tscEstimateMetricMetaMsgSize(pCmd);
S
slguan 已提交
1600 1601 1602 1603
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for metric meter msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1604 1605 1606 1607 1608

  pStart = pCmd->payload + tsRpcHeadSize;
  pMsg = pStart;

  SMgmtHead *pMgmt = (SMgmtHead *)pMsg;
H
hjxilinx 已提交
1609
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pMgmt->db);
H
hzcheng 已提交
1610 1611 1612

  pMsg += sizeof(SMgmtHead);

S
slguan 已提交
1613
  pMetaMsg = (SSuperTableMetaMsg *)pMsg;
S
slguan 已提交
1614
  pMetaMsg->numOfTables = htonl(pQueryInfo->numOfTables);
S
slguan 已提交
1615

S
slguan 已提交
1616
  pMsg += sizeof(SSuperTableMetaMsg);
S
slguan 已提交
1617 1618 1619 1620 1621

  int32_t offset = pMsg - (char *)pMetaMsg;
  pMetaMsg->join = htonl(offset);

  // todo refactor
S
slguan 已提交
1622
  pMetaMsg->joinCondLen = htonl((TSDB_TABLE_ID_LEN + sizeof(int16_t)) * 2);
H
hzcheng 已提交
1623

S
slguan 已提交
1624
  memcpy(pMsg, pTagCond->joinInfo.left.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1625
  pMsg += TSDB_TABLE_ID_LEN;
H
hzcheng 已提交
1626

S
slguan 已提交
1627 1628 1629
  *(int16_t *)pMsg = pTagCond->joinInfo.left.tagCol;
  pMsg += sizeof(int16_t);

S
slguan 已提交
1630
  memcpy(pMsg, pTagCond->joinInfo.right.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1631
  pMsg += TSDB_TABLE_ID_LEN;
S
slguan 已提交
1632 1633 1634 1635

  *(int16_t *)pMsg = pTagCond->joinInfo.right.tagCol;
  pMsg += sizeof(int16_t);

1636
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
1637
    pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
H
hjxilinx 已提交
1638
    uint64_t uid = pTableMetaInfo->pTableMeta->uid;
S
slguan 已提交
1639 1640 1641 1642

    offset = pMsg - (char *)pMetaMsg;
    pMetaMsg->metaElem[i] = htonl(offset);

S
slguan 已提交
1643 1644
    SSuperTableMetaElemMsg *pElem = (SSuperTableMetaElemMsg *)pMsg;
    pMsg += sizeof(SSuperTableMetaElemMsg);
S
slguan 已提交
1645 1646 1647 1648

    // convert to unicode before sending to mnode for metric query
    int32_t condLen = 0;
    if (pTagCond->numOfTagCond > 0) {
1649
      SCond *pCond = tsGetSTableQueryCond(pTagCond, uid);
H
hjxilinx 已提交
1650
      if (pCond != NULL && pCond->cond != NULL) {
H
hjxilinx 已提交
1651
        condLen = strlen(pCond->cond) + 1;
1652

H
hjxilinx 已提交
1653
        bool ret = taosMbsToUcs4(pCond->cond, condLen, pMsg, condLen * TSDB_NCHAR_SIZE);
S
slguan 已提交
1654
        if (!ret) {
1655
          tscError("%p mbs to ucs4 failed:%s", pSql, tsGetSTableQueryCond(pTagCond, uid));
S
slguan 已提交
1656 1657 1658
          return 0;
        }
      }
H
hzcheng 已提交
1659 1660
    }

S
slguan 已提交
1661
    pElem->condLen = htonl(condLen);
H
hzcheng 已提交
1662

S
slguan 已提交
1663 1664 1665
    offset = pMsg - (char *)pMetaMsg;
    pElem->cond = htonl(offset);
    pMsg += condLen * TSDB_NCHAR_SIZE;
H
hzcheng 已提交
1666

S
slguan 已提交
1667 1668 1669
    pElem->rel = htons(pTagCond->relType);
    if (pTagCond->tbnameCond.uid == uid) {
      offset = pMsg - (char *)pMetaMsg;
H
hzcheng 已提交
1670

S
slguan 已提交
1671
      pElem->tableCond = htonl(offset);
H
hjxilinx 已提交
1672 1673 1674 1675 1676 1677 1678
      
      uint32_t len = 0;
      if (pTagCond->tbnameCond.cond != NULL) {
        len = strlen(pTagCond->tbnameCond.cond);
        memcpy(pMsg, pTagCond->tbnameCond.cond, len);
      }
      
H
hjxilinx 已提交
1679 1680
      pElem->tableCondLen = htonl(len);
      pMsg += len;
S
slguan 已提交
1681 1682
    }

1683
    SSqlGroupbyExpr *pGroupby = &pQueryInfo->groupbyExpr;
S
slguan 已提交
1684

H
hjxilinx 已提交
1685
    if (pGroupby->tableIndex != i && pGroupby->numOfGroupCols > 0) {
S
slguan 已提交
1686 1687 1688 1689 1690
      pElem->orderType = 0;
      pElem->orderIndex = 0;
      pElem->numOfGroupCols = 0;
    } else {
      pElem->numOfGroupCols = htons(pGroupby->numOfGroupCols);
H
hjxilinx 已提交
1691 1692
      for (int32_t j = 0; j < pTableMetaInfo->numOfTags; ++j) {
        pElem->tagCols[j] = htons(pTableMetaInfo->tagColumnIndex[j]);
S
slguan 已提交
1693 1694 1695 1696 1697 1698 1699 1700
      }

      if (pGroupby->numOfGroupCols != 0) {
        pElem->orderIndex = htons(pGroupby->orderIndex);
        pElem->orderType = htons(pGroupby->orderType);
        offset = pMsg - (char *)pMetaMsg;

        pElem->groupbyTagColumnList = htonl(offset);
1701
        for (int32_t j = 0; j < pQueryInfo->groupbyExpr.numOfGroupCols; ++j) {
1702 1703
          SColIndex *pCol = &pQueryInfo->groupbyExpr.columnInfo[j];
          SColIndex *pDestCol = (SColIndex *)pMsg;
1704

H
hjxilinx 已提交
1705
          pDestCol->colIdxInBuf = 0;
1706
          pDestCol->colIndex = htons(pCol->colIndex);
H
hjxilinx 已提交
1707 1708
          pDestCol->colId = htons(pDestCol->colId);
          pDestCol->flag = htons(pDestCol->flag);
H
hjxilinx 已提交
1709
          strncpy(pDestCol->name, pCol->name, tListLen(pCol->name));
1710

1711
          pMsg += sizeof(SColIndex);
S
slguan 已提交
1712 1713
        }
      }
H
hzcheng 已提交
1714
    }
S
slguan 已提交
1715

H
hjxilinx 已提交
1716 1717
    strcpy(pElem->tableId, pTableMetaInfo->name);
    pElem->numOfTags = htons(pTableMetaInfo->numOfTags);
S
slguan 已提交
1718 1719 1720

    int16_t len = pMsg - (char *)pElem;
    pElem->elemLen = htons(len);  // redundant data for integrate check
H
hzcheng 已提交
1721 1722 1723 1724
  }

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1725
  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
H
hzcheng 已提交
1726
  assert(msgLen + minMsgSize() <= size);
H
hjxilinx 已提交
1727
#endif
1728
  
H
hjxilinx 已提交
1729
  SSqlCmd *pCmd = &pSql->cmd;
H
hjxilinx 已提交
1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742
  
  char* pMsg = pCmd->payload;
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
  
  SCMSTableVgroupMsg *pStableVgroupMsg = (SCMSTableVgroupMsg *) pMsg;
  pStableVgroupMsg->numOfTables = htonl(pQueryInfo->numOfTables);
  pMsg += sizeof(SCMSTableVgroupMsg);
  
  for(int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
    STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
    strncpy(pMsg, pTableMetaInfo->name, TSDB_TABLE_ID_LEN);
    pMsg += TSDB_TABLE_ID_LEN;
  }
H
hjxilinx 已提交
1743 1744

  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
H
hjxilinx 已提交
1745
  pCmd->payloadLen = (pMsg - pCmd->payload);
H
hjxilinx 已提交
1746

1747
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1748 1749
}

1750
int tscEstimateHeartBeatMsgLength(SSqlObj *pSql) {
H
hzcheng 已提交
1751 1752 1753
  int      size = 0;
  STscObj *pObj = pSql->pTscObj;

guanshengliang's avatar
guanshengliang 已提交
1754
  size += tsRpcHeadSize;
S
slguan 已提交
1755
  size += sizeof(SQqueryList);
H
hzcheng 已提交
1756 1757 1758

  SSqlObj *tpSql = pObj->sqlList;
  while (tpSql) {
S
slguan 已提交
1759
    size += sizeof(SQueryDesc);
H
hzcheng 已提交
1760 1761 1762
    tpSql = tpSql->next;
  }

S
slguan 已提交
1763
  size += sizeof(SStreamList);
H
hzcheng 已提交
1764 1765
  SSqlStream *pStream = pObj->streamList;
  while (pStream) {
S
slguan 已提交
1766
    size += sizeof(SStreamDesc);
H
hzcheng 已提交
1767 1768 1769 1770 1771 1772
    pStream = pStream->next;
  }

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1773
int tscBuildHeartBeatMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1774 1775 1776 1777 1778 1779 1780 1781 1782
  char *pMsg, *pStart;
  int   msgLen = 0;
  int   size = 0;

  SSqlCmd *pCmd = &pSql->cmd;
  STscObj *pObj = pSql->pTscObj;

  pthread_mutex_lock(&pObj->mutex);

1783
  size = tscEstimateHeartBeatMsgLength(pSql);
S
slguan 已提交
1784
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
1785
    pthread_mutex_unlock(&pObj->mutex);
S
slguan 已提交
1786 1787 1788
    tscError("%p failed to malloc for heartbeat msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1789

guanshengliang's avatar
guanshengliang 已提交
1790
  pMsg = pCmd->payload;
H
hzcheng 已提交
1791 1792 1793 1794 1795 1796 1797
  pStart = pMsg;

  pMsg = tscBuildQueryStreamDesc(pMsg, pObj);
  pthread_mutex_unlock(&pObj->mutex);

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1798
  pCmd->msgType = TSDB_MSG_TYPE_CM_HEARTBEAT;
H
hzcheng 已提交
1799 1800

  assert(msgLen + minMsgSize() <= size);
1801
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1802 1803
}

1804 1805
int tscProcessTableMetaRsp(SSqlObj *pSql) {
  STableMetaMsg *pMetaMsg = (STableMetaMsg *)pSql->res.pRsp;
H
hzcheng 已提交
1806

1807 1808
  pMetaMsg->sid = htonl(pMetaMsg->sid);
  pMetaMsg->sversion = htons(pMetaMsg->sversion);
H
hjxilinx 已提交
1809 1810 1811
  
  pMetaMsg->vgroup.vgId = htonl(pMetaMsg->vgroup.vgId);
  
1812 1813
  pMetaMsg->uid = htobe64(pMetaMsg->uid);
  pMetaMsg->contLen = htons(pMetaMsg->contLen);
1814
  pMetaMsg->numOfColumns = htons(pMetaMsg->numOfColumns);
H
hzcheng 已提交
1815

H
hjxilinx 已提交
1816 1817
  if (pMetaMsg->sid < 0 || pMetaMsg->vgroup.numOfIps < 0) {
    tscError("invalid meter vgId:%d, sid%d", pMetaMsg->vgroup.numOfIps, pMetaMsg->sid);
H
hzcheng 已提交
1818 1819 1820
    return TSDB_CODE_INVALID_VALUE;
  }

1821 1822
  if (pMetaMsg->numOfTags > TSDB_MAX_TAGS || pMetaMsg->numOfTags < 0) {
    tscError("invalid numOfTags:%d", pMetaMsg->numOfTags);
H
hzcheng 已提交
1823 1824 1825
    return TSDB_CODE_INVALID_VALUE;
  }

1826 1827
  if (pMetaMsg->numOfColumns > TSDB_MAX_COLUMNS || pMetaMsg->numOfColumns <= 0) {
    tscError("invalid numOfColumns:%d", pMetaMsg->numOfColumns);
H
hzcheng 已提交
1828 1829 1830
    return TSDB_CODE_INVALID_VALUE;
  }

H
hjxilinx 已提交
1831 1832
  for (int i = 0; i < pMetaMsg->vgroup.numOfIps; ++i) {
    pMetaMsg->vgroup.ipAddr[i].port = htons(pMetaMsg->vgroup.ipAddr[i].port);
H
hzcheng 已提交
1833 1834
  }

1835
  SSchema* pSchema = pMetaMsg->schema;
H
hzcheng 已提交
1836

1837
  int32_t numOfTotalCols = pMetaMsg->numOfColumns + pMetaMsg->numOfTags;
H
hzcheng 已提交
1838 1839 1840
  for (int i = 0; i < numOfTotalCols; ++i) {
    pSchema->bytes = htons(pSchema->bytes);
    pSchema->colId = htons(pSchema->colId);
H
hjxilinx 已提交
1841 1842
    
    assert(pSchema->type >= TSDB_DATA_TYPE_BOOL && pSchema->type <= TSDB_DATA_TYPE_NCHAR);
H
hzcheng 已提交
1843 1844 1845
    pSchema++;
  }

1846 1847
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
H
hzcheng 已提交
1848

1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859
#if 0
  // if current table is created according to super table, get the table meta of super table
  if (pTableMeta->tableType == TSDB_CHILD_TABLE) {
    char id[TSDB_TABLE_ID_LEN + 1] = {0};
    strncpy(id, pMetaMsg->stableId, TSDB_TABLE_ID_LEN);
  
    // NOTE: if the table meta of super table is not cached at client side yet, the pSTable is NULL
    pTableMeta->pSTable = taosCacheAcquireByName(tscCacheHandle, id);
  }
#endif
  
H
hzcheng 已提交
1860
  // todo add one more function: taosAddDataIfNotExists();
1861
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
H
hjxilinx 已提交
1862
  assert(pTableMetaInfo->pTableMeta == NULL);
H
hzcheng 已提交
1863

H
hjxilinx 已提交
1864
  pTableMetaInfo->pTableMeta =
H
hjxilinx 已提交
1865
      (STableMeta *) taosCachePut(tscCacheHandle, pTableMetaInfo->name, pTableMeta, size, tsTableMetaKeepTimer);
H
hjxilinx 已提交
1866
  
1867
  // todo handle out of memory case
1868
  if (pTableMetaInfo->pTableMeta == NULL) {
H
hjxilinx 已提交
1869
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
1870
  }
H
hzcheng 已提交
1871

H
Haojun Liao 已提交
1872
  tscTrace("%p recv table meta: %"PRId64 ", tid:%d, name:%s", pSql, pTableMeta->uid, pTableMeta->sid, pTableMetaInfo->name);
1873
  free(pTableMeta);
1874
  
H
hjxilinx 已提交
1875
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1876 1877
}

S
slguan 已提交
1878
/**
1879
 *  multi table meta rsp pkg format:
1880
 *  | STaosRsp | ieType | SCMMultiTableInfoMsg | SMeterMeta0 | SSchema0 | SMeterMeta1 | SSchema1 | SMeterMeta2 | SSchema2
S
slguan 已提交
1881 1882 1883
 *  |...... 1B        1B            4B
 **/
int tscProcessMultiMeterMetaRsp(SSqlObj *pSql) {
1884
#if 0
S
slguan 已提交
1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896
  char *rsp = pSql->res.pRsp;

  ieType = *rsp;
  if (ieType != TSDB_IE_TYPE_META) {
    tscError("invalid ie type:%d", ieType);
    pSql->res.code = TSDB_CODE_INVALID_IE;
    pSql->res.numOfTotal = 0;
    return TSDB_CODE_OTHERS;
  }

  rsp++;

1897
  SCMMultiTableInfoMsg *pInfo = (SCMMultiTableInfoMsg *)rsp;
S
slguan 已提交
1898
  totalNum = htonl(pInfo->numOfTables);
1899
  rsp += sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1900 1901

  for (i = 0; i < totalNum; i++) {
S
slguan 已提交
1902
    SMultiTableMeta *pMultiMeta = (SMultiTableMeta *)rsp;
1903
    STableMeta *     pMeta = pMultiMeta->metas;
S
slguan 已提交
1904 1905 1906

    pMeta->sid = htonl(pMeta->sid);
    pMeta->sversion = htons(pMeta->sversion);
H
hjxilinx 已提交
1907
    pMeta->vgId = htonl(pMeta->vgId);
S
slguan 已提交
1908 1909
    pMeta->uid = htobe64(pMeta->uid);

H
hjxilinx 已提交
1910 1911
    if (pMeta->sid <= 0 || pMeta->vgId < 0) {
      tscError("invalid meter vgId:%d, sid%d", pMeta->vgId, pMeta->sid);
S
slguan 已提交
1912 1913 1914 1915 1916
      pSql->res.code = TSDB_CODE_INVALID_VALUE;
      pSql->res.numOfTotal = i;
      return TSDB_CODE_OTHERS;
    }

H
hjxilinx 已提交
1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939
    //    pMeta->numOfColumns = htons(pMeta->numOfColumns);
    //
    //    if (pMeta->numOfTags > TSDB_MAX_TAGS || pMeta->numOfTags < 0) {
    //      tscError("invalid tag value count:%d", pMeta->numOfTags);
    //      pSql->res.code = TSDB_CODE_INVALID_VALUE;
    //      pSql->res.numOfTotal = i;
    //      return TSDB_CODE_OTHERS;
    //    }
    //
    //    if (pMeta->numOfTags > TSDB_MAX_TAGS || pMeta->numOfTags < 0) {
    //      tscError("invalid numOfTags:%d", pMeta->numOfTags);
    //      pSql->res.code = TSDB_CODE_INVALID_VALUE;
    //      pSql->res.numOfTotal = i;
    //      return TSDB_CODE_OTHERS;
    //    }
    //
    //    if (pMeta->numOfColumns > TSDB_MAX_COLUMNS || pMeta->numOfColumns < 0) {
    //      tscError("invalid numOfColumns:%d", pMeta->numOfColumns);
    //      pSql->res.code = TSDB_CODE_INVALID_VALUE;
    //      pSql->res.numOfTotal = i;
    //      return TSDB_CODE_OTHERS;
    //    }
    //
H
hjxilinx 已提交
1940
    //    for (int j = 0; j < TSDB_REPLICA_MAX_NUM; ++j) {
H
hjxilinx 已提交
1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974
    //      pMeta->vpeerDesc[j].vnode = htonl(pMeta->vpeerDesc[j].vnode);
    //    }
    //
    //    pMeta->rowSize = 0;
    //    rsp += sizeof(SMultiTableMeta);
    //    pSchema = (SSchema *)rsp;
    //
    //    int32_t numOfTotalCols = pMeta->numOfColumns + pMeta->numOfTags;
    //    for (int j = 0; j < numOfTotalCols; ++j) {
    //      pSchema->bytes = htons(pSchema->bytes);
    //      pSchema->colId = htons(pSchema->colId);
    //
    //      // ignore the tags length
    //      if (j < pMeta->numOfColumns) {
    //        pMeta->rowSize += pSchema->bytes;
    //      }
    //      pSchema++;
    //    }
    //
    //    rsp += numOfTotalCols * sizeof(SSchema);
    //
    //    int32_t  tagLen = 0;
    //    SSchema *pTagsSchema = tscGetTableTagSchema(pMeta);
    //
    //    if (pMeta->tableType == TSDB_CHILD_TABLE) {
    //      for (int32_t j = 0; j < pMeta->numOfTags; ++j) {
    //        tagLen += pTagsSchema[j].bytes;
    //      }
    //    }
    //
    //    rsp += tagLen;
    //    int32_t size = (int32_t)(rsp - ((char *)pMeta));  // Consistent with STableMeta in cache
    //
    //    pMeta->index = 0;
H
hjxilinx 已提交
1975
    //    (void)taosCachePut(tscCacheHandle, pMeta->tableId, (char *)pMeta, size, tsTableMetaKeepTimer);
H
hjxilinx 已提交
1976
    //  }
S
slguan 已提交
1977
  }
H
hjxilinx 已提交
1978
  
S
slguan 已提交
1979 1980 1981
  pSql->res.code = TSDB_CODE_SUCCESS;
  pSql->res.numOfTotal = i;
  tscTrace("%p load multi-metermeta resp complete num:%d", pSql, pSql->res.numOfTotal);
1982 1983
#endif
  
S
slguan 已提交
1984 1985 1986
  return TSDB_CODE_SUCCESS;
}

H
hjxilinx 已提交
1987 1988
int tscProcessSTableVgroupRsp(SSqlObj *pSql) {
#if 0
S
slguan 已提交
1989 1990
  void **      metricMetaList = NULL;
  int32_t *    sizes = NULL;
H
hjxilinx 已提交
1991
  
S
slguan 已提交
1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007
  int32_t num = htons(*(int16_t *)rsp);
  rsp += sizeof(int16_t);

  metricMetaList = calloc(1, POINTER_BYTES * num);
  sizes = calloc(1, sizeof(int32_t) * num);

  // return with error code
  if (metricMetaList == NULL || sizes == NULL) {
    tfree(metricMetaList);
    tfree(sizes);
    pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;

    return pSql->res.code;
  }

  for (int32_t k = 0; k < num; ++k) {
S
slguan 已提交
2008
    pMeta = (SSuperTableMeta *)rsp;
S
slguan 已提交
2009 2010

    size_t size = (size_t)pSql->res.rspLen - 1;
S
slguan 已提交
2011
    rsp = rsp + sizeof(SSuperTableMeta);
S
slguan 已提交
2012

S
slguan 已提交
2013
    pMeta->numOfTables = htonl(pMeta->numOfTables);
S
slguan 已提交
2014 2015 2016
    pMeta->numOfVnodes = htonl(pMeta->numOfVnodes);
    pMeta->tagLen = htons(pMeta->tagLen);

2017
    size += pMeta->numOfVnodes * sizeof(SVnodeSidList *) + pMeta->numOfTables * sizeof(STableIdInfo *);
H
hzcheng 已提交
2018

2019 2020
    char *pBuf = calloc(1, size);
    if (pBuf == NULL) {
S
slguan 已提交
2021 2022 2023
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2024

S
slguan 已提交
2025
    SSuperTableMeta *pNewMetricMeta = (SSuperTableMeta *)pBuf;
S
slguan 已提交
2026
    metricMetaList[k] = pNewMetricMeta;
H
hzcheng 已提交
2027

S
slguan 已提交
2028
    pNewMetricMeta->numOfTables = pMeta->numOfTables;
S
slguan 已提交
2029 2030
    pNewMetricMeta->numOfVnodes = pMeta->numOfVnodes;
    pNewMetricMeta->tagLen = pMeta->tagLen;
H
hzcheng 已提交
2031

S
slguan 已提交
2032
    pBuf = pBuf + sizeof(SSuperTableMeta) + pNewMetricMeta->numOfVnodes * sizeof(SVnodeSidList *);
H
hzcheng 已提交
2033

S
slguan 已提交
2034 2035
    for (int32_t i = 0; i < pMeta->numOfVnodes; ++i) {
      SVnodeSidList *pSidLists = (SVnodeSidList *)rsp;
2036
      memcpy(pBuf, pSidLists, sizeof(SVnodeSidList));
H
hzcheng 已提交
2037

2038 2039
      pNewMetricMeta->list[i] = pBuf - (char *)pNewMetricMeta;  // offset value
      SVnodeSidList *pLists = (SVnodeSidList *)pBuf;
H
hzcheng 已提交
2040

S
slguan 已提交
2041
      tscTrace("%p metricmeta:vid:%d,numOfTables:%d", pSql, i, pLists->numOfSids);
H
hzcheng 已提交
2042

2043
      pBuf += sizeof(SVnodeSidList) + sizeof(STableIdInfo *) * pSidLists->numOfSids;
S
slguan 已提交
2044
      rsp += sizeof(SVnodeSidList);
H
hzcheng 已提交
2045

2046
      size_t elemSize = sizeof(STableIdInfo) + pNewMetricMeta->tagLen;
S
slguan 已提交
2047
      for (int32_t j = 0; j < pSidLists->numOfSids; ++j) {
2048 2049
        pLists->pSidExtInfoList[j] = pBuf - (char *)pLists;
        memcpy(pBuf, rsp, elemSize);
2050

2051 2052
        ((STableIdInfo *)pBuf)->uid = htobe64(((STableIdInfo *)pBuf)->uid);
        ((STableIdInfo *)pBuf)->sid = htonl(((STableIdInfo *)pBuf)->sid);
2053

2054 2055
        rsp += elemSize;
        pBuf += elemSize;
S
slguan 已提交
2056
      }
H
hzcheng 已提交
2057
    }
S
slguan 已提交
2058

2059
    sizes[k] = pBuf - (char *)pNewMetricMeta;
H
hzcheng 已提交
2060 2061
  }

2062
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
S
slguan 已提交
2063 2064 2065
  for (int32_t i = 0; i < num; ++i) {
    char name[TSDB_MAX_TAGS_LEN + 1] = {0};

H
hjxilinx 已提交
2066
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2067
    tscGetMetricMetaCacheKey(pQueryInfo, name, pTableMetaInfo->pTableMeta->uid);
H
hzcheng 已提交
2068

S
slguan 已提交
2069 2070 2071
#ifdef _DEBUG_VIEW
    printf("generate the metric key:%s, index:%d\n", name, i);
#endif
H
hzcheng 已提交
2072

S
slguan 已提交
2073
    // release the used metricmeta
H
hjxilinx 已提交
2074 2075
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), false);
    pTableMetaInfo->pMetricMeta = (SSuperTableMeta *)taosCachePut(tscCacheHandle, name, (char *)metricMetaList[i],
S
slguan 已提交
2076 2077 2078 2079
                                                                      sizes[i], tsMetricMetaKeepTimer);
    tfree(metricMetaList[i]);

    // failed to put into cache
H
hjxilinx 已提交
2080
    if (pTableMetaInfo->pMetricMeta == NULL) {
S
slguan 已提交
2081 2082 2083
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2084 2085
  }

S
slguan 已提交
2086 2087 2088 2089 2090 2091 2092 2093
_error_clean:
  // free allocated resource
  for (int32_t i = 0; i < num; ++i) {
    tfree(metricMetaList[i]);
  }

  free(sizes);
  free(metricMetaList);
H
hjxilinx 已提交
2094
#endif
2095
  SSqlRes* pRes = &pSql->res;
H
hjxilinx 已提交
2096
  
H
hjxilinx 已提交
2097
  // NOTE: the order of several table must be preserved.
2098
  SCMSTableVgroupRspMsg *pStableVgroup = (SCMSTableVgroupRspMsg *)pRes->pRsp;
H
hjxilinx 已提交
2099 2100
  pStableVgroup->numOfTables = htonl(pStableVgroup->numOfTables);
  char* pMsg = pRes->pRsp + sizeof(SCMSTableVgroupRspMsg);
H
hjxilinx 已提交
2101
  
2102 2103 2104
  // master sqlObj locates in param
  SSqlObj* parent = pSql->param;
  assert(parent != NULL);
H
hjxilinx 已提交
2105
  
2106
  SSqlCmd* pCmd = &parent->cmd;
H
hjxilinx 已提交
2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127
  for(int32_t i = 0; i < pStableVgroup->numOfTables; ++i) {
    STableMetaInfo *pInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
    SVgroupsInfo *  pVgroupInfo = (SVgroupsInfo *)pMsg;
    pVgroupInfo->numOfVgroups = htonl(pVgroupInfo->numOfVgroups);

    size_t size = sizeof(SCMVgroupInfo) * pVgroupInfo->numOfVgroups + sizeof(SVgroupsInfo);
    pInfo->vgroupList = calloc(1, size);
    assert(pInfo->vgroupList != NULL);

    memcpy(pInfo->vgroupList, pVgroupInfo, size);
    for (int32_t j = 0; j < pInfo->vgroupList->numOfVgroups; ++j) {
      SCMVgroupInfo *pVgroups = &pInfo->vgroupList->vgroups[j];

      pVgroups->vgId = htonl(pVgroups->vgId);
      assert(pVgroups->numOfIps >= 1);

      for (int32_t k = 0; k < pVgroups->numOfIps; ++k) {
        pVgroups->ipAddr[k].port = htons(pVgroups->ipAddr[k].port);
      }

      pMsg += size;
2128
    }
H
hjxilinx 已提交
2129 2130
  }
  
S
slguan 已提交
2131
  return pSql->res.code;
H
hzcheng 已提交
2132 2133 2134 2135 2136 2137
}

/*
 * current process do not use the cache at all
 */
int tscProcessShowRsp(SSqlObj *pSql) {
H
hjxilinx 已提交
2138
  STableMetaMsg * pMetaMsg;
2139
  SCMShowRsp *pShow;
S
slguan 已提交
2140
  SSchema *    pSchema;
H
hzcheng 已提交
2141 2142
  char         key[20];

2143 2144 2145
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

H
hjxilinx 已提交
2146
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
2147

H
hjxilinx 已提交
2148
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
2149

2150
  pShow = (SCMShowRsp *)pRes->pRsp;
S
slguan 已提交
2151
  pShow->qhandle = htobe64(pShow->qhandle);
H
hzcheng 已提交
2152 2153
  pRes->qhandle = pShow->qhandle;

S
slguan 已提交
2154
  tscResetForNextRetrieve(pRes);
H
hjxilinx 已提交
2155
  pMetaMsg = &(pShow->tableMeta);
H
hzcheng 已提交
2156

H
hjxilinx 已提交
2157
  pMetaMsg->numOfColumns = ntohs(pMetaMsg->numOfColumns);
H
hzcheng 已提交
2158

H
hjxilinx 已提交
2159
  pSchema = pMetaMsg->schema;
H
hjxilinx 已提交
2160 2161
  pMetaMsg->sid = ntohs(pMetaMsg->sid);
  for (int i = 0; i < pMetaMsg->numOfColumns; ++i) {
H
hzcheng 已提交
2162 2163 2164 2165
    pSchema->bytes = htons(pSchema->bytes);
    pSchema++;
  }

2166
  key[0] = pCmd->msgType + 'a';
H
hzcheng 已提交
2167 2168
  strcpy(key + 1, "showlist");

H
hjxilinx 已提交
2169
  taosCacheRelease(tscCacheHandle, (void *)&(pTableMetaInfo->pTableMeta), false);
H
hjxilinx 已提交
2170
  
H
hjxilinx 已提交
2171 2172 2173
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
  
H
hjxilinx 已提交
2174
  pTableMetaInfo->pTableMeta = taosCachePut(tscCacheHandle, key, (char *)pTableMeta, size, tsTableMetaKeepTimer);
H
hjxilinx 已提交
2175
  SSchema *pTableSchema = tscGetTableSchema(pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2176

2177 2178 2179 2180
  if (pQueryInfo->colList == NULL) {
    pQueryInfo->colList = taosArrayInit(4, POINTER_BYTES);
  }
  
H
hjxilinx 已提交
2181 2182
  SFieldInfo* pFieldInfo = &pQueryInfo->fieldsInfo;
  
S
slguan 已提交
2183
  SColumnIndex index = {0};
H
hjxilinx 已提交
2184 2185 2186
  pSchema = pMetaMsg->schema;
  
  for (int16_t i = 0; i < pMetaMsg->numOfColumns; ++i, ++pSchema) {
S
slguan 已提交
2187
    index.columnIndex = i;
2188 2189
    tscColumnListInsert(pQueryInfo->colList, &index);
    
H
hjxilinx 已提交
2190 2191
    TAOS_FIELD f = tscCreateField(pSchema->type, pSchema->name, pSchema->bytes);
    SFieldSupInfo* pInfo = tscFieldInfoAppend(pFieldInfo, &f);
H
hjxilinx 已提交
2192
    
H
hjxilinx 已提交
2193
    pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index,
2194
                     pTableSchema[i].type, pTableSchema[i].bytes, pTableSchema[i].bytes, false);
H
hzcheng 已提交
2195
  }
H
hjxilinx 已提交
2196 2197
  
  pCmd->numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2198
  tscFieldInfoUpdateOffset(pQueryInfo);
H
hjxilinx 已提交
2199 2200
  
  tfree(pTableMeta);
H
hzcheng 已提交
2201 2202 2203 2204
  return 0;
}

int tscProcessConnectRsp(SSqlObj *pSql) {
S
slguan 已提交
2205
  char temp[TSDB_TABLE_ID_LEN * 2];
H
hzcheng 已提交
2206 2207 2208
  STscObj *pObj = pSql->pTscObj;
  SSqlRes *pRes = &pSql->res;

2209
  SCMConnectRsp *pConnect = (SCMConnectRsp *)pRes->pRsp;
H
hzcheng 已提交
2210
  strcpy(pObj->acctId, pConnect->acctId);  // copy acctId from response
2211 2212
  int32_t len = sprintf(temp, "%s%s%s", pObj->acctId, TS_PATH_DELIMITER, pObj->db);

H
hjxilinx 已提交
2213 2214 2215
  assert(len <= tListLen(pObj->db));
  strncpy(pObj->db, temp, tListLen(pObj->db));
  
2216 2217
  if (pConnect->ipList.numOfIps > 0) 
    tscSetMgmtIpList(&pConnect->ipList);
H
hzcheng 已提交
2218

S
slguan 已提交
2219
  strcpy(pObj->sversion, pConnect->serverVersion);
H
hzcheng 已提交
2220 2221
  pObj->writeAuth = pConnect->writeAuth;
  pObj->superAuth = pConnect->superAuth;
S
scripts  
slguan 已提交
2222
  taosTmrReset(tscProcessActivityTimer, tsShellActivityTimer * 500, pObj, tscTmr, &pObj->pTimer);
H
hzcheng 已提交
2223 2224 2225 2226 2227

  return 0;
}

int tscProcessUseDbRsp(SSqlObj *pSql) {
S
slguan 已提交
2228
  STscObj *       pObj = pSql->pTscObj;
2229
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
S
slguan 已提交
2230

H
hjxilinx 已提交
2231
  strcpy(pObj->db, pTableMetaInfo->name);
H
hzcheng 已提交
2232 2233 2234 2235
  return 0;
}

int tscProcessDropDbRsp(SSqlObj *UNUSED_PARAM(pSql)) {
2236
  taosCacheEmpty(tscCacheHandle);
H
hzcheng 已提交
2237 2238 2239 2240
  return 0;
}

int tscProcessDropTableRsp(SSqlObj *pSql) {
2241
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
S
slguan 已提交
2242

H
hjxilinx 已提交
2243 2244
  STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMeta == NULL) {
H
hzcheng 已提交
2245 2246 2247 2248 2249 2250 2251 2252
    /* not in cache, abort */
    return 0;
  }

  /*
   * 1. if a user drops one table, which is the only table in a vnode, remove operation will incur vnode to be removed.
   * 2. Then, a user creates a new metric followed by a table with identical name of removed table but different schema,
   * here the table will reside in a new vnode.
S
slguan 已提交
2253 2254
   * The cached information is expired, however, we may have lost the ref of original meter. So, clear whole cache
   * instead.
H
hzcheng 已提交
2255
   */
H
hjxilinx 已提交
2256 2257
  tscTrace("%p force release metermeta after drop table:%s", pSql, pTableMetaInfo->name);
  taosCacheRelease(tscCacheHandle, (void **)&pTableMeta, true);
H
hzcheng 已提交
2258

H
hjxilinx 已提交
2259 2260
  if (pTableMetaInfo->pTableMeta) {
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hzcheng 已提交
2261 2262 2263 2264 2265 2266
  }

  return 0;
}

int tscProcessAlterTableMsgRsp(SSqlObj *pSql) {
2267
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
S
slguan 已提交
2268

H
hjxilinx 已提交
2269 2270
  STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMeta == NULL) { /* not in cache, abort */
H
hzcheng 已提交
2271 2272 2273
    return 0;
  }

H
hjxilinx 已提交
2274 2275
  tscTrace("%p force release metermeta in cache after alter-table: %s", pSql, pTableMetaInfo->name);
  taosCacheRelease(tscCacheHandle, (void **)&pTableMeta, true);
H
hzcheng 已提交
2276

H
hjxilinx 已提交
2277
  if (pTableMetaInfo->pTableMeta) {
weixin_48148422's avatar
weixin_48148422 已提交
2278
    bool isSuperTable = UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo);
H
hzcheng 已提交
2279

H
hjxilinx 已提交
2280
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2281
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2282

2283
    if (isSuperTable) {  // if it is a super table, reset whole query cache
H
hjxilinx 已提交
2284
      tscTrace("%p reset query cache since table:%s is stable", pSql, pTableMetaInfo->name);
2285
      taosCacheEmpty(tscCacheHandle);
H
hzcheng 已提交
2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299
    }
  }

  return 0;
}

int tscProcessAlterDbMsgRsp(SSqlObj *pSql) {
  UNUSED(pSql);
  return 0;
}

int tscProcessQueryRsp(SSqlObj *pSql) {
  SSqlRes *pRes = &pSql->res;

2300
  SQueryTableRsp *pQuery = (SQueryTableRsp *)pRes->pRsp;
S
slguan 已提交
2301 2302 2303
  pQuery->qhandle = htobe64(pQuery->qhandle);
  pRes->qhandle = pQuery->qhandle;

H
hzcheng 已提交
2304
  pRes->data = NULL;
S
slguan 已提交
2305
  tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
2306 2307 2308
  return 0;
}

H
hjxilinx 已提交
2309
int tscProcessRetrieveRspFromNode(SSqlObj *pSql) {
S
slguan 已提交
2310 2311 2312
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

S
slguan 已提交
2313
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2314 2315 2316

  pRes->numOfRows = htonl(pRetrieve->numOfRows);
  pRes->precision = htons(pRetrieve->precision);
2317 2318
  pRes->offset    = htobe64(pRetrieve->offset);
  pRes->useconds  = htobe64(pRetrieve->useconds);
H
hjxilinx 已提交
2319
  pRes->completed = (pRetrieve->completed == 1);
2320
  pRes->data      = pRetrieve->data;
H
hjxilinx 已提交
2321
  
2322
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
2323 2324 2325 2326
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
    return pRes->code;
  }
  
weixin_48148422's avatar
weixin_48148422 已提交
2327
  if (pSql->pSubscription != NULL) {
H
hjxilinx 已提交
2328
    int32_t numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2329
    
H
hjxilinx 已提交
2330
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, numOfCols - 1);
H
hjxilinx 已提交
2331 2332
    int16_t     offset = tscFieldInfoGetOffset(pQueryInfo, numOfCols - 1);
    
weixin_48148422's avatar
weixin_48148422 已提交
2333 2334
    char* p = pRes->data + (pField->bytes + offset) * pRes->numOfRows;

S
slguan 已提交
2335
    int32_t numOfTables = htonl(*(int32_t*)p);
weixin_48148422's avatar
weixin_48148422 已提交
2336
    p += sizeof(int32_t);
S
slguan 已提交
2337
    for (int i = 0; i < numOfTables; i++) {
weixin_48148422's avatar
weixin_48148422 已提交
2338 2339
      int64_t uid = htobe64(*(int64_t*)p);
      p += sizeof(int64_t);
weixin_48148422's avatar
weixin_48148422 已提交
2340
      p += sizeof(int32_t); // skip tid
weixin_48148422's avatar
weixin_48148422 已提交
2341 2342
      TSKEY key = htobe64(*(TSKEY*)p);
      p += sizeof(TSKEY);
2343
      tscUpdateSubscriptionProgress(pSql->pSubscription, uid, key);
weixin_48148422's avatar
weixin_48148422 已提交
2344
    }
2345 2346
  }

H
hzcheng 已提交
2347
  pRes->row = 0;
S
slguan 已提交
2348
  tscTrace("%p numOfRows:%d, offset:%d", pSql, pRes->numOfRows, pRes->offset);
H
hzcheng 已提交
2349 2350 2351 2352 2353

  return 0;
}

int tscProcessRetrieveRspFromLocal(SSqlObj *pSql) {
2354 2355
  SSqlRes *   pRes = &pSql->res;
  SSqlCmd *   pCmd = &pSql->cmd;
2356
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
2357

S
slguan 已提交
2358
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2359 2360 2361

  pRes->numOfRows = htonl(pRetrieve->numOfRows);
  pRes->data = pRetrieve->data;
2362

2363
  tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
2364 2365 2366 2367
  pRes->row = 0;
  return 0;
}

H
hjxilinx 已提交
2368
void tscTableMetaCallBack(void *param, TAOS_RES *res, int code);
H
hzcheng 已提交
2369

2370
static int32_t getTableMetaFromMgmt(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo) {
S
slguan 已提交
2371 2372
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
  if (NULL == pNew) {
2373
    tscError("%p malloc failed for new sqlobj to get table meta", pSql);
S
slguan 已提交
2374 2375
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
2376

H
hzcheng 已提交
2377 2378 2379
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;
  pNew->cmd.command = TSDB_SQL_META;
S
slguan 已提交
2380

2381
  tscAddSubqueryInfo(&pNew->cmd);
2382 2383 2384 2385

  SQueryInfo *pNewQueryInfo = NULL;
  tscGetQueryInfoDetailSafely(&pNew->cmd, 0, &pNewQueryInfo);

H
hjxilinx 已提交
2386
  pNew->cmd.autoCreated = pSql->cmd.autoCreated;  // create table if not exists
B
Bomin Zhang 已提交
2387
  if (TSDB_CODE_SUCCESS != tscAllocPayload(&pNew->cmd, TSDB_DEFAULT_PAYLOAD_SIZE + pSql->cmd.payloadLen)) {
2388
    tscError("%p malloc failed for payload to get table meta", pSql);
S
slguan 已提交
2389
    free(pNew);
2390

S
slguan 已提交
2391 2392 2393
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

H
hjxilinx 已提交
2394
  STableMetaInfo *pNewMeterMetaInfo = tscAddEmptyMetaInfo(pNewQueryInfo);
2395
  assert(pNew->cmd.numOfClause == 1 && pNewQueryInfo->numOfTables == 1);
H
hzcheng 已提交
2396

2397
  strncpy(pNewMeterMetaInfo->name, pTableMetaInfo->name, tListLen(pNewMeterMetaInfo->name));
B
Bomin Zhang 已提交
2398 2399
  memcpy(pNew->cmd.payload, pSql->cmd.payload, pSql->cmd.payloadLen);  // tag information if table does not exists.
  pNew->cmd.payloadLen = pSql->cmd.payloadLen;
2400
  tscTrace("%p new pSqlObj:%p to get tableMeta, auto create:%d", pSql, pNew, pNew->cmd.autoCreated);
H
hzcheng 已提交
2401

H
hjxilinx 已提交
2402 2403
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
H
hzcheng 已提交
2404

H
hjxilinx 已提交
2405 2406 2407
  int32_t code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2408 2409 2410 2411 2412
  }

  return code;
}

H
hjxilinx 已提交
2413
int32_t tscGetTableMeta(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo) {
H
hjxilinx 已提交
2414
  assert(strlen(pTableMetaInfo->name) != 0);
S
slguan 已提交
2415

2416
  // If this STableMetaInfo owns a table meta, release it first
H
hjxilinx 已提交
2417 2418
  if (pTableMetaInfo->pTableMeta != NULL) {
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), false);
2419 2420
  }
  
H
hjxilinx 已提交
2421 2422
  pTableMetaInfo->pTableMeta = (STableMeta *)taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMetaInfo->pTableMeta != NULL) {
H
hjxilinx 已提交
2423
    STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
2424 2425
    tscTrace("%p retrieve table Meta from cache, the number of columns:%d, numOfTags:%d, %p", pSql, tinfo.numOfColumns,
             tinfo.numOfTags, pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2426 2427 2428

    return TSDB_CODE_SUCCESS;
  }
2429 2430
  
  return getTableMetaFromMgmt(pSql, pTableMetaInfo);
H
hzcheng 已提交
2431 2432
}

H
hjxilinx 已提交
2433
int tscGetMeterMetaEx(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo, bool createIfNotExists) {
H
hjxilinx 已提交
2434
  pSql->cmd.autoCreated = createIfNotExists;
H
hjxilinx 已提交
2435
  return tscGetTableMeta(pSql, pTableMetaInfo);
H
hzcheng 已提交
2436 2437 2438 2439
}

/*
 * in handling the renew metermeta problem during insertion,
S
slguan 已提交
2440
 *
H
hzcheng 已提交
2441 2442 2443 2444 2445
 * If the meter is created on demand during insertion, the routine usually waits for a short
 * period to re-issue the getMeterMeta msg, in which makes a greater change that vnode has
 * successfully created the corresponding table.
 */
static void tscWaitingForCreateTable(SSqlCmd *pCmd) {
S
slguan 已提交
2446
  if (pCmd->command == TSDB_SQL_INSERT) {
H
hzcheng 已提交
2447 2448 2449 2450 2451 2452 2453
    taosMsleep(50);  // todo: global config
  }
}

/**
 * in renew metermeta, do not retrieve metadata in cache.
 * @param pSql          sql object
S
slguan 已提交
2454
 * @param tableId       meter id
H
hzcheng 已提交
2455 2456
 * @return              status code
 */
S
slguan 已提交
2457
int tscRenewMeterMeta(SSqlObj *pSql, char *tableId) {
2458 2459
  int code = 0;

H
hjxilinx 已提交
2460
  // handle table meta renew process
H
hzcheng 已提交
2461
  SSqlCmd *pCmd = &pSql->cmd;
2462 2463

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
2464
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
2465 2466

  /*
S
slguan 已提交
2467
   * 1. only update the metermeta in force model metricmeta is not updated
H
hzcheng 已提交
2468 2469
   * 2. if get metermeta failed, still get the metermeta
   */
2470
  if (pTableMetaInfo->pTableMeta == NULL || !tscQueryOnSTable(pCmd)) {
2471
    STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
2472
    if (pTableMetaInfo->pTableMeta) {
2473 2474
      tscTrace("%p update table meta, old: numOfTags:%d, numOfCols:%d, uid:%" PRId64 ", addr:%p", pSql,
               tscGetNumOfTags(pTableMeta), tscGetNumOfColumns(pTableMeta), pTableMeta->uid, pTableMeta);
H
hzcheng 已提交
2475
    }
2476

2477
    tscWaitingForCreateTable(pCmd);
H
hjxilinx 已提交
2478
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hzcheng 已提交
2479

2480
    code = getTableMetaFromMgmt(pSql, pTableMetaInfo);  // todo ??
H
hzcheng 已提交
2481
  } else {
H
hjxilinx 已提交
2482
    tscTrace("%p metric query not update metric meta, numOfTags:%d, numOfCols:%d, uid:%" PRId64 ", addr:%p", pSql,
H
hjxilinx 已提交
2483 2484
             tscGetNumOfTags(pTableMetaInfo->pTableMeta), pCmd->numOfCols, pTableMetaInfo->pTableMeta->uid,
             pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2485 2486 2487 2488 2489
  }

  return code;
}

H
hjxilinx 已提交
2490
static bool allVgroupInfoRetrieved(SSqlCmd* pCmd, int32_t clauseIndex) {
2491
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2492
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2493
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2494 2495
    if (pTableMetaInfo->vgroupList == NULL) {
      return false;
S
slguan 已提交
2496 2497
    }
  }
H
hjxilinx 已提交
2498 2499 2500 2501
  
  // all super tables vgroupinfo are retrieved, no need to retrieve vgroup info anymore
  return true;
}
H
hzcheng 已提交
2502

H
hjxilinx 已提交
2503 2504 2505 2506 2507
int tscGetSTableVgroupInfo(SSqlObj *pSql, int32_t clauseIndex) {
  int      code = TSDB_CODE_NETWORK_UNAVAIL;
  SSqlCmd *pCmd = &pSql->cmd;
  
  if (allVgroupInfoRetrieved(pCmd, clauseIndex)) {
H
hzcheng 已提交
2508 2509
    return TSDB_CODE_SUCCESS;
  }
H
hjxilinx 已提交
2510

S
slguan 已提交
2511
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
H
hzcheng 已提交
2512 2513 2514
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;

H
hjxilinx 已提交
2515
  pNew->cmd.command = TSDB_SQL_STABLEVGROUP;
2516 2517
  
  SQueryInfo *pNewQueryInfo = NULL;
2518 2519 2520
  if ((code = tscGetQueryInfoDetailSafely(&pNew->cmd, 0, &pNewQueryInfo)) != TSDB_CODE_SUCCESS) {
    return code;
  }
2521
  
H
hjxilinx 已提交
2522
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2523
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2524 2525 2526
    STableMetaInfo *pMInfo = tscGetMetaInfo(pQueryInfo, i);
    STableMeta *pTableMeta = taosCacheAcquireByData(tscCacheHandle, pMInfo->pTableMeta);
    tscAddTableMetaInfo(pNewQueryInfo, pMInfo->name, pTableMeta, NULL, pMInfo->tagColList);
S
slguan 已提交
2527 2528 2529 2530 2531 2532
  }

  if ((code = tscAllocPayload(&pNew->cmd, TSDB_DEFAULT_PAYLOAD_SIZE)) != TSDB_CODE_SUCCESS) {
    tscFreeSqlObj(pNew);
    return code;
  }
H
hzcheng 已提交
2533

2534
  pNewQueryInfo->numOfTables = pQueryInfo->numOfTables;
H
hjxilinx 已提交
2535
  tscTrace("%p new sqlObj:%p to get vgroupInfo, numOfTables:%d", pSql, pNew, pNewQueryInfo->numOfTables);
H
hzcheng 已提交
2536

2537 2538 2539 2540 2541
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
  code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2542 2543 2544 2545 2546
  }

  return code;
}

2547
void tscInitMsgsFp() {
S
slguan 已提交
2548 2549
  tscBuildMsg[TSDB_SQL_SELECT] = tscBuildQueryMsg;
  tscBuildMsg[TSDB_SQL_INSERT] = tscBuildSubmitMsg;
J
jtao1735 已提交
2550
  tscBuildMsg[TSDB_SQL_FETCH] = tscBuildFetchMsg;
H
hzcheng 已提交
2551 2552

  tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
2553
  tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
H
hzcheng 已提交
2554

2555 2556
  tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
  tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
H
hzcheng 已提交
2557 2558

  tscBuildMsg[TSDB_SQL_CREATE_TABLE] = tscBuildCreateTableMsg;
S
[TD-16]  
slguan 已提交
2559
  tscBuildMsg[TSDB_SQL_DROP_USER] = tscBuildDropUserMsg;
H
hzcheng 已提交
2560 2561 2562
  tscBuildMsg[TSDB_SQL_DROP_ACCT] = tscBuildDropAcctMsg;
  tscBuildMsg[TSDB_SQL_DROP_DB] = tscBuildDropDbMsg;
  tscBuildMsg[TSDB_SQL_DROP_TABLE] = tscBuildDropTableMsg;
2563
  tscBuildMsg[TSDB_SQL_ALTER_USER] = tscBuildUserMsg;
S
slguan 已提交
2564 2565 2566
  tscBuildMsg[TSDB_SQL_CREATE_DNODE] = tscBuildCreateDnodeMsg;
  tscBuildMsg[TSDB_SQL_DROP_DNODE] = tscBuildDropDnodeMsg;
  tscBuildMsg[TSDB_SQL_CFG_DNODE] = tscBuildCfgDnodeMsg;
H
hzcheng 已提交
2567 2568 2569 2570 2571
  tscBuildMsg[TSDB_SQL_ALTER_TABLE] = tscBuildAlterTableMsg;
  tscBuildMsg[TSDB_SQL_ALTER_DB] = tscAlterDbMsg;

  tscBuildMsg[TSDB_SQL_CONNECT] = tscBuildConnectMsg;
  tscBuildMsg[TSDB_SQL_USE_DB] = tscBuildUseDbMsg;
H
hjxilinx 已提交
2572
  tscBuildMsg[TSDB_SQL_META] = tscBuildTableMetaMsg;
H
hjxilinx 已提交
2573
  tscBuildMsg[TSDB_SQL_STABLEVGROUP] = tscBuildSTableVgroupMsg;
S
slguan 已提交
2574
  tscBuildMsg[TSDB_SQL_MULTI_META] = tscBuildMultiMeterMetaMsg;
H
hzcheng 已提交
2575 2576 2577 2578

  tscBuildMsg[TSDB_SQL_HB] = tscBuildHeartBeatMsg;
  tscBuildMsg[TSDB_SQL_SHOW] = tscBuildShowMsg;
  tscBuildMsg[TSDB_SQL_RETRIEVE] = tscBuildRetrieveFromMgmtMsg;
2579 2580 2581
  tscBuildMsg[TSDB_SQL_KILL_QUERY] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_STREAM] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_CONNECTION] = tscBuildKillMsg;
H
hzcheng 已提交
2582 2583

  tscProcessMsgRsp[TSDB_SQL_SELECT] = tscProcessQueryRsp;
H
hjxilinx 已提交
2584
  tscProcessMsgRsp[TSDB_SQL_FETCH] = tscProcessRetrieveRspFromNode;
H
hzcheng 已提交
2585 2586 2587 2588 2589

  tscProcessMsgRsp[TSDB_SQL_DROP_DB] = tscProcessDropDbRsp;
  tscProcessMsgRsp[TSDB_SQL_DROP_TABLE] = tscProcessDropTableRsp;
  tscProcessMsgRsp[TSDB_SQL_CONNECT] = tscProcessConnectRsp;
  tscProcessMsgRsp[TSDB_SQL_USE_DB] = tscProcessUseDbRsp;
2590
  tscProcessMsgRsp[TSDB_SQL_META] = tscProcessTableMetaRsp;
H
hjxilinx 已提交
2591
  tscProcessMsgRsp[TSDB_SQL_STABLEVGROUP] = tscProcessSTableVgroupRsp;
S
slguan 已提交
2592
  tscProcessMsgRsp[TSDB_SQL_MULTI_META] = tscProcessMultiMeterMetaRsp;
H
hzcheng 已提交
2593 2594

  tscProcessMsgRsp[TSDB_SQL_SHOW] = tscProcessShowRsp;
H
hjxilinx 已提交
2595
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE] = tscProcessRetrieveRspFromNode;  // rsp handled by same function.
H
hzcheng 已提交
2596
  tscProcessMsgRsp[TSDB_SQL_DESCRIBE_TABLE] = tscProcessDescribeTableRsp;
2597

H
hjxilinx 已提交
2598 2599 2600 2601 2602
  tscProcessMsgRsp[TSDB_SQL_CURRENT_DB] = tscProcessTagRetrieveRsp;
  tscProcessMsgRsp[TSDB_SQL_CURRENT_USER] = tscProcessTagRetrieveRsp;
  tscProcessMsgRsp[TSDB_SQL_SERV_VERSION] = tscProcessTagRetrieveRsp;
  tscProcessMsgRsp[TSDB_SQL_CLI_VERSION] = tscProcessTagRetrieveRsp;
  tscProcessMsgRsp[TSDB_SQL_SERV_STATUS] = tscProcessTagRetrieveRsp;
2603

H
hzcheng 已提交
2604 2605
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_EMPTY_RESULT] = tscProcessEmptyResultRsp;

H
hjxilinx 已提交
2606
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_LOCALMERGE] = tscProcessRetrieveLocalMergeRsp;
H
hzcheng 已提交
2607 2608 2609 2610 2611 2612 2613 2614 2615 2616

  tscProcessMsgRsp[TSDB_SQL_ALTER_TABLE] = tscProcessAlterTableMsgRsp;
  tscProcessMsgRsp[TSDB_SQL_ALTER_DB] = tscProcessAlterDbMsgRsp;

  tscKeepConn[TSDB_SQL_SHOW] = 1;
  tscKeepConn[TSDB_SQL_RETRIEVE] = 1;
  tscKeepConn[TSDB_SQL_SELECT] = 1;
  tscKeepConn[TSDB_SQL_FETCH] = 1;
  tscKeepConn[TSDB_SQL_HB] = 1;
}