tscServer.c 88.2 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 143 144 145 146 147 148 149 150 151 152 153 154 155 156
    }
  } else {
    tscTrace("heart beat failed, code:%d", code);
  }

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

    // ignore the error information returned from mnode when set ignore flag in sql
S
slguan 已提交
330
    if (pRes->code == TSDB_CODE_DB_ALREADY_EXIST && pCmd->existsCheck && pRes->rspType == TSDB_MSG_TYPE_CM_CREATE_DB_RSP) {
H
hzcheng 已提交
331 332 333 334 335 336 337
      pRes->code = TSDB_CODE_SUCCESS;
    }

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

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

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

374
    if (shouldFree) {
375
      tscTrace("%p sqlObj is automatically freed", pSql);
sangshuduo's avatar
sangshuduo 已提交
376
      tscFreeSqlObj(pSql);
H
hzcheng 已提交
377 378 379
    }
  }

380
  rpcFreeCont(rpcMsg->pCont);
H
hzcheng 已提交
381 382
}

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

404
  code = tscSendMsgToServer(pSql);
405
  if (code != TSDB_CODE_SUCCESS) {
S
slguan 已提交
406
    pRes->code = code;
H
hjxilinx 已提交
407
    tscQueueAsyncRes(pSql);
S
slguan 已提交
408
  }
H
hjxilinx 已提交
409 410
  
  return TSDB_CODE_SUCCESS;
S
slguan 已提交
411 412 413
}

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

422
  if (pQueryInfo != NULL) {
H
hjxilinx 已提交
423
    pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
424 425
    if (pTableMetaInfo != NULL) {
      name = pTableMetaInfo->name;
426
    }
427

428
    type = pQueryInfo->type;
429
  
430
    // for heartbeat, numOfTables == 0;
431
    assert((pQueryInfo->numOfTables == 0 && pQueryInfo->command == TSDB_SQL_HB) || pQueryInfo->numOfTables > 0);
S
slguan 已提交
432
  }
433

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

S
slguan 已提交
447
  // todo handle async situation
448 449
  if (QUERY_IS_JOIN_QUERY(type)) {
    if ((pQueryInfo->type & TSDB_QUERY_TYPE_SUBQUERY) == 0) {
H
hjxilinx 已提交
450
      return tscHandleMasterJoinQuery(pSql);
S
slguan 已提交
451 452
    } else {
      // for first stage sub query, iterate all vnodes to get all timestamp
453
      if ((pQueryInfo->type & TSDB_QUERY_TYPE_JOIN_SEC_STAGE) != TSDB_QUERY_TYPE_JOIN_SEC_STAGE) {
S
slguan 已提交
454 455 456 457
        return doProcessSql(pSql);
      }
    }
  }
458
  
H
hjxilinx 已提交
459 460 461 462 463
  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 已提交
464 465
    return pSql->res.code;
  }
466
  
S
slguan 已提交
467 468
  return doProcessSql(pSql);
}
H
hzcheng 已提交
469

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

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

S
slguan 已提交
481
    if (pSub == NULL) {
H
hzcheng 已提交
482 483
      continue;
    }
S
slguan 已提交
484

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

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

512
int tscBuildRetrieveMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
513 514 515 516 517
  char *pMsg, *pStart;

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

518
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg *)pMsg;
S
slguan 已提交
519
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
S
slguan 已提交
520 521
  pMsg += sizeof(pSql->res.qhandle);

522
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
S
slguan 已提交
523
  pRetrieveMsg->free = htons(pQueryInfo->type);
524
  pMsg += sizeof(pQueryInfo->type);
H
hzcheng 已提交
525

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

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

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

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

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

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

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

H
hjxilinx 已提交
603
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
604 605 606 607 608 609 610 611
  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);
612 613 614
  return pMsg;
}

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

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

S
slguan 已提交
620 621 622 623
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for query msg", pSql);
    return -1;
  }
624
  
625
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
626
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
627
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
628
  
629
  if (taosArrayGetSize(pQueryInfo->colList) <= 0) {
630 631 632
    tscError("%p illegal value of numOfCols in query msg: %d", pSql, tscGetNumOfColumns(pTableMeta));
    return -1;
  }
633 634 635 636 637 638 639 640 641 642
  
  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;
  }
643 644

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

S
slguan 已提交
646
  SQueryTableMsg *pQueryMsg = (SQueryTableMsg *)pStart;
H
hzcheng 已提交
647

648
  int32_t msgLen      = 0;
S
slguan 已提交
649
  int32_t numOfTables = 0;
650
  int32_t numOfTags   = taosArrayGetSize(pTableMetaInfo->tagColList);
H
hzcheng 已提交
651

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

680
  if (pQueryInfo->order.order == TSDB_ORDER_ASC) {
H
hjxilinx 已提交
681 682
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.skey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.ekey);
H
hzcheng 已提交
683
  } else {
H
hjxilinx 已提交
684 685
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.ekey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.skey);
H
hzcheng 已提交
686 687
  }

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

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

719
    if (pCol->colIndex.columnIndex >= tscGetNumOfColumns(pTableMeta) || pColSchema->type < TSDB_DATA_TYPE_BOOL ||
720 721 722 723 724
        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);

725
      return TSDB_CODE_INVALID_SQL;
726
    }
H
hzcheng 已提交
727 728 729

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

S
slguan 已提交
733 734 735
    // 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 已提交
736

S
slguan 已提交
737
      SColumnFilterInfo *pFilterMsg = (SColumnFilterInfo *)pMsg;
738
      pFilterMsg->filterstr = htons(pColFilter->filterstr);
S
slguan 已提交
739 740

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

742
      if (pColFilter->filterstr) {
S
slguan 已提交
743 744 745 746 747 748 749 750 751 752
        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 已提交
753

S
slguan 已提交
754 755 756 757 758
      if (pColFilter->lowerRelOptr == TSDB_RELATION_INVALID && pColFilter->upperRelOptr == TSDB_RELATION_INVALID) {
        tscError("invalid filter info");
        return -1;
      }
    }
H
hzcheng 已提交
759 760
  }

H
hjxilinx 已提交
761
  SSqlFuncMsg *pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hjxilinx 已提交
762
  for (int32_t i = 0; i < tscSqlExprNumOfExprs(pQueryInfo); ++i) {
763
    SSqlExpr *pExpr = tscSqlExprGet(pQueryInfo, i);
H
hzcheng 已提交
764

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

771 772 773
    pSqlFuncExpr->colInfo.colId    = htons(pExpr->colInfo.colId);
    pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
    pSqlFuncExpr->colInfo.flag     = htons(pExpr->colInfo.flag);
H
hzcheng 已提交
774

775
    pSqlFuncExpr->functionId  = htons(pExpr->functionId);
H
hzcheng 已提交
776
    pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
H
hjxilinx 已提交
777
    pMsg += sizeof(SSqlFuncMsg);
H
hzcheng 已提交
778 779 780 781 782 783 784

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

H
hjxilinx 已提交
791
    pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hzcheng 已提交
792 793
  }

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

797
  SSqlGroupbyExpr *pGroupbyExpr = &pQueryInfo->groupbyExpr;
