tscServer.c 87.0 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
static void tscSetDnodeIpList(SSqlObj* pSql, SCMVgroupInfo* pVgroupInfo) {
49 50
  SRpcIpSet* pIpList = &pSql->ipList;
  
51
  pIpList->numOfIps = pVgroupInfo->numOfIps;
52 53
  pIpList->inUse    = 0;
  
54 55 56
  for(int32_t i = 0; i < pVgroupInfo->numOfIps; ++i) {
    strcpy(pIpList->fqdn[i], pVgroupInfo->ipAddr[i].fqdn);
    pIpList->port[i] = pVgroupInfo->ipAddr[i].port;
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
    for (int i = 0; i < tscMgmtIpSet.numOfIps; ++i) {
J
jtao1735 已提交
65
      tscTrace("mgmt index:%d %s:%d", i, tscMgmtIpSet.fqdn[i], tscMgmtIpSet.port[i]);
S
slguan 已提交
66
    }
S
slguan 已提交
67 68 69
  }
}

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

void tscSetMgmtIpListFromEdge() {
S
slguan 已提交
79 80 81
  if (tscMgmtIpSet.numOfIps != 1) {
    tscMgmtIpSet.numOfIps = 1;
    tscMgmtIpSet.inUse = 0;
H
hjxilinx 已提交
82
    taosGetFqdnPortFromEp(tsFirst, tscMgmtIpSet.fqdn[0], &tscMgmtIpSet.port[0]);
S
slguan 已提交
83 84 85 86 87
    tscTrace("edge mgmt IP list:");
    tscPrintMgmtIp();
  }
}

S
slguan 已提交
88
void tscUpdateIpSet(void *ahandle, SRpcIpSet *pIpSet) {
S
slguan 已提交
89
  tscMgmtIpSet = *pIpSet;
S
slguan 已提交
90 91 92 93
  tscTrace("mgmt IP list is changed for ufp is called, numOfIps:%d inUse:%d", tscMgmtIpSet.numOfIps, tscMgmtIpSet.inUse);
  for (int32_t i = 0; i < tscMgmtIpSet.numOfIps; ++i) {
    tscTrace("index:%d fqdn:%s port:%d", i, tscMgmtIpSet.fqdn[i], tscMgmtIpSet.port[i]);
  }
S
slguan 已提交
94 95
}

