tscServer.c 87.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"
H
hzcheng 已提交
30 31 32

#define TSC_MGMT_VNODE 999

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

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

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

42 43 44
TSKEY tscGetSubscriptionProgress(void* sub, int64_t uid);
void tscUpdateSubscriptionProgress(void* sub, int64_t uid, TSKEY ts);
void tscSaveSubscriptionProgress(void* sub);
H
hzcheng 已提交
45

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

48 49 50
static void tscSetDnodeIpList(SSqlObj* pSql, STableMeta* pTableMeta) {
  SRpcIpSet* pIpList = &pSql->ipList;
  
H
hjxilinx 已提交
51
  pIpList->numOfIps = pTableMeta->vgroupInfo.numOfIps;
52 53 54
  pIpList->port     = tsDnodeShellPort;
  pIpList->inUse    = 0;
  
H
hjxilinx 已提交
55 56
  for(int32_t i = 0; i < pTableMeta->vgroupInfo.numOfIps; ++i) {
    pIpList->ip[i] = pTableMeta->vgroupInfo.ipAddr[i].ip;
57 58 59
  }
}

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

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

void tscSetMgmtIpListFromEdge() {
S
slguan 已提交
80 81 82 83 84
  if (tscMgmtIpSet.numOfIps != 1) {
    tscMgmtIpSet.numOfIps = 1;
    tscMgmtIpSet.inUse = 0;
    tscMgmtIpSet.port = tsMnodeShellPort;
    tscMgmtIpSet.ip[0] = inet_addr(tsMasterIp);
S
slguan 已提交
85 86 87 88 89
    tscTrace("edge mgmt IP list:");
    tscPrintMgmtIp();
  }
}

S
slguan 已提交
90 91
void tscUpdateIpSet(void *ahandle, SRpcIpSet *pIpSet) {
  tscTrace("mgmt IP list is changed for ufp is called");
S
slguan 已提交
92
  tscMgmtIpSet = *pIpSet;
S
slguan 已提交
93 94
}

S
slguan 已提交
95
void tscSetMgmtIpList(SRpcIpSet *pIpList) {
S
slguan 已提交
96 97 98 99 100 101 102 103
  /*
    * The iplist returned by the cluster edition is the current management nodes
    * and the iplist returned by the edge edition is empty
    */
  if (pIpList->numOfIps != 0) {
    tscSetMgmtIpListFromCluster(pIpList);
  } else {
    tscSetMgmtIpListFromEdge();
S
slguan 已提交
104 105 106
  }
}

H
hjxilinx 已提交
107 108 109 110 111 112 113
/*
 * 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.
 */
114
UNUSED_FUNC
H
hjxilinx 已提交
115 116
static int32_t tscGetMgmtConnMaxRetryTimes() {
  int32_t factor = 2;
S
slguan 已提交
117
  return tscMgmtIpSet.numOfIps * factor;
H
hjxilinx 已提交
118 119
}

H
hzcheng 已提交
120 121 122 123 124 125 126 127 128 129 130 131
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) {
132
    SCMHeartBeatRsp *pRsp = (SCMHeartBeatRsp *)pRes->pRsp;
S
slguan 已提交
133
    SRpcIpSet *      pIpList = &pRsp->ipList;
S
slguan 已提交
134
    tscSetMgmtIpList(pIpList);
S
slguan 已提交
135

H
hzcheng 已提交
136 137 138
    if (pRsp->killConnection) {
      tscKillConnection(pObj);
    } else {
S
slguan 已提交
139 140
      if (pRsp->queryId) tscKillQuery(pObj, htonl(pRsp->queryId));
      if (pRsp->streamId) tscKillStream(pObj, htonl(pRsp->streamId));
H
hzcheng 已提交
141 142
    }
  } else {
H
hjxilinx 已提交
143
    tscTrace("heart beat failed, code:%s", tstrerror(code));
H
hzcheng 已提交
144 145 146 147 148 149 150 151 152 153 154 155 156
  }

  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 已提交
157 158 159
    SSqlObj *pSql = (SSqlObj *)calloc(1, sizeof(SSqlObj));
    if (NULL == pSql) return;

H
hzcheng 已提交
160
    pSql->fp = tscProcessHeartBeatRsp;
H
hjxilinx 已提交
161
    
162 163 164 165
    SQueryInfo *pQueryInfo = NULL;
    tscGetQueryInfoDetailSafely(&pSql->cmd, 0, &pQueryInfo);
    pQueryInfo->command = TSDB_SQL_HB;
    
166
    pSql->cmd.command = TSDB_SQL_HB;
S
slguan 已提交
167 168 169 170 171
    if (TSDB_CODE_SUCCESS != tscAllocPayload(&(pSql->cmd), TSDB_DEFAULT_PAYLOAD_SIZE)) {
      tfree(pSql);
      return;
    }

172
    pSql->cmd.command = TSDB_SQL_HB;
H
hzcheng 已提交
173 174 175 176
    pSql->param = pObj;
    pSql->pTscObj = pObj;
    pSql->signature = pSql;
    pObj->pHb = pSql;
177
    tscAddSubqueryInfo(&pObj->pHb->cmd);
178

S
slguan 已提交
179
    tscTrace("%p pHb is allocated, pObj:%p", pObj->pHb, pObj);
H
hzcheng 已提交
180 181 182
  }

  if (tscShouldFreeHeatBeat(pObj->pHb)) {
S
slguan 已提交
183
    tscTrace("%p free HB object and release connection", pObj);
H
hzcheng 已提交
184 185 186 187 188 189 190 191 192
    tscFreeSqlObj(pObj->pHb);
    tscCloseTscObj(pObj);
    return;
  }

  tscProcessSql(pObj->pHb);
}

int tscSendMsgToServer(SSqlObj *pSql) {
193
  STscObj* pObj = pSql->pTscObj;
H
hjxilinx 已提交
194 195 196
  SSqlCmd* pCmd = &pSql->cmd;
  
  char *pMsg = rpcMallocCont(pCmd->payloadLen);
S
slguan 已提交
197
  if (NULL == pMsg) {
S
slguan 已提交
198 199
    tscError("%p msg:%s malloc fail", pSql, taosMsg[pSql->cmd.msgType]);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
H
hzcheng 已提交
200 201
  }

S
slguan 已提交
202
  if (pSql->cmd.command < TSDB_SQL_MGMT) {
203
    tscTrace("%p msg:%s is sent to server %d", pSql, taosMsg[pSql->cmd.msgType], pSql->ipList.port);
S
slguan 已提交
204
    memcpy(pMsg, pSql->cmd.payload + tsRpcHeadSize, pSql->cmd.payloadLen);
205 206 207 208 209 210

    SRpcMsg rpcMsg = {
      .msgType = pSql->cmd.msgType,
      .pCont   = pMsg,
      .contLen = pSql->cmd.payloadLen,
      .handle  = pSql,
H
hjxilinx 已提交
211
      .code    = 0
212
    };
H
hjxilinx 已提交
213
    rpcSendRequest(pVnodeConn, &pSql->ipList, &rpcMsg);
S
slguan 已提交
214
  } else {
S
slguan 已提交
215
    pSql->ipList = tscMgmtIpSet;
H
hjxilinx 已提交
216 217 218
    pSql->ipList.port = tsMnodeShellPort;
    
    tscTrace("%p msg:%s is sent to server %d", pSql, taosMsg[pSql->cmd.msgType], pSql->ipList.port);
S
slguan 已提交
219
    memcpy(pMsg, pSql->cmd.payload, pSql->cmd.payloadLen);
220 221 222 223 224 225 226
    SRpcMsg rpcMsg = {
        .msgType = pSql->cmd.msgType,
        .pCont   = pMsg,
        .contLen = pSql->cmd.payloadLen,
        .handle  = pSql,
        .code   = 0
    };
227
    rpcSendRequest(pObj->pMgmtConn, &pSql->ipList, &rpcMsg);
H
hzcheng 已提交
228 229
  }

S
slguan 已提交
230
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
231 232
}

233 234
void tscProcessMsgFromServer(SRpcMsg *rpcMsg) {
  SSqlObj *pSql = (SSqlObj *)rpcMsg->handle;
S
slguan 已提交
235
  if (pSql == NULL || pSql->signature != pSql) {
H
hzcheng 已提交
236
    tscError("%p sql is already released, signature:%p", pSql, pSql->signature);
S
slguan 已提交
237
    return;
H
hzcheng 已提交
238 239
  }

S
slguan 已提交
240 241 242
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;
  STscObj *pObj = pSql->pTscObj;
243
  tscTrace("%p msg:%p is received from server", pSql, rpcMsg->pCont);
H
hzcheng 已提交
244 245

  if (pSql->freed || pObj->signature != pObj) {
S
slguan 已提交
246 247
    tscTrace("%p sql is already released or DB connection is closed, freed:%d pObj:%p signature:%p", pSql, pSql->freed,
             pObj, pObj->signature);
H
hzcheng 已提交
248
    tscFreeSqlObj(pSql);
249
    rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
250
    return;
H
hzcheng 已提交
251 252
  }

253 254
  if (rpcMsg->pCont == NULL) {
    rpcMsg->code = TSDB_CODE_NETWORK_UNAVAIL;
S
slguan 已提交
255
  } else {
256
    STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
257 258
    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 已提交
259
        rpcMsg->code == TSDB_CODE_NETWORK_UNAVAIL || rpcMsg->code == TSDB_CODE_NOT_ACTIVE_TABLE ||
260
        rpcMsg->code == TSDB_CODE_TABLE_ID_MISMATCH) {
S
slguan 已提交
261 262 263 264 265 266 267 268 269 270 271
      /*
       * 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) {
272 273
        rpcMsg->code = TSDB_CODE_NETWORK_UNAVAIL;
        rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
274 275
        return;
      } else if (pCmd->command == TSDB_SQL_HB) {
276 277
        rpcMsg->code = TSDB_CODE_NOT_READY;
        rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
278 279
        return;
      } else {
280
        tscWarn("%p it shall renew table meta, code:%s, retry:%d", pSql, tstrerror(rpcMsg->code), ++pSql->retry);
281
        
282
        pSql->res.code = rpcMsg->code;  // keep the previous error code
283 284 285 286 287 288 289 290
        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);
          }
  
291
          rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
292 293
          return;
        }
H
hzcheng 已提交
294 295
      }
    }
S
slguan 已提交
296
  }
H
hzcheng 已提交
297

298 299 300 301 302
  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;
  }
  
H
hzcheng 已提交
303
  pRes->rspLen = 0;
304
  
H
hzcheng 已提交
305
  if (pRes->code != TSDB_CODE_QUERY_CANCELLED) {
306
    pRes->code = (rpcMsg->code != TSDB_CODE_SUCCESS) ? rpcMsg->code : TSDB_CODE_NETWORK_UNAVAIL;
H
hzcheng 已提交
307
  } else {
H
hjxilinx 已提交
308
    tscTrace("%p query is cancelled, code:%d", pSql, tstrerror(pRes->code));
H
hzcheng 已提交
309 310
  }

S
slguan 已提交
311
  if (pRes->code != TSDB_CODE_QUERY_CANCELLED) {
312
    assert(rpcMsg->msgType == pCmd->msgType + 1);
313
    pRes->code    = rpcMsg->code;
314
    pRes->rspType = rpcMsg->msgType;
315
    pRes->rspLen  = rpcMsg->contLen;
H
hzcheng 已提交
316

317 318 319 320 321 322
    if (pRes->rspLen > 0) {
      char *tmp = (char *)realloc(pRes->pRsp, pRes->rspLen);
      if (tmp == NULL) {
        pRes->code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      } else {
        pRes->pRsp = tmp;
323
        memcpy(pRes->pRsp, rpcMsg->pCont, pRes->rspLen);
S
slguan 已提交
324
      }
325 326
    } else {
      pRes->pRsp = NULL;
S
slguan 已提交
327 328
    }

H
hzcheng 已提交
329 330 331 332
    /*
     * There is not response callback function for submit response.
     * The actual inserted number of points is the first number.
     */
333
    if (rpcMsg->msgType == TSDB_MSG_TYPE_SUBMIT_RSP && pRes->pRsp != NULL) {
334
      SShellSubmitRspMsg *pMsg = (SShellSubmitRspMsg*)pRes->pRsp;
S
slguan 已提交
335 336 337 338 339 340 341
      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;
342
      tscTrace("%p cmd:%d code:%s, inserted rows:%d, rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code),
H
hjxilinx 已提交
343
          pMsg->affectedRows, pRes->rspLen);
S
slguan 已提交
344
    } else {
H
hjxilinx 已提交
345
      tscTrace("%p cmd:%d code:%s rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code), pRes->rspLen);
H
hzcheng 已提交
346 347 348
    }
  }

349 350
  if (pRes->code == TSDB_CODE_SUCCESS && tscProcessMsgRsp[pCmd->command])
    rpcMsg->code = (*tscProcessMsgRsp[pCmd->command])(pSql);
H
hjxilinx 已提交
351
  
352
  if (rpcMsg->code != TSDB_CODE_ACTION_IN_PROGRESS) {
353
    void *taosres = tscKeepConn[pCmd->command] ? pSql : NULL;
H
[td-32]  
hjxilinx 已提交
354
    rpcMsg->code = pRes->code ? pRes->code : pRes->numOfRows;
355
    
356
    tscTrace("%p SQL result:%s res:%p", pSql, tstrerror(pRes->code), pSql);
H
hzcheng 已提交
357

358 359
    /*
     * Whether to free sqlObj or not should be decided before call the user defined function, since this SqlObj
H
hjxilinx 已提交
360 361
     * 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.
362 363
     *
     * If this block of memory is re-allocated for an insert thread, in which tscKeepConn[command] equals to 0,
H
hjxilinx 已提交
364
     * the tscShouldBeFreed will success and tscFreeSqlObj free it immediately.
365
     */
H
hjxilinx 已提交
366
    bool shouldFree = tscShouldBeFreed(pSql);
H
hjxilinx 已提交
367
    (*pSql->fp)(pSql->param, taosres, rpcMsg->code);
H
hzcheng 已提交
368

369
    if (shouldFree) {
370
      tscTrace("%p sqlObj is automatically freed", pSql);
sangshuduo's avatar
sangshuduo 已提交
371
      tscFreeSqlObj(pSql);
H
hzcheng 已提交
372 373 374
    }
  }

375
  rpcFreeCont(rpcMsg->pCont);
H
hzcheng 已提交
376 377
}

S
slguan 已提交
378 379 380
int doProcessSql(SSqlObj *pSql) {
  SSqlCmd *pCmd = &pSql->cmd;
  SSqlRes *pRes = &pSql->res;
381 382
  int32_t code = TSDB_CODE_SUCCESS;
  
H
hjxilinx 已提交
383 384 385 386 387 388 389
  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 已提交
390
      pCmd->command == TSDB_SQL_STABLEVGROUP) {
391 392 393 394 395 396
    pRes->code = tscBuildMsg[pCmd->command](pSql, NULL);
  }
  
  if (pRes->code != TSDB_CODE_SUCCESS) {
    tscQueueAsyncRes(pSql);
    return pRes->code;
S
slguan 已提交
397
  }
398

399
  code = tscSendMsgToServer(pSql);
400
  if (code != TSDB_CODE_SUCCESS) {
S
slguan 已提交
401
    pRes->code = code;
H
hjxilinx 已提交
402
    tscQueueAsyncRes(pSql);
S
slguan 已提交
403
  }
H
hjxilinx 已提交
404 405
  
  return TSDB_CODE_SUCCESS;
S
slguan 已提交
406 407 408
}

