tscServer.c 87.9 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

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

#define TSC_MGMT_VNODE 999

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

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

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

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

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

48 49 50
static void tscSetDnodeIpList(SSqlObj* pSql, STableMeta* pTableMeta) {
  SRpcIpSet* pIpList = &pSql->ipList;
  
H
hjxilinx 已提交
51
  pIpList->numOfIps = pTableMeta->vgroupInfo.numOfIps;
52 53
  pIpList->inUse    = 0;
  
H
hjxilinx 已提交
54
  for(int32_t i = 0; i < pTableMeta->vgroupInfo.numOfIps; ++i) {
J
jtao1735 已提交
55 56
    strcpy(pIpList->fqdn[i], pTableMeta->vgroupInfo.ipAddr[i].fqdn);
    pIpList->port[i] = pTableMeta->vgroupInfo.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;
S
slguan 已提交
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 89
void tscUpdateIpSet(void *ahandle, SRpcIpSet *pIpSet) {
  tscTrace("mgmt IP list is changed for ufp is called");
S
slguan 已提交
90
  tscMgmtIpSet = *pIpSet;
S
slguan 已提交
91 92
}

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

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

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

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

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

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

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

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

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

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

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

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

  tscProcessSql(pObj->pHb);
}

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

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

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

S
slguan 已提交
226
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
227 228
}

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

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

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

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

S
slguan 已提交
306 307 308 309 310
  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 已提交
311
  if (pRes->code != TSDB_CODE_QUERY_CANCELLED) {
312
    assert(rpcMsg->msgType == pCmd->msgType + 1);
313
    pRes->code    = rpcMsg->code;
314
    pRes->rspType = rpcMsg->msgType;
315
    pRes->rspLen  = rpcMsg->contLen;
H
hzcheng 已提交
316

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

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

    /*
     * There is not response callback function for submit response.
     * The actual inserted number of points is the first number.
     */
338
    if (rpcMsg->msgType == TSDB_MSG_TYPE_SUBMIT_RSP && pRes->pRsp != NULL) {
339
      SShellSubmitRspMsg *pMsg = (SShellSubmitRspMsg*)pRes->pRsp;
S
slguan 已提交
340 341 342 343 344 345 346
      pMsg->code = htonl(pMsg->code);
      pMsg->numOfRows = htonl(pMsg->numOfRows);
      pMsg->affectedRows = htonl(pMsg->affectedRows);
      pMsg->failedRows = htonl(pMsg->failedRows);
      pMsg->numOfFailedBlocks = htonl(pMsg->numOfFailedBlocks);

      pRes->numOfRows += pMsg->affectedRows;
347
      tscTrace("%p cmd:%d code:%s, inserted rows:%d, rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code),
H
hjxilinx 已提交
348
          pMsg->affectedRows, pRes->rspLen);
S
slguan 已提交
349
    } else {
H
hjxilinx 已提交
350
      tscTrace("%p cmd:%d code:%s rsp len:%d", pSql, pCmd->command, tstrerror(pRes->code), pRes->rspLen);
H
hzcheng 已提交
351 352 353
    }
  }

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

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

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

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

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

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

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

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

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

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

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

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

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

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

H
hzcheng 已提交
485 486 487 488 489
    /*
     * here, we cannot set the command = TSDB_SQL_KILL_QUERY. Otherwise, it may cause
     * sub-queries not correctly released and master sql object of metric query reaches an abnormal state.
     */
    pSql->pSubs[i]->res.code = TSDB_CODE_QUERY_CANCELLED;
S
slguan 已提交
490
    //taosStopRpcConn(pSql->pSubs[i]->thandle);
H
hzcheng 已提交
491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511
  }

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

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

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

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

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

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

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

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

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

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

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

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

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

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

H
hjxilinx 已提交
603
  STableMeta * pTableMeta = pTableMetaInfo->pTableMeta;
H
hjxilinx 已提交
604 605 606 607 608 609 610 611
  tscTrace("%p vgId:%d, query on table:%s, uid:%" PRIu64, pSql, vgId, pTableMetaInfo->name, pTableMeta->uid);
  
  STableIdInfo *pTableIdInfo = (STableIdInfo *)pMsg;
  pTableIdInfo->sid = htonl(pTableMeta->sid);
  pTableIdInfo->uid = htobe64(pTableMeta->uid);
  pTableIdInfo->key = htobe64(tscGetSubscriptionProgress(pSql->pSubscription, pTableMeta->uid));
  
  pMsg += sizeof(STableIdInfo);
612 613 614
  return pMsg;
}

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

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

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

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

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

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

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

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

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

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