S
slguan 已提交
96
void tscSetMgmtIpList(SRpcIpSet *pIpList) {
S
slguan 已提交
97 98 99 100 101 102 103 104
  /*
    * 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 已提交
105 106 107
  }
}

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

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

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

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

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

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

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

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

  tscProcessSql(pObj->pHb);
}

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

S
slguan 已提交
203
  if (pSql->cmd.command < TSDB_SQL_MGMT) {
S
slguan 已提交
204
    memcpy(pMsg, pSql->cmd.payload + tsRpcHeadSize, pSql->cmd.payloadLen);
J
jtao1735 已提交
205 206 207 208 209
  } else {
    pSql->ipList = tscMgmtIpSet;
    memcpy(pMsg, pSql->cmd.payload, pSql->cmd.payloadLen);
  }

210
  tscTrace("%p msg:%s is sent to server", pSql, taosMsg[pSql->cmd.msgType]);
211

J
jtao1735 已提交
212
  SRpcMsg rpcMsg = {
213 214 215 216
      .msgType = pSql->cmd.msgType,
      .pCont   = pMsg,
      .contLen = pSql->cmd.payloadLen,
      .handle  = pSql,
H
hjxilinx 已提交
217
      .code    = 0
J
jtao1735 已提交
218
  };
219
  rpcSendRequest(pObj->pDnodeConn, &pSql->ipList, &rpcMsg);
H
hzcheng 已提交
220

S
slguan 已提交
221
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
222 223
}

J
jtao1735 已提交
224
void tscProcessMsgFromServer(SRpcMsg *rpcMsg, SRpcIpSet *pIpSet) {
225
  SSqlObj *pSql = (SSqlObj *)rpcMsg->handle;
226 227 228 229 230
  if (pSql == NULL) {
    tscError("%p sql is already released", pSql->signature);
    return;
  }
  if (pSql->signature != pSql) {
H
hzcheng 已提交
231
    tscError("%p sql is already released, signature:%p", pSql, pSql->signature);
S
slguan 已提交
232
    return;
H
hzcheng 已提交
233 234
  }

S
slguan 已提交
235 236 237
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;
  STscObj *pObj = pSql->pTscObj;
238
  tscTrace("%p msg:%p is received from server", pSql, rpcMsg->pCont);
H
hzcheng 已提交
239 240

  if (pSql->freed || pObj->signature != pObj) {
S
slguan 已提交
241 242
    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 已提交
243
    tscFreeSqlObj(pSql);
244
    rpcFreeCont(rpcMsg->pCont);
S
slguan 已提交
245
    return;
H
hzcheng 已提交
246 247
  }

J
jtao1735 已提交
248 249 250 251 252 253
  if (pCmd->command < TSDB_SQL_MGMT) {
    if (pIpSet) pSql->ipList = *pIpSet;
  } else {
    if (pIpSet) tscMgmtIpSet = *pIpSet;
  }

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

S
slguan 已提交
307 308 309 310 311
  if (pRes->code == TSDB_CODE_SUCCESS) {
    tscTrace("%p reset retry counter to be 0 due to success rsp, old:%d", pSql, pSql->retry);
    pSql->retry = 0;
  }

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

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

H
hzcheng 已提交
330 331 332 333
    /*
     * There is not response callback function for submit response.
     * The actual inserted number of points is the first number.
     */
334
    if (rpcMsg->msgType == TSDB_MSG_TYPE_SUBMIT_RSP && pRes->pRsp != NULL) {
335
      SShellSubmitRspMsg *pMsg = (SShellSubmitRspMsg*)pRes->pRsp;
S
slguan 已提交
336 337 338 339 340 341 342
      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;
343
      tscTrace("%p cmd:%d code:%s, inserted rows:%d, rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code),
H
hjxilinx 已提交
344
          pMsg->affectedRows, pRes->rspLen);
S
slguan 已提交
345
    } else {
H
hjxilinx 已提交
346
      tscTrace("%p cmd:%d code:%s rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code), pRes->rspLen);
H
hzcheng 已提交
347 348
    }
  }
349
  
350 351
  if (pRes->code == TSDB_CODE_SUCCESS && tscProcessMsgRsp[pCmd->command])
    rpcMsg->code = (*tscProcessMsgRsp[pCmd->command])(pSql);
H
hjxilinx 已提交
352
  
353
  if (rpcMsg->code != TSDB_CODE_ACTION_IN_PROGRESS) {
354
    void *taosres = tscKeepConn[pCmd->command] ? pSql : NULL;
H
[td-32]  
hjxilinx 已提交
355
    rpcMsg->code = pRes->code ? pRes->code : pRes->numOfRows;
356
    
357
    tscTrace("%p SQL result:%s res:%p", pSql, tstrerror(pRes->code), pSql);
H
hzcheng 已提交
358

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

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

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

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

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

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

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

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

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

H
hjxilinx 已提交
444
void tscKillSTableQuery(SSqlObj *pSql) {
445 446 447
  SSqlCmd* pCmd = &pSql->cmd;
  
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
448
  if (!tscIsTwoStageSTableQuery(pQueryInfo, 0)) {
H
hzcheng 已提交
449 450 451 452 453 454
    return;
  }

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

S
slguan 已提交
455
    if (pSub == NULL) {
H
hzcheng 已提交
456 457
      continue;
    }
S
slguan 已提交
458

H
hzcheng 已提交
459 460 461 462 463
    /*
     * 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 已提交
464
    //taosStopRpcConn(pSql->pSubs[i]->thandle);
H
hzcheng 已提交
465 466 467 468 469 470 471 472 473 474 475
  }

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

H
hjxilinx 已提交
476
  while (pCmd->command != TSDB_SQL_RETRIEVE_LOCALMERGE && pCmd->command != TSDB_SQL_RETRIEVE_EMPTY_RESULT) {
H
hzcheng 已提交
477 478 479 480 481 482
    taosMsleep(100);
    if (taosGetTimestampMs() - stime > MAX_WAITING_TIME) {
      break;
    }
  }

H
hjxilinx 已提交
483
  tscTrace("%p super table query cancelled", pSql);
H
hzcheng 已提交
484 485
}

J
jtao1735 已提交
486
int tscBuildFetchMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
487 488 489 490 491
  char *pMsg, *pStart;

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

492
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg *)pMsg;
S
slguan 已提交
493
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
S
slguan 已提交
494 495
  pMsg += sizeof(pSql->res.qhandle);

496
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
S
slguan 已提交
497
  pRetrieveMsg->free = htons(pQueryInfo->type);
498
  pMsg += sizeof(pQueryInfo->type);
H
hzcheng 已提交
499

500
  // todo valid the vgroupId at the client side
H
hjxilinx 已提交
501 502
  STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
  
weixin_48148422's avatar
weixin_48148422 已提交
503
  if (UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo)) {
H
hjxilinx 已提交
504
    int32_t vgIndex = pTableMetaInfo->vgroupIndex;
505
    
H
hjxilinx 已提交
506 507
    SVgroupsInfo* pVgroupInfo = pTableMetaInfo->vgroupList;
    pRetrieveMsg->header.vgId = htonl(pVgroupInfo->vgroups[vgIndex].vgId);
508
  } else {
H
hjxilinx 已提交
509
    STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
510
    pRetrieveMsg->header.vgId = htonl(pTableMeta->vgroupInfo.vgId);
511 512
  }
  
513 514 515 516
  pMsg += sizeof(SRetrieveTableMsg);
  
  pRetrieveMsg->header.contLen = htonl(pSql->cmd.payloadLen);
  
J
jtao1735 已提交
517
  pSql->cmd.msgType = TSDB_MSG_TYPE_FETCH;
518
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
519 520
}

521
int tscBuildSubmitMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hjxilinx 已提交
522
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);
H
hjxilinx 已提交
523
  STableMeta* pTableMeta = tscGetMetaInfo(pQueryInfo, 0)->pTableMeta;
H
hjxilinx 已提交
524
  
525 526 527 528
  char* pMsg = pSql->cmd.payload + tsRpcHeadSize;
  
  // NOTE: shell message size should not include SMsgDesc
  int32_t size = pSql->cmd.payloadLen - sizeof(SMsgDesc);
H
hjxilinx 已提交
529
  
530
  SMsgDesc* pMsgDesc = (SMsgDesc*) pMsg;
531 532
  
  pMsgDesc->numOfVnodes = htonl(1);       //todo set the right number of vnodes
533
  pMsg += sizeof(SMsgDesc);
H
hjxilinx 已提交
534
  
535
  SSubmitMsg *pShellMsg = (SSubmitMsg *)pMsg;
H
hjxilinx 已提交
536
  int32_t vgId = pTableMeta->vgroupInfo.vgId;
537
  
H
hjxilinx 已提交
538
  pShellMsg->header.vgId = htonl(vgId);
539
  pShellMsg->header.contLen = htonl(size);
540
  pShellMsg->length = pShellMsg->header.contLen;
H
hjxilinx 已提交
541
  
542
  pShellMsg->numOfBlocks = htonl(pSql->cmd.numOfTablesInSubmit);  // number of meters to be inserted
H
hzcheng 已提交
543

H
hjxilinx 已提交
544
  // pSql->cmd.payloadLen is set during copying data into payload
S
slguan 已提交
545
  pSql->cmd.msgType = TSDB_MSG_TYPE_SUBMIT;
546
  tscSetDnodeIpList(pSql, &pTableMeta->vgroupInfo);
547
  
S
slguan 已提交
548
  tscTrace("%p build submit msg, vgId:%d numOfVgroup:%d numberOfIP:%d", pSql, vgId, htonl(pMsgDesc->numOfVnodes), pSql->ipList.numOfIps);
549
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
550 551 552 553 554 555
}

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

560
  int32_t srcColListSize = taosArrayGetSize(pQueryInfo->colList) * sizeof(SColumnInfo);
H
hjxilinx 已提交
561 562 563 564
  
  size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
  int32_t exprSize = sizeof(SSqlFuncMsg) * numOfExprs;
  
565
  //STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
566

567
  // table query without tags values
weixin_48148422's avatar
weixin_48148422 已提交
568
  //if (!UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo)) {
569 570
    return MIN_QUERY_MSG_PKT_SIZE + minMsgSize() + sizeof(SQueryTableMsg) + srcColListSize + exprSize + 4096;
  //}
H
hjxilinx 已提交
571
  
572 573
  //int32_t size = 4096;
  //return size;
H
hzcheng 已提交
574 575
}

576
static char *doSerializeTableInfo(SQueryTableMsg* pQueryMsg, SSqlObj *pSql, char *pMsg) {
577
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, pSql->cmd.clauseIndex, 0);
578

H
hjxilinx 已提交
579
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
weixin_48148422's avatar
weixin_48148422 已提交
580
  if (UTIL_TABLE_IS_NORMAL_TABLE(pTableMetaInfo) || pTableMetaInfo->pVgroupTables == NULL) {
581 582
    
    SCMVgroupInfo* pVgroupInfo = NULL;
weixin_48148422's avatar
weixin_48148422 已提交
583
    if (UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo)) {
584 585
      int32_t index = pTableMetaInfo->vgroupIndex;
      assert(index >= 0);
H
hjxilinx 已提交
586
  
587 588
      pVgroupInfo = &pTableMetaInfo->vgroupList->vgroups[index];
      tscTrace("%p query on stable, vgIndex:%d, numOfVgroups:%d", pSql, index, pTableMetaInfo->vgroupList->numOfVgroups);
H
hjxilinx 已提交
589 590
    } else {
      pVgroupInfo = &pTableMeta->vgroupInfo;
591 592 593 594 595 596 597 598 599
    }
    
    tscSetDnodeIpList(pSql, pVgroupInfo);
    pQueryMsg->head.vgId = htonl(pVgroupInfo->vgId);
    
    STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
    pTableIdInfo->tid = htonl(pTableMeta->sid);
    pTableIdInfo->uid = htobe64(pTableMeta->uid);
    pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pTableMeta->uid));
H
hjxilinx 已提交
600
  
601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
    pQueryMsg->numOfTables = htonl(1);  // set the number of tables
    
    pMsg += sizeof(STableIdInfo);
  } else {
    int32_t index = pTableMetaInfo->vgroupIndex;
    int32_t numOfVgroups = taosArrayGetSize(pTableMetaInfo->pVgroupTables);
    assert(index >= 0 && index < numOfVgroups);
  
    tscTrace("%p query on stable, vgIndex:%d, numOfVgroups:%d", pSql, index, numOfVgroups);
    
    SVgroupTableInfo* pTableIdList = taosArrayGet(pTableMetaInfo->pVgroupTables, index);
    
    // set the vgroup info
    tscSetDnodeIpList(pSql, &pTableIdList->vgInfo);
    pQueryMsg->head.vgId = htonl(pTableIdList->vgInfo.vgId);
    
    int32_t numOfTables = taosArrayGetSize(pTableIdList->itemList);
    pQueryMsg->numOfTables = htonl(numOfTables);  // set the number of tables
  
    // serialize each table id info
    for(int32_t i = 0; i < numOfTables; ++i) {
      STableIdInfo* pItem = taosArrayGet(pTableIdList->itemList, i);
      
      STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
      pTableIdInfo->tid = htonl(pItem->tid);
      pTableIdInfo->uid = htobe64(pItem->uid);
weixin_48148422's avatar
weixin_48148422 已提交
627
      pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pTableMeta->uid));
628 629 630 631 632 633
      pMsg += sizeof(STableIdInfo);
    }
  }
  
  tscTrace("%p vgId:%d, query on table:%s, uid:%" PRIu64, pSql, htonl(pQueryMsg->head.vgId), pTableMetaInfo->name,
      pTableMeta->uid);
H
hjxilinx 已提交
634
  
635 636 637
  return pMsg;
}

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

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

S
slguan 已提交
643 644 645 646
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for query msg", pSql);
    return -1;
  }
647
  
648
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hjxilinx 已提交
649
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hjxilinx 已提交
650
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
651
  
H
hjxilinx 已提交
652
  if (taosArrayGetSize(pQueryInfo->colList) <= 0 && !tscQueryTags(pQueryInfo)) {
653 654 655
    tscError("%p illegal value of numOfCols in query msg: %d", pSql, tscGetNumOfColumns(pTableMeta));
    return -1;
  }
656 657 658 659 660 661 662 663 664 665
  
  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;
  }
666 667

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

S
slguan 已提交
669
  SQueryTableMsg *pQueryMsg = (SQueryTableMsg *)pStart;
H
hzcheng 已提交
670

671
  int32_t numOfTags = taosArrayGetSize(pTableMetaInfo->tagColList);
672
  
673
  if (pQueryInfo->order.order == TSDB_ORDER_ASC) {
H
hjxilinx 已提交
674 675
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.skey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.ekey);
H
hzcheng 已提交
676
  } else {
H
hjxilinx 已提交
677 678
    pQueryMsg->window.skey = htobe64(pQueryInfo->window.ekey);
    pQueryMsg->window.ekey = htobe64(pQueryInfo->window.skey);
H
hzcheng 已提交
679 680
  }

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

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

711
    if (pCol->colIndex.columnIndex >= tscGetNumOfColumns(pTableMeta) || pColSchema->type < TSDB_DATA_TYPE_BOOL ||
712 713 714 715 716
        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);

717
      return TSDB_CODE_INVALID_SQL;
718
    }
H
hzcheng 已提交
719 720 721

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

S
slguan 已提交
725 726 727
    // 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 已提交
728

S
slguan 已提交
729
      SColumnFilterInfo *pFilterMsg = (SColumnFilterInfo *)pMsg;
730
      pFilterMsg->filterstr = htons(pColFilter->filterstr);
S
slguan 已提交
731 732

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

734
      if (pColFilter->filterstr) {
S
slguan 已提交
735 736 737 738 739 740 741 742 743 744
        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 已提交
745

S
slguan 已提交
746 747 748 749 750
      if (pColFilter->lowerRelOptr == TSDB_RELATION_INVALID && pColFilter->upperRelOptr == TSDB_RELATION_INVALID) {
        tscError("invalid filter info");
        return -1;
      }
    }
H
hzcheng 已提交
751 752
  }

H
hjxilinx 已提交
753
  SSqlFuncMsg *pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hjxilinx 已提交
754
  for (int32_t i = 0; i < tscSqlExprNumOfExprs(pQueryInfo); ++i) {
755
    SSqlExpr *pExpr = tscSqlExprGet(pQueryInfo, i);
H
hzcheng 已提交
756

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

763 764 765
    pSqlFuncExpr->colInfo.colId    = htons(pExpr->colInfo.colId);
    pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
    pSqlFuncExpr->colInfo.flag     = htons(pExpr->colInfo.flag);
H
hzcheng 已提交
766

767
    pSqlFuncExpr->functionId  = htons(pExpr->functionId);
H
hzcheng 已提交
768
    pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
H
hjxilinx 已提交
769
    pMsg += sizeof(SSqlFuncMsg);
H
hzcheng 已提交
770 771 772 773 774 775 776

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

H
hjxilinx 已提交
783
    pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
H
hzcheng 已提交
784
  }
785
  
786
  // serialize the table info (sid, uid, tags)
787 788
  pMsg = doSerializeTableInfo(pQueryMsg, pSql, pMsg);
  
789
  SSqlGroupbyExpr *pGroupbyExpr = &pQueryInfo->groupbyExpr;
790
  if (pGroupbyExpr->numOfGroupCols > 0) {
S
slguan 已提交
791
    pQueryMsg->orderByIdx = htons(pGroupbyExpr->orderIndex);
H
hzcheng 已提交
792 793
    pQueryMsg->orderType = htons(pGroupbyExpr->orderType);

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

800 801
      *((int16_t *)pMsg) += pCol->colIndex;
      pMsg += sizeof(pCol->colIndex);
S
slguan 已提交
802 803 804

      *((int16_t *)pMsg) += pCol->flag;
      pMsg += sizeof(pCol->flag);
H
hjxilinx 已提交
805 806 807
      
      memcpy(pMsg, pCol->name, tListLen(pCol->name));
      pMsg += tListLen(pCol->name);
S
slguan 已提交
808 809 810
    }
  }

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

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

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

    // todo refactor
859 860
    fseek(pQueryInfo->tsBuf->f, pBlockInfo->offset, SEEK_SET);
    fread(pMsg, pBlockInfo->compLen, 1, pQueryInfo->tsBuf->f);
S
slguan 已提交
861 862 863 864

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

S
slguan 已提交
867 868
  pQueryMsg->tsLen = htonl(tsLen);
  pQueryMsg->tsNumOfBlocks = htonl(numOfBlocks);
869 870
  if (pQueryInfo->tsBuf != NULL) {
    pQueryMsg->tsOrder = htonl(pQueryInfo->tsBuf->tsOrder);
H
hzcheng 已提交
871 872
  }

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

894
  int32_t msgLen = pMsg - pStart;
H
hzcheng 已提交
895 896 897

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

  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
904 905
}

906 907
int32_t tscBuildCreateDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
908
  pCmd->payloadLen = sizeof(SCMCreateDbMsg);
S
slguan 已提交
909
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DB;
H
hzcheng 已提交
910

911
  SCMCreateDbMsg *pCreateDbMsg = (SCMCreateDbMsg*)pCmd->payload;
912

913
  assert(pCmd->numOfClause == 1);
914
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
915
  strncpy(pCreateDbMsg->db, pTableMetaInfo->name, tListLen(pCreateDbMsg->db));
H
hzcheng 已提交
916

917
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
918 919
}

920 921
int32_t tscBuildCreateDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
922
  pCmd->payloadLen = sizeof(SCMCreateDnodeMsg);
S
slguan 已提交
923 924 925 926
  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 已提交
927

928
  SCMCreateDnodeMsg *pCreate = (SCMCreateDnodeMsg *)pCmd->payload;
J
jtao1735 已提交
929 930
  strncpy(pCreate->ep, pInfo->pDCLInfo->a[0].z, pInfo->pDCLInfo->a[0].n);
  
S
slguan 已提交
931
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_DNODE;
H
hzcheng 已提交
932

933
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
934 935
}

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

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

946 947
  SSQLToken *pName = &pInfo->pDCLInfo->user.user;
  SSQLToken *pPwd = &pInfo->pDCLInfo->user.passwd;
H
hzcheng 已提交
948

949 950
  strncpy(pAlterMsg->user, pName->z, pName->n);
  strncpy(pAlterMsg->pass, pPwd->z, pPwd->n);
S
slguan 已提交
951

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

954 955 956 957 958 959 960 961
  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 已提交
962

963 964 965 966 967 968 969 970 971 972 973 974 975
  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 已提交
976

S
slguan 已提交
977
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_ACCT;
978
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
979 980
}

981 982
int32_t tscBuildUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
983
  pCmd->payloadLen = sizeof(SCMCreateUserMsg);
S
slguan 已提交
984

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

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

992 993 994
  SUserInfo *pUser = &pInfo->pDCLInfo->user;
  strncpy(pAlterMsg->user, pUser->user.z, pUser->user.n);
  pAlterMsg->flag = pUser->type;
H
hzcheng 已提交
995

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

1004
  if (pUser->type == TSDB_ALTER_USER_PASSWD || pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
S
slguan 已提交
1005
    pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_USER;
1006
  } else {
S
slguan 已提交
1007
    pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_USER;
1008
  }
H
hzcheng 已提交
1009

1010
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1011 1012
}

1013 1014
int32_t tscBuildCfgDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1015
  pCmd->payloadLen = sizeof(SCMCfgDnodeMsg);
guanshengliang's avatar
guanshengliang 已提交
1016
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONFIG_DNODE;
1017 1018
  return TSDB_CODE_SUCCESS;
}
S
slguan 已提交
1019

1020 1021
int32_t tscBuildDropDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1022
  pCmd->payloadLen = sizeof(SCMDropDbMsg);
H
hzcheng 已提交
1023

S
slguan 已提交
1024 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;
  }

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

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

S
slguan 已提交
1035
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DB;
1036
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1037 1038
}

1039 1040
int32_t tscBuildDropTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1041
  pCmd->payloadLen = sizeof(SCMDropTableMsg);
H
hzcheng 已提交
1042

S
slguan 已提交
1043 1044 1045 1046
  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 已提交
1047

1048
  SCMDropTableMsg *pDropTableMsg = (SCMDropTableMsg*)pCmd->payload;
1049
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1050
  strcpy(pDropTableMsg->tableId, pTableMetaInfo->name);
S
slguan 已提交
1051
  pDropTableMsg->igNotExists = pInfo->pDCLInfo->existsCheck ? 1 : 0;
H
hzcheng 已提交
1052

S
slguan 已提交
1053
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_TABLE;
1054
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1055 1056
}

1057
int32_t tscBuildDropDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1058
  SSqlCmd *pCmd = &pSql->cmd;
1059
  pCmd->payloadLen = sizeof(SCMDropDnodeMsg);
S
slguan 已提交
1060 1061 1062 1063
  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 已提交
1064

1065
  SCMDropDnodeMsg *pDrop = (SCMDropDnodeMsg *)pCmd->payload;
1066
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
J
jtao1735 已提交
1067
  strcpy(pDrop->ep, pTableMetaInfo->name);
S
slguan 已提交
1068
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_DNODE;
H
hzcheng 已提交
1069

1070
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1071 1072
}

S
[TD-16]  
slguan 已提交
1073
int32_t tscBuildDropUserMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1074
  SSqlCmd *pCmd = &pSql->cmd;
1075
  pCmd->payloadLen = sizeof(SCMDropUserMsg);
S
slguan 已提交
1076
  pCmd->msgType = TSDB_MSG_TYPE_CM_DROP_USER;
H
hzcheng 已提交
1077

S
slguan 已提交
1078 1079 1080 1081
  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 已提交
1082

1083
  SCMDropUserMsg *pDropMsg = (SCMDropUserMsg*)pCmd->payload;
1084
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1085
  strcpy(pDropMsg->user, pTableMetaInfo->name);
H
hzcheng 已提交
1086

1087
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1088 1089
}

S
[TD-16]  
slguan 已提交
1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106
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;
}

1107 1108
int32_t tscBuildUseDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
1109
  pCmd->payloadLen = sizeof(SCMUseDbMsg);
H
hzcheng 已提交
1110

S
slguan 已提交
1111 1112 1113 1114
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
1115

1116
  SCMUseDbMsg *pUseDbMsg = (SCMUseDbMsg*)pCmd->payload;
1117
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1118
  strcpy(pUseDbMsg->db, pTableMetaInfo->name);
S
slguan 已提交
1119
  pCmd->msgType = TSDB_MSG_TYPE_CM_USE_DB;
H
hzcheng 已提交
1120

1121
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1122 1123
}

1124
int32_t tscBuildShowMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1125
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1126
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1127
  pCmd->msgType = TSDB_MSG_TYPE_CM_SHOW;
1128
  pCmd->payloadLen = sizeof(SCMShowMsg) + 100;
H
hzcheng 已提交
1129

S
slguan 已提交
1130 1131 1132
  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 已提交
1133
  }
H
hzcheng 已提交
1134

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

1137
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1138
  size_t nameLen = strlen(pTableMetaInfo->name);
S
slguan 已提交
1139
  if (nameLen > 0) {
H
hjxilinx 已提交
1140
    strcpy(pShowMsg->db, pTableMetaInfo->name);  // prefix is set here
H
hzcheng 已提交
1141
  } else {
S
slguan 已提交
1142
    strcpy(pShowMsg->db, pObj->db);
H
hzcheng 已提交
1143 1144
  }

1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156
  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 已提交
1157

1158 1159 1160 1161
    strncpy(pShowMsg->payload, pIpAddr->z, pIpAddr->n);
    pShowMsg->payloadLen = htons(pIpAddr->n);
  }

1162
  pCmd->payloadLen = sizeof(SCMShowMsg) + pShowMsg->payloadLen;
1163
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1164 1165
}

1166
int32_t tscBuildKillMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1167
  SSqlCmd *pCmd = &pSql->cmd;
1168
  pCmd->payloadLen = sizeof(SCMKillQueryMsg);
H
hzcheng 已提交
1169

S
slguan 已提交
1170 1171 1172 1173
  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 已提交
1174

1175
  SCMKillQueryMsg *pKill = (SCMKillQueryMsg*)pCmd->payload;
1176 1177 1178
  strncpy(pKill->queryId, pInfo->pDCLInfo->ip.z, pInfo->pDCLInfo->ip.n);
  switch (pCmd->command) {
    case TSDB_SQL_KILL_QUERY:
S
slguan 已提交
1179
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_QUERY;
1180 1181
      break;
    case TSDB_SQL_KILL_CONNECTION:
S
slguan 已提交
1182
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_CONN;
1183 1184
      break;
    case TSDB_SQL_KILL_STREAM:
S
slguan 已提交
1185
      pCmd->msgType = TSDB_MSG_TYPE_CM_KILL_STREAM;
1186 1187 1188
      break;
  }
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1189 1190
}

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

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

1196
  SCreateTableSQL *pCreateTableInfo = pInfo->pCreateTableInfo;
1197
  if (pCreateTableInfo->type == TSQL_CREATE_TABLE_FROM_STABLE) {
H
hzcheng 已提交
1198 1199
    size += sizeof(STagData);
  } else {
S
slguan 已提交
1200
    size += sizeof(SSchema) * (pCmd->numOfCols + pCmd->count);
H
hzcheng 已提交
1201
  }
1202

1203 1204 1205
  if (pCreateTableInfo->pSelect != NULL) {
    size += (pCreateTableInfo->pSelect->selectToken.n + 1);
  }
H
hzcheng 已提交
1206 1207 1208 1209

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1210
int tscBuildCreateTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1211
  int              msgLen = 0;
S
slguan 已提交
1212
  SSchema *        pSchema;
H
hzcheng 已提交
1213
  int              size = 0;
1214 1215 1216
  SSqlCmd *pCmd = &pSql->cmd;

  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
H
hjxilinx 已提交
1217
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1218 1219

  // Reallocate the payload size
1220
  size = tscEstimateCreateTableMsgLength(pSql, pInfo);
S
slguan 已提交
1221 1222
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for create table msg", pSql);
1223
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
S
slguan 已提交
1224
  }
H
hzcheng 已提交
1225 1226


1227
  SCMCreateTableMsg *pCreateTableMsg = (SCMCreateTableMsg *)pCmd->payload;
H
hjxilinx 已提交
1228
  strcpy(pCreateTableMsg->tableId, pTableMetaInfo->name);
1229 1230

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

1233 1234 1235
  SCreateTableSQL *pCreateTable = pInfo->pCreateTableInfo;

  pCreateTableMsg->igExists = pCreateTable->existCheck ? 1 : 0;
H
hzcheng 已提交
1236 1237 1238 1239
  pCreateTableMsg->numOfColumns = htons(pCmd->numOfCols);
  pCreateTableMsg->numOfTags = htons(pCmd->count);

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

1242 1243 1244
  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 已提交
1245
    pMsg += sizeof(STagData);
1246
  } else {  // create (super) table
1247
    pSchema = (SSchema *)pCreateTableMsg->schema;
1248

H
hzcheng 已提交
1249
    for (int i = 0; i < pCmd->numOfCols + pCmd->count; ++i) {
H
hjxilinx 已提交
1250
      TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1251 1252 1253 1254

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

H
hzcheng 已提交
1256 1257 1258 1259
      pSchema++;
    }

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

1263 1264 1265
      strncpy(pMsg, pQuerySql->selectToken.z, pQuerySql->selectToken.n + 1);
      pCreateTableMsg->sqlLen = htons(pQuerySql->selectToken.n + 1);
      pMsg += pQuerySql->selectToken.n + 1;
H
hzcheng 已提交
1266 1267 1268
    }
  }

H
hjxilinx 已提交
1269
  tscFieldInfoClear(&pQueryInfo->fieldsInfo);
H
hzcheng 已提交
1270

S
slguan 已提交
1271
  msgLen = pMsg - (char*)pCreateTableMsg;
S
slguan 已提交
1272
  pCreateTableMsg->contLen = htonl(msgLen);
H
hzcheng 已提交
1273
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1274
  pCmd->msgType = TSDB_MSG_TYPE_CM_CREATE_TABLE;
H
hzcheng 已提交
1275 1276

  assert(msgLen + minMsgSize() <= size);
1277
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1278 1279 1280
}

int tscEstimateAlterTableMsgLength(SSqlCmd *pCmd) {
1281
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
guanshengliang's avatar
guanshengliang 已提交
1282
  return minMsgSize() + sizeof(SCMAlterTableMsg) + sizeof(SSchema) * tscNumOfFields(pQueryInfo) +
H
hzcheng 已提交
1283 1284 1285
         TSDB_EXTRA_PAYLOAD_SIZE;
}

1286
int tscBuildAlterTableMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1287
  SCMAlterTableMsg *pAlterTableMsg;
1288
  char *          pMsg;
H
hzcheng 已提交
1289 1290 1291
  int             msgLen = 0;
  int             size = 0;

1292 1293 1294
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);

H
hjxilinx 已提交
1295
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
H
hzcheng 已提交
1296 1297

  size = tscEstimateAlterTableMsgLength(pCmd);
S
slguan 已提交
1298 1299 1300 1301
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for alter table msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1302

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

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

1307 1308
  SAlterTableSQL *pAlterInfo = pInfo->pAlterInfo;

H
hjxilinx 已提交
1309
  strcpy(pAlterTableMsg->tableId, pTableMetaInfo->name);
1310
  pAlterTableMsg->type = htons(pAlterInfo->type);
1311

S
slguan 已提交
1312
  pAlterTableMsg->numOfCols = tscNumOfFields(pQueryInfo);
1313
  memcpy(pAlterTableMsg->tagVal, pAlterInfo->tagData.data, TSDB_MAX_TAGS_LEN);
H
hzcheng 已提交
1314

S
slguan 已提交
1315
  SSchema *pSchema = pAlterTableMsg->schema;
S
slguan 已提交
1316
  for (int i = 0; i < pAlterTableMsg->numOfCols; ++i) {
H
hjxilinx 已提交
1317
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
H
hzcheng 已提交
1318 1319 1320 1321 1322 1323 1324 1325 1326

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

  pMsg = (char *)pSchema;

S
slguan 已提交
1327
  msgLen = pMsg - (char*)pAlterTableMsg;
H
hzcheng 已提交
1328
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1329
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_TABLE;
H
hzcheng 已提交
1330 1331

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

1333
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1334 1335
}

1336
int tscAlterDbMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1337
  SSqlCmd *pCmd = &pSql->cmd;
1338
  pCmd->payloadLen = sizeof(SCMAlterDbMsg);
S
slguan 已提交
1339
  pCmd->msgType = TSDB_MSG_TYPE_CM_ALTER_DB;
H
hzcheng 已提交
1340

1341
  SCMAlterDbMsg *pAlterDbMsg = (SCMAlterDbMsg*)pCmd->payload;
1342
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
H
hjxilinx 已提交
1343
  strcpy(pAlterDbMsg->db, pTableMetaInfo->name);
H
hzcheng 已提交
1344

1345
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1346 1347
}

1348
int tscBuildRetrieveFromMgmtMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
S
slguan 已提交
1349
  SSqlCmd *pCmd = &pSql->cmd;
J
jtao1735 已提交
1350
  pCmd->msgType = TSDB_MSG_TYPE_CM_RETRIEVE;
S
slguan 已提交
1351
  pCmd->payloadLen = sizeof(SRetrieveTableMsg);
S
slguan 已提交
1352

S
slguan 已提交
1353 1354 1355
  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 已提交
1356
  }
S
slguan 已提交
1357

S
slguan 已提交
1358 1359 1360 1361
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
  SRetrieveTableMsg *pRetrieveMsg = (SRetrieveTableMsg*)pCmd->payload;
  pRetrieveMsg->qhandle = htobe64(pSql->res.qhandle);
  pRetrieveMsg->free = htons(pQueryInfo->type);
H
hzcheng 已提交
1362

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

1366
static int tscSetResultPointer(SQueryInfo *pQueryInfo, SSqlRes *pRes) {
1367
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
H
hzcheng 已提交
1368 1369 1370
    return pRes->code;
  }

H
hjxilinx 已提交
1371
  for (int i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
H
hjxilinx 已提交
1372
    int16_t offset = tscFieldInfoGetOffset(pQueryInfo, i);
H
hjxilinx 已提交
1373
    pRes->tsrow[i] = ((char*) pRes->data + offset * pRes->numOfRows);
H
hzcheng 已提交
1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
  }

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

1389
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
1390

H
hzcheng 已提交
1391 1392 1393 1394 1395 1396 1397
  pRes->code = TSDB_CODE_SUCCESS;

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

1398
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1399
  } else {
S
slguan 已提交
1400
    tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415
  }

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

H
hjxilinx 已提交
1419
  STableComInfo tinfo = tscGetTableInfo(pTableMetaInfo->pTableMeta);
H
hjxilinx 已提交
1420 1421
  
  int32_t numOfRes = tinfo.numOfColumns + tinfo.numOfTags;
H
hzcheng 已提交
1422 1423 1424 1425
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

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

1428 1429
//  SQueryInfo *    pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
//  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, 0);
S
slguan 已提交
1430 1431

  int32_t numOfRes = 0;
H
hjxilinx 已提交
1432
#if 0
1433
  if (tscSqlExprGet(pQueryInfo, 0)->functionId == TSDB_FUNC_TAGPRJ) {
H
hjxilinx 已提交
1434
    numOfRes = pTableMetaInfo->pMetricMeta->numOfTables;
H
hzcheng 已提交
1435 1436 1437
  } else {
    numOfRes = 1;  // for count function, there is only one output.
  }
H
hjxilinx 已提交
1438 1439 1440
  
#endif

H
hzcheng 已提交
1441 1442 1443
  return tscLocalResultCommonBuilder(pSql, numOfRes);
}

H
hjxilinx 已提交
1444
int tscProcessRetrieveLocalMergeRsp(SSqlObj *pSql) {
H
hzcheng 已提交
1445 1446 1447
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

1448
  pRes->code = tscDoLocalMerge(pSql);
H
hjxilinx 已提交
1449
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
H
hzcheng 已提交
1450 1451

  if (pRes->code == TSDB_CODE_SUCCESS && pRes->numOfRows > 0) {
1452
    tscSetResultPointer(pQueryInfo, pRes);
H
hzcheng 已提交
1453 1454 1455 1456
  }

  pRes->row = 0;

1457
  uint8_t code = pRes->code;
H
hjxilinx 已提交
1458 1459 1460 1461
  if (pRes->code == TSDB_CODE_SUCCESS) {
    (*pSql->fp)(pSql->param, pSql, pRes->numOfRows);
  } else {
    tscQueueAsyncRes(pSql);
H
hzcheng 已提交
1462 1463 1464 1465 1466
  }

  return code;
}

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

1469
int tscBuildConnectMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1470
  STscObj *pObj = pSql->pTscObj;
S
slguan 已提交
1471
  SSqlCmd *pCmd = &pSql->cmd;
S
slguan 已提交
1472
  pCmd->msgType = TSDB_MSG_TYPE_CM_CONNECT;
1473
  pCmd->payloadLen = sizeof(SCMConnectMsg);
H
hzcheng 已提交
1474

S
slguan 已提交
1475 1476 1477 1478 1479
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

1480
  SCMConnectMsg *pConnect = (SCMConnectMsg*)pCmd->payload;
H
hzcheng 已提交
1481 1482 1483 1484 1485

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

1489
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1490 1491
}

H
hjxilinx 已提交
1492
int tscBuildTableMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
1493
  SCMTableInfoMsg *pInfoMsg;
S
slguan 已提交
1494
  char *         pMsg;
H
hzcheng 已提交
1495 1496 1497 1498 1499
  int            msgLen = 0;

  char *tmpData = 0;
  if (pSql->cmd.allocSize > 0) {
    tmpData = calloc(1, pSql->cmd.allocSize);
1500 1501 1502 1503
    if (NULL == tmpData) {
      return TSDB_CODE_CLI_OUT_OF_MEMORY;
    }

H
hzcheng 已提交
1504 1505 1506 1507
    // STagData is in binary format, strncpy is not available
    memcpy(tmpData, pSql->cmd.payload, pSql->cmd.allocSize);
  }

1508 1509 1510
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

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

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

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

H
hjxilinx 已提交
1519
  if (pSql->cmd.autoCreated) {
H
hzcheng 已提交
1520 1521 1522 1523
    memcpy(pInfoMsg->tags, tmpData, sizeof(STagData));
    pMsg += sizeof(STagData);
  }

H
hjxilinx 已提交
1524
  pCmd->payloadLen = pMsg - (char*)pInfoMsg;;
S
slguan 已提交
1525
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLE_META;
H
hzcheng 已提交
1526 1527 1528 1529

  tfree(tmpData);

  assert(msgLen + minMsgSize() <= pCmd->allocSize);
1530
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1531 1532
}

S
slguan 已提交
1533
/**
1534
 *  multi table meta req pkg format:
1535
 *  | SMgmtHead | SCMMultiTableInfoMsg | tableId0 | tableId1 | tableId2 | ......
S
slguan 已提交
1536 1537
 *      no used         4B
 **/
1538
int tscBuildMultiMeterMetaMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
guanshengliang's avatar
guanshengliang 已提交
1539
#if 0
S
slguan 已提交
1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551
  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 已提交
1552
  memset(pMgmt->db, 0, TSDB_TABLE_ID_LEN);  // server don't need the db
S
slguan 已提交
1553

1554
  SCMMultiTableInfoMsg *pInfoMsg = (SCMMultiTableInfoMsg *)(pCmd->payload + tsRpcHeadSize + sizeof(SMgmtHead));
S
slguan 已提交
1555
  pInfoMsg->numOfTables = htonl((int32_t)pCmd->count);
S
slguan 已提交
1556 1557

  if (pCmd->payloadLen > 0) {
S
slguan 已提交
1558
    memcpy(pInfoMsg->tableIds, tmpData, pCmd->payloadLen);
S
slguan 已提交
1559 1560 1561 1562
  }

  tfree(tmpData);

1563
  pCmd->payloadLen += sizeof(SMgmtHead) + sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1564
  pCmd->msgType = TSDB_MSG_TYPE_CM_TABLES_META;
S
slguan 已提交
1565 1566 1567

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

S
slguan 已提交
1568
  tscTrace("%p build load multi-metermeta msg completed, numOfTables:%d, msg size:%d", pSql, pCmd->count,
S
slguan 已提交
1569 1570 1571
           pCmd->payloadLen);

  return pCmd->payloadLen;
guanshengliang's avatar
guanshengliang 已提交
1572 1573
#endif
  return 0;  
S
slguan 已提交
1574 1575
}

H
hjxilinx 已提交
1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601
//static UNUSED_FUNC int32_t tscEstimateMetricMetaMsgSize(SSqlCmd *pCmd) {
////  const int32_t defaultSize =
////      minMsgSize() + sizeof(SSuperTableMetaMsg) + sizeof(SMgmtHead) + sizeof(int16_t) * TSDB_MAX_TAGS;
////  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
////
////  int32_t n = 0;
////  size_t size = taosArrayGetSize(pQueryInfo->tagCond.pCond);
////  for (int32_t i = 0; i < size; ++i) {
////    assert(0);
//////    n += strlen(pQueryInfo->tagCond.cond[i].cond);
////  }
////
////  int32_t tagLen = n * TSDB_NCHAR_SIZE;
////  if (pQueryInfo->tagCond.tbnameCond.cond != NULL) {
////    tagLen += strlen(pQueryInfo->tagCond.tbnameCond.cond) * TSDB_NCHAR_SIZE;
////  }
////
////  int32_t joinCondLen = (TSDB_TABLE_ID_LEN + sizeof(int16_t)) * 2;
////  int32_t elemSize = sizeof(SSuperTableMetaElemMsg) * pQueryInfo->numOfTables;
////
////  int32_t colSize = pQueryInfo->groupbyExpr.numOfGroupCols*sizeof(SColIndex);
////
////  int32_t len = tagLen + joinCondLen + elemSize + colSize + defaultSize;
////
////  return MAX(len, TSDB_DEFAULT_PAYLOAD_SIZE);
//}
H
hzcheng 已提交
1602

H
hjxilinx 已提交
1603 1604 1605
int tscBuildSTableVgroupMsg(SSqlObj *pSql, SSqlInfo *pInfo) {

#if 0
S
slguan 已提交
1606
  SSuperTableMetaMsg *pMetaMsg;
H
hzcheng 已提交
1607 1608
  char *          pMsg, *pStart;
  int             msgLen = 0;
S
slguan 已提交
1609
  int             tableIndex = 0;
H
hzcheng 已提交
1610

1611 1612 1613
  SSqlCmd *   pCmd = &pSql->cmd;
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0);

1614
  STagCond *pTagCond = &pQueryInfo->tagCond;
S
slguan 已提交
1615

H
hjxilinx 已提交
1616
  STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
H
hzcheng 已提交
1617 1618

  int32_t size = tscEstimateMetricMetaMsgSize(pCmd);
S
slguan 已提交
1619 1620 1621 1622
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
    tscError("%p failed to malloc for metric meter msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1623 1624 1625 1626 1627

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

  SMgmtHead *pMgmt = (SMgmtHead *)pMsg;
H
hjxilinx 已提交
1628
  tscGetDBInfoFromMeterId(pTableMetaInfo->name, pMgmt->db);
H
hzcheng 已提交
1629 1630 1631

  pMsg += sizeof(SMgmtHead);

S
slguan 已提交
1632
  pMetaMsg = (SSuperTableMetaMsg *)pMsg;
S
slguan 已提交
1633
  pMetaMsg->numOfTables = htonl(pQueryInfo->numOfTables);
S
slguan 已提交
1634

S
slguan 已提交
1635
  pMsg += sizeof(SSuperTableMetaMsg);
S
slguan 已提交
1636 1637 1638 1639 1640

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

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

S
slguan 已提交
1643
  memcpy(pMsg, pTagCond->joinInfo.left.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1644
  pMsg += TSDB_TABLE_ID_LEN;
H
hzcheng 已提交
1645

S
slguan 已提交
1646 1647 1648
  *(int16_t *)pMsg = pTagCond->joinInfo.left.tagCol;
  pMsg += sizeof(int16_t);

S
slguan 已提交
1649
  memcpy(pMsg, pTagCond->joinInfo.right.tableId, TSDB_TABLE_ID_LEN);
S
slguan 已提交
1650
  pMsg += TSDB_TABLE_ID_LEN;
S
slguan 已提交
1651 1652 1653 1654

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

1655
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
1656
    pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
H
hjxilinx 已提交
1657
    uint64_t uid = pTableMetaInfo->pTableMeta->uid;
S
slguan 已提交
1658 1659 1660 1661

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

S
slguan 已提交
1662 1663
    SSuperTableMetaElemMsg *pElem = (SSuperTableMetaElemMsg *)pMsg;
    pMsg += sizeof(SSuperTableMetaElemMsg);
S
slguan 已提交
1664 1665 1666 1667

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

H
hjxilinx 已提交
1672
        bool ret = taosMbsToUcs4(pCond->cond, condLen, pMsg, condLen * TSDB_NCHAR_SIZE);
S
slguan 已提交
1673
        if (!ret) {
1674
          tscError("%p mbs to ucs4 failed:%s", pSql, tsGetSTableQueryCond(pTagCond, uid));
S
slguan 已提交
1675 1676 1677
          return 0;
        }
      }
H
hzcheng 已提交
1678 1679
    }

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

S
slguan 已提交
1682 1683 1684
    offset = pMsg - (char *)pMetaMsg;
    pElem->cond = htonl(offset);
    pMsg += condLen * TSDB_NCHAR_SIZE;
H
hzcheng 已提交
1685

S
slguan 已提交
1686 1687 1688
    pElem->rel = htons(pTagCond->relType);
    if (pTagCond->tbnameCond.uid == uid) {
      offset = pMsg - (char *)pMetaMsg;
H
hzcheng 已提交
1689

S
slguan 已提交
1690
      pElem->tableCond = htonl(offset);
H
hjxilinx 已提交
1691 1692 1693 1694 1695 1696 1697
      
      uint32_t len = 0;
      if (pTagCond->tbnameCond.cond != NULL) {
        len = strlen(pTagCond->tbnameCond.cond);
        memcpy(pMsg, pTagCond->tbnameCond.cond, len);
      }
      
H
hjxilinx 已提交
1698 1699
      pElem->tableCondLen = htonl(len);
      pMsg += len;
S
slguan 已提交
1700 1701
    }

1702
    SSqlGroupbyExpr *pGroupby = &pQueryInfo->groupbyExpr;
S
slguan 已提交
1703

H
hjxilinx 已提交
1704
    if (pGroupby->tableIndex != i && pGroupby->numOfGroupCols > 0) {
S
slguan 已提交
1705 1706 1707 1708 1709
      pElem->orderType = 0;
      pElem->orderIndex = 0;
      pElem->numOfGroupCols = 0;
    } else {
      pElem->numOfGroupCols = htons(pGroupby->numOfGroupCols);
H
hjxilinx 已提交
1710 1711
      for (int32_t j = 0; j < pTableMetaInfo->numOfTags; ++j) {
        pElem->tagCols[j] = htons(pTableMetaInfo->tagColumnIndex[j]);
S
slguan 已提交
1712 1713 1714 1715 1716 1717 1718 1719
      }

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

        pElem->groupbyTagColumnList = htonl(offset);
1720
        for (int32_t j = 0; j < pQueryInfo->groupbyExpr.numOfGroupCols; ++j) {
1721 1722
          SColIndex *pCol = &pQueryInfo->groupbyExpr.columnInfo[j];
          SColIndex *pDestCol = (SColIndex *)pMsg;
1723

H
hjxilinx 已提交
1724
          pDestCol->colIdxInBuf = 0;
1725
          pDestCol->colIndex = htons(pCol->colIndex);
H
hjxilinx 已提交
1726 1727
          pDestCol->colId = htons(pDestCol->colId);
          pDestCol->flag = htons(pDestCol->flag);
H
hjxilinx 已提交
1728
          strncpy(pDestCol->name, pCol->name, tListLen(pCol->name));
1729

1730
          pMsg += sizeof(SColIndex);
S
slguan 已提交
1731 1732
        }
      }
H
hzcheng 已提交
1733
    }
S
slguan 已提交
1734

H
hjxilinx 已提交
1735 1736
    strcpy(pElem->tableId, pTableMetaInfo->name);
    pElem->numOfTags = htons(pTableMetaInfo->numOfTags);
S
slguan 已提交
1737 1738 1739

    int16_t len = pMsg - (char *)pElem;
    pElem->elemLen = htons(len);  // redundant data for integrate check
H
hzcheng 已提交
1740 1741 1742 1743
  }

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1744
  pCmd->msgType = TSDB_MSG_TYPE_CM_STABLE_VGROUP;