798
  if (pGroupbyExpr->numOfGroupCols > 0) {
S
slguan 已提交
799
    pQueryMsg->orderByIdx = htons(pGroupbyExpr->orderIndex);
H
hzcheng 已提交
800 801
    pQueryMsg->orderType = htons(pGroupbyExpr->orderType);

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

808 809
      *((int16_t *)pMsg) += pCol->colIndex;
      pMsg += sizeof(pCol->colIndex);
S
slguan 已提交
810 811 812

      *((int16_t *)pMsg) += pCol->flag;
      pMsg += sizeof(pCol->flag);
H
hjxilinx 已提交
813 814 815
      
      memcpy(pMsg, pCol->name, tListLen(pCol->name));
      pMsg += tListLen(pCol->name);
S
slguan 已提交
816 817 818
    }
  }

819
  if (pQueryInfo->interpoType != TSDB_INTERPO_NONE) {
H
hjxilinx 已提交
820
    for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
821 822
      *((int64_t *)pMsg) = htobe64(pQueryInfo->defaultVal[i]);
      pMsg += sizeof(pQueryInfo->defaultVal[0]);
S
slguan 已提交
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 852 853 854 855
  
  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 已提交
856 857 858 859 860 861

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

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

    // todo refactor
867 868
    fseek(pQueryInfo->tsBuf->f, pBlockInfo->offset, SEEK_SET);
    fread(pMsg, pBlockInfo->compLen, 1, pQueryInfo->tsBuf->f);
S
slguan 已提交
869 870 871 872

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

S
slguan 已提交
875 876
  pQueryMsg->tsLen = htonl(tsLen);
  pQueryMsg->tsNumOfBlocks = htonl(numOfBlocks);
877 878
  if (pQueryInfo->tsBuf != NULL) {
    pQueryMsg->tsOrder = htonl(pQueryInfo->tsBuf->tsOrder);
H
hzcheng 已提交
879 880
  }

H
hjxilinx 已提交
881
  // serialize tag column query condition
882
  if (pQueryInfo->tagCond.pCond != NULL && taosArrayGetSize(pQueryInfo->tagCond.pCond) > 0) {
883 884
    STagCond* pTagCond = &pQueryInfo->tagCond;
    
885
    SCond *pCond = tsGetSTableQueryCond(pTagCond, pTableMeta->uid);
886
    if (pCond != NULL && pCond->cond != NULL) {
887 888
      pQueryMsg->tagCondLen = htons(pCond->len);
      memcpy(pMsg, pCond->cond, pCond->len);
889
      
890
      pMsg += pCond->len;
891 892 893
    }
  }
  
weixin_48148422's avatar
weixin_48148422 已提交
894 895 896 897 898 899 900 901
  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 已提交
902
  // tbname in/like query expression should be sent to mgmt node
H
hzcheng 已提交
903 904 905 906
  msgLen = pMsg - pStart;

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

  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
913 914
}

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

920
  SCMCreateDbMsg *pCreateDbMsg = (SCMCreateDbMsg*)pCmd->payload;
921

922
  assert(pCmd->numOfClause == 1);
923
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
924
  strncpy(pCreateDbMsg->db, pTableMetaInfo->name, tListLen(pCreateDbMsg->db));
H
hzcheng 已提交
925

926
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
927 928
}

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

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

941
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
942 943
}

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

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

954 955
  SSQLToken *pName = &pInfo->pDCLInfo->user.user;
  SSQLToken *pPwd = &pInfo->pDCLInfo->user.passwd;
H
hzcheng 已提交
956

957 958
  strncpy(pAlterMsg->user, pName->z, pName->n);
  strncpy(pAlterMsg->pass, pPwd->z, pPwd->n);
S
slguan 已提交
959

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

962 963 964 965 966 967 968 969
  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 已提交
970

971 972 973 974 975 976 977 978 979 980 981 982 983
  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 已提交
984

S
slguan 已提交
985
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_ACCT;
986
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
987 988
}

989 990
int32_t tscBuildUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
991
  pCmd->payloadLen = sizeof(SCMCreateUserMsg);
S
slguan 已提交
992

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

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

1000 1001 1002
  SUserInfo *pUser = &pInfo->pDCLInfo->user;
  strncpy(pAlterMsg->user, pUser->user.z, pUser->user.n);
  pAlterMsg->flag = pUser->type;
H
hzcheng 已提交
1003

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

1012
  if (pUser->type == TSDB_ALTER_USER_PASSWD || pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
S
slguan 已提交
1013
    pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_USER;
1014
  } else {
S
slguan 已提交
1015
    pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_USER;
1016
  }
H
hzcheng 已提交
1017

1018
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1019 1020
}

1021 1022
int32_t tscBuildCfgDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1023
  pCmd->payloadLen = sizeof(SCMCfgDnodeMsg);
H
hzcheng 已提交
1024

S
slguan 已提交
1025 1026 1027 1028
  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 已提交
1029

S
slguan 已提交
1030
  pCmd->msgType = TSDB_MSG_TYPE_MD_CONFIG_DNODE;
1031 1032
  return TSDB_CODE_SUCCESS;
}
S
slguan 已提交
1033

1034 1035
int32_t tscBuildDropDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1036
  pCmd->payloadLen = sizeof(SCMDropDbMsg);
H
hzcheng 已提交
1037

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

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

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

S
slguan 已提交
1049
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DB;
1050
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1051 1052
}

1053 1054
int32_t tscBuildDropTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1055
  pCmd->payloadLen = sizeof(SCMDropTableMsg);
H
hzcheng 已提交
1056

S
slguan 已提交
1057 1058 1059 1060
  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 已提交
1061

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

S
slguan 已提交
1067
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_TABLE;
1068
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1069 1070
}

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

1079
  SCMDropDnodeMsg *pDrop = (SCMDropDnodeMsg *)pCmd->payload;
1080
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1081
  strcpy(pDrop->ip, pTableMetaInfo->name);
S
slguan 已提交
1082
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DNODE;
H
hzcheng 已提交
1083

1084
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1085 1086
}

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

S
slguan 已提交
1092 1093 1094 1095
  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 已提交
1096

1097
  SCMDropUserMsg *pDropMsg = (SCMDropUserMsg*)pCmd->payload;
1098
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1099
  strcpy(pDropMsg->user, pTableMetaInfo->name);
H
hzcheng 已提交
1100

1101
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1102 1103
}

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

1121 1122
int32_t tscBuildUseDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1123
  pCmd->payloadLen = sizeof(SCMUseDbMsg);
H
hzcheng 已提交
1124

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

1130
  SCMUseDbMsg *pUseDbMsg = (SCMUseDbMsg*)pCmd->payload;
1131
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1132
  strcpy(pUseDbMsg->db, pTableMetaInfo->name);
S
slguan 已提交
1133
  pCmd->msgType = TSDB_MSG_TYPE_CM_USE_DB;
H
hzcheng 已提交
1134

1135
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1136 1137
}

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

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

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

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

1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170
  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 已提交
1171

1172 1173 1174 1175
    strncpy(pShowMsg->payload, pIpAddr->z, pIpAddr->n);
    pShowMsg->payloadLen = htons(pIpAddr->n);
  }

1176
  pCmd->payloadLen = sizeof(SCMShowMsg) + pShowMsg->payloadLen;
1177
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1178 1179
}