int tscProcessSql(SSqlObj *pSql) {
409 410 411
  char *   name = NULL;
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;
412 413
  
  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
414
  STableMetaInfo *pTableMetaInfo = NULL;
415
  uint16_t        type = 0;
416

417
  if (pQueryInfo != NULL) {
H
hjxilinx 已提交
418
    pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
419 420
    if (pTableMetaInfo != NULL) {
      name = pTableMetaInfo->name;
421
    }
422

423
    type = pQueryInfo->type;
424
  
H
hjxilinx 已提交
425 426 427
    // while numOfTables equals to 0, it must be Heartbeat
    assert((pQueryInfo->numOfTables == 0 && pQueryInfo->command == TSDB_SQL_HB) ||
            pQueryInfo->numOfTables > 0);
S
slguan 已提交
428
  }
429

430
  tscTrace("%p SQL cmd:%d will be processed, name:%s, type:%d", pSql, pCmd->command, name, type);
H
hzcheng 已提交
431
  if (pSql->cmd.command < TSDB_SQL_MGMT) {
H
hjxilinx 已提交
432 433
    // the pTableMetaInfo cannot be NULL
    if (pTableMetaInfo == NULL) {
H
hjxilinx 已提交
434 435 436
      pSql->res.code = TSDB_CODE_OTHERS;
      return pSql->res.code;
    }
H
hzcheng 已提交
437
  } else if (pSql->cmd.command < TSDB_SQL_LOCAL) {
S
slguan 已提交
438
    pSql->ipList = tscMgmtIpSet;
H
hzcheng 已提交
439 440 441 442
  } else {  // local handler
    return (*tscProcessMsgRsp[pCmd->command])(pSql);
  }

S
slguan 已提交
443
  // todo handle async situation
444 445
  if (QUERY_IS_JOIN_QUERY(type)) {
    if ((pQueryInfo->type & TSDB_QUERY_TYPE_SUBQUERY) == 0) {
H
hjxilinx 已提交
446
      return tscHandleMasterJoinQuery(pSql);
S
slguan 已提交
447 448
    } else {
      // for first stage sub query, iterate all vnodes to get all timestamp
449
      if ((pQueryInfo->type & TSDB_QUERY_TYPE_JOIN_SEC_STAGE) != TSDB_QUERY_TYPE_JOIN_SEC_STAGE) {
S
slguan 已提交
450 451 452 453
        return doProcessSql(pSql);
      }
    }
  }
454
  
H
hjxilinx 已提交
455 456 457 458 459
  if (tscIsTwoStageSTableQuery(pQueryInfo, 0)) { // super table query
    tscHandleMasterSTableQuery(pSql);
    return pRes->code;
  } else if (pSql->fp == (void(*)())tscHandleMultivnodeInsert) {  // multi-vnodes insertion
    tscHandleMultivnodeInsert(pSql);
H
hzcheng 已提交
460 461
    return pSql->res.code;
  }
462
  
S
slguan 已提交
463 464
  return doProcessSql(pSql);
}
H
hzcheng 已提交
465

H
hjxilinx 已提交
466
void tscKillSTableQuery(SSqlObj *pSql) {
467 468 469
  SSqlCmd* pCmd = &pSql->cmd;
  
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
470
  if (!tscIsTwoStageSTableQuery(pQueryInfo, 0)) {
H
hzcheng 已提交
471 472 473 474 475 476
    return;
  }

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

S
slguan 已提交
477
    if (pSub == NULL) {
H
hzcheng 已提交
478 479
      continue;
    }
S
slguan 已提交
480

H
hzcheng 已提交
481 482 483 484 485
    /*
     * 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 已提交
486
    //taosStopRpcConn(pSql->pSubs[i]->thandle);
H
hzcheng 已提交
487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507
  }

  /*
   * 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();

  while (pSql->cmd.command != TSDB_SQL_RETRIEVE_METRIC && pSql->cmd.command != TSDB_SQL_RETRIEVE_EMPTY_RESULT) {
    taosMsleep(100);
    if (taosGetTimestampMs() - stime > MAX_WAITING_TIME) {
      break;
    }
  }

  tscTrace("%p metric query is cancelled", pSql);
}

508
int tscBuildRetrieveMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
509 510 511 512 513
  char *pMsg, *pStart;

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

514
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg *)pMsg;
S
slguan 已提交
515
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
S
slguan 已提交
516 517
  pMsg += sizeof(pSql->res.qhandle);

518
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
S
slguan 已提交
519
  pRetrieveMsg->free = htons(pQueryInfo->type);
520
  pMsg += sizeof(pQueryInfo->type);
H
hzcheng 已提交
521

522 523 524
  // todo valid the vgroupId at the client side
  if (UTIL_TABLE_IS_SUPERTABLE(pQueryInfo->pTableMetaInfo[0])) {
    SVgroupsInfo* pVgroupInfo = pQueryInfo->pTableMetaInfo[0]->vgroupList;
H
hjxilinx 已提交
525
    assert(pVgroupInfo->numOfVgroups == 1); // todo fix me
526
    
H
hjxilinx 已提交
527
    pRetrieveMsg->header.vgId = htonl(pVgroupInfo->vgroups[0].vgId);
528 529
  } else {
    STableMeta* pTableMeta = pQueryInfo->pTableMetaInfo[0]->pTableMeta;
H
hjxilinx 已提交
530
    pRetrieveMsg->header.vgId = htonl(pTableMeta->vgroupInfo.vgId);
531 532
  }
  
533 534 535 536
  pMsg += sizeof(SRetrieveTableMsg);
  
  pRetrieveMsg->header.contLen = htonl(pSql->cmd.payloadLen);
  
S
slguan 已提交
537
  pSql->cmd.msgType = TSDB_MSG_TYPE_RETRIEVE;
538
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
539 540
}

541
int tscBuildSubmitMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hjxilinx 已提交
542
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
H
hjxilinx 已提交
543
  STableMeta* pTableMeta = tscGetMetaInfo(pQueryInfo, 0)->pTableMeta;
H
hjxilinx 已提交
544
  
545 546 547 548
  char* pMsg = pSql->cmd.payload + tsRpcHeadSize;
  
  // NOTE: shell message size should not include SMsgDesc
  int32_t size = pSql->cmd.payloadLen - sizeof(SMsgDesc);
H
hjxilinx 已提交
549
  
550
  SMsgDesc* pMsgDesc = (SMsgDesc*) pMsg;
551 552
  
  pMsgDesc->numOfVnodes = htonl(1);       //todo set the right number of vnodes
553
  pMsg += sizeof(SMsgDesc);
H
hjxilinx 已提交
554
  
555
  SSubmitMsg *pShellMsg = (SSubmitMsg *)pMsg;
H
hjxilinx 已提交
556
  int32_t vgId = pTableMeta->vgroupInfo.vgId;
557
  
H
hjxilinx 已提交
558
  pShellMsg->header.vgId = htonl(vgId);
559
  pShellMsg->header.contLen = htonl(size);
560
  pShellMsg->length = pShellMsg->header.contLen;
H
hjxilinx 已提交
561
  
562
  pShellMsg->numOfBlocks = htonl(pSql->cmd.numOfTablesInSubmit);  // number of meters to be inserted
H
hzcheng 已提交
563

H
hjxilinx 已提交
564
  // pSql->cmd.payloadLen is set during copying data into payload
S
slguan 已提交
565
  pSql->cmd.msgType = TSDB_MSG_TYPE_SUBMIT;
566
  tscSetDnodeIpList(pSql, pTableMeta);
567
  
H
hjxilinx 已提交
568
  tscTrace("%p build submit msg, vgId:%d numOfVgroup:%d", pSql, vgId, htonl(pMsgDesc->numOfVnodes));
569
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
570 571 572 573 574 575
}

/*
 * for meter query, simply return the size <= 1k
 * for metric query, estimate size according to meter tags
 */
576
static int32_t tscEstimateQueryMsgSize(SSqlCmd *pCmd, int32_t clauseIndex) {
H
hzcheng 已提交
577
  const static int32_t MIN_QUERY_MSG_PKT_SIZE = TSDB_MAX_BYTES_PER_ROW * 5;
578
  SQueryInfo *         pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
H
hzcheng 已提交
579

580
  int32_t srcColListSize = taosArrayGetSize(pQueryInfo->colList) * sizeof(SColumnInfo);
H
hjxilinx 已提交
581 582 583 584
  
  size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
  int32_t exprSize = sizeof(SSqlFuncMsg) * numOfExprs;
  
H
hjxilinx 已提交
585
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
586 587

  // meter query without tags values
H
hjxilinx 已提交
588
  if (!UTIL_TABLE_IS_SUPERTABLE(pTableMetaInfo)) {
S
slguan 已提交
589
    return MIN_QUERY_MSG_PKT_SIZE + minMsgSize() + sizeof(SQueryTableMsg) + srcColListSize + exprSize;
H
hzcheng 已提交
590
  }
H
hjxilinx 已提交
591 592
  
  int32_t size = 4096;
S
slguan 已提交
593
  return size;
H
hzcheng 已提交
594 595
}

H
hjxilinx 已提交
596
static char *doSerializeTableInfo(SSqlObj *pSql, int32_t vgId, char *pMsg) {
597
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, pSql->cmd.clauseIndex, 0);
598

H
hjxilinx 已提交
599
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
600 601 602 603 604 605 606 607
  tscTrace("%p vgId:%d, query on table:%s, uid:%" PRIu64, pSql, vgId, pTableMetaInfo->name, pTableMeta->uid);
  
  STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
  pTableIdInfo->sid = htonl(pTableMeta->sid);
  pTableIdInfo->uid = htobe64(pTableMeta->uid);
  pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pTableMeta->uid));
  
  pMsg += sizeof(STableIdInfo);
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
  
625
  if (taosArrayGetSize(pQueryInfo->colList) <= 0) {
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 msgLen      = 0;
S
slguan 已提交
645
  int32_t numOfTables = 0;
646
  int32_t numOfTags   = taosArrayGetSize(pTableMetaInfo->tagColList);
H
hzcheng 已提交
647

H
hjxilinx 已提交
648
  if (UTIL_TABLE_IS_NOMRAL_TABLE(pTableMetaInfo)) {
S
slguan 已提交
649
    numOfTables = 1;
650
    tscSetDnodeIpList(pSql, pTableMeta);
H
hjxilinx 已提交
651
    pQueryMsg->head.vgId = htonl(pTableMeta->vgroupInfo.vgId);
H
hjxilinx 已提交
652
    tscTrace("%p queried tables:%d, table id: %s", pSql, 1, pTableMetaInfo->name);
653
  } else {  // query super table
654 655 656
    int32_t index = pTableMetaInfo->vgroupIndex;
    if (index < 0) {
      tscError("%p error vgroupIndex:%d", pSql, index);
H
hzcheng 已提交
657 658
      return -1;
    }
H
hjxilinx 已提交
659
    
660 661 662
    SCMVgroupInfo* pVgroupInfo = &pTableMetaInfo->vgroupList->vgroups[index];
    
    pSql->ipList.numOfIps = pVgroupInfo->numOfIps; // todo fix me
663 664
    pSql->ipList.port     = tsDnodeShellPort;
    pSql->ipList.inUse    = 0;
665
  
666 667 668
    for(int32_t i = 0; i < pVgroupInfo->numOfIps; ++i) {
      pSql->ipList.ip[i] = pVgroupInfo->ipAddr[i].ip;
    }
H
hjxilinx 已提交
669
    
670
    tscTrace("%p query on super table, numOfVgroup:%d, vgroupIndex:%d", pSql, pTableMetaInfo->vgroupList->numOfVgroups, index);
671
    
H
hjxilinx 已提交
672
    pQueryMsg->head.vgId = htonl(pVgroupInfo->vgId);
H
hjxilinx 已提交
673
    numOfTables = 1;
H
hzcheng 已提交
674 675
  }

676
  if (pQueryInfo->order.order == TSDB_ORDER_ASC) {
H
hjxilinx 已提交
677 678
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.skey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.ekey);
H
hzcheng 已提交
679
  } else {
H
hjxilinx 已提交
680 681
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.ekey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.skey);
H
hzcheng 已提交
682 683
  }