H
hzcheng 已提交
1745
  assert(msgLen + minMsgSize() <= size);
H
hjxilinx 已提交
1746
#endif
1747
  
H
hjxilinx 已提交
1748
  SSqlCmd *pCmd = &pSql->cmd;
H
hjxilinx 已提交
1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761
  
  char* pMsg = pCmd->payload;
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, 0);
  
  SCMSTableVgroupMsg *pStableVgroupMsg = (SCMSTableVgroupMsg *) pMsg;
  pStableVgroupMsg->numOfTables = htonl(pQueryInfo->numOfTables);
  pMsg += sizeof(SCMSTableVgroupMsg);
  
  for(int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
    STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
    strncpy(pMsg, pTableMetaInfo->name, TSDB_TABLE_ID_LEN);
    pMsg += TSDB_TABLE_ID_LEN;
  }
H
hjxilinx 已提交
1762 1763

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

1766
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1767 1768
}

1769
int tscEstimateHeartBeatMsgLength(SSqlObj *pSql) {
H
hzcheng 已提交
1770 1771 1772
  int      size = 0;
  STscObj *pObj = pSql->pTscObj;

guanshengliang's avatar
guanshengliang 已提交
1773
  size += tsRpcHeadSize;
S
slguan 已提交
1774
  size += sizeof(SQqueryList);
H
hzcheng 已提交
1775 1776 1777

  SSqlObj *tpSql = pObj->sqlList;
  while (tpSql) {
S
slguan 已提交
1778
    size += sizeof(SQueryDesc);
H
hzcheng 已提交
1779 1780 1781
    tpSql = tpSql->next;
  }

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

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

1792
int tscBuildHeartBeatMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
H
hzcheng 已提交
1793 1794 1795 1796 1797 1798 1799 1800 1801
  char *pMsg, *pStart;
  int   msgLen = 0;
  int   size = 0;

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

  pthread_mutex_lock(&pObj->mutex);

1802
  size = tscEstimateHeartBeatMsgLength(pSql);
S
slguan 已提交
1803
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, size)) {
1804
    pthread_mutex_unlock(&pObj->mutex);
S
slguan 已提交
1805 1806 1807
    tscError("%p failed to malloc for heartbeat msg", pSql);
    return -1;
  }