1180
int32_t tscBuildKillMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1181
  SSqlCmd *pCmd = &pSql->cmd;
1182
  pCmd->payloadLen = sizeof(SCMKillQueryMsg);
H
hzcheng 已提交
1183

S
slguan 已提交
1184 1185 1186 1187
  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 已提交
1188

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

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

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

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

1217 1218 1219
  if (pCreateTableInfo->pSelect != NULL) {
    size += (pCreateTableInfo->pSelect->selectToken.n + 1);
  }
H
hzcheng 已提交
1220 1221 1222 1223

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

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

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
1231
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1232 1233

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


1241
  SCMCreateTableMsg *pCreateTableMsg = (SCMCreateTableMsg *)pCmd->payload;
H
hjxilinx 已提交
1242
  strcpy(pCreateTableMsg->tableId, pTableMetaInfo->name);
1243 1244

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

1247 1248 1249
  SCreateTableSQL *pCreateTable = pInfo->pCreateTableInfo;

  pCreateTableMsg->igExists = pCreateTable->existCheck ? 1 : 0;
H
hzcheng 已提交
1250 1251 1252 1253
  pCreateTableMsg->numOfColumns = htons(pCmd->numOfCols);
  pCreateTableMsg->numOfTags = htons(pCmd->count);

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

1256 1257 1258
  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 已提交
1259
    pMsg += sizeof(STagData);
1260
  } else {  // create (super) table
1261
    pSchema = (SSchema *)pCreateTableMsg->schema;
1262

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

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

H
hzcheng 已提交
1270 1271 1272 1273
      pSchema++;
    }

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

1277 1278 1279
      strncpy(pMsg, pQuerySql->selectToken.z, pQuerySql->selectToken.n + 1);
      pCreateTableMsg->sqlLen = htons(pQuerySql->selectToken.n + 1);
      pMsg += pQuerySql->selectToken.n + 1;
H
hzcheng 已提交
1280 1281 1282
    }
  }

H
hjxilinx 已提交
1283
  tscFieldInfoClear(&pQueryInfo->fieldsInfo);
H
hzcheng 已提交
1284

S
slguan 已提交
1285
  msgLen = pMsg - (char*)pCreateTableMsg;
S
slguan 已提交
1286
  pCreateTableMsg->contLen = htonl(msgLen);
H
hzcheng 已提交
1287
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1288
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_TABLE;
H
hzcheng 已提交
1289 1290

  assert(msgLen + minMsgSize() <= size);
1291
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1292 1293 1294
}

int tscEstimateAlterTableMsgLength(SSqlCmd *pCmd) {
1295
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
1296
  return minMsgSize() + sizeof(SMgmtHead) + sizeof(SCMAlterTableMsg) + sizeof(SSchema) * tscNumOfFields(pQueryInfo) +
H
hzcheng 已提交
1297 1298 1299
         TSDB_EXTRA_PAYLOAD_SIZE;
}

1300
int tscBuildAlterTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1301
  SCMAlterTableMsg *pAlterTableMsg;
1302
  char *          pMsg;
H
hzcheng 已提交
1303 1304 1305
  int             msgLen = 0;
  int             size = 0;

1306 1307 1308
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);

H
hjxilinx 已提交
1309
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1310 1311

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

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

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

1321 1322
  SAlterTableSQL *pAlterInfo = pInfo->pAlterInfo;

H
hjxilinx 已提交
1323
  strcpy(pAlterTableMsg->tableId, pTableMetaInfo->name);
1324
  pAlterTableMsg->type = htons(pAlterInfo->type);
1325

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

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

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

  pMsg = (char *)pSchema;

S
slguan 已提交
1341
  msgLen = pMsg - (char*)pAlterTableMsg;
H
hzcheng 已提交
1342
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1343
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_TABLE;
H
hzcheng 已提交
1344 1345

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

1347
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1348 1349
}

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

S
slguan 已提交
1355 1356 1357 1358
  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 已提交
1359

1360
  SCMAlterDbMsg *pAlterDbMsg = (SCMAlterDbMsg*)pCmd->payload;
1361
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1362
  strcpy(pAlterDbMsg->db, pTableMetaInfo->name);
H
hzcheng 已提交
1363

1364
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1365 1366
}

1367
int tscBuildRetrieveFromMgmtMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1368 1369 1370
  SSqlCmd *pCmd = &pSql->cmd;
  pCmd->msgType = TSDB_MSG_TYPE_RETRIEVE;
  pCmd->payloadLen = sizeof(SRetrieveTableMsg);
S
slguan 已提交
1371

S
slguan 已提交
1372 1373 1374
  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 已提交
1375
  }
S
slguan 已提交
1376

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

1382
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1383 1384
}

1385
static int tscSetResultPointer(SQueryInfo *pQueryInfo, SSqlRes *pRes) {
1386
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
H
hzcheng 已提交
1387 1388 1389
    return pRes->code;
  }

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

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

1408
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
1409

H
hzcheng 已提交
1410 1411 1412 1413 1414 1415 1416
  pRes->code = TSDB_CODE_SUCCESS;

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

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

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

H
hjxilinx 已提交
1438
  STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
H
hjxilinx 已提交
1439 1440
  
  int32_t numOfRes = tinfo.numOfColumns + tinfo.numOfTags;
H
hzcheng 已提交
1441 1442 1443 1444
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

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

1447 1448
//  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
//  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
S
slguan 已提交
1449 1450

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

H
hzcheng 已提交
1460 1461 1462 1463 1464 1465 1466
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

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

H
hjxilinx 已提交
1467 1468
  pRes->code = tscDoLocalreduce(pSql);
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hzcheng 已提交
1469 1470

  if (pRes->code == TSDB_CODE_SUCCESS && pRes->numOfRows > 0) {
1471
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1472 1473 1474 1475
  }

  pRes->row = 0;

1476
  uint8_t code = pRes->code;
H
hzcheng 已提交
1477
  if (pSql->fp) {  // async retrieve metric data
1478 1479
    if (pRes->code == TSDB_CODE_SUCCESS) {
      (*pSql->fp)(pSql->param, pSql, pRes->numOfRows);
H
hzcheng 已提交
1480 1481 1482 1483 1484 1485 1486 1487
    } else {
      tscQueueAsyncRes(pSql);
    }
  }

  return code;
}

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

1490
int tscBuildConnectMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1491
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1492
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1493
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONNECT;
1494
  pCmd->payloadLen = sizeof(SCMConnectMsg);
H
hzcheng 已提交
1495

S
slguan 已提交
1496 1497 1498 1499 1500
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

1501
  SCMConnectMsg *pConnect = (SCMConnectMsg*)pCmd->payload;
H
hzcheng 已提交
1502 1503 1504 1505 1506

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

1510
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1511 1512
}

H
hjxilinx 已提交
1513
int tscBuildTableMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1514
  SCMTableInfoMsg *pInfoMsg;
S
slguan 已提交
1515
  char *         pMsg;
H
hzcheng 已提交
1516 1517 1518 1519 1520
  int            msgLen = 0;

  char *tmpData = 0;
  if (pSql->cmd.allocSize > 0) {
    tmpData = calloc(1, pSql->cmd.allocSize);
1521 1522 1523 1524
    if (NULL == tmpData) {
      return TSDB_CODE_CLI_OUT_OF_MEMORY;
    }

H
hzcheng 已提交
1525 1526 1527 1528
    // STagData is in binary format, strncpy is not available
    memcpy(tmpData, pSql->cmd.payload, pSql->cmd.allocSize);
  }