684
  pQueryMsg->numOfTables    = htonl(numOfTables);
685 686 687 688 689
  pQueryMsg->order          = htons(pQueryInfo->order.order);
  pQueryMsg->orderColId     = htons(pQueryInfo->order.orderColId);
  pQueryMsg->interpoType    = htons(pQueryInfo->interpoType);
  pQueryMsg->limit          = htobe64(pQueryInfo->limit.limit);
  pQueryMsg->offset         = htobe64(pQueryInfo->limit.offset);
690
  pQueryMsg->numOfCols      = htons(taosArrayGetSize(pQueryInfo->colList));
691 692
  pQueryMsg->intervalTime   = htobe64(pQueryInfo->intervalTime);
  pQueryMsg->slidingTime    = htobe64(pQueryInfo->slidingTime);
H
hjxilinx 已提交
693
  pQueryMsg->slidingTimeUnit = pQueryInfo->slidingTimeUnit;
694
  pQueryMsg->numOfGroupCols = htons(pQueryInfo->groupbyExpr.numOfGroupCols);
695
  pQueryMsg->numOfTags      = htonl(numOfTags);
weixin_48148422's avatar
weixin_48148422 已提交
696
  pQueryMsg->tagNameRelType = htons(pQueryInfo->tagCond.relType);
697
  pQueryMsg->queryType      = htons(pQueryInfo->type);
H
hjxilinx 已提交
698 699 700
  
  size_t numOfOutput = tscSqlExprNumOfExprs(pQueryInfo);
  pQueryMsg->numOfOutput = htons(numOfOutput);
H
hjxilinx 已提交
701 702
  if (numOfOutput < 0) {
    tscError("%p illegal value of number of output columns in query msg: %d", pSql, numOfOutput);
H
hzcheng 已提交
703 704 705 706
    return -1;
  }

  // set column list ids
707 708
  size_t numOfCols = taosArrayGetSize(pQueryInfo->colList);
  char *pMsg = (char *)(pQueryMsg->colList) + numOfCols * sizeof(SColumnInfo);
H
hjxilinx 已提交
709
  SSchema *pSchema = tscGetTableSchema(pTableMeta);
710
  
711 712 713
  for (int32_t i = 0; i < numOfCols; ++i) {
    SColumn *pCol = taosArrayGetP(pQueryInfo->colList, i);
    SSchema *pColSchema = &pSchema[pCol->colIndex.columnIndex];
H
hzcheng 已提交
714

715
    if (pCol->colIndex.columnIndex >= tscGetNumOfColumns(pTableMeta) || pColSchema->type < TSDB_DATA_TYPE_BOOL ||
716 717 718 719 720
        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);

721
      return TSDB_CODE_INVALID_SQL;
722
    }
H
hzcheng 已提交
723 724 725

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

S
slguan 已提交
729 730 731
    // 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 已提交
732

S
slguan 已提交
733
      SColumnFilterInfo *pFilterMsg = (SColumnFilterInfo *)pMsg;
734
      pFilterMsg->filterstr = htons(pColFilter->filterstr);
S
slguan 已提交
735 736

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

738
      if (pColFilter->filterstr) {
S
slguan 已提交
739 740 741 742 743 744 745 746 747 748
        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 已提交
749

S
slguan 已提交
750 751 752 753 754
      if (pColFilter->lowerRelOptr == TSDB_RELATION_INVALID && pColFilter->upperRelOptr == TSDB_RELATION_INVALID) {
        tscError("invalid filter info");
        return -1;
      }
    }
H
hzcheng 已提交
755 756
  }

H
hjxilinx 已提交
757
  SSqlFuncMsg *pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hjxilinx 已提交
758
  for (int32_t i = 0; i < tscSqlExprNumOfExprs(pQueryInfo); ++i) {
759
    SSqlExpr *pExpr = tscSqlExprGet(pQueryInfo, i);
H
hzcheng 已提交
760

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

767 768 769
    pSqlFuncExpr->colInfo.colId    = htons(pExpr->colInfo.colId);
    pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
    pSqlFuncExpr->colInfo.flag     = htons(pExpr->colInfo.flag);
H
hzcheng 已提交
770

771
    pSqlFuncExpr->functionId  = htons(pExpr->functionId);
H
hzcheng 已提交
772
    pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
H
hjxilinx 已提交
773
    pMsg += sizeof(SSqlFuncMsg);
H
hzcheng 已提交
774 775 776 777 778 779 780

    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);
781
        pMsg += pExpr->param[j].nLen;
H
hzcheng 已提交
782 783 784 785 786
      } else {
        pSqlFuncExpr->arg[j].argValue.i64 = htobe64(pExpr->param[j].i64Key);
      }
    }

H
hjxilinx 已提交
787
    pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hzcheng 已提交
788 789
  }

790
  // serialize the table info (sid, uid, tags)
H
hjxilinx 已提交
791
  pMsg = doSerializeTableInfo(pSql, htons(pQueryMsg->head.vgId), pMsg);
H
hzcheng 已提交
792

793
  SSqlGroupbyExpr *pGroupbyExpr = &pQueryInfo->groupbyExpr;
794
  if (pGroupbyExpr->numOfGroupCols > 0) {
S
slguan 已提交
795
    pQueryMsg->orderByIdx = htons(pGroupbyExpr->orderIndex);
H
hzcheng 已提交
796 797
    pQueryMsg->orderType = htons(pGroupbyExpr->orderType);

S
slguan 已提交
798
    for (int32_t j = 0; j < pGroupbyExpr->numOfGroupCols; ++j) {
799 800
      SColIndex* pCol = taosArrayGet(pGroupbyExpr->columnInfo, j);
  
S
slguan 已提交
801 802 803
      *((int16_t *)pMsg) = pCol->colId;
      pMsg += sizeof(pCol->colId);

804 805
      *((int16_t *)pMsg) += pCol->colIndex;
      pMsg += sizeof(pCol->colIndex);
S
slguan 已提交
806 807 808

      *((int16_t *)pMsg) += pCol->flag;
      pMsg += sizeof(pCol->flag);
H
hjxilinx 已提交
809 810 811
      
      memcpy(pMsg, pCol->name, tListLen(pCol->name));
      pMsg += tListLen(pCol->name);
S
slguan 已提交
812 813 814
    }
  }

815
  if (pQueryInfo->interpoType != TSDB_INTERPO_NONE) {
H
hjxilinx 已提交
816
    for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
817 818
      *((int64_t *)pMsg) = htobe64(pQueryInfo->defaultVal[i]);
      pMsg += sizeof(pQueryInfo->defaultVal[0]);
S
slguan 已提交
819 820
    }
  }
821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851
  
  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) {
      SColumn *pCol = taosArrayGetP(pQueryInfo->colList, i);
      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 已提交
852 853 854 855 856 857

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

858
  if (pQueryInfo->tsBuf != NULL) {
H
hjxilinx 已提交
859
    STSVnodeBlockInfo *pBlockInfo = tsBufGetVnodeBlockInfo(pQueryInfo->tsBuf, pTableMetaInfo->vgroupIndex);
860
    assert(QUERY_IS_JOIN_QUERY(pQueryInfo->type) && pBlockInfo != NULL);  // this query should not be sent
S
slguan 已提交
861 862

    // todo refactor
863 864
    fseek(pQueryInfo->tsBuf->f, pBlockInfo->offset, SEEK_SET);
    fread(pMsg, pBlockInfo->compLen, 1, pQueryInfo->tsBuf->f);
S
slguan 已提交
865 866 867 868

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

S
slguan 已提交
871 872
  pQueryMsg->tsLen = htonl(tsLen);
  pQueryMsg->tsNumOfBlocks = htonl(numOfBlocks);
873 874
  if (pQueryInfo->tsBuf != NULL) {
    pQueryMsg->tsOrder = htonl(pQueryInfo->tsBuf->tsOrder);
H
hzcheng 已提交
875 876
  }

H
hjxilinx 已提交
877
  // serialize tag column query condition
878
  if (pQueryInfo->tagCond.pCond != NULL && taosArrayGetSize(pQueryInfo->tagCond.pCond) > 0) {
879 880
    STagCond* pTagCond = &pQueryInfo->tagCond;
    
881
    SCond *pCond = tsGetSTableQueryCond(pTagCond, pTableMeta->uid);
882
    if (pCond != NULL && pCond->cond != NULL) {
883 884
      pQueryMsg->tagCondLen = htons(pCond->len);
      memcpy(pMsg, pCond->cond, pCond->len);
885
      
886
      pMsg += pCond->len;
887 888 889
    }
  }
  
weixin_48148422's avatar
weixin_48148422 已提交
890 891 892 893 894 895 896 897
  if (pQueryInfo->tagCond.tbnameCond.cond == NULL) {
    *pMsg = 0;
    pMsg++;
  } else {
    strcpy(pMsg, pQueryInfo->tagCond.tbnameCond.cond);
    pMsg += strlen(pQueryInfo->tagCond.tbnameCond.cond) + 1;
  }

H
hjxilinx 已提交
898
  // tbname in/like query expression should be sent to mgmt node
H
hzcheng 已提交
899 900 901 902
  msgLen = pMsg - pStart;

  tscTrace("%p msg built success,len:%d bytes", pSql, msgLen);
  pCmd->payloadLen = msgLen;
S
slguan 已提交
903
  pSql->cmd.msgType = TSDB_MSG_TYPE_QUERY;
H
hjxilinx 已提交
904
  
905
  pQueryMsg->head.contLen = htonl(msgLen);
H
hzcheng 已提交
906
  assert(msgLen + minMsgSize() <= size);
907 908

  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
909 910
}

911 912
int32_t tscBuildCreateDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
913
  pCmd->payloadLen = sizeof(SCMCreateDbMsg);
S
slguan 已提交
914
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DB;
H
hzcheng 已提交
915

916
  SCMCreateDbMsg *pCreateDbMsg = (SCMCreateDbMsg*)pCmd->payload;
917

918
  assert(pCmd->numOfClause == 1);
919
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
920
  strncpy(pCreateDbMsg->db, pTableMetaInfo->name, tListLen(pCreateDbMsg->db));
H
hzcheng 已提交
921

922
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
923 924
}

925 926
int32_t tscBuildCreateDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
927
  pCmd->payloadLen = sizeof(SCMCreateDnodeMsg);
S
slguan 已提交
928 929 930 931
  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 已提交
932

933
  SCMCreateDnodeMsg *pCreate = (SCMCreateDnodeMsg *)pCmd->payload;
934
  strncpy(pCreate->ip, pInfo->pDCLInfo->a[0].z, pInfo->pDCLInfo->a[0].n);
S
slguan 已提交
935
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DNODE;
H
hzcheng 已提交
936

937
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
938 939
}

940 941
int32_t tscBuildAcctMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
942
  pCmd->payloadLen = sizeof(SCMCreateAcctMsg);
S
slguan 已提交
943 944 945 946
  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 已提交
947

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

950 951
  SSQLToken *pName = &pInfo->pDCLInfo->user.user;
  SSQLToken *pPwd = &pInfo->pDCLInfo->user.passwd;
H
hzcheng 已提交
952

953 954
  strncpy(pAlterMsg->user, pName->z, pName->n);
  strncpy(pAlterMsg->pass, pPwd->z, pPwd->n);
S
slguan 已提交
955

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

958 959 960 961 962 963 964 965
  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 已提交
966

967 968 969 970 971 972 973 974 975 976 977 978 979
  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 已提交
980

S
slguan 已提交
981
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_ACCT;
982
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
983 984
}

985 986
int32_t tscBuildUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
987
  pCmd->payloadLen = sizeof(SCMCreateUserMsg);
S
slguan 已提交
988

S
slguan 已提交
989 990 991 992 993
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

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

996 997 998
  SUserInfo *pUser = &pInfo->pDCLInfo->user;
  strncpy(pAlterMsg->user, pUser->user.z, pUser->user.n);
  pAlterMsg->flag = pUser->type;
H
hzcheng 已提交
999

1000 1001 1002 1003
  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 已提交
1004 1005
  } else { // create user password info
    strncpy(pAlterMsg->pass, pUser->passwd.z, pUser->passwd.n);
1006
  }
H
hzcheng 已提交
1007