H
hzcheng 已提交
1808

guanshengliang's avatar
guanshengliang 已提交
1809
  pMsg = pCmd->payload;
H
hzcheng 已提交
1810 1811 1812 1813 1814 1815 1816
  pStart = pMsg;

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

  msgLen = pMsg - pStart;
  pCmd->payloadLen = msgLen;
S
slguan 已提交
1817
  pCmd->msgType = TSDB_MSG_TYPE_CM_HEARTBEAT;
H
hzcheng 已提交
1818 1819

  assert(msgLen + minMsgSize() <= size);
1820
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1821 1822
}

1823 1824
int tscProcessTableMetaRsp(SSqlObj *pSql) {
  STableMetaMsg *pMetaMsg = (STableMetaMsg *)pSql->res.pRsp;
H
hzcheng 已提交
1825

1826 1827
  pMetaMsg->sid = htonl(pMetaMsg->sid);
  pMetaMsg->sversion = htons(pMetaMsg->sversion);
H
hjxilinx 已提交
1828 1829 1830
  
  pMetaMsg->vgroup.vgId = htonl(pMetaMsg->vgroup.vgId);
  
1831 1832
  pMetaMsg->uid = htobe64(pMetaMsg->uid);
  pMetaMsg->contLen = htons(pMetaMsg->contLen);
1833
  pMetaMsg->numOfColumns = htons(pMetaMsg->numOfColumns);
H
hzcheng 已提交
1834

H
hjxilinx 已提交
1835 1836
  if (pMetaMsg->sid < 0 || pMetaMsg->vgroup.numOfIps < 0) {
    tscError("invalid meter vgId:%d, sid%d", pMetaMsg->vgroup.numOfIps, pMetaMsg->sid);
H
hzcheng 已提交
1837 1838 1839
    return TSDB_CODE_INVALID_VALUE;
  }

1840 1841
  if (pMetaMsg->numOfTags > TSDB_MAX_TAGS || pMetaMsg->numOfTags < 0) {
    tscError("invalid numOfTags:%d", pMetaMsg->numOfTags);
H
hzcheng 已提交
1842 1843 1844
    return TSDB_CODE_INVALID_VALUE;
  }

1845 1846
  if (pMetaMsg->numOfColumns > TSDB_MAX_COLUMNS || pMetaMsg->numOfColumns <= 0) {
    tscError("invalid numOfColumns:%d", pMetaMsg->numOfColumns);
H
hzcheng 已提交
1847 1848 1849
    return TSDB_CODE_INVALID_VALUE;
  }

H
hjxilinx 已提交
1850 1851
  for (int i = 0; i < pMetaMsg->vgroup.numOfIps; ++i) {
    pMetaMsg->vgroup.ipAddr[i].port = htons(pMetaMsg->vgroup.ipAddr[i].port);
H
hzcheng 已提交
1852 1853
  }

1854
  SSchema* pSchema = pMetaMsg->schema;
H
hzcheng 已提交
1855

1856
  int32_t numOfTotalCols = pMetaMsg->numOfColumns + pMetaMsg->numOfTags;
H
hzcheng 已提交
1857 1858 1859
  for (int i = 0; i < numOfTotalCols; ++i) {
    pSchema->bytes = htons(pSchema->bytes);
    pSchema->colId = htons(pSchema->colId);
H
hjxilinx 已提交
1860 1861
    
    assert(pSchema->type >= TSDB_DATA_TYPE_BOOL && pSchema->type <= TSDB_DATA_TYPE_NCHAR);
H
hzcheng 已提交
1862 1863 1864
    pSchema++;
  }

1865 1866
  size_t size = 0;
  STableMeta* pTableMeta = tscCreateTableMetaFromMsg(pMetaMsg, &size);
H
hzcheng 已提交
1867

1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878
#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 已提交
1879
  // todo add one more function: taosAddDataIfNotExists();
1880
  STableMetaInfo *pTableMetaInfo = tscGetTableMetaInfoFromCmd(&pSql->cmd, 0, 0);
H
hjxilinx 已提交
1881
  assert(pTableMetaInfo->pTableMeta == NULL);
H
hzcheng 已提交
1882

H
hjxilinx 已提交
1883
  pTableMetaInfo->pTableMeta =
H
hjxilinx 已提交
1884
      (STableMeta *) taosCachePut(tscCacheHandle, pTableMetaInfo->name, pTableMeta, size, tsTableMetaKeepTimer);
H
hjxilinx 已提交
1885
  
1886
  // todo handle out of memory case
1887
  if (pTableMetaInfo->pTableMeta == NULL) {
H
hjxilinx 已提交
1888
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
1889
  }
H
hzcheng 已提交
1890

1891
  free(pTableMeta);
1892
  
H
hjxilinx 已提交
1893
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1894 1895
}