1529 1530 1531
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

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

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

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

H
hjxilinx 已提交
1540
  if (pSql->cmd.autoCreated) {
H
hzcheng 已提交
1541 1542 1543 1544
    memcpy(pInfoMsg->tags, tmpData, sizeof(STagData));
    pMsg += sizeof(STagData);
  }

H
hjxilinx 已提交
1545
  pCmd->payloadLen = pMsg - (char*)pInfoMsg;;
S
slguan 已提交
1546
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLE_META;
H
hzcheng 已提交
1547 1548 1549 1550

  tfree(tmpData);

  assert(msgLen + minMsgSize() <= pCmd->allocSize);
1551
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1552 1553
}

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

1574
  SCMMultiTableInfoMsg *pInfoMsg = (SCMMultiTableInfoMsg *)(pCmd->payload + tsRpcHeadSize + sizeof(SMgmtHead));
S
slguan 已提交
1575
  pInfoMsg->numOfTables = htonl((int32_t)pCmd->count);
S
slguan 已提交
1576 1577

  if (pCmd->payloadLen > 0) {
S
slguan 已提交
1578
    memcpy(pInfoMsg->tableIds, tmpData, pCmd->payloadLen);
S
slguan 已提交
1579 1580 1581 1582
  }

  tfree(tmpData);

1583
  pCmd->payloadLen += sizeof(SMgmtHead) + sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1584
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLES_META;
S
slguan 已提交
1585 1586 1587

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

S
slguan 已提交
1588
  tscTrace("%p build load multi-metermeta msg completed, numOfTables:%d, msg size:%d", pSql, pCmd->count,
S
slguan 已提交
1589 1590 1591 1592 1593
           pCmd->payloadLen);

  return pCmd->payloadLen;
}

1594
static UNUSED_FUNC int32_t tscEstimateMetricMetaMsgSize(SSqlCmd *pCmd) {
H
hzcheng 已提交
1595
  const int32_t defaultSize =
S
slguan 已提交
1596
      minMsgSize() + sizeof(SSuperTableMetaMsg) + sizeof(SMgmtHead) + sizeof(int16_t) * TSDB_MAX_TAGS;
1597
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hzcheng 已提交
1598

S
slguan 已提交
1599
  int32_t n = 0;
1600 1601 1602 1603
  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 已提交
1604
  }
S
slguan 已提交
1605

H
hjxilinx 已提交
1606
  int32_t tagLen = n * TSDB_NCHAR_SIZE;
1607 1608
  if (pQueryInfo->tagCond.tbnameCond.cond != NULL) {
    tagLen += strlen(pQueryInfo->tagCond.tbnameCond.cond) * TSDB_NCHAR_SIZE;
H
hjxilinx 已提交
1609
  }
1610

S
slguan 已提交
1611
  int32_t joinCondLen = (TSDB_TABLE_ID_LEN + sizeof(int16_t)) * 2;
S
slguan 已提交
1612
  int32_t elemSize = sizeof(SSuperTableMetaElemMsg) * pQueryInfo->numOfTables;
H
hjxilinx 已提交
1613
  
1614
  int32_t colSize = pQueryInfo->groupbyExpr.numOfGroupCols*sizeof(SColIndex);
S
slguan 已提交
1615

H
hjxilinx 已提交
1616
  int32_t len = tagLen + joinCondLen + elemSize + colSize + defaultSize;
S
slguan 已提交
1617 1618

  return MAX(len, TSDB_DEFAULT_PAYLOAD_SIZE);
H
hzcheng 已提交
1619 1620
}

H
hjxilinx 已提交
1621 1622 1623
int tscBuildSTableVgroupMsg(SSqlObj *pSql, SSqlInfo *pInfo) {

#if 0
S
slguan 已提交
1624
  SSuperTableMetaMsg *pMetaMsg;
H
hzcheng 已提交
1625 1626
  char *          pMsg, *pStart;
  int             msgLen = 0;
S
slguan 已提交
1627
  int             tableIndex = 0;
H
hzcheng 已提交
1628

1629 1630 1631
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

1632
  STagCond *pTagCond = &pQueryInfo->tagCond;
S
slguan 已提交
1633

H
hjxilinx 已提交
1634
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
H
hzcheng 已提交
1635 1636

  int32_t size = tscEstimateMetricMetaMsgSize(pCmd);
S
slguan 已提交
1637 1638 1639 1640
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for metric meter msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1641 1642 1643 1644 1645

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

  SMgmtHead *pMgmt = (SMgmtHead *)pMsg;
H
hjxilinx 已提交
1646
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pMgmt->db);
H
hzcheng 已提交
1647 1648 1649

  pMsg += sizeof(SMgmtHead);

S
slguan 已提交
1650
  pMetaMsg = (SSuperTableMetaMsg *)pMsg;
S
slguan 已提交
1651
  pMetaMsg->numOfTables = htonl(pQueryInfo->numOfTables);
S
slguan 已提交
1652

S
slguan 已提交
1653
  pMsg += sizeof(SSuperTableMetaMsg);
S
slguan 已提交
1654 1655 1656 1657 1658

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

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

S
slguan 已提交
1661
  memcpy(pMsg, pTagCond->joinInfo.left.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1662
  pMsg += TSDB_TABLE_ID_LEN;
H
hzcheng 已提交
1663

S
slguan 已提交
1664 1665 1666
  *(int16_t *)pMsg = pTagCond->joinInfo.left.tagCol;
  pMsg += sizeof(int16_t);

S
slguan 已提交
1667
  memcpy(pMsg, pTagCond->joinInfo.right.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1668
  pMsg += TSDB_TABLE_ID_LEN;
S
slguan 已提交
1669 1670 1671 1672

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

1673
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
1674
    pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
H
hjxilinx 已提交
1675
    uint64_t uid = pTableMetaInfo->pTableMeta->uid;
S
slguan 已提交
1676 1677 1678 1679

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

S
slguan 已提交
1680 1681
    SSuperTableMetaElemMsg *pElem = (SSuperTableMetaElemMsg *)pMsg;
    pMsg += sizeof(SSuperTableMetaElemMsg);
S
slguan 已提交
1682 1683 1684 1685

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

H
hjxilinx 已提交
1690
        bool ret = taosMbsToUcs4(pCond->cond, condLen, pMsg, condLen * TSDB_NCHAR_SIZE);
S
slguan 已提交
1691
        if (!ret) {
1692
          tscError("%p mbs to ucs4 failed:%s", pSql, tsGetSTableQueryCond(pTagCond, uid));
S
slguan 已提交
1693 1694 1695
          return 0;
        }
      }
H
hzcheng 已提交
1696 1697
    }

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

S
slguan 已提交
1700 1701 1702
    offset = pMsg - (char *)pMetaMsg;
    pElem->cond = htonl(offset);
    pMsg += condLen * TSDB_NCHAR_SIZE;
H
hzcheng 已提交
1703