1008
  if (pUser->type == TSDB_ALTER_USER_PASSWD || pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
S
slguan 已提交
1009
    pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_USER;
1010
  } else {
S
slguan 已提交
1011
    pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_USER;
1012
  }
H
hzcheng 已提交
1013

1014
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1015 1016
}

1017 1018
int32_t tscBuildCfgDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1019
  pCmd->payloadLen = sizeof(SCMCfgDnodeMsg);
H
hzcheng 已提交
1020

S
slguan 已提交
1021 1022 1023 1024
  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 已提交
1025

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

1030 1031
int32_t tscBuildDropDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1032
  pCmd->payloadLen = sizeof(SCMDropDbMsg);
H
hzcheng 已提交
1033

S
slguan 已提交
1034 1035 1036 1037 1038
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

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

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

S
slguan 已提交
1045
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DB;
1046
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1047 1048
}

1049 1050
int32_t tscBuildDropTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1051
  pCmd->payloadLen = sizeof(SCMDropTableMsg);
H
hzcheng 已提交
1052

S
slguan 已提交
1053 1054 1055 1056
  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 已提交
1057

1058
  SCMDropTableMsg *pDropTableMsg = (SCMDropTableMsg*)pCmd->payload;
1059
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1060
  strcpy(pDropTableMsg->tableId, pTableMetaInfo->name);
S
slguan 已提交
1061
  pDropTableMsg->igNotExists = pInfo->pDCLInfo->existsCheck ? 1 : 0;
H
hzcheng 已提交
1062

S
slguan 已提交
1063
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_TABLE;
1064
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1065 1066
}

1067
int32_t tscBuildDropDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1068
  SSqlCmd *pCmd = &pSql->cmd;
1069
  pCmd->payloadLen = sizeof(SCMDropDnodeMsg);
S
slguan 已提交
1070 1071 1072 1073
  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 已提交
1074

1075
  SCMDropDnodeMsg *pDrop = (SCMDropDnodeMsg *)pCmd->payload;
1076
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1077
  strcpy(pDrop->ip, pTableMetaInfo->name);
S
slguan 已提交
1078
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DNODE;
H
hzcheng 已提交
1079

1080
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1081 1082
}

S
[TD-16]  
slguan 已提交
1083
int32_t tscBuildDropUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1084
  SSqlCmd *pCmd = &pSql->cmd;
1085
  pCmd->payloadLen = sizeof(SCMDropUserMsg);
S
slguan 已提交
1086
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_USER;
H
hzcheng 已提交
1087

S
slguan 已提交
1088 1089 1090 1091
  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 已提交
1092

1093
  SCMDropUserMsg *pDropMsg = (SCMDropUserMsg*)pCmd->payload;
1094
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1095
  strcpy(pDropMsg->user, pTableMetaInfo->name);
H
hzcheng 已提交
1096

1097
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1098 1099
}

S
[TD-16]  
slguan 已提交
1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
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;
}

1117 1118
int32_t tscBuildUseDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1119
  pCmd->payloadLen = sizeof(SCMUseDbMsg);
H
hzcheng 已提交
1120

S
slguan 已提交
1121 1122 1123 1124
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
1125

1126
  SCMUseDbMsg *pUseDbMsg = (SCMUseDbMsg*)pCmd->payload;
1127
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1128
  strcpy(pUseDbMsg->db, pTableMetaInfo->name);
S
slguan 已提交
1129
  pCmd->msgType = TSDB_MSG_TYPE_CM_USE_DB;
H
hzcheng 已提交
1130

1131
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1132 1133
}

1134
int32_t tscBuildShowMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1135
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1136
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1137
  pCmd->msgType = TSDB_MSG_TYPE_CM_SHOW;
1138
  pCmd->payloadLen = sizeof(SCMShowMsg) + 100;
H
hzcheng 已提交
1139

S
slguan 已提交
1140 1141 1142
  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 已提交
1143
  }
H
hzcheng 已提交
1144

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

1147
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1148
  size_t nameLen = strlen(pTableMetaInfo->name);
S
slguan 已提交
1149
  if (nameLen > 0) {
H
hjxilinx 已提交
1150
    strcpy(pShowMsg->db, pTableMetaInfo->name);  // prefix is set here
H
hzcheng 已提交
1151
  } else {
S
slguan 已提交
1152
    strcpy(pShowMsg->db, pObj->db);
H
hzcheng 已提交
1153 1154
  }

1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166
  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 已提交
1167

1168 1169 1170 1171
    strncpy(pShowMsg->payload, pIpAddr->z, pIpAddr->n);
    pShowMsg->payloadLen = htons(pIpAddr->n);
  }

1172
  pCmd->payloadLen = sizeof(SCMShowMsg) + pShowMsg->payloadLen;
1173
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1174 1175
}

1176
int32_t tscBuildKillMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1177
  SSqlCmd *pCmd = &pSql->cmd;
1178
  pCmd->payloadLen = sizeof(SCMKillQueryMsg);
H
hzcheng 已提交
1179

S
slguan 已提交
1180 1181 1182 1183
  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 已提交
1184

1185
  SCMKillQueryMsg *pKill = (SCMKillQueryMsg*)pCmd->payload;
1186 1187 1188
  strncpy(pKill->queryId, pInfo->pDCLInfo->ip.z, pInfo->pDCLInfo->ip.n);
  switch (pCmd->command) {
    case TSDB_SQL_KILL_QUERY:
S
slguan 已提交
1189
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_QUERY;
1190 1191
      break;
    case TSDB_SQL_KILL_CONNECTION:
S
slguan 已提交
1192
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_CONN;
1193 1194
      break;
    case TSDB_SQL_KILL_STREAM:
S
slguan 已提交
1195
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_STREAM;
1196 1197 1198
      break;
  }
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1199 1200
}

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

1204
  int32_t size = minMsgSize() + sizeof(SMgmtHead) + sizeof(SCMCreateTableMsg);
H
hzcheng 已提交
1205

1206
  SCreateTableSQL *pCreateTableInfo = pInfo->pCreateTableInfo;
1207
  if (pCreateTableInfo->type == TSQL_CREATE_TABLE_FROM_STABLE) {
H
hzcheng 已提交
1208 1209
    size += sizeof(STagData);
  } else {
S
slguan 已提交
1210
    size += sizeof(SSchema) * (pCmd->numOfCols + pCmd->count);
H
hzcheng 已提交
1211
  }
1212

1213 1214 1215
  if (pCreateTableInfo->pSelect != NULL) {
    size += (pCreateTableInfo->pSelect->selectToken.n + 1);
  }
H
hzcheng 已提交
1216 1217 1218 1219

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1220
int tscBuildCreateTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1221
  int              msgLen = 0;
S
slguan 已提交
1222
  SSchema *        pSchema;
H
hzcheng 已提交
1223
  int              size = 0;
1224 1225 1226
  SSqlCmd *pCmd = &pSql->cmd;

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
1227
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1228 1229

  // Reallocate the payload size
1230
  size = tscEstimateCreateTableMsgLength(pSql, pInfo);
S
slguan 已提交
1231 1232
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for create table msg", pSql);
1233
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
S
slguan 已提交
1234
  }
H
hzcheng 已提交
1235 1236


1237
  SCMCreateTableMsg *pCreateTableMsg = (SCMCreateTableMsg *)pCmd->payload;
H
hjxilinx 已提交
1238
  strcpy(pCreateTableMsg->tableId, pTableMetaInfo->name);
1239 1240

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

1243 1244 1245
  SCreateTableSQL *pCreateTable = pInfo->pCreateTableInfo;

  pCreateTableMsg->igExists = pCreateTable->existCheck ? 1 : 0;
H
hzcheng 已提交
1246 1247 1248 1249
  pCreateTableMsg->numOfColumns = htons(pCmd->numOfCols);
  pCreateTableMsg->numOfTags = htons(pCmd->count);

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

1252 1253 1254
  int8_t type = pInfo->pCreateTableInfo->type;
  if (type == TSQL_CREATE_TABLE_FROM_STABLE) {  // create by using super table, tags value
    memcpy(pMsg, &pInfo->pCreateTableInfo->usingInfo.tagdata, sizeof(STagData));
H
hzcheng 已提交
1255
    pMsg += sizeof(STagData);
1256
  } else {  // create (super) table
1257
    pSchema = (SSchema *)pCreateTableMsg->schema;
1258

H
hzcheng 已提交
1259
    for (int i = 0; i < pCmd->numOfCols + pCmd->count; ++i) {
H
hjxilinx 已提交
1260
      TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1261 1262 1263 1264

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

H
hzcheng 已提交
1266 1267 1268 1269
      pSchema++;
    }

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

1273 1274 1275
      strncpy(pMsg, pQuerySql->selectToken.z, pQuerySql->selectToken.n + 1);
      pCreateTableMsg->sqlLen = htons(pQuerySql->selectToken.n + 1);
      pMsg += pQuerySql->selectToken.n + 1;
H
hzcheng 已提交
1276 1277 1278
    }
  }

H
hjxilinx 已提交
1279
  tscFieldInfoClear(&pQueryInfo->fieldsInfo);
H
hzcheng 已提交
1280

S
slguan 已提交
1281
  msgLen = pMsg - (char*)pCreateTableMsg;
S
slguan 已提交
1282
  pCreateTableMsg->contLen = htonl(msgLen);
H
hzcheng 已提交
1283
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1284
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_TABLE;
H
hzcheng 已提交
1285 1286

  assert(msgLen + minMsgSize() <= size);
1287
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1288 1289 1290
}

int tscEstimateAlterTableMsgLength(SSqlCmd *pCmd) {
1291
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
1292
  return minMsgSize() + sizeof(SMgmtHead) + sizeof(SCMAlterTableMsg) + sizeof(SSchema) * tscNumOfFields(pQueryInfo) +
H
hzcheng 已提交
1293 1294 1295
         TSDB_EXTRA_PAYLOAD_SIZE;
}

1296
int tscBuildAlterTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1297
  SCMAlterTableMsg *pAlterTableMsg;
1298
  char *          pMsg;
H
hzcheng 已提交
1299 1300 1301
  int             msgLen = 0;
  int             size = 0;

1302 1303 1304
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);

H
hjxilinx 已提交
1305
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1306 1307

  size = tscEstimateAlterTableMsgLength(pCmd);
S
slguan 已提交
1308 1309 1310 1311
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for alter table msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1312

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

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

1317 1318
  SAlterTableSQL *pAlterInfo = pInfo->pAlterInfo;

H
hjxilinx 已提交
1319
  strcpy(pAlterTableMsg->tableId, pTableMetaInfo->name);
1320
  pAlterTableMsg->type = htons(pAlterInfo->type);
1321

1322
  pAlterTableMsg->numOfCols = htons(tscNumOfFields(pQueryInfo));
1323
  memcpy(pAlterTableMsg->tagVal, pAlterInfo->tagData.data, TSDB_MAX_TAGS_LEN);
H
hzcheng 已提交
1324

S
slguan 已提交
1325
  SSchema *pSchema = pAlterTableMsg->schema;
1326
  for (int i = 0; i < tscNumOfFields(pQueryInfo); ++i) {
H
hjxilinx 已提交
1327
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1328 1329 1330 1331 1332 1333 1334 1335 1336

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

  pMsg = (char *)pSchema;

S
slguan 已提交
1337
  msgLen = pMsg - (char*)pAlterTableMsg;
H
hzcheng 已提交
1338
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1339
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_TABLE;
H
hzcheng 已提交
1340 1341

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

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

1346
int tscAlterDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1347
  SSqlCmd *pCmd = &pSql->cmd;
1348
  pCmd->payloadLen = sizeof(SCMAlterDbMsg);
S
slguan 已提交
1349
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_DB;
H
hzcheng 已提交
1350

S
slguan 已提交
1351 1352 1353 1354
  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 已提交
1355

1356
  SCMAlterDbMsg *pAlterDbMsg = (SCMAlterDbMsg*)pCmd->payload;
1357
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1358
  strcpy(pAlterDbMsg->db, pTableMetaInfo->name);
H
hzcheng 已提交
1359

1360
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1361 1362
}

1363
int tscBuildRetrieveFromMgmtMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1364 1365 1366
  SSqlCmd *pCmd = &pSql->cmd;
  pCmd->msgType = TSDB_MSG_TYPE_RETRIEVE;
  pCmd->payloadLen = sizeof(SRetrieveTableMsg);
S
slguan 已提交
1367

S
slguan 已提交
1368 1369 1370
  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 已提交
1371
  }
S
slguan 已提交
1372

S
slguan 已提交
1373 1374 1375 1376
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg*)pCmd->payload;
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
  pRetrieveMsg->free = htons(pQueryInfo->type);
H
hzcheng 已提交
1377

1378
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1379 1380
}

1381
static int tscSetResultPointer(SQueryInfo *pQueryInfo, SSqlRes *pRes) {
1382
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
H
hzcheng 已提交
1383 1384 1385
    return pRes->code;
  }

H
hjxilinx 已提交
1386
  for (int i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
H
hjxilinx 已提交
1387
    int16_t offset = tscFieldInfoGetOffset(pQueryInfo, i);
H
hjxilinx 已提交
1388
    pRes->tsrow[i] = ((char*) pRes->data + offset * pRes->numOfRows);
H
hzcheng 已提交
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
  }

  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;
1403

1404
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
1405

H
hzcheng 已提交
1406 1407 1408 1409 1410 1411 1412
  pRes->code = TSDB_CODE_SUCCESS;

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