S
slguan 已提交
1896
/**
1897
 *  multi table meta rsp pkg format:
1898
 *  | STaosRsp | ieType | SCMMultiTableInfoMsg | SMeterMeta0 | SSchema0 | SMeterMeta1 | SSchema1 | SMeterMeta2 | SSchema2
S
slguan 已提交
1899 1900 1901
 *  |...... 1B        1B            4B
 **/
int tscProcessMultiMeterMetaRsp(SSqlObj *pSql) {
1902
#if 0
S
slguan 已提交
1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914
  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++;

1915
  SCMMultiTableInfoMsg *pInfo = (SCMMultiTableInfoMsg *)rsp;
S
slguan 已提交
1916
  totalNum = htonl(pInfo->numOfTables);
1917
  rsp += sizeof(SCMMultiTableInfoMsg);
S
slguan 已提交
1918 1919

  for (i = 0; i < totalNum; i++) {
S
slguan 已提交
1920
    SMultiTableMeta *pMultiMeta = (SMultiTableMeta *)rsp;
1921
    STableMeta *     pMeta = pMultiMeta->metas;
S
slguan 已提交
1922 1923 1924

    pMeta->sid = htonl(pMeta->sid);
    pMeta->sversion = htons(pMeta->sversion);
H
hjxilinx 已提交
1925
    pMeta->vgId = htonl(pMeta->vgId);
S
slguan 已提交
1926 1927
    pMeta->uid = htobe64(pMeta->uid);

H
hjxilinx 已提交
1928 1929
    if (pMeta->sid <= 0 || pMeta->vgId < 0) {
      tscError("invalid meter vgId:%d, sid%d", pMeta->vgId, pMeta->sid);
S
slguan 已提交
1930 1931 1932 1933 1934
      pSql->res.code = TSDB_CODE_INVALID_VALUE;
      pSql->res.numOfTotal = i;
      return TSDB_CODE_OTHERS;
    }

H
hjxilinx 已提交
1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957
    //    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 已提交
1958
    //    for (int j = 0; j < TSDB_REPLICA_MAX_NUM; ++j) {
H
hjxilinx 已提交
1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
    //      pMeta->vpeerDesc[j].vnode = htonl(pMeta->vpeerDesc[j].vnode);
    //    }
    //
    //    pMeta->rowSize = 0;
    //    rsp += sizeof(SMultiTableMeta);
    //    pSchema = (SSchema *)rsp;
    //
    //    int32_t numOfTotalCols = pMeta->numOfColumns + pMeta->numOfTags;
    //    for (int j = 0; j < numOfTotalCols; ++j) {
    //      pSchema->bytes = htons(pSchema->bytes);
    //      pSchema->colId = htons(pSchema->colId);
    //
    //      // ignore the tags length
    //      if (j < pMeta->numOfColumns) {
    //        pMeta->rowSize += pSchema->bytes;
    //      }
    //      pSchema++;
    //    }
    //
    //    rsp += numOfTotalCols * sizeof(SSchema);
    //
    //    int32_t  tagLen = 0;
    //    SSchema *pTagsSchema = tscGetTableTagSchema(pMeta);
    //
    //    if (pMeta->tableType == TSDB_CHILD_TABLE) {
    //      for (int32_t j = 0; j < pMeta->numOfTags; ++j) {
    //        tagLen += pTagsSchema[j].bytes;
    //      }
    //    }
    //
    //    rsp += tagLen;
    //    int32_t size = (int32_t)(rsp - ((char *)pMeta));  // Consistent with STableMeta in cache
    //
    //    pMeta->index = 0;
H
hjxilinx 已提交
1993
    //    (void)taosCachePut(tscCacheHandle, pMeta->tableId, (char *)pMeta, size, tsTableMetaKeepTimer);
H
hjxilinx 已提交
1994
    //  }
S
slguan 已提交
1995
  }
H
hjxilinx 已提交
1996
  
S
slguan 已提交
1997 1998 1999
  pSql->res.code = TSDB_CODE_SUCCESS;
  pSql->res.numOfTotal = i;
  tscTrace("%p load multi-metermeta resp complete num:%d", pSql, pSql->res.numOfTotal);
2000 2001
#endif
  
S
slguan 已提交
2002 2003 2004
  return TSDB_CODE_SUCCESS;
}