719 720 721 722 723 724
    if (pCol->colIndex.columnIndex >= tscGetNumOfColumns(pTableMeta) || pColSchema->type < TSDB_DATA_TYPE_BOOL ||
        pColSchema->type > TSDB_DATA_TYPE_NCHAR) {
      tscError("%p sid:%d uid:%" PRIu64" id:%s, column index out of range, numOfColumns:%d, index:%d, column name:%s",
          pSql, pTableMeta->sid, pTableMeta->uid, pTableMetaInfo->name, tscGetNumOfColumns(pTableMeta), pCol->colIndex,
               pColSchema->name);

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

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

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

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

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

742
      if (pColFilter->filterstr) {
S
slguan 已提交
743 744 745 746 747 748 749 750 751 752
        pFilterMsg->len = htobe64(pColFilter->len);
        memcpy(pMsg, (void *)pColFilter->pz, pColFilter->len + 1);
        pMsg += (pColFilter->len + 1);  // append the additional filter binary info
      } else {
        pFilterMsg->lowerBndi = htobe64(pColFilter->lowerBndi);
        pFilterMsg->upperBndi = htobe64(pColFilter->upperBndi);
      }

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

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

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

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

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

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

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

      if (pExpr->param[j].nType == TSDB_DATA_TYPE_BINARY) {
        memcpy(pMsg, pExpr->param[j].pz, pExpr->param[j].nLen);
785
        pMsg += pExpr->param[j].nLen;
H
hzcheng 已提交
786 787 788 789 790
      } else {
        pSqlFuncExpr->arg[j].argValue.i64 = htobe64(pExpr->param[j].i64Key);
      }
    }

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

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

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

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

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

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

819
  if (pQueryInfo->interpoType != TSDB_INTERPO_NONE) {
H
hjxilinx 已提交
820
    for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
821 822
      *((int64_t *)pMsg) = htobe64(pQueryInfo->defaultVal[i]);
      pMsg += sizeof(pQueryInfo->defaultVal[0]);
S
slguan 已提交
823 824
    }
  }
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855
  
  if (numOfTags != 0) {
    int32_t numOfColumns = tscGetNumOfColumns(pTableMeta);
    int32_t numOfTagColumns = tscGetNumOfTags(pTableMeta);
    int32_t total = numOfTagColumns + numOfColumns;
    
    pSchema = tscGetTableTagSchema(pTableMeta);
    
    for (int32_t i = 0; i < numOfTags; ++i) {
      SColumn *pCol = taosArrayGetP(pQueryInfo->colList, i);
      SSchema *pColSchema = &pSchema[pCol->colIndex.columnIndex];

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

        return TSDB_CODE_INVALID_SQL;
      }
  
      SColumnInfo* pTagCol = (SColumnInfo*) pMsg;
  
      pTagCol->colId = htons(pColSchema->colId);
      pTagCol->bytes = htons(pColSchema->bytes);
      pTagCol->type  = htons(pColSchema->type);
      pTagCol->numOfFilters = 0;
      
      pMsg += sizeof(SColumnInfo);
    }
  }
S
slguan 已提交
856 857 858 859 860 861

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

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

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

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

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

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

H
hjxilinx 已提交
902
  // tbname in/like query expression should be sent to mgmt node
H
hzcheng 已提交
903 904 905 906
  msgLen = pMsg - pStart;

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

  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
913 914
}

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

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

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

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

929 930
int32_t tscBuildCreateDnodeMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
  SSqlCmd *pCmd = &pSql->cmd;
931
  pCmd->payloadLen = sizeof(SCMCreateDnodeMsg);
S
slguan 已提交
932 933 934 935
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }
H
hzcheng 已提交
936

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

S
slguan 已提交
1145 1146 1147
  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 已提交
1148
  }
H
hzcheng 已提交
1149

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

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

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

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

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

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

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

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

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

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

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

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

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

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

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

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


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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1322 1323
  SAlterTableSQL *pAlterInfo = pInfo->pAlterInfo;

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

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

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

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

  pMsg = (char *)pSchema;

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

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

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

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

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

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

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

S
slguan 已提交
1368 1369 1370
  if (TSDB_CODE_SUCCESS != tscAllocPayload(pCmd, pCmd->payloadLen)) {
    tscError("%p failed to malloc for query msg", pSql);
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
H
hzcheng 已提交
1371
  }
S
slguan 已提交
1372

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

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

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

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

  return 0;
}

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

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

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

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

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

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

  return code;
}

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

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

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

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

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

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

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

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

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

  pRes->row = 0;

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

  return code;
}

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

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

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

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

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