1413
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1414
  } else {
S
slguan 已提交
1415
    tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430
  }

  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 已提交
1431
  SSqlCmd *       pCmd = &pSql->cmd;
1432
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hzcheng 已提交
1433

H
hjxilinx 已提交
1434
  STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
H
hjxilinx 已提交
1435 1436
  
  int32_t numOfRes = tinfo.numOfColumns + tinfo.numOfTags;
H
hzcheng 已提交
1437 1438 1439 1440
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

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

1443 1444
//  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
//  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
S
slguan 已提交
1445 1446

  int32_t numOfRes = 0;
H
hjxilinx 已提交
1447
#if 0
1448
  if (tscSqlExprGet(pQueryInfo, 0)->functionId == TSDB_FUNC_TAGPRJ) {
H
hjxilinx 已提交
1449
    numOfRes = pTableMetaInfo->pMetricMeta->numOfTables;
H
hzcheng 已提交
1450 1451 1452
  } else {
    numOfRes = 1;  // for count function, there is only one output.
  }
H
hjxilinx 已提交
1453 1454 1455
  
#endif

H
hzcheng 已提交
1456 1457 1458 1459 1460 1461 1462
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

int tscProcessRetrieveMetricRsp(SSqlObj *pSql) {
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

H
hjxilinx 已提交
1463 1464
  pRes->code = tscDoLocalreduce(pSql);
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hzcheng 已提交
1465 1466

  if (pRes->code == TSDB_CODE_SUCCESS && pRes->numOfRows > 0) {
1467
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1468 1469 1470 1471
  }

  pRes->row = 0;

1472
  uint8_t code = pRes->code;
H
hjxilinx 已提交
1473 1474 1475 1476
  if (pRes->code == TSDB_CODE_SUCCESS) {
    (*pSql->fp)(pSql->param, pSql, pRes->numOfRows);
  } else {
    tscQueueAsyncRes(pSql);
H
hzcheng 已提交
1477 1478 1479 1480 1481
  }

  return code;
}

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

1484
int tscBuildConnectMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1485
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1486
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1487
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONNECT;
1488
  pCmd->payloadLen = sizeof(SCMConnectMsg);
H
hzcheng 已提交
1489

S
slguan 已提交
1490 1491 1492 1493 1494
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

1495
  SCMConnectMsg *pConnect = (SCMConnectMsg*)pCmd->payload;
H
hzcheng 已提交
1496 1497 1498 1499 1500

  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 已提交
1501
  strcpy(pConnect->clientVersion, version);
S
slguan 已提交
1502
  strcpy(pConnect->msgVersion, "");
H
hzcheng 已提交
1503

1504
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1505 1506
}

H
hjxilinx 已提交
1507
int tscBuildTableMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1508
  SCMTableInfoMsg *pInfoMsg;
S
slguan 已提交
1509
  char *         pMsg;
H
hzcheng 已提交
1510 1511 1512 1513 1514
  int            msgLen = 0;

  char *tmpData = 0;
  if (pSql->cmd.allocSize > 0) {
    tmpData = calloc(1, pSql->cmd.allocSize);
1515 1516 1517 1518
    if (NULL == tmpData) {
      return TSDB_CODE_CLI_OUT_OF_MEMORY;
    }

H
hzcheng 已提交
1519 1520 1521 1522
    // STagData is in binary format, strncpy is not available
    memcpy(tmpData, pSql->cmd.payload, pSql->cmd.allocSize);
  }

1523 1524 1525
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

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

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

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

H
hjxilinx 已提交
1534
  if (pSql->cmd.autoCreated) {
H
hzcheng 已提交
1535 1536 1537 1538
    memcpy(pInfoMsg->tags, tmpData, sizeof(STagData));
    pMsg += sizeof(STagData);
  }

H
hjxilinx 已提交
1539
  pCmd->payloadLen = pMsg - (char*)pInfoMsg;;
S
slguan 已提交
1540
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLE_META;
H
hzcheng 已提交
1541 1542 1543 1544

  tfree(tmpData);

  assert(msgLen + minMsgSize() <= pCmd->allocSize);
1545
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1546 1547
}

S
slguan 已提交
1548
/**
1549
 *  multi table meta req pkg format:
1550
 *  | SMgmtHead | SCMMultiTableInfoMsg | tableId0 | tableId1 | tableId2 | ......
S
slguan 已提交
1551 1552
 *      no used         4B
 **/
1553
int tscBuildMultiMeterMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
  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 已提交
1566
  memset(pMgmt->db, 0, TSDB_TABLE_ID_LEN);  // server don't need the db
S
slguan 已提交
1567

1568
  SCMMultiTableInfoMsg *pInfoMsg = (SCMMultiTableInfoMsg *)(pCmd->payload + tsRpcHeadSize + sizeof(SMgmtHead));
S
slguan 已提交
1569
  pInfoMsg->numOfTables = htonl((int32_t)pCmd->count);
S
slguan 已提交
1570 1571

  if (pCmd->payloadLen > 0) {
S
slguan 已提交
1572
    memcpy(pInfoMsg->tableIds, tmpData, pCmd->payloadLen);
S
slguan 已提交
1573 1574 1575 1576
  }

  tfree(tmpData);

1577
  pCmd->payloadLen += sizeof(SMgmtHead) + sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1578
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLES_META;
S
slguan 已提交
1579 1580 1581

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

S
slguan 已提交
1582
  tscTrace("%p build load multi-metermeta msg completed, numOfTables:%d, msg size:%d", pSql, pCmd->count,
S
slguan 已提交
1583 1584 1585 1586 1587
           pCmd->payloadLen);

  return pCmd->payloadLen;
}

1588
static UNUSED_FUNC int32_t tscEstimateMetricMetaMsgSize(SSqlCmd *pCmd) {
H
hzcheng 已提交
1589
  const int32_t defaultSize =
S
slguan 已提交
1590
      minMsgSize() + sizeof(SSuperTableMetaMsg) + sizeof(SMgmtHead) + sizeof(int16_t) * TSDB_MAX_TAGS;
1591
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hzcheng 已提交
1592

S
slguan 已提交
1593
  int32_t n = 0;
1594 1595 1596 1597
  size_t size = taosArrayGetSize(pQueryInfo->tagCond.pCond);
  for (int32_t i = 0; i < size; ++i) {
    assert(0);
//    n += strlen(pQueryInfo->tagCond.cond[i].cond);
H
hzcheng 已提交
1598
  }
S
slguan 已提交
1599

H
hjxilinx 已提交
1600
  int32_t tagLen = n * TSDB_NCHAR_SIZE;
1601 1602
  if (pQueryInfo->tagCond.tbnameCond.cond != NULL) {
    tagLen += strlen(pQueryInfo->tagCond.tbnameCond.cond) * TSDB_NCHAR_SIZE;
H
hjxilinx 已提交
1603
  }
1604

S
slguan 已提交
1605
  int32_t joinCondLen = (TSDB_TABLE_ID_LEN + sizeof(int16_t)) * 2;
S
slguan 已提交
1606
  int32_t elemSize = sizeof(SSuperTableMetaElemMsg) * pQueryInfo->numOfTables;
H
hjxilinx 已提交
1607
  
1608
  int32_t colSize = pQueryInfo->groupbyExpr.numOfGroupCols*sizeof(SColIndex);
S
slguan 已提交
1609

H
hjxilinx 已提交
1610
  int32_t len = tagLen + joinCondLen + elemSize + colSize + defaultSize;
S
slguan 已提交
1611 1612

  return MAX(len, TSDB_DEFAULT_PAYLOAD_SIZE);
H
hzcheng 已提交
1613 1614
}

H
hjxilinx 已提交
1615 1616 1617
int tscBuildSTableVgroupMsg(SSqlObj *pSql, SSqlInfo *pInfo) {

#if 0
S
slguan 已提交
1618
  SSuperTableMetaMsg *pMetaMsg;
H
hzcheng 已提交
1619 1620
  char *          pMsg, *pStart;
  int             msgLen = 0;
S
slguan 已提交
1621
  int             tableIndex = 0;
H
hzcheng 已提交
1622

1623 1624 1625
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

1626
  STagCond *pTagCond = &pQueryInfo->tagCond;
S
slguan 已提交
1627

H
hjxilinx 已提交
1628
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
H
hzcheng 已提交
1629 1630

  int32_t size = tscEstimateMetricMetaMsgSize(pCmd);
S
slguan 已提交
1631 1632 1633 1634
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for metric meter msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1635 1636 1637 1638 1639

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

  SMgmtHead *pMgmt = (SMgmtHead *)pMsg;
H
hjxilinx 已提交
1640
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pMgmt->db);
H
hzcheng 已提交
1641 1642 1643

  pMsg += sizeof(SMgmtHead);

S
slguan 已提交
1644
  pMetaMsg = (SSuperTableMetaMsg *)pMsg;
S
slguan 已提交
1645
  pMetaMsg->numOfTables = htonl(pQueryInfo->numOfTables);
S
slguan 已提交
1646

S
slguan 已提交
1647
  pMsg += sizeof(SSuperTableMetaMsg);
S
slguan 已提交
1648 1649 1650 1651 1652

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

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

S
slguan 已提交
1655
  memcpy(pMsg, pTagCond->joinInfo.left.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1656
  pMsg += TSDB_TABLE_ID_LEN;
H
hzcheng 已提交
1657

S
slguan 已提交
1658 1659 1660
  *(int16_t *)pMsg = pTagCond->joinInfo.left.tagCol;
  pMsg += sizeof(int16_t);

S
slguan 已提交
1661
  memcpy(pMsg, pTagCond->joinInfo.right.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1662
  pMsg += TSDB_TABLE_ID_LEN;
S
slguan 已提交
1663 1664 1665 1666

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

1667
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
1668
    pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
H
hjxilinx 已提交
1669
    uint64_t uid = pTableMetaInfo->pTableMeta->uid;
S
slguan 已提交
1670 1671 1672 1673

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

S
slguan 已提交
1674 1675
    SSuperTableMetaElemMsg *pElem = (SSuperTableMetaElemMsg *)pMsg;
    pMsg += sizeof(SSuperTableMetaElemMsg);
S
slguan 已提交
1676 1677 1678 1679

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

H
hjxilinx 已提交
1684
        bool ret = taosMbsToUcs4(pCond->cond, condLen, pMsg, condLen * TSDB_NCHAR_SIZE);
S
slguan 已提交
1685
        if (!ret) {
1686
          tscError("%p mbs to ucs4 failed:%s", pSql, tsGetSTableQueryCond(pTagCond, uid));
S
slguan 已提交
1687 1688 1689
          return 0;
        }
      }
H
hzcheng 已提交
1690 1691
    }

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

S
slguan 已提交
1694 1695 1696
    offset = pMsg - (char *)pMetaMsg;
    pElem->cond = htonl(offset);
    pMsg += condLen * TSDB_NCHAR_SIZE;
H
hzcheng 已提交
1697

S
slguan 已提交
1698 1699 1700
    pElem->rel = htons(pTagCond->relType);
    if (pTagCond->tbnameCond.uid == uid) {
      offset = pMsg - (char *)pMetaMsg;
H
hzcheng 已提交
1701

S
slguan 已提交
1702
      pElem->tableCond = htonl(offset);
H
hjxilinx 已提交
1703 1704 1705 1706 1707 1708 1709
      
      uint32_t len = 0;
      if (pTagCond->tbnameCond.cond != NULL) {
        len = strlen(pTagCond->tbnameCond.cond);
        memcpy(pMsg, pTagCond->tbnameCond.cond, len);
      }
      
H
hjxilinx 已提交
1710 1711
      pElem->tableCondLen = htonl(len);
      pMsg += len;
S
slguan 已提交
1712 1713
    }

1714
    SSqlGroupbyExpr *pGroupby = &pQueryInfo->groupbyExpr;
S
slguan 已提交
1715

H
hjxilinx 已提交
1716
    if (pGroupby->tableIndex != i && pGroupby->numOfGroupCols > 0) {
S
slguan 已提交
1717 1718 1719 1720 1721
      pElem->orderType = 0;
      pElem->orderIndex = 0;
      pElem->numOfGroupCols = 0;
    } else {
      pElem->numOfGroupCols = htons(pGroupby->numOfGroupCols);
H
hjxilinx 已提交
1722 1723
      for (int32_t j = 0; j < pTableMetaInfo->numOfTags; ++j) {
        pElem->tagCols[j] = htons(pTableMetaInfo->tagColumnIndex[j]);
S
slguan 已提交
1724 1725 1726 1727 1728 1729 1730 1731
      }

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

        pElem->groupbyTagColumnList = htonl(offset);
1732
        for (int32_t j = 0; j < pQueryInfo->groupbyExpr.numOfGroupCols; ++j) {
1733 1734
          SColIndex *pCol = &pQueryInfo->groupbyExpr.columnInfo[j];
          SColIndex *pDestCol = (SColIndex *)pMsg;
1735

H
hjxilinx 已提交
1736
          pDestCol->colIdxInBuf = 0;
1737
          pDestCol->colIndex = htons(pCol->colIndex);
H
hjxilinx 已提交
1738 1739
          pDestCol->colId = htons(pDestCol->colId);
          pDestCol->flag = htons(pDestCol->flag);
H
hjxilinx 已提交
1740
          strncpy(pDestCol->name, pCol->name, tListLen(pCol->name));
1741

1742
          pMsg += sizeof(SColIndex);
S
slguan 已提交
1743 1744
        }
      }