H
hjxilinx 已提交
2005 2006
int tscProcessSTableVgroupRsp(SSqlObj *pSql) {
#if 0
S
slguan 已提交
2007 2008
  void **      metricMetaList = NULL;
  int32_t *    sizes = NULL;
H
hjxilinx 已提交
2009
  
S
slguan 已提交
2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025
  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 已提交
2026
    pMeta = (SSuperTableMeta *)rsp;
S
slguan 已提交
2027 2028

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

S
slguan 已提交
2031
    pMeta->numOfTables = htonl(pMeta->numOfTables);
S
slguan 已提交
2032 2033 2034
    pMeta->numOfVnodes = htonl(pMeta->numOfVnodes);
    pMeta->tagLen = htons(pMeta->tagLen);

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

2037 2038
    char *pBuf = calloc(1, size);
    if (pBuf == NULL) {
S
slguan 已提交
2039 2040 2041
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2042

S
slguan 已提交
2043
    SSuperTableMeta *pNewMetricMeta = (SSuperTableMeta *)pBuf;
S
slguan 已提交
2044
    metricMetaList[k] = pNewMetricMeta;
H
hzcheng 已提交
2045

S
slguan 已提交
2046
    pNewMetricMeta->numOfTables = pMeta->numOfTables;
S
slguan 已提交
2047 2048
    pNewMetricMeta->numOfVnodes = pMeta->numOfVnodes;
    pNewMetricMeta->tagLen = pMeta->tagLen;
H
hzcheng 已提交
2049

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

S
slguan 已提交
2052 2053
    for (int32_t i = 0; i < pMeta->numOfVnodes; ++i) {
      SVnodeSidList *pSidLists = (SVnodeSidList *)rsp;
2054
      memcpy(pBuf, pSidLists, sizeof(SVnodeSidList));
H
hzcheng 已提交
2055

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

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

2061
      pBuf += sizeof(SVnodeSidList) + sizeof(STableIdInfo *) * pSidLists->numOfSids;
S
slguan 已提交
2062
      rsp += sizeof(SVnodeSidList);
H
hzcheng 已提交
2063

2064
      size_t elemSize = sizeof(STableIdInfo) + pNewMetricMeta->tagLen;
S
slguan 已提交
2065
      for (int32_t j = 0; j < pSidLists->numOfSids; ++j) {
2066 2067
        pLists->pSidExtInfoList[j] = pBuf - (char *)pLists;
        memcpy(pBuf, rsp, elemSize);
2068

2069 2070
        ((STableIdInfo *)pBuf)->uid = htobe64(((STableIdInfo *)pBuf)->uid);
        ((STableIdInfo *)pBuf)->sid = htonl(((STableIdInfo *)pBuf)->sid);
2071

2072 2073
        rsp += elemSize;
        pBuf += elemSize;
S
slguan 已提交
2074
      }
H
hzcheng 已提交
2075
    }
S
slguan 已提交
2076

2077
    sizes[k] = pBuf - (char *)pNewMetricMeta;
H
hzcheng 已提交
2078 2079
  }

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

H
hjxilinx 已提交
2084
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2085
    tscGetMetricMetaCacheKey(pQueryInfo, name, pTableMetaInfo->pTableMeta->uid);
H
hzcheng 已提交
2086

S
slguan 已提交
2087 2088 2089
#ifdef _DEBUG_VIEW
    printf("generate the metric key:%s, index:%d\n", name, i);
#endif
H
hzcheng 已提交
2090

S
slguan 已提交
2091
    // release the used metricmeta
H
hjxilinx 已提交
2092 2093
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), false);
    pTableMetaInfo->pMetricMeta = (SSuperTableMeta *)taosCachePut(tscCacheHandle, name, (char *)metricMetaList[i],
S
slguan 已提交
2094 2095 2096 2097
                                                                      sizes[i], tsMetricMetaKeepTimer);
    tfree(metricMetaList[i]);

    // failed to put into cache