1506
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1507 1508
}

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

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

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

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

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

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

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

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

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

  tfree(tmpData);

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

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

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

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

  tfree(tmpData);

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

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

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

  return pCmd->payloadLen;
}

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

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

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

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

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

  return MAX(len, TSDB_DEFAULT_PAYLOAD_SIZE);
H
hzcheng 已提交
1615 1616
}

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

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

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

1628
  STagCond *pTagCond = &pQueryInfo->tagCond;
S
slguan 已提交
1629

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

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

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

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

  pMsg += sizeof(SMgmtHead);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1716
    SSqlGroupbyExpr *pGroupby = &pQueryInfo->groupbyExpr;
S
slguan 已提交
1717

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

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

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

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

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

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

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

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

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

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

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

1772
  return TSDB_CODE_SUCCESS;
H
hzcheng 已提交
1773 1774
}

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

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

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

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

  return size + TSDB_EXTRA_PAYLOAD_SIZE;
}

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

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

  pthread_mutex_lock(&pObj->mutex);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  free(sizes);
  free(metricMetaList);
H
hjxilinx 已提交
2122
#endif
2123
  SSqlRes* pRes = &pSql->res;
H
hjxilinx 已提交
2124
  
2125
  SCMSTableVgroupRspMsg *pStableVgroup = (SCMSTableVgroupRspMsg *)pRes->pRsp;
H
hjxilinx 已提交
2126
  pStableVgroup->numOfVgroups = htonl(pStableVgroup->numOfVgroups);
H
hjxilinx 已提交
2127
  
2128 2129 2130
  // master sqlObj locates in param
  SSqlObj* parent = pSql->param;
  assert(parent != NULL);
H
hjxilinx 已提交
2131
  
2132
  SSqlCmd* pCmd = &parent->cmd;
H
hjxilinx 已提交
2133 2134
  STableMetaInfo* pInfo = tscGetTableMetaInfoFromCmd(pCmd, pCmd->clauseIndex, 0);
  
2135 2136 2137
  pInfo->vgroupList = malloc(pRes->rspLen);
  memcpy(pInfo->vgroupList, pStableVgroup, pRes->rspLen);
  
H
hjxilinx 已提交
2138 2139
  for(int32_t i = 0; i < pInfo->vgroupList->numOfVgroups; ++i) {
    SCMVgroupInfo* pVgroups = &pInfo->vgroupList->vgroups[i];
2140
    
H
hjxilinx 已提交
2141
    pVgroups->vgId = htonl(pVgroups->vgId);
2142
    assert(pVgroups->numOfIps >= 1);
2143
    
2144
    for(int32_t j = 0; j < pVgroups->numOfIps; ++j) {
H
hjxilinx 已提交
2145
      pVgroups->ipAddr[j].port = htons(pVgroups->ipAddr[j].port);
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, tsMeterMetaKeepTimer);
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 2239 2240 2241 2242 2243 2244
  pObj->writeAuth = pConnect->writeAuth;
  pObj->superAuth = pConnect->superAuth;
  taosTmrReset(tscProcessActivityTimer, tsShellActivityTimer * 500, pObj, tscTmr, &pObj->pTimer);

  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
hjxilinx 已提交
2278
//    taosCacheRelease(tscCacheHandle, (void **)&(pTableMetaInfo->pMetricMeta), true);
H
hzcheng 已提交
2279 2280 2281 2282 2283 2284
  }

  return 0;
}

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

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

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

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

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

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

  return 0;
}

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

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

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

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

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

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

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

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

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

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

  return 0;
}

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

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

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

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

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

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

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

2396
  tscAddSubqueryInfo(&pNew->cmd);
2397 2398 2399 2400

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

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

S
slguan 已提交
2406 2407 2408
    return TSDB_CODE_CLI_OUT_OF_MEMORY;
  }

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

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

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

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

  return code;
}

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

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

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

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

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

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

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

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

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

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

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

  return code;
}

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

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

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

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

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

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

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

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

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

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

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

2565 2566
  pNewQueryInfo->groupbyExpr = pQueryInfo->groupbyExpr;
  pNewQueryInfo->numOfTables = pQueryInfo->numOfTables;
H
hzcheng 已提交
2567

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

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

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

  return code;
}

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

  tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
2598
  tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
H
hzcheng 已提交
2599

2600 2601
  tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
  tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
H
hzcheng 已提交
2602 2603

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

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

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

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

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

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

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

  tscProcessMsgRsp[TSDB_SQL_RETRIEVE_METRIC] = tscProcessRetrieveMetricRsp;

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

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