H
hzcheng 已提交
1745
    }
S
slguan 已提交
1746

H
hjxilinx 已提交
1747 1748
    strcpy(pElem->tableId, pTableMetaInfo->name);
    pElem->numOfTags = htons(pTableMetaInfo->numOfTags);
S
slguan 已提交
1749 1750 1751

    int16_t len = pMsg - (char *)pElem;
    pElem->elemLen = htons(len);  // redundant data for integrate check
H
hzcheng 已提交
1752 1753 1754 1755
  }

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1756
  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
H
hzcheng 已提交
1757
  assert(msgLen + minMsgSize() <= size);
H
hjxilinx 已提交
1758
#endif
1759
  
H
hjxilinx 已提交
1760 1761 1762 1763 1764 1765 1766 1767 1768 1769
  SSqlCmd *pCmd = &pSql->cmd;

  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);

  SCMSTableVgroupMsg *pStableVgroupMsg = (SCMSTableVgroupMsg *) pCmd->payload;
  strncpy(pStableVgroupMsg->tableId, pTableMetaInfo->name, tListLen(pStableVgroupMsg->tableId));

  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
  pCmd->payloadLen = sizeof(SCMSTableVgroupMsg);

1770
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1771 1772
}

1773
int tscEstimateHeartBeatMsgLength(SSqlObj *pSql) {
H
hzcheng 已提交
1774 1775 1776 1777
  int      size = 0;
  STscObj *pObj = pSql->pTscObj;

  size += tsRpcHeadSize + sizeof(SMgmtHead);
S
slguan 已提交
1778
  size += sizeof(SQqueryList);
H
hzcheng 已提交
1779 1780 1781

  SSqlObj *tpSql = pObj->sqlList;
  while (tpSql) {
S
slguan 已提交
1782
    size += sizeof(SQueryDesc);
H
hzcheng 已提交
1783 1784 1785
    tpSql = tpSql->next;
  }

S
slguan 已提交
1786
  size += sizeof(SStreamList);
H
hzcheng 已提交
1787 1788
  SSqlStream *pStream = pObj->streamList;
  while (pStream) {
S
slguan 已提交
1789
    size += sizeof(SStreamDesc);
H
hzcheng 已提交
1790 1791 1792 1793 1794 1795
    pStream = pStream->next;
  }

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1796
int tscBuildHeartBeatMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1797 1798 1799 1800 1801 1802 1803 1804 1805
  char *pMsg, *pStart;
  int   msgLen = 0;
  int   size = 0;

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

  pthread_mutex_lock(&pObj->mutex);

1806
  size = tscEstimateHeartBeatMsgLength(pSql);
S
slguan 已提交
1807
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
1808
    pthread_mutex_unlock(&pObj->mutex);
S
slguan 已提交
1809 1810 1811
    tscError("%p failed to malloc for heartbeat msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824

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

  SMgmtHead *pMgmt = (SMgmtHead *)pMsg;
  strcpy(pMgmt->db, pObj->db);
  pMsg += sizeof(SMgmtHead);

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

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1825
  pCmd->msgType = TSDB_MSG_TYPE_CM_HEARTBEAT;
H
hzcheng 已提交
1826 1827

  assert(msgLen + minMsgSize() <= size);
1828
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1829 1830
}

1831 1832
int tscProcessTableMetaRsp(SSqlObj *pSql) {
  STableMetaMsg *pMetaMsg = (STableMetaMsg *)pSql->res.pRsp;
H
hzcheng 已提交
1833

1834 1835
  pMetaMsg->sid = htonl(pMetaMsg->sid);
  pMetaMsg->sversion = htons(pMetaMsg->sversion);
H
hjxilinx 已提交
1836 1837 1838
  
  pMetaMsg->vgroup.vgId = htonl(pMetaMsg->vgroup.vgId);
  
1839 1840
  pMetaMsg->uid = htobe64(pMetaMsg->uid);
  pMetaMsg->contLen = htons(pMetaMsg->contLen);
1841
  pMetaMsg->numOfColumns = htons(pMetaMsg->numOfColumns);
H
hzcheng 已提交
1842

H
hjxilinx 已提交
1843 1844
  if (pMetaMsg->sid < 0 || pMetaMsg->vgroup.numOfIps < 0) {
    tscError("invalid meter vgId:%d, sid%d", pMetaMsg->vgroup.numOfIps, pMetaMsg->sid);
H
hzcheng 已提交
1845 1846 1847
    return TSDB_CODE_INVALID_VALUE;
  }

1848 1849
  if (pMetaMsg->numOfTags > TSDB_MAX_TAGS || pMetaMsg->numOfTags < 0) {
    tscError("invalid numOfTags:%d", pMetaMsg->numOfTags);
H
hzcheng 已提交
1850 1851 1852
    return TSDB_CODE_INVALID_VALUE;
  }

1853 1854
  if (pMetaMsg->numOfColumns > TSDB_MAX_COLUMNS || pMetaMsg->numOfColumns <= 0) {
    tscError("invalid numOfColumns:%d", pMetaMsg->numOfColumns);
H
hzcheng 已提交
1855 1856 1857
    return TSDB_CODE_INVALID_VALUE;
  }

H
hjxilinx 已提交
1858 1859 1860 1861 1862
  for (int i = 0; i < pMetaMsg->vgroup.numOfIps; ++i) {
    pMetaMsg->vgroup.ipAddr[i].ip = htonl(pMetaMsg->vgroup.ipAddr[i].ip);
    pMetaMsg->vgroup.ipAddr[i].port = htons(pMetaMsg->vgroup.ipAddr[i].port);
    
    assert(pMetaMsg->vgroup.ipAddr[i].ip != 0);
H
hzcheng 已提交
1863 1864
  }

1865
  SSchema* pSchema = pMetaMsg->schema;
H
hzcheng 已提交
1866

1867
  int32_t numOfTotalCols = pMetaMsg->numOfColumns + pMetaMsg->numOfTags;
H
hzcheng 已提交
1868 1869 1870
  for (int i = 0; i < numOfTotalCols; ++i) {
    pSchema->bytes = htons(pSchema->bytes);
    pSchema->colId = htons(pSchema->colId);
H
hjxilinx 已提交
1871 1872
    
    assert(pSchema->type >= TSDB_DATA_TYPE_BOOL && pSchema->type <= TSDB_DATA_TYPE_NCHAR);
H
hzcheng 已提交
1873 1874 1875
    pSchema++;
  }

1876 1877
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
H
hzcheng 已提交
1878

1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889
#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 已提交
1890
  // todo add one more function: taosAddDataIfNotExists();
1891
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
H
hjxilinx 已提交
1892
  assert(pTableMetaInfo->pTableMeta == NULL);
H
hzcheng 已提交
1893

H
hjxilinx 已提交
1894 1895 1896
  pTableMetaInfo->pTableMeta =
      (STableMeta *) taosCachePut(tscCacheHandle, pTableMetaInfo->name, pTableMeta, size, tsMeterMetaKeepTimer);
  
1897
  // todo handle out of memory case
1898
  if (pTableMetaInfo->pTableMeta == NULL) {
H
hjxilinx 已提交
1899
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
1900
  }
H
hzcheng 已提交
1901

1902
  free(pTableMeta);
1903
  
H
hjxilinx 已提交
1904
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1905 1906
}

S
slguan 已提交
1907
/**
1908
 *  multi table meta rsp pkg format:
1909
 *  | STaosRsp | ieType | SCMMultiTableInfoMsg | SMeterMeta0 | SSchema0 | SMeterMeta1 | SSchema1 | SMeterMeta2 | SSchema2
S
slguan 已提交
1910 1911 1912
 *  |...... 1B        1B            4B
 **/
int tscProcessMultiMeterMetaRsp(SSqlObj *pSql) {
1913
#if 0
S
slguan 已提交
1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925
  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++;

1926
  SCMMultiTableInfoMsg *pInfo = (SCMMultiTableInfoMsg *)rsp;
S
slguan 已提交
1927
  totalNum = htonl(pInfo->numOfTables);
1928
  rsp += sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1929 1930

  for (i = 0; i < totalNum; i++) {
S
slguan 已提交
1931
    SMultiTableMeta *pMultiMeta = (SMultiTableMeta *)rsp;
1932
    STableMeta *     pMeta = pMultiMeta->metas;
S
slguan 已提交
1933 1934 1935

    pMeta->sid = htonl(pMeta->sid);
    pMeta->sversion = htons(pMeta->sversion);
H
hjxilinx 已提交
1936
    pMeta->vgId = htonl(pMeta->vgId);
S
slguan 已提交
1937 1938
    pMeta->uid = htobe64(pMeta->uid);

H
hjxilinx 已提交
1939 1940
    if (pMeta->sid <= 0 || pMeta->vgId < 0) {
      tscError("invalid meter vgId:%d, sid%d", pMeta->vgId, pMeta->sid);
S
slguan 已提交
1941 1942 1943 1944 1945
      pSql->res.code = TSDB_CODE_INVALID_VALUE;
      pSql->res.numOfTotal = i;
      return TSDB_CODE_OTHERS;
    }

H
hjxilinx 已提交
1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968
    //    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 已提交
1969
    //    for (int j = 0; j < TSDB_REPLICA_MAX_NUM; ++j) {
H
hjxilinx 已提交
1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005
    //      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;
    //    (void)taosCachePut(tscCacheHandle, pMeta->tableId, (char *)pMeta, size, tsMeterMetaKeepTimer);
    //  }
S
slguan 已提交
2006
  }
H
hjxilinx 已提交
2007
  
S
slguan 已提交
2008 2009 2010
  pSql->res.code = TSDB_CODE_SUCCESS;
  pSql->res.numOfTotal = i;
  tscTrace("%p load multi-metermeta resp complete num:%d", pSql, pSql->res.numOfTotal);
2011 2012
#endif
  
S
slguan 已提交
2013 2014 2015
  return TSDB_CODE_SUCCESS;
}

H
hjxilinx 已提交
2016 2017
int tscProcessSTableVgroupRsp(SSqlObj *pSql) {
#if 0
S
slguan 已提交
2018 2019
  void **      metricMetaList = NULL;
  int32_t *    sizes = NULL;
H
hjxilinx 已提交
2020
  
S
slguan 已提交
2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
  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 已提交
2037
    pMeta = (SSuperTableMeta *)rsp;
S
slguan 已提交
2038 2039

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

S
slguan 已提交
2042
    pMeta->numOfTables = htonl(pMeta->numOfTables);
S
slguan 已提交
2043 2044 2045
    pMeta->numOfVnodes = htonl(pMeta->numOfVnodes);
    pMeta->tagLen = htons(pMeta->tagLen);

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

2048 2049
    char *pBuf = calloc(1, size);
    if (pBuf == NULL) {
S
slguan 已提交
2050 2051 2052
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2053

S
slguan 已提交
2054
    SSuperTableMeta *pNewMetricMeta = (SSuperTableMeta *)pBuf;
S
slguan 已提交
2055
    metricMetaList[k] = pNewMetricMeta;
H
hzcheng 已提交
2056

S
slguan 已提交
2057
    pNewMetricMeta->numOfTables = pMeta->numOfTables;
S
slguan 已提交
2058 2059
    pNewMetricMeta->numOfVnodes = pMeta->numOfVnodes;
    pNewMetricMeta->tagLen = pMeta->tagLen;
H
hzcheng 已提交
2060

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

S
slguan 已提交
2063 2064
    for (int32_t i = 0; i < pMeta->numOfVnodes; ++i) {
      SVnodeSidList *pSidLists = (SVnodeSidList *)rsp;
2065
      memcpy(pBuf, pSidLists, sizeof(SVnodeSidList));
H
hzcheng 已提交
2066

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

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

2072
      pBuf += sizeof(SVnodeSidList) + sizeof(STableIdInfo *) * pSidLists->numOfSids;
S
slguan 已提交
2073
      rsp += sizeof(SVnodeSidList);
H
hzcheng 已提交
2074

2075
      size_t elemSize = sizeof(STableIdInfo) + pNewMetricMeta->tagLen;
S
slguan 已提交
2076
      for (int32_t j = 0; j < pSidLists->numOfSids; ++j) {
2077 2078
        pLists->pSidExtInfoList[j] = pBuf - (char *)pLists;
        memcpy(pBuf, rsp, elemSize);
2079

2080 2081
        ((STableIdInfo *)pBuf)->uid = htobe64(((STableIdInfo *)pBuf)->uid);
        ((STableIdInfo *)pBuf)->sid = htonl(((STableIdInfo *)pBuf)->sid);
2082

2083 2084
        rsp += elemSize;
        pBuf += elemSize;
S
slguan 已提交
2085
      }
H
hzcheng 已提交
2086
    }
S
slguan 已提交
2087

2088
    sizes[k] = pBuf - (char *)pNewMetricMeta;
H
hzcheng 已提交
2089 2090
  }

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

H
hjxilinx 已提交
2095
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2096
    tscGetMetricMetaCacheKey(pQueryInfo, name, pTableMetaInfo->pTableMeta->uid);
H
hzcheng 已提交
2097

S
slguan 已提交
2098 2099 2100
#ifdef _DEBUG_VIEW
    printf("generate the metric key:%s, index:%d\n", name, i);
#endif
H
hzcheng 已提交
2101

S
slguan 已提交
2102
    // release the used metricmeta
H
hjxilinx 已提交
2103 2104
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), false);
    pTableMetaInfo->pMetricMeta = (SSuperTableMeta *)taosCachePut(tscCacheHandle, name, (char *)metricMetaList[i],
S
slguan 已提交
2105 2106 2107 2108
                                                                      sizes[i], tsMetricMetaKeepTimer);
    tfree(metricMetaList[i]);

    // failed to put into cache