S
slguan 已提交
1704 1705 1706
    pElem->rel = htons(pTagCond->relType);
    if (pTagCond->tbnameCond.uid == uid) {
      offset = pMsg - (char *)pMetaMsg;
H
hzcheng 已提交
1707

S
slguan 已提交
1708
      pElem->tableCond = htonl(offset);
H
hjxilinx 已提交
1709 1710 1711 1712 1713 1714 1715
      
      uint32_t len = 0;
      if (pTagCond->tbnameCond.cond != NULL) {
        len = strlen(pTagCond->tbnameCond.cond);
        memcpy(pMsg, pTagCond->tbnameCond.cond, len);
      }
      
H
hjxilinx 已提交
1716 1717
      pElem->tableCondLen = htonl(len);
      pMsg += len;
S
slguan 已提交
1718 1719
    }

1720
    SSqlGroupbyExpr *pGroupby = &pQueryInfo->groupbyExpr;
S
slguan 已提交
1721

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

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

        pElem->groupbyTagColumnList = htonl(offset);
1738
        for (int32_t j = 0; j < pQueryInfo->groupbyExpr.numOfGroupCols; ++j) {
1739 1740
          SColIndex *pCol = &pQueryInfo->groupbyExpr.columnInfo[j];
          SColIndex *pDestCol = (SColIndex *)pMsg;
1741

H
hjxilinx 已提交
1742
          pDestCol->colIdxInBuf = 0;
1743
          pDestCol->colIndex = htons(pCol->colIndex);
H
hjxilinx 已提交
1744 1745
          pDestCol->colId = htons(pDestCol->colId);
          pDestCol->flag = htons(pDestCol->flag);
H
hjxilinx 已提交
1746
          strncpy(pDestCol->name, pCol->name, tListLen(pCol->name));
1747

1748
          pMsg += sizeof(SColIndex);
S
slguan 已提交
1749 1750
        }
      }
H
hzcheng 已提交
1751
    }
S
slguan 已提交
1752

H
hjxilinx 已提交
1753 1754
    strcpy(pElem->tableId, pTableMetaInfo->name);
    pElem->numOfTags = htons(pTableMetaInfo->numOfTags);
S
slguan 已提交
1755 1756 1757

    int16_t len = pMsg - (char *)pElem;
    pElem->elemLen = htons(len);  // redundant data for integrate check
H
hzcheng 已提交
1758 1759 1760 1761
  }

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1762
  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
H
hzcheng 已提交
1763
  assert(msgLen + minMsgSize() <= size);
H
hjxilinx 已提交
1764
#endif
1765
  
H
hjxilinx 已提交
1766 1767 1768 1769 1770 1771 1772 1773 1774 1775
  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);

1776
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1777 1778
}

1779
int tscEstimateHeartBeatMsgLength(SSqlObj *pSql) {
H
hzcheng 已提交
1780 1781 1782 1783
  int      size = 0;
  STscObj *pObj = pSql->pTscObj;

  size += tsRpcHeadSize + sizeof(SMgmtHead);
S
slguan 已提交
1784
  size += sizeof(SQqueryList);
H
hzcheng 已提交
1785 1786 1787

  SSqlObj *tpSql = pObj->sqlList;
  while (tpSql) {
S
slguan 已提交
1788
    size += sizeof(SQueryDesc);
H
hzcheng 已提交
1789 1790 1791
    tpSql = tpSql->next;
  }

S
slguan 已提交
1792
  size += sizeof(SStreamList);
H
hzcheng 已提交
1793 1794
  SSqlStream *pStream = pObj->streamList;
  while (pStream) {
S
slguan 已提交
1795
    size += sizeof(SStreamDesc);
H
hzcheng 已提交
1796 1797 1798 1799 1800 1801
    pStream = pStream->next;
  }

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1802
int tscBuildHeartBeatMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1803 1804 1805 1806 1807 1808 1809 1810 1811
  char *pMsg, *pStart;
  int   msgLen = 0;
  int   size = 0;

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

  pthread_mutex_lock(&pObj->mutex);

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

  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 已提交
1831
  pCmd->msgType = TSDB_MSG_TYPE_CM_HEARTBEAT;
H
hzcheng 已提交
1832 1833

  assert(msgLen + minMsgSize() <= size);
1834
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1835 1836
}

1837 1838
int tscProcessTableMetaRsp(SSqlObj *pSql) {
  STableMetaMsg *pMetaMsg = (STableMetaMsg *)pSql->res.pRsp;
H
hzcheng 已提交
1839

1840 1841
  pMetaMsg->sid = htonl(pMetaMsg->sid);
  pMetaMsg->sversion = htons(pMetaMsg->sversion);
H
hjxilinx 已提交
1842 1843 1844
  
  pMetaMsg->vgroup.vgId = htonl(pMetaMsg->vgroup.vgId);
  
1845 1846
  pMetaMsg->uid = htobe64(pMetaMsg->uid);
  pMetaMsg->contLen = htons(pMetaMsg->contLen);
1847
  pMetaMsg->numOfColumns = htons(pMetaMsg->numOfColumns);
H
hzcheng 已提交
1848

H
hjxilinx 已提交
1849 1850
  if (pMetaMsg->sid < 0 || pMetaMsg->vgroup.numOfIps < 0) {
    tscError("invalid meter vgId:%d, sid%d", pMetaMsg->vgroup.numOfIps, pMetaMsg->sid);
H
hzcheng 已提交
1851 1852 1853
    return TSDB_CODE_INVALID_VALUE;
  }

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

1859 1860
  if (pMetaMsg->numOfColumns > TSDB_MAX_COLUMNS || pMetaMsg->numOfColumns <= 0) {
    tscError("invalid numOfColumns:%d", pMetaMsg->numOfColumns);
H
hzcheng 已提交
1861 1862 1863
    return TSDB_CODE_INVALID_VALUE;
  }

H
hjxilinx 已提交
1864 1865 1866 1867 1868
  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 已提交
1869 1870
  }

1871
  SSchema* pSchema = pMetaMsg->schema;
H
hzcheng 已提交
1872

1873
  int32_t numOfTotalCols = pMetaMsg->numOfColumns + pMetaMsg->numOfTags;
H
hzcheng 已提交
1874 1875 1876
  for (int i = 0; i < numOfTotalCols; ++i) {
    pSchema->bytes = htons(pSchema->bytes);
    pSchema->colId = htons(pSchema->colId);
H
hjxilinx 已提交
1877 1878
    
    assert(pSchema->type >= TSDB_DATA_TYPE_BOOL && pSchema->type <= TSDB_DATA_TYPE_NCHAR);
H
hzcheng 已提交
1879 1880 1881
    pSchema++;
  }

1882 1883
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
H
hzcheng 已提交
1884

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

H
hjxilinx 已提交
1900 1901 1902
  pTableMetaInfo->pTableMeta =
      (STableMeta *) taosCachePut(tscCacheHandle, pTableMetaInfo->name, pTableMeta, size, tsMeterMetaKeepTimer);
  
1903
  // todo handle out of memory case
1904
  if (pTableMetaInfo->pTableMeta == NULL) {
H
hjxilinx 已提交
1905
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
1906
  }
H
hzcheng 已提交
1907

1908
  free(pTableMeta);
1909
  
H
hjxilinx 已提交
1910
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1911 1912
}

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