H
hjxilinx 已提交
2098
    if (pTableMetaInfo->pMetricMeta == NULL) {
S
slguan 已提交
2099 2100 2101
      pSql->res.code = TSDB_CODE_CLI_OUT_OF_MEMORY;
      goto _error_clean;
    }
H
hzcheng 已提交
2102 2103
  }

S
slguan 已提交
2104 2105 2106 2107 2108 2109 2110 2111
_error_clean:
  // free allocated resource
  for (int32_t i = 0; i < num; ++i) {
    tfree(metricMetaList[i]);
  }

  free(sizes);
  free(metricMetaList);
H
hjxilinx 已提交
2112
#endif
2113
  SSqlRes* pRes = &pSql->res;
H
hjxilinx 已提交
2114
  
H
hjxilinx 已提交
2115
  // NOTE: the order of several table must be preserved.
2116
  SCMSTableVgroupRspMsg *pStableVgroup = (SCMSTableVgroupRspMsg *)pRes->pRsp;
H
hjxilinx 已提交
2117 2118
  pStableVgroup->numOfTables = htonl(pStableVgroup->numOfTables);
  char* pMsg = pRes->pRsp + sizeof(SCMSTableVgroupRspMsg);
H
hjxilinx 已提交
2119
  
2120 2121 2122
  // master sqlObj locates in param
  SSqlObj* parent = pSql->param;
  assert(parent != NULL);
H
hjxilinx 已提交
2123
  
2124
  SSqlCmd* pCmd = &parent->cmd;
H
hjxilinx 已提交
2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145
  for(int32_t i = 0; i < pStableVgroup->numOfTables; ++i) {
    STableMetaInfo *pInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, i);
    SVgroupsInfo *  pVgroupInfo = (SVgroupsInfo *)pMsg;
    pVgroupInfo->numOfVgroups = htonl(pVgroupInfo->numOfVgroups);

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

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

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

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

      pMsg += size;
2146
    }
H
hjxilinx 已提交
2147 2148
  }
  
S
slguan 已提交
2149
  return pSql->res.code;
H
hzcheng 已提交
2150 2151 2152 2153 2154 2155
}

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

2161 2162 2163
  SSqlRes *pRes = &pSql->res;
  SSqlCmd *pCmd = &pSql->cmd;

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

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

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

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

H
hjxilinx 已提交
2175
  pMetaMsg->numOfColumns = ntohs(pMetaMsg->numOfColumns);
H
hzcheng 已提交
2176

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

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

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

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

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

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

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

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

  return 0;
}

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

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

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

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

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

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

  return 0;
}

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

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

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

H
hjxilinx 已提交
2294
  if (pTableMetaInfo->pTableMeta) {
weixin_48148422's avatar
weixin_48148422 已提交
2295
    bool isSuperTable = UTIL_TABLE_IS_SUPER_TABLE(pTableMetaInfo);
H
hzcheng 已提交
2296

H
hjxilinx 已提交
2297
    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pTableMeta), true);
H
hjxilinx 已提交
2298
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2299

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

  return 0;
}

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

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

2317
  SQueryTableRsp *pQuery = (SQueryTableRsp *)pRes->pRsp;