H
hjxilinx 已提交
2109
    if (pTableMetaInfo->pMetricMeta == NULL) {
S
slguan 已提交
2110 2111 2112
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2113 2114
  }

S
slguan 已提交
2115 2116 2117 2118 2119 2120 2121 2122
_error_clean:
  // free allocated resource
  for (int32_t i = 0; i < num; ++i) {
    tfree(metricMetaList[i]);
  }

  free(sizes);
  free(metricMetaList);
H
hjxilinx 已提交
2123
#endif
2124
  SSqlRes* pRes = &pSql->res;
H
hjxilinx 已提交
2125
  
2126
  SCMSTableVgroupRspMsg *pStableVgroup = (SCMSTableVgroupRspMsg *)pRes->pRsp;
H
hjxilinx 已提交
2127
  pStableVgroup->numOfVgroups = htonl(pStableVgroup->numOfVgroups);
H
hjxilinx 已提交
2128
  
2129 2130 2131
  // master sqlObj locates in param
  SSqlObj* parent = pSql->param;
  assert(parent != NULL);
H
hjxilinx 已提交
2132
  
2133
  SSqlCmd* pCmd = &parent->cmd;
H
hjxilinx 已提交
2134 2135
  STableMetaInfo* pInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
  
2136 2137 2138
  pInfo->vgroupList = malloc(pRes->rspLen);
  memcpy(pInfo->vgroupList, pStableVgroup, pRes->rspLen);
  
H
hjxilinx 已提交
2139 2140
  for(int32_t i = 0; i < pInfo->vgroupList->numOfVgroups; ++i) {
    SCMVgroupInfo* pVgroups = &pInfo->vgroupList->vgroups[i];
2141
    
H
hjxilinx 已提交
2142
    pVgroups->vgId = htonl(pVgroups->vgId);
2143
    assert(pVgroups->numOfIps >= 1);
2144
    
2145
    for(int32_t j = 0; j < pVgroups->numOfIps; ++j) {
H
hjxilinx 已提交
2146 2147
      pVgroups->ipAddr[j].ip = htonl(pVgroups->ipAddr[j].ip);
      pVgroups->ipAddr[j].port = htons(pVgroups->ipAddr[j].port);
2148
    }
H
hjxilinx 已提交
2149 2150
  }
  
S
slguan 已提交
2151
  return pSql->res.code;
H
hzcheng 已提交
2152 2153 2154 2155 2156 2157
}

/*
 * current process do not use the cache at all
 */
int tscProcessShowRsp(SSqlObj *pSql) {
H
hjxilinx 已提交
2158
  STableMetaMsg * pMetaMsg;
2159
  SCMShowRsp *pShow;
S
slguan 已提交
2160
  SSchema *    pSchema;
H
hzcheng 已提交
2161 2162
  char         key[20];

2163 2164 2165
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

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

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

2170
  pShow = (SCMShowRsp *)pRes->pRsp;
S
slguan 已提交
2171
  pShow->qhandle = htobe64(pShow->qhandle);
H
hzcheng 已提交
2172 2173
  pRes->qhandle = pShow->qhandle;

S
slguan 已提交
2174
  tscResetForNextRetrieve(pRes);
H
hjxilinx 已提交
2175
  pMetaMsg = &(pShow->tableMeta);
H
hzcheng 已提交
2176

H
hjxilinx 已提交
2177
  pMetaMsg->numOfColumns = ntohs(pMetaMsg->numOfColumns);
H
hzcheng 已提交
2178

H
hjxilinx 已提交
2179
  pSchema = pMetaMsg->schema;
H
hjxilinx 已提交
2180 2181
  pMetaMsg->sid = ntohs(pMetaMsg->sid);
  for (int i = 0; i < pMetaMsg->numOfColumns; ++i) {
H
hzcheng 已提交
2182 2183 2184 2185
    pSchema->bytes = htons(pSchema->bytes);
    pSchema++;
  }

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

H
hjxilinx 已提交
2189
  taosCacheRelease(tscCacheHandle, (void *)&(pTableMetaInfo->pTableMeta), false);
H
hjxilinx 已提交
2190
  
H
hjxilinx 已提交
2191 2192 2193
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
  
H
hjxilinx 已提交
2194
  pTableMetaInfo->pTableMeta = taosCachePut(tscCacheHandle, key, (char *)pTableMeta, size, tsMeterMetaKeepTimer);
H
hjxilinx 已提交
2195
  SSchema *pTableSchema = tscGetTableSchema(pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2196

2197 2198 2199 2200
  if (pQueryInfo->colList == NULL) {
    pQueryInfo->colList = taosArrayInit(4, POINTER_BYTES);
  }
  
H
hjxilinx 已提交
2201 2202
  SFieldInfo* pFieldInfo = &pQueryInfo->fieldsInfo;
  
S
slguan 已提交
2203
  SColumnIndex index = {0};
H
hjxilinx 已提交
2204 2205 2206
  pSchema = pMetaMsg->schema;
  
  for (int16_t i = 0; i < pMetaMsg->numOfColumns; ++i, ++pSchema) {
S
slguan 已提交
2207
    index.columnIndex = i;
2208 2209
    tscColumnListInsert(pQueryInfo->colList, &index);
    
H
hjxilinx 已提交
2210 2211
    TAOS_FIELD f = tscCreateField(pSchema->type, pSchema->name, pSchema->bytes);
    SFieldSupInfo* pInfo = tscFieldInfoAppend(pFieldInfo, &f);
H
hjxilinx 已提交
2212
    
H
hjxilinx 已提交
2213
    pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index,
2214
                     pTableSchema[i].type, pTableSchema[i].bytes, pTableSchema[i].bytes, false);
H
hzcheng 已提交
2215
  }
H
hjxilinx 已提交
2216 2217
  
  pCmd->numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2218
  tscFieldInfoUpdateOffset(pQueryInfo);
H
hjxilinx 已提交
2219 2220
  
  tfree(pTableMeta);
H
hzcheng 已提交
2221 2222 2223 2224
  return 0;
}

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

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

H
hjxilinx 已提交
2233 2234 2235
  assert(len <= tListLen(pObj->db));
  strncpy(pObj->db, temp, tListLen(pObj->db));
  
S
slguan 已提交
2236
  tscSetMgmtIpList(&pConnect->ipList);
H
hzcheng 已提交
2237

S
slguan 已提交
2238
  strcpy(pObj->sversion, pConnect->serverVersion);
H
hzcheng 已提交
2239 2240 2241 2242 2243 2244 2245 2246
  pObj->writeAuth = pConnect->writeAuth;
  pObj->superAuth = pConnect->superAuth;
  taosTmrReset(tscProcessActivityTimer, tsShellActivityTimer * 500, pObj, tscTmr, &pObj->pTimer);

  return 0;
}

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

H
hjxilinx 已提交
2250
  strcpy(pObj->db, pTableMetaInfo->name);
H
hzcheng 已提交
2251 2252 2253 2254
  return 0;
}

int tscProcessDropDbRsp(SSqlObj *UNUSED_PARAM(pSql)) {
2255
  taosCacheEmpty(tscCacheHandle);
H
hzcheng 已提交
2256 2257 2258 2259
  return 0;
}

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

H
hjxilinx 已提交
2262 2263
  STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMeta == NULL) {
H
hzcheng 已提交
2264 2265 2266 2267 2268 2269 2270 2271
    /* 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 已提交
2272 2273
   * The cached information is expired, however, we may have lost the ref of original meter. So, clear whole cache
   * instead.
H
hzcheng 已提交
2274
   */
H
hjxilinx 已提交
2275 2276
  tscTrace("%p force release metermeta after drop table:%s", pSql, pTableMetaInfo->name);
  taosCacheRelease(tscCacheHandle, (void **)&pTableMeta, true);
H
hzcheng 已提交
2277

H
hjxilinx 已提交
2278 2279
  if (pTableMetaInfo->pTableMeta) {
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2280
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2281 2282 2283 2284 2285 2286
  }

  return 0;
}

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

H
hjxilinx 已提交
2289 2290
  STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMeta == NULL) { /* not in cache, abort */
H
hzcheng 已提交
2291 2292 2293
    return 0;
  }

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

H
hjxilinx 已提交
2297
  if (pTableMetaInfo->pTableMeta) {
H
hjxilinx 已提交
2298
    bool isSuperTable = UTIL_TABLE_IS_SUPERTABLE(pTableMetaInfo);
H
hzcheng 已提交
2299

H
hjxilinx 已提交
2300
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2301
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2302

2303
    if (isSuperTable) {  // if it is a super table, reset whole query cache
H
hjxilinx 已提交
2304
      tscTrace("%p reset query cache since table:%s is stable", pSql, pTableMetaInfo->name);
2305
      taosCacheEmpty(tscCacheHandle);
H
hzcheng 已提交
2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319
    }
  }

  return 0;
}

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

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

2320
  SQueryTableRsp *pQuery = (SQueryTableRsp *)pRes->pRsp;
S
slguan 已提交
2321 2322 2323
  pQuery->qhandle = htobe64(pQuery->qhandle);
  pRes->qhandle = pQuery->qhandle;

H
hzcheng 已提交
2324
  pRes->data = NULL;
S
slguan 已提交
2325
  tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
2326 2327 2328
  return 0;
}

H
hjxilinx 已提交
2329
int tscProcessRetrieveRspFromNode(SSqlObj *pSql) {
S
slguan 已提交
2330 2331 2332
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

S
slguan 已提交
2333
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2334 2335 2336

  pRes->numOfRows = htonl(pRetrieve->numOfRows);
  pRes->precision = htons(pRetrieve->precision);
2337 2338
  pRes->offset    = htobe64(pRetrieve->offset);
  pRes->useconds  = htobe64(pRetrieve->useconds);
H
hjxilinx 已提交
2339
  pRes->completed = (pRetrieve->completed == 1);
2340
  pRes->data      = pRetrieve->data;
H
hjxilinx 已提交
2341
  
2342 2343
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
  tscSetResultPointer(pQueryInfo, pRes);
2344

weixin_48148422's avatar
weixin_48148422 已提交
2345
  if (pSql->pSubscription != NULL) {
H
hjxilinx 已提交
2346
    int32_t numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2347
    
H
hjxilinx 已提交
2348
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, numOfCols - 1);
H
hjxilinx 已提交
2349 2350
    int16_t     offset = tscFieldInfoGetOffset(pQueryInfo, numOfCols - 1);
    
weixin_48148422's avatar
weixin_48148422 已提交
2351 2352
    char* p = pRes->data + (pField->bytes + offset) * pRes->numOfRows;

S
slguan 已提交
2353
    int32_t numOfTables = htonl(*(int32_t*)p);
weixin_48148422's avatar
weixin_48148422 已提交
2354
    p += sizeof(int32_t);
S
slguan 已提交
2355
    for (int i = 0; i < numOfTables; i++) {
weixin_48148422's avatar
weixin_48148422 已提交
2356 2357 2358 2359
      int64_t uid = htobe64(*(int64_t*)p);
      p += sizeof(int64_t);
      TSKEY key = htobe64(*(TSKEY*)p);
      p += sizeof(TSKEY);
2360
      tscUpdateSubscriptionProgress(pSql->pSubscription, uid, key);
weixin_48148422's avatar
weixin_48148422 已提交
2361
    }
2362 2363
  }

H
hzcheng 已提交
2364
  pRes->row = 0;
S
slguan 已提交
2365
  tscTrace("%p numOfRows:%d, offset:%d", pSql, pRes->numOfRows, pRes->offset);
H
hzcheng 已提交
2366 2367 2368 2369 2370

  return 0;
}

int tscProcessRetrieveRspFromLocal(SSqlObj *pSql) {
2371 2372
  SSqlRes *   pRes = &pSql->res;
  SSqlCmd *   pCmd = &pSql->cmd;
2373
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
2374

S
slguan 已提交
2375
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2376 2377 2378

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

2380
  tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
2381 2382 2383 2384
  pRes->row = 0;
  return 0;
}

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