1932
  SCMMultiTableInfoMsg *pInfo = (SCMMultiTableInfoMsg *)rsp;
S
slguan 已提交
1933
  totalNum = htonl(pInfo->numOfTables);
1934
  rsp += sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1935 1936

  for (i = 0; i < totalNum; i++) {
S
slguan 已提交
1937
    SMultiTableMeta *pMultiMeta = (SMultiTableMeta *)rsp;
1938
    STableMeta *     pMeta = pMultiMeta->metas;
S
slguan 已提交
1939 1940 1941

    pMeta->sid = htonl(pMeta->sid);
    pMeta->sversion = htons(pMeta->sversion);
H
hjxilinx 已提交
1942
    pMeta->vgId = htonl(pMeta->vgId);
S
slguan 已提交
1943 1944
    pMeta->uid = htobe64(pMeta->uid);

H
hjxilinx 已提交
1945 1946
    if (pMeta->sid <= 0 || pMeta->vgId < 0) {
      tscError("invalid meter vgId:%d, sid%d", pMeta->vgId, pMeta->sid);
S
slguan 已提交
1947 1948 1949 1950 1951
      pSql->res.code = TSDB_CODE_INVALID_VALUE;
      pSql->res.numOfTotal = i;
      return TSDB_CODE_OTHERS;
    }

H
hjxilinx 已提交
1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974
    //    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 已提交
1975
    //    for (int j = 0; j < TSDB_REPLICA_MAX_NUM; ++j) {
H
hjxilinx 已提交
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 2006 2007 2008 2009 2010 2011
    //      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 已提交
2012
  }
H
hjxilinx 已提交
2013
  
S
slguan 已提交
2014 2015 2016
  pSql->res.code = TSDB_CODE_SUCCESS;
  pSql->res.numOfTotal = i;
  tscTrace("%p load multi-metermeta resp complete num:%d", pSql, pSql->res.numOfTotal);
2017 2018
#endif
  
S
slguan 已提交
2019 2020 2021
  return TSDB_CODE_SUCCESS;
}

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

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

S
slguan 已提交
2048
    pMeta->numOfTables = htonl(pMeta->numOfTables);
S
slguan 已提交
2049 2050 2051
    pMeta->numOfVnodes = htonl(pMeta->numOfVnodes);
    pMeta->tagLen = htons(pMeta->tagLen);

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

2054 2055
    char *pBuf = calloc(1, size);
    if (pBuf == NULL) {
S
slguan 已提交
2056 2057 2058
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2059

S
slguan 已提交
2060
    SSuperTableMeta *pNewMetricMeta = (SSuperTableMeta *)pBuf;
S
slguan 已提交
2061
    metricMetaList[k] = pNewMetricMeta;
H
hzcheng 已提交
2062

S
slguan 已提交
2063
    pNewMetricMeta->numOfTables = pMeta->numOfTables;
S
slguan 已提交
2064 2065
    pNewMetricMeta->numOfVnodes = pMeta->numOfVnodes;
    pNewMetricMeta->tagLen = pMeta->tagLen;
H
hzcheng 已提交
2066

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

S
slguan 已提交
2069 2070
    for (int32_t i = 0; i < pMeta->numOfVnodes; ++i) {
      SVnodeSidList *pSidLists = (SVnodeSidList *)rsp;
2071
      memcpy(pBuf, pSidLists, sizeof(SVnodeSidList));
H
hzcheng 已提交
2072

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

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

2078
      pBuf += sizeof(SVnodeSidList) + sizeof(STableIdInfo *) * pSidLists->numOfSids;
S
slguan 已提交
2079
      rsp += sizeof(SVnodeSidList);
H
hzcheng 已提交
2080

2081
      size_t elemSize = sizeof(STableIdInfo) + pNewMetricMeta->tagLen;
S
slguan 已提交
2082
      for (int32_t j = 0; j < pSidLists->numOfSids; ++j) {
2083 2084
        pLists->pSidExtInfoList[j] = pBuf - (char *)pLists;
        memcpy(pBuf, rsp, elemSize);
2085

2086 2087
        ((STableIdInfo *)pBuf)->uid = htobe64(((STableIdInfo *)pBuf)->uid);
        ((STableIdInfo *)pBuf)->sid = htonl(((STableIdInfo *)pBuf)->sid);
2088

2089 2090
        rsp += elemSize;
        pBuf += elemSize;
S
slguan 已提交
2091
      }
H
hzcheng 已提交
2092
    }
S
slguan 已提交
2093

2094
    sizes[k] = pBuf - (char *)pNewMetricMeta;
H
hzcheng 已提交
2095 2096
  }

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

H
hjxilinx 已提交
2101
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2102
    tscGetMetricMetaCacheKey(pQueryInfo, name, pTableMetaInfo->pTableMeta->uid);
H
hzcheng 已提交
2103

S
slguan 已提交
2104 2105 2106
#ifdef _DEBUG_VIEW
    printf("generate the metric key:%s, index:%d\n", name, i);
#endif
H
hzcheng 已提交
2107

S
slguan 已提交
2108
    // release the used metricmeta
H
hjxilinx 已提交
2109 2110
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), false);
    pTableMetaInfo->pMetricMeta = (SSuperTableMeta *)taosCachePut(tscCacheHandle, name, (char *)metricMetaList[i],
S
slguan 已提交
2111 2112 2113 2114
                                                                      sizes[i], tsMetricMetaKeepTimer);
    tfree(metricMetaList[i]);

    // failed to put into cache
H
hjxilinx 已提交
2115
    if (pTableMetaInfo->pMetricMeta == NULL) {
S
slguan 已提交
2116 2117 2118
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2119 2120
  }

S
slguan 已提交
2121 2122 2123 2124 2125 2126 2127 2128
_error_clean:
  // free allocated resource
  for (int32_t i = 0; i < num; ++i) {
    tfree(metricMetaList[i]);
  }

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

/*
 * current process do not use the cache at all
 */