S
slguan 已提交
2318 2319 2320
  pQuery->qhandle = htobe64(pQuery->qhandle);
  pRes->qhandle = pQuery->qhandle;

H
hzcheng 已提交
2321
  pRes->data = NULL;
S
slguan 已提交
2322
  tscResetForNextRetrieve(pRes);
H
hzcheng 已提交
2323 2324 2325
  return 0;
}

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

S
slguan 已提交
2330
  SRetrieveTableRsp *pRetrieve = (SRetrieveTableRsp *)pRes->pRsp;
H
hzcheng 已提交
2331 2332 2333

  pRes->numOfRows = htonl(pRetrieve->numOfRows);
  pRes->precision = htons(pRetrieve->precision);
2334 2335
  pRes->offset    = htobe64(pRetrieve->offset);
  pRes->useconds  = htobe64(pRetrieve->useconds);
H
hjxilinx 已提交
2336
  pRes->completed = (pRetrieve->completed == 1);
2337
  pRes->data      = pRetrieve->data;
H
hjxilinx 已提交
2338
  
2339
  SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
2340 2341 2342 2343
  if (tscCreateResPointerInfo(pRes, pQueryInfo) != TSDB_CODE_SUCCESS) {
    return pRes->code;
  }
  
weixin_48148422's avatar
weixin_48148422 已提交
2344
  if (pSql->pSubscription != NULL) {
H
hjxilinx 已提交
2345
    int32_t numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
H
hjxilinx 已提交
2346
    
H
hjxilinx 已提交
2347
    TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, numOfCols - 1);
H
hjxilinx 已提交
2348 2349
    int16_t     offset = tscFieldInfoGetOffset(pQueryInfo, numOfCols - 1);
    
weixin_48148422's avatar
weixin_48148422 已提交
2350 2351
    char* p = pRes->data + (pField->bytes + offset) * pRes->numOfRows;

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

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

  return 0;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  return code;
}

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

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

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

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

/*
 * in handling the renew metermeta problem during insertion,
S
slguan 已提交
2456
 *
H
hzcheng 已提交
2457 2458 2459 2460 2461
 * If the meter is created on demand during insertion, the routine usually waits for a short
 * period to re-issue the getMeterMeta msg, in which makes a greater change that vnode has
 * successfully created the corresponding table.
 */
static void tscWaitingForCreateTable(SSqlCmd *pCmd) {
S
slguan 已提交
2462
  if (pCmd->command == TSDB_SQL_INSERT) {
H
hzcheng 已提交
2463 2464 2465 2466 2467 2468 2469
    taosMsleep(50);  // todo: global config
  }
}

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

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

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

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

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

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

  return code;
}

H
hjxilinx 已提交
2506
static bool allVgroupInfoRetrieved(SSqlCmd* pCmd, int32_t clauseIndex) {
2507
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2508
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2509
    STableMetaInfo *pTableMetaInfo = tscGetMetaInfo(pQueryInfo, i);
H
hjxilinx 已提交
2510 2511
    if (pTableMetaInfo->vgroupList == NULL) {
      return false;
S
slguan 已提交
2512 2513
    }
  }
H
hjxilinx 已提交
2514 2515 2516 2517
  
  // all super tables vgroupinfo are retrieved, no need to retrieve vgroup info anymore
  return true;
}
H
hzcheng 已提交
2518

H
hjxilinx 已提交
2519 2520 2521 2522 2523
int tscGetSTableVgroupInfo(SSqlObj *pSql, int32_t clauseIndex) {
  int      code = TSDB_CODE_NETWORK_UNAVAIL;
  SSqlCmd *pCmd = &pSql->cmd;
  
  if (allVgroupInfoRetrieved(pCmd, clauseIndex)) {
H
hzcheng 已提交
2524 2525
    return TSDB_CODE_SUCCESS;
  }
H
hjxilinx 已提交
2526

S
slguan 已提交
2527
  SSqlObj *pNew = calloc(1, sizeof(SSqlObj));
H
hzcheng 已提交
2528 2529 2530
  pNew->pTscObj = pSql->pTscObj;
  pNew->signature = pNew;

H
hjxilinx 已提交
2531
  pNew->cmd.command = TSDB_SQL_STABLEVGROUP;
2532 2533
  
  SQueryInfo *pNewQueryInfo = NULL;
2534 2535 2536
  if ((code = tscGetQueryInfoDetailSafely(&pNew->cmd, 0, &pNewQueryInfo)) != TSDB_CODE_SUCCESS) {
    return code;
  }
2537
  
H
hjxilinx 已提交
2538
  SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, clauseIndex);
2539
  for (int32_t i = 0; i < pQueryInfo->numOfTables; ++i) {
H
hjxilinx 已提交
2540 2541 2542
    STableMetaInfo *pMInfo = tscGetMetaInfo(pQueryInfo, i);
    STableMeta *pTableMeta = taosCacheAcquireByData(tscCacheHandle, pMInfo->pTableMeta);
    tscAddTableMetaInfo(pNewQueryInfo, pMInfo->name, pTableMeta, NULL, pMInfo->tagColList);
S
slguan 已提交
2543 2544 2545 2546 2547 2548
  }

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

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

2553 2554 2555 2556 2557
  pNew->fp = tscTableMetaCallBack;
  pNew->param = pSql;
  code = tscProcessSql(pNew);
  if (code == TSDB_CODE_SUCCESS) {
    code = TSDB_CODE_ACTION_IN_PROGRESS;
H
hzcheng 已提交
2558 2559 2560 2561 2562
  }

  return code;
}

2563
void tscInitMsgsFp() {
S
slguan 已提交
2564 2565
  tscBuildMsg[TSDB_SQL_SELECT] = tscBuildQueryMsg;
  tscBuildMsg[TSDB_SQL_INSERT] = tscBuildSubmitMsg;
J
jtao1735 已提交
2566
  tscBuildMsg[TSDB_SQL_FETCH] = tscBuildFetchMsg;
H
hzcheng 已提交
2567 2568

  tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
2569
  tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
H
hzcheng 已提交
2570

2571 2572
  tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
  tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
H
hzcheng 已提交
2573 2574

  tscBuildMsg[TSDB_SQL_CREATE_TABLE] = tscBuildCreateTableMsg;
S
[TD-16]  
slguan 已提交
2575
  tscBuildMsg[TSDB_SQL_DROP_USER] = tscBuildDropUserMsg;
H
hzcheng 已提交
2576 2577 2578
  tscBuildMsg[TSDB_SQL_DROP_ACCT] = tscBuildDropAcctMsg;
  tscBuildMsg[TSDB_SQL_DROP_DB] = tscBuildDropDbMsg;
  tscBuildMsg[TSDB_SQL_DROP_TABLE] = tscBuildDropTableMsg;
2579
  tscBuildMsg[TSDB_SQL_ALTER_USER] = tscBuildUserMsg;
S
slguan 已提交
2580 2581 2582
  tscBuildMsg[TSDB_SQL_CREATE_DNODE] = tscBuildCreateDnodeMsg;
  tscBuildMsg[TSDB_SQL_DROP_DNODE] = tscBuildDropDnodeMsg;
  tscBuildMsg[TSDB_SQL_CFG_DNODE] = tscBuildCfgDnodeMsg;
H
hzcheng 已提交
2583 2584 2585 2586 2587
  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 已提交
2588
  tscBuildMsg[TSDB_SQL_META] = tscBuildTableMetaMsg;
H
hjxilinx 已提交
2589
  tscBuildMsg[TSDB_SQL_STABLEVGROUP] = tscBuildSTableVgroupMsg;
S
slguan 已提交
2590
  tscBuildMsg[TSDB_SQL_MULTI_META] = tscBuildMultiMeterMetaMsg;
H
hzcheng 已提交
2591 2592 2593 2594

  tscBuildMsg[TSDB_SQL_HB] = tscBuildHeartBeatMsg;
  tscBuildMsg[TSDB_SQL_SHOW] = tscBuildShowMsg;
  tscBuildMsg[TSDB_SQL_RETRIEVE] = tscBuildRetrieveFromMgmtMsg;
2595 2596 2597
  tscBuildMsg[TSDB_SQL_KILL_QUERY] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_STREAM] = tscBuildKillMsg;
  tscBuildMsg[TSDB_SQL_KILL_CONNECTION] = tscBuildKillMsg;
H
hzcheng 已提交
2598 2599

  tscProcessMsgRsp[TSDB_SQL_SELECT] = tscProcessQueryRsp;
H
hjxilinx 已提交
2600
  tscProcessMsgRsp[TSDB_SQL_FETCH] = tscProcessRetrieveRspFromNode;
H
hzcheng 已提交
2601 2602 2603 2604 2605

  tscProcessMsgRsp[TSDB_SQL_DROP_DB] = tscProcessDropDbRsp;
  tscProcessMsgRsp[TSDB_SQL_DROP_TABLE] = tscProcessDropTableRsp;
  tscProcessMsgRsp[TSDB_SQL_CONNECT] = tscProcessConnectRsp;
  tscProcessMsgRsp[TSDB_SQL_USE_DB] = tscProcessUseDbRsp;
2606
  tscProcessMsgRsp[TSDB_SQL_META] = tscProcessTableMetaRsp;
H
hjxilinx 已提交
2607
  tscProcessMsgRsp[TSDB_SQL_STABLEVGROUP] = tscProcessSTableVgroupRsp;
S
slguan 已提交
2608
  tscProcessMsgRsp[TSDB_SQL_MULTI_META] = tscProcessMultiMeterMetaRsp;
H
hzcheng 已提交
2609 2610

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

H
hjxilinx 已提交
2614 2615 2616 2617 2618
  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;
2619

H
hzcheng 已提交
2620 2621
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_EMPTY_RESULT] = tscProcessEmptyResultRsp;

H
hjxilinx 已提交
2622
  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_LOCALMERGE] = tscProcessRetrieveLocalMergeRsp;
H
hzcheng 已提交
2623 2624 2625 2626 2627 2628 2629 2630 2631 2632

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