2387
static int32_t getTableMetaFromMgmt(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo) {
S
slguan 已提交
2388 2389
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
  if (NULL == pNew) {
2390
    tscError("%p malloc failed for new sqlobj to get table meta", pSql);
S
slguan 已提交
2391 2392
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
2393

H
hzcheng 已提交
2394 2395 2396
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;
  pNew->cmd.command = TSDB_SQL_META;
S
slguan 已提交
2397

2398
  tscAddSubqueryInfo(&pNew->cmd);
2399 2400 2401 2402

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

H
hjxilinx 已提交
2403
  pNew->cmd.autoCreated = pSql->cmd.autoCreated;  // create table if not exists
S
slguan 已提交
2404
  if (TSDB_CODE_SUCCESS != tscAllocPayload(&pNew->cmd, TSDB_DEFAULT_PAYLOAD_SIZE)) {
2405
    tscError("%p malloc failed for payload to get table meta", pSql);
S
slguan 已提交
2406
    free(pNew);
2407

S
slguan 已提交
2408 2409 2410
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

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

2414
  strncpy(pNewMeterMetaInfo->name, pTableMetaInfo->name, tListLen(pNewMeterMetaInfo->name));
2415
  memcpy(pNew->cmd.payload, pSql->cmd.payload, TSDB_DEFAULT_PAYLOAD_SIZE);  // tag information if table does not exists.
S
slguan 已提交
2416
  tscTrace("%p new pSqlObj:%p to get tableMeta", pSql, pNew);
H
hzcheng 已提交
2417

H
hjxilinx 已提交
2418 2419
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
H
hzcheng 已提交
2420

H
hjxilinx 已提交
2421 2422 2423
  int32_t code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2424 2425 2426 2427 2428
  }

  return code;
}

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

2432
  // If this STableMetaInfo owns a table meta, release it first
H
hjxilinx 已提交
2433 2434
  if (pTableMetaInfo->pTableMeta != NULL) {
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), false);
2435 2436
  }
  
H
hjxilinx 已提交
2437 2438
  pTableMetaInfo->pTableMeta = (STableMeta *)taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMetaInfo->pTableMeta != NULL) {
H
hjxilinx 已提交
2439
    STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
2440 2441
    tscTrace("%p retrieve table Meta from cache, the number of columns:%d, numOfTags:%d, %p", pSql, tinfo.numOfColumns,
             tinfo.numOfTags, pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2442 2443 2444

    return TSDB_CODE_SUCCESS;
  }
2445 2446
  
  return getTableMetaFromMgmt(pSql, pTableMetaInfo);
H
hzcheng 已提交
2447 2448
}

H
hjxilinx 已提交
2449
int tscGetMeterMetaEx(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo, bool createIfNotExists) {
H
hjxilinx 已提交
2450
  pSql->cmd.autoCreated = createIfNotExists;
H
hjxilinx 已提交
2451
  return tscGetTableMeta(pSql, pTableMetaInfo);
H
hzcheng 已提交
2452 2453 2454 2455
}

/*
 * in handling the renew metermeta problem during insertion,
S
slguan 已提交
2456
 *
H
hzcheng 已提交
2457 2458 2459 2460 2461
 * 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 已提交
2462
  if (pCmd->command == TSDB_SQL_INSERT) {
H
hzcheng 已提交
2463 2464 2465 2466 2467 2468 2469
    taosMsleep(50);  // todo: global config
  }
}

/**
 * in renew metermeta, do not retrieve metadata in cache.
 * @param pSql          sql object
S
slguan 已提交
2470
 * @param tableId       meter id
H
hzcheng 已提交
2471 2472
 * @return              status code
 */
S
slguan 已提交
2473
int tscRenewMeterMeta(SSqlObj *pSql, char *tableId) {
2474 2475
  int code = 0;

H
hjxilinx 已提交
2476
  // handle table meta renew process
H
hzcheng 已提交
2477
  SSqlCmd *pCmd = &pSql->cmd;
2478 2479

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
2480
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
2481 2482

  /*
S
slguan 已提交
2483
   * 1. only update the metermeta in force model metricmeta is not updated
H
hzcheng 已提交
2484 2485
   * 2. if get metermeta failed, still get the metermeta
   */
2486
  if (pTableMetaInfo->pTableMeta == NULL || !tscQueryOnSTable(pCmd)) {
2487
    STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
2488
    if (pTableMetaInfo->pTableMeta) {
2489 2490
      tscTrace("%p update table meta, old: numOfTags:%d, numOfCols:%d, uid:%" PRId64 ", addr:%p", pSql,
               tscGetNumOfTags(pTableMeta), tscGetNumOfColumns(pTableMeta), pTableMeta->uid, pTableMeta);
H
hzcheng 已提交
2491
    }
2492

2493
    tscWaitingForCreateTable(pCmd);
H
hjxilinx 已提交
2494
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hzcheng 已提交
2495

2496
    code = getTableMetaFromMgmt(pSql, pTableMetaInfo);  // todo ??
H
hzcheng 已提交
2497
  } else {
H
hjxilinx 已提交
2498
    tscTrace("%p metric query not update metric meta, numOfTags:%d, numOfCols:%d, uid:%" PRId64 ", addr:%p", pSql,
H
hjxilinx 已提交
2499 2500
             tscGetNumOfTags(pTableMetaInfo->pTableMeta), pCmd->numOfCols, pTableMetaInfo->pTableMeta->uid,
             pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2501 2502 2503 2504 2505
  }

  return code;
}

H
hjxilinx 已提交
2506
int tscGetSTableVgroupInfo(SSqlObj *pSql, int32_t clauseIndex) {
S
slguan 已提交
2507 2508
  int      code = TSDB_CODE_NETWORK_UNAVAIL;
  SSqlCmd *pCmd = &pSql->cmd;
H
hzcheng 已提交
2509

H
hjxilinx 已提交
2510
  //the query condition is serialized into pCmd->payload, we need to rebuild key for stable meta info in cache.
2511
//  bool required = false;
2512

2513
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2514
  if (pQueryInfo->pTableMetaInfo[0]->vgroupList != NULL) {
H
hjxilinx 已提交
2515 2516 2517 2518
    return TSDB_CODE_SUCCESS;
  }
  
#if 0
2519
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
S
slguan 已提交
2520 2521
    char tagstr[TSDB_MAX_TAGS_LEN + 1] = {0};

H
hjxilinx 已提交
2522
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2523
    tscGetMetricMetaCacheKey(pQueryInfo, tagstr, pTableMetaInfo->pTableMeta->uid);
S
slguan 已提交
2524

H
hjxilinx 已提交
2525
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), false);
S
slguan 已提交
2526

2527
    SSuperTableMeta *ppMeta = (SSuperTableMeta *)taosCacheAcquireByName(tscCacheHandle, tagstr);
S
slguan 已提交
2528
    if (ppMeta == NULL) {
2529
      required = true;
S
slguan 已提交
2530 2531
      break;
    } else {
H
hjxilinx 已提交
2532
//      pTableMetaInfo->pMetricMeta = ppMeta;
S
slguan 已提交
2533 2534
    }
  }
H
hzcheng 已提交
2535

2536 2537
  // all metricmeta for one clause are retrieved from cache, no need to retrieve metricmeta from management node
  if (!required) {
H
hzcheng 已提交
2538 2539
    return TSDB_CODE_SUCCESS;
  }
H
hjxilinx 已提交
2540 2541
#endif
  
S
slguan 已提交
2542
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
H
hzcheng 已提交
2543 2544 2545
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;

H
hjxilinx 已提交
2546
  pNew->cmd.command = TSDB_SQL_STABLEVGROUP;
2547 2548
  
  SQueryInfo *pNewQueryInfo = NULL;
2549 2550 2551
  if ((code = tscGetQueryInfoDetailSafely(&pNew->cmd, 0, &pNewQueryInfo)) != TSDB_CODE_SUCCESS) {
    return code;
  }
2552
  
2553
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2554
    STableMetaInfo *pMMInfo = tscGetMetaInfo(pQueryInfo, i);
2555

H
hjxilinx 已提交
2556
    STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pMMInfo->name);
2557
    tscAddTableMetaInfo(pNewQueryInfo, pMMInfo->name, pTableMeta, NULL, pMMInfo->tagColList);
S
slguan 已提交
2558 2559 2560 2561 2562 2563
  }

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

2565
  tscTagCondCopy(&pNewQueryInfo->tagCond, &pQueryInfo->tagCond);
H
hzcheng 已提交
2566

2567 2568
  pNewQueryInfo->groupbyExpr = pQueryInfo->groupbyExpr;
  pNewQueryInfo->numOfTables = pQueryInfo->numOfTables;
H
hzcheng 已提交
2569

2570 2571
  pNewQueryInfo->slimit = pQueryInfo->slimit;
  pNewQueryInfo->order = pQueryInfo->order;
H
hjxilinx 已提交
2572 2573 2574 2575 2576
  
  STagCond* pTagCond = &pNewQueryInfo->tagCond;
  tscTrace("%p new sqlobj:%p info, numOfTables:%d, slimit:%" PRId64 ", soffset:%" PRId64 ", order:%d, tbname cond:%s",
      pSql, pNew, pNewQueryInfo->numOfTables, pNewQueryInfo->slimit.limit, pNewQueryInfo->slimit.offset,
      pNewQueryInfo->order.order, pTagCond->tbnameCond.cond)
H
hzcheng 已提交
2577

2578 2579
//  if (pSql->fp != NULL && pSql->pStream == NULL) {
//    pCmd->pDataBlocks = tscDestroyBlockArrayList(pCmd->pDataBlocks);
2580
//    tscFreeQueryInfo(pCmd);
2581
//  }
H
hzcheng 已提交
2582

H
hjxilinx 已提交
2583
  tscTrace("%p allocate new pSqlObj:%p to get stable vgroupInfo", pSql, pNew);
2584 2585 2586 2587 2588
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
  code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2589 2590 2591 2592 2593
  }

  return code;
}

2594
void tscInitMsgsFp() {
S
slguan 已提交
2595 2596 2597
  tscBuildMsg[TSDB_SQL_SELECT] = tscBuildQueryMsg;
  tscBuildMsg[TSDB_SQL_INSERT] = tscBuildSubmitMsg;
  tscBuildMsg[TSDB_SQL_FETCH] = tscBuildRetrieveMsg;
H
hzcheng 已提交
2598 2599

  tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
2600
  tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
H
hzcheng 已提交
2601

2602 2603
  tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
  tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
H
hzcheng 已提交
2604 2605

  tscBuildMsg[TSDB_SQL_CREATE_TABLE] = tscBuildCreateTableMsg;
S
[TD-16]  
slguan 已提交
2606
  tscBuildMsg[TSDB_SQL_DROP_USER] = tscBuildDropUserMsg;
H
hzcheng 已提交
2607 2608 2609
  tscBuildMsg[TSDB_SQL_DROP_ACCT] = tscBuildDropAcctMsg;
  tscBuildMsg[TSDB_SQL_DROP_DB] = tscBuildDropDbMsg;
  tscBuildMsg[TSDB_SQL_DROP_TABLE] = tscBuildDropTableMsg;
2610
  tscBuildMsg[TSDB_SQL_ALTER_USER] = tscBuildUserMsg;
S
slguan 已提交
2611 2612 2613
  tscBuildMsg[TSDB_SQL_CREATE_DNODE] = tscBuildCreateDnodeMsg;
  tscBuildMsg[TSDB_SQL_DROP_DNODE] = tscBuildDropDnodeMsg;
  tscBuildMsg[TSDB_SQL_CFG_DNODE] = tscBuildCfgDnodeMsg;
H
hzcheng 已提交
2614 2615 2616 2617 2618
  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 已提交
2619
  tscBuildMsg[TSDB_SQL_META] = tscBuildTableMetaMsg;
H
hjxilinx 已提交
2620
  tscBuildMsg[TSDB_SQL_STABLEVGROUP] = tscBuildSTableVgroupMsg;
S
slguan 已提交
2621
  tscBuildMsg[TSDB_SQL_MULTI_META] = tscBuildMultiMeterMetaMsg;
H
hzcheng 已提交
2622 2623 2624 2625

  tscBuildMsg[TSDB_SQL_HB] = tscBuildHeartBeatMsg;
  tscBuildMsg[TSDB_SQL_SHOW] = tscBuildShowMsg;
  tscBuildMsg[TSDB_SQL_RETRIEVE] = tscBuildRetrieveFromMgmtMsg;
2626 2627 2628
  tscBuildMsg[TSDB_SQL_KILL_QUERY] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_STREAM] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_CONNECTION] = tscBuildKillMsg;
H
hzcheng 已提交
2629 2630

  tscProcessMsgRsp[TSDB_SQL_SELECT] = tscProcessQueryRsp;
H
hjxilinx 已提交
2631
  tscProcessMsgRsp[TSDB_SQL_FETCH] = tscProcessRetrieveRspFromNode;
H
hzcheng 已提交
2632 2633 2634 2635 2636

  tscProcessMsgRsp[TSDB_SQL_DROP_DB] = tscProcessDropDbRsp;
  tscProcessMsgRsp[TSDB_SQL_DROP_TABLE] = tscProcessDropTableRsp;
  tscProcessMsgRsp[TSDB_SQL_CONNECT] = tscProcessConnectRsp;
  tscProcessMsgRsp[TSDB_SQL_USE_DB] = tscProcessUseDbRsp;
2637
  tscProcessMsgRsp[TSDB_SQL_META] = tscProcessTableMetaRsp;
H
hjxilinx 已提交
2638
  tscProcessMsgRsp[TSDB_SQL_STABLEVGROUP] = tscProcessSTableVgroupRsp;
S
slguan 已提交
2639
  tscProcessMsgRsp[TSDB_SQL_MULTI_META] = tscProcessMultiMeterMetaRsp;
H
hzcheng 已提交
2640 2641

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

H
hzcheng 已提交
2645
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_TAGS] = tscProcessTagRetrieveRsp;
H
hjxilinx 已提交
2646 2647 2648 2649 2650
  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;
2651

H
hzcheng 已提交
2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_EMPTY_RESULT] = tscProcessEmptyResultRsp;

  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_METRIC] = tscProcessRetrieveMetricRsp;

  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;
}