int tscProcessShowRsp(SSqlObj *pSql) {
H
hjxilinx 已提交
2164
  STableMetaMsg * pMetaMsg;
2165
  SCMShowRsp *pShow;
S
slguan 已提交
2166
  SSchema *    pSchema;
H
hzcheng 已提交
2167 2168
  char         key[20];

2169 2170 2171
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

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

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

2176
  pShow = (SCMShowRsp *)pRes->pRsp;
S
slguan 已提交
2177
  pShow->qhandle = htobe64(pShow->qhandle);
H
hzcheng 已提交
2178 2179
  pRes->qhandle = pShow->qhandle;

S
slguan 已提交
2180
  tscResetForNextRetrieve(pRes);
H
hjxilinx 已提交
2181
  pMetaMsg = &(pShow->tableMeta);
H
hzcheng 已提交
2182

H
hjxilinx 已提交
2183
  pMetaMsg->numOfColumns = ntohs(pMetaMsg->numOfColumns);
H
hzcheng 已提交
2184

H
hjxilinx 已提交
2185
  pSchema = pMetaMsg->schema;
H
hjxilinx 已提交
2186 2187
  pMetaMsg->sid = ntohs(pMetaMsg->sid);
  for (int i = 0; i < pMetaMsg->numOfColumns; ++i) {
H
hzcheng 已提交
2188 2189 2190 2191
    pSchema->bytes = htons(pSchema->bytes);
    pSchema++;
  }

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

H
hjxilinx 已提交
2195
  taosCacheRelease(tscCacheHandle, (void *)&(pTableMetaInfo->pTableMeta), false);
H
hjxilinx 已提交
2196
  
H
hjxilinx 已提交
2197 2198 2199
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
  
H
hjxilinx 已提交
2200
  pTableMetaInfo->pTableMeta = taosCachePut(tscCacheHandle, key, (char *)pTableMeta, size, tsMeterMetaKeepTimer);
H
hjxilinx 已提交
2201
  SSchema *pTableSchema = tscGetTableSchema(pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2202

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

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

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

H
hjxilinx 已提交
2239 2240 2241
  assert(len <= tListLen(pObj->db));
  strncpy(pObj->db, temp, tListLen(pObj->db));
  
S
slguan 已提交
2242
  tscSetMgmtIpList(&pConnect->ipList);
H
hzcheng 已提交
2243

S
slguan 已提交
2244
  strcpy(pObj->sversion, pConnect->serverVersion);
H
hzcheng 已提交
2245 2246 2247 2248 2249 2250 2251 2252
  pObj->writeAuth = pConnect->writeAuth;
  pObj->superAuth = pConnect->superAuth;
  taosTmrReset(tscProcessActivityTimer, tsShellActivityTimer * 500, pObj, tscTmr, &pObj->pTimer);

  return 0;
}

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

H
hjxilinx 已提交
2256
  strcpy(pObj->db, pTableMetaInfo->name);
H
hzcheng 已提交
2257 2258 2259 2260
  return 0;
}

int tscProcessDropDbRsp(SSqlObj *UNUSED_PARAM(pSql)) {
2261
  taosCacheEmpty(tscCacheHandle);
H
hzcheng 已提交
2262 2263 2264 2265
  return 0;
}

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

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

H
hjxilinx 已提交
2284 2285
  if (pTableMetaInfo->pTableMeta) {
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2286
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2287 2288 2289 2290 2291 2292
  }

  return 0;
}

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

H
hjxilinx 已提交
2295 2296
  STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pTableMetaInfo->name);
  if (pTableMeta == NULL) { /* not in cache, abort */
H
hzcheng 已提交
2297 2298 2299
    return 0;
  }

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

H
hjxilinx 已提交
2303
  if (pTableMetaInfo->pTableMeta) {
H
hjxilinx 已提交
2304
    bool isSuperTable = UTIL_TABLE_IS_SUPERTABLE(pTableMetaInfo);
H
hzcheng 已提交
2305

H
hjxilinx 已提交
2306
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2307
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2308

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

  return 0;
}

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

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

2326
  SQueryTableRsp *pQuery = (SQueryTableRsp *)pRes->pRsp;
S
slguan 已提交
2327 2328 2329
  pQuery->qhandle = htobe64(pQuery->qhandle);
  pRes->qhandle = pQuery->qhandle;

H
hzcheng 已提交
2330
  pRes->data = NULL;
S
slguan 已提交
2331
  tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
2332 2333 2334
  return 0;
}

H
hjxilinx 已提交
2335
int tscProcessRetrieveRspFromNode(SSqlObj *pSql) {
S
slguan 已提交
2336 2337 2338
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

S
slguan 已提交
2339
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2340 2341 2342

  pRes->numOfRows = htonl(pRetrieve->numOfRows);
  pRes->precision = htons(pRetrieve->precision);
2343 2344
  pRes->offset    = htobe64(pRetrieve->offset);
  pRes->useconds  = htobe64(pRetrieve->useconds);
H
hjxilinx 已提交
2345
  pRes->completed = (pRetrieve->completed == 1);
2346
  pRes->data      = pRetrieve->data;
H
hjxilinx 已提交
2347
  
2348 2349
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
  tscSetResultPointer(pQueryInfo, pRes);
2350

weixin_48148422's avatar
weixin_48148422 已提交
2351
  if (pSql->pSubscription != NULL) {
H
hjxilinx 已提交
2352
    int32_t numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2353
    
H
hjxilinx 已提交
2354
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, numOfCols - 1);
H
hjxilinx 已提交
2355 2356
    int16_t     offset = tscFieldInfoGetOffset(pQueryInfo, numOfCols - 1);
    
weixin_48148422's avatar
weixin_48148422 已提交
2357 2358
    char* p = pRes->data + (pField->bytes + offset) * pRes->numOfRows;

S
slguan 已提交
2359
    int32_t numOfTables = htonl(*(int32_t*)p);
weixin_48148422's avatar
weixin_48148422 已提交
2360
    p += sizeof(int32_t);
S
slguan 已提交
2361
    for (int i = 0; i < numOfTables; i++) {
weixin_48148422's avatar
weixin_48148422 已提交
2362 2363 2364 2365
      int64_t uid = htobe64(*(int64_t*)p);
      p += sizeof(int64_t);
      TSKEY key = htobe64(*(TSKEY*)p);
      p += sizeof(TSKEY);
2366
      tscUpdateSubscriptionProgress(pSql->pSubscription, uid, key);
weixin_48148422's avatar
weixin_48148422 已提交
2367
    }
2368 2369
  }

H
hzcheng 已提交
2370
  pRes->row = 0;
S
slguan 已提交
2371
  tscTrace("%p numOfRows:%d, offset:%d", pSql, pRes->numOfRows, pRes->offset);
H
hzcheng 已提交
2372 2373 2374 2375 2376

  return 0;
}

int tscProcessRetrieveRspFromLocal(SSqlObj *pSql) {
2377 2378
  SSqlRes *   pRes = &pSql->res;
  SSqlCmd *   pCmd = &pSql->cmd;
2379
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
2380

S
slguan 已提交
2381
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2382 2383 2384

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

2386
  tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
2387 2388 2389 2390
  pRes->row = 0;
  return 0;
}

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

2393
static int32_t getTableMetaFromMgmt(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo) {
S
slguan 已提交
2394 2395
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
  if (NULL == pNew) {
2396
    tscError("%p malloc failed for new sqlobj to get table meta", pSql);
S
slguan 已提交
2397 2398
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
2399

H
hzcheng 已提交
2400 2401 2402
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;
  pNew->cmd.command = TSDB_SQL_META;
S
slguan 已提交
2403

2404
  tscAddSubqueryInfo(&pNew->cmd);
2405 2406 2407 2408

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

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

S
slguan 已提交
2414 2415 2416
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

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

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

H
hjxilinx 已提交
2424 2425
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
H
hzcheng 已提交
2426

H
hjxilinx 已提交
2427 2428 2429
  int32_t code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2430 2431 2432 2433 2434
  }

  return code;
}

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

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

    return TSDB_CODE_SUCCESS;
  }
2451 2452
  
  return getTableMetaFromMgmt(pSql, pTableMetaInfo);
H
hzcheng 已提交
2453 2454
}

H
hjxilinx 已提交
2455
int tscGetMeterMetaEx(SSqlObj *pSql, STableMetaInfo *pTableMetaInfo, bool createIfNotExists) {
H
hjxilinx 已提交
2456
  pSql->cmd.autoCreated = createIfNotExists;
H
hjxilinx 已提交
2457
  return tscGetTableMeta(pSql, pTableMetaInfo);
H
hzcheng 已提交
2458 2459 2460 2461
}

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

/**
 * in renew metermeta, do not retrieve metadata in cache.
 * @param pSql          sql object
S
slguan 已提交
2476
 * @param tableId       meter id
H
hzcheng 已提交
2477 2478
 * @return              status code
 */
S
slguan 已提交
2479
int tscRenewMeterMeta(SSqlObj *pSql, char *tableId) {
2480 2481
  int code = 0;

H
hjxilinx 已提交
2482
  // handle table meta renew process
H
hzcheng 已提交
2483
  SSqlCmd *pCmd = &pSql->cmd;
2484 2485

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
2486
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
2487 2488

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

2499
    tscWaitingForCreateTable(pCmd);
H
hjxilinx 已提交
2500
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hzcheng 已提交
2501

2502
    code = getTableMetaFromMgmt(pSql, pTableMetaInfo);  // todo ??
H
hzcheng 已提交
2503
  } else {
H
hjxilinx 已提交
2504
    tscTrace("%p metric query not update metric meta, numOfTags:%d, numOfCols:%d, uid:%" PRId64 ", addr:%p", pSql,
H
hjxilinx 已提交
2505 2506
             tscGetNumOfTags(pTableMetaInfo->pTableMeta), pCmd->numOfCols, pTableMetaInfo->pTableMeta->uid,
             pTableMetaInfo->pTableMeta);
H
hzcheng 已提交
2507 2508 2509 2510 2511
  }

  return code;
}

H
hjxilinx 已提交
2512
int tscGetSTableVgroupInfo(SSqlObj *pSql, int32_t clauseIndex) {
S
slguan 已提交
2513 2514
  int      code = TSDB_CODE_NETWORK_UNAVAIL;
  SSqlCmd *pCmd = &pSql->cmd;
H
hzcheng 已提交
2515

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

2519
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2520
  if (pQueryInfo->pTableMetaInfo[0]->vgroupList != NULL) {
H
hjxilinx 已提交
2521 2522 2523 2524
    return TSDB_CODE_SUCCESS;
  }
  
#if 0
2525
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
S
slguan 已提交
2526 2527
    char tagstr[TSDB_MAX_TAGS_LEN + 1] = {0};

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

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

2533
    SSuperTableMeta *ppMeta = (SSuperTableMeta *)taosCacheAcquireByName(tscCacheHandle, tagstr);
S
slguan 已提交
2534
    if (ppMeta == NULL) {
2535
      required = true;
S
slguan 已提交
2536 2537
      break;
    } else {
H
hjxilinx 已提交
2538
//      pTableMetaInfo->pMetricMeta = ppMeta;
S
slguan 已提交
2539 2540
    }
  }
H
hzcheng 已提交
2541

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

H
hjxilinx 已提交
2552
  pNew->cmd.command = TSDB_SQL_STABLEVGROUP;
2553 2554
  
  SQueryInfo *pNewQueryInfo = NULL;
2555 2556 2557
  if ((code = tscGetQueryInfoDetailSafely(&pNew->cmd, 0, &pNewQueryInfo)) != TSDB_CODE_SUCCESS) {
    return code;
  }
2558
  
2559
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2560
    STableMetaInfo *pMMInfo = tscGetMetaInfo(pQueryInfo, i);
2561

H
hjxilinx 已提交
2562
    STableMeta *pTableMeta = taosCacheAcquireByName(tscCacheHandle, pMMInfo->name);
2563
    tscAddTableMetaInfo(pNewQueryInfo, pMMInfo->name, pTableMeta, NULL, pMMInfo->tagColList);
S
slguan 已提交
2564 2565 2566 2567 2568 2569
  }

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

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

2573 2574
  pNewQueryInfo->groupbyExpr = pQueryInfo->groupbyExpr;
  pNewQueryInfo->numOfTables = pQueryInfo->numOfTables;
H
hzcheng 已提交
2575

2576 2577
  pNewQueryInfo->slimit = pQueryInfo->slimit;
  pNewQueryInfo->order = pQueryInfo->order;
H
hjxilinx 已提交
2578 2579 2580 2581 2582
  
  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 已提交
2583

2584 2585
//  if (pSql->fp != NULL && pSql->pStream == NULL) {
//    pCmd->pDataBlocks = tscDestroyBlockArrayList(pCmd->pDataBlocks);
2586
//    tscFreeQueryInfo(pCmd);
2587
//  }
H
hzcheng 已提交
2588

H
hjxilinx 已提交
2589
  tscTrace("%p allocate new pSqlObj:%p to get stable vgroupInfo", pSql, pNew);
2590 2591 2592 2593 2594
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
  code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2595 2596 2597 2598 2599
  }

  return code;
}

2600
void tscInitMsgsFp() {
S
slguan 已提交
2601 2602 2603
  tscBuildMsg[TSDB_SQL_SELECT] = tscBuildQueryMsg;
  tscBuildMsg[TSDB_SQL_INSERT] = tscBuildSubmitMsg;
  tscBuildMsg[TSDB_SQL_FETCH] = tscBuildRetrieveMsg;
H
hzcheng 已提交
2604 2605

  tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
2606
  tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
H
hzcheng 已提交
2607

2608 2609
  tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
  tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
H
hzcheng 已提交
2610 2611

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

  tscBuildMsg[TSDB_SQL_HB] = tscBuildHeartBeatMsg;
  tscBuildMsg[TSDB_SQL_SHOW] = tscBuildShowMsg;
  tscBuildMsg[TSDB_SQL_RETRIEVE] = tscBuildRetrieveFromMgmtMsg;
2632 2633 2634
  tscBuildMsg[TSDB_SQL_KILL_QUERY] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_STREAM] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_CONNECTION] = tscBuildKillMsg;
H
hzcheng 已提交
2635 2636

  tscProcessMsgRsp[TSDB_SQL_SELECT] = tscProcessQueryRsp;
H
hjxilinx 已提交
2637
  tscProcessMsgRsp[TSDB_SQL_FETCH] = tscProcessRetrieveRspFromNode;
H
hzcheng 已提交
2638 2639 2640 2641 2642

  tscProcessMsgRsp[TSDB_SQL_DROP_DB] = tscProcessDropDbRsp;
  tscProcessMsgRsp[TSDB_SQL_DROP_TABLE] = tscProcessDropTableRsp;
  tscProcessMsgRsp[TSDB_SQL_CONNECT] = tscProcessConnectRsp;
  tscProcessMsgRsp[TSDB_SQL_USE_DB] = tscProcessUseDbRsp;
2643
  tscProcessMsgRsp[TSDB_SQL_META] = tscProcessTableMetaRsp;
H
hjxilinx 已提交
2644
  tscProcessMsgRsp[TSDB_SQL_STABLEVGROUP] = tscProcessSTableVgroupRsp;
S
slguan 已提交
2645
  tscProcessMsgRsp[TSDB_SQL_MULTI_META] = tscProcessMultiMeterMetaRsp;
H
hzcheng 已提交
2646 2647

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

H
hzcheng 已提交
2651
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_TAGS] = tscProcessTagRetrieveRsp;
H
hjxilinx 已提交
2652 2653 2654 2655 2656
  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;
2657

H
hzcheng 已提交
2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670
  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;
}