vnodeSvr.c 39.5 KB
Newer Older
H
Hongze Cheng 已提交
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/>.
 */

H
Hongze Cheng 已提交
16
#include "vnd.h"
H
Hongze Cheng 已提交
17

18 19 20 21 22 23 24 25 26
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessAlterConfirmReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
27 28
static int32_t vnodeProcessAlterHashRangeReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
29
static int32_t vnodeProcessDropTtlTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
S
Shengliang Guan 已提交
30
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
31
static int32_t vnodeProcessDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
32
static int32_t vnodeProcessBatchDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
33

34
int32_t vnodePreProcessWriteMsg(SVnode *pVnode, SRpcMsg *pMsg) {
S
Shengliang Guan 已提交
35
  int32_t  code = 0;
H
Hongze Cheng 已提交
36
  SDecoder dc = {0};
H
Hongze Cheng 已提交
37

H
Hongze Cheng 已提交
38 39 40 41 42 43
  switch (pMsg->msgType) {
    case TDMT_VND_CREATE_TABLE: {
      int64_t ctime = taosGetTimestampMs();
      int32_t nReqs;

      tDecoderInit(&dc, (uint8_t *)pMsg->pCont + sizeof(SMsgHead), pMsg->contLen - sizeof(SMsgHead));
H
Hongze Cheng 已提交
44 45 46 47
      if (tStartDecode(&dc) < 0) {
        code = TSDB_CODE_INVALID_MSG;
        return code;
      }
H
Hongze Cheng 已提交
48

H
Hongze Cheng 已提交
49 50 51 52
      if (tDecodeI32v(&dc, &nReqs) < 0) {
        code = TSDB_CODE_INVALID_MSG;
        goto _err;
      }
H
Hongze Cheng 已提交
53 54
      for (int32_t iReq = 0; iReq < nReqs; iReq++) {
        tb_uid_t uid = tGenIdPI64();
H
Hongze Cheng 已提交
55
        char    *name = NULL;
H
Hongze Cheng 已提交
56 57 58 59
        if (tStartDecode(&dc) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          goto _err;
        }
H
Hongze Cheng 已提交
60

H
Hongze Cheng 已提交
61 62 63 64 65 66 67 68
        if (tDecodeI32v(&dc, NULL) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          return code;
        }
        if (tDecodeCStr(&dc, &name) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          return code;
        }
H
Hongze Cheng 已提交
69 70 71
        *(int64_t *)(dc.data + dc.pos) = uid;
        *(int64_t *)(dc.data + dc.pos + 8) = ctime;

72
        vTrace("vgId:%d, table:%s uid:%" PRId64 " is generated", pVnode->config.vgId, name, uid);
H
Hongze Cheng 已提交
73 74 75 76 77 78 79 80 81 82 83
        tEndDecode(&dc);
      }

      tEndDecode(&dc);
      tDecoderClear(&dc);
    } break;
    case TDMT_VND_SUBMIT: {
      SSubmitMsgIter msgIter = {0};
      SSubmitReq    *pSubmitReq = (SSubmitReq *)pMsg->pCont;
      SSubmitBlk    *pBlock = NULL;
      int64_t        ctime = taosGetTimestampMs();
H
Hongze Cheng 已提交
84
      tb_uid_t       uid;
H
Hongze Cheng 已提交
85

H
Hongze Cheng 已提交
86 87 88 89
      if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
        code = terrno;
        goto _err;
      }
H
Hongze Cheng 已提交
90 91 92 93 94 95

      for (;;) {
        tGetSubmitMsgNext(&msgIter, &pBlock);
        if (pBlock == NULL) break;

        if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
96
          char *name = NULL;
H
Hongze Cheng 已提交
97

H
Hongze Cheng 已提交
98
          tDecoderInit(&dc, pBlock->data, msgIter.schemaLen);
H
Hongze Cheng 已提交
99 100 101 102
          if (tStartDecode(&dc) < 0) {
            code = TSDB_CODE_INVALID_MSG;
            return code;
          }
H
Hongze Cheng 已提交
103

H
Hongze Cheng 已提交
104 105 106 107 108 109 110 111
          if (tDecodeI32v(&dc, NULL) < 0) {
            code = TSDB_CODE_INVALID_MSG;
            return code;
          }
          if (tDecodeCStr(&dc, &name) < 0) {
            code = TSDB_CODE_INVALID_MSG;
            return code;
          }
H
Hongze Cheng 已提交
112 113 114 115 116

          uid = metaGetTableEntryUidByName(pVnode->pMeta, name);
          if (uid == 0) {
            uid = tGenIdPI64();
          }
H
Hongze Cheng 已提交
117
          *(int64_t *)(dc.data + dc.pos) = uid;
H
Hongze Cheng 已提交
118
          *(int64_t *)(dc.data + dc.pos + 8) = ctime;
H
Hongze Cheng 已提交
119
          pBlock->uid = htobe64(uid);
H
Hongze Cheng 已提交
120 121 122 123 124 125 126

          tEndDecode(&dc);
          tDecoderClear(&dc);
        }
      }

    } break;
H
Hongze Cheng 已提交
127 128 129 130 131 132 133 134 135 136
    case TDMT_VND_DELETE: {
      int32_t     size;
      int32_t     ret;
      uint8_t    *pCont;
      SEncoder   *pCoder = &(SEncoder){0};
      SDeleteRes  res = {0};
      SReadHandle handle = {
          .meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};

      code = qWorkerProcessDeleteMsg(&handle, pVnode->pQuery, pMsg, &res);
wmmhello's avatar
wmmhello 已提交
137 138 139
      if (code) {
        goto _err;
      }
H
Hongze Cheng 已提交
140 141 142 143 144

      // malloc and encode
      tEncodeSize(tEncodeDeleteRes, &res, size, ret);
      pCont = rpcMallocCont(size + sizeof(SMsgHead));

145 146
      ((SMsgHead *)pCont)->contLen = size + sizeof(SMsgHead);
      ((SMsgHead *)pCont)->vgId = TD_VID(pVnode);
H
Hongze Cheng 已提交
147 148 149 150 151 152 153 154 155 156 157

      tEncoderInit(pCoder, pCont + sizeof(SMsgHead), size);
      tEncodeDeleteRes(pCoder, &res);
      tEncoderClear(pCoder);

      rpcFreeCont(pMsg->pCont);
      pMsg->pCont = pCont;
      pMsg->contLen = size + sizeof(SMsgHead);

      taosArrayDestroy(res.uidList);
    } break;
H
Hongze Cheng 已提交
158 159 160
    default:
      break;
  }
H
Hongze Cheng 已提交
161

S
Shengliang Guan 已提交
162
  return code;
H
Hongze Cheng 已提交
163 164 165 166

_err:
  vError("vgId%d, preprocess request failed since %s", TD_VID(pVnode), tstrerror(code));
  return code;
H
Hongze Cheng 已提交
167 168
}

169
int32_t vnodeProcessWriteMsg(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
170 171 172 173
  void   *ptr = NULL;
  void   *pReq;
  int32_t len;
  int32_t ret;
H
Hongze Cheng 已提交
174

175
  if (!pVnode->inUse) {
176
    terrno = TSDB_CODE_VND_NO_AVAIL_BUFPOOL;
H
Hongze Cheng 已提交
177 178
    vError("vgId:%d, not ready to write since %s", TD_VID(pVnode), terrstr());
    return -1;
179 180
  }

181 182 183
  if (version <= pVnode->state.applied) {
    vError("vgId:%d, duplicate write request. version: %" PRId64 ", applied: %" PRId64 "", TD_VID(pVnode), version,
           pVnode->state.applied);
184
    terrno = TSDB_CODE_VND_DUP_REQUEST;
185 186 187
    return -1;
  }

188
  vDebug("vgId:%d, start to process write request %s, index:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
189
         version);
H
Hongze Cheng 已提交
190

191 192
  ASSERT(pVnode->state.applyTerm <= pMsg->info.conn.applyTerm);
  ASSERT(pVnode->state.applied + 1 == version);
193

H
Hongze Cheng 已提交
194
  pVnode->state.applied = version;
H
Hongze Cheng 已提交
195
  pVnode->state.applyTerm = pMsg->info.conn.applyTerm;
H
Hongze Cheng 已提交
196

197 198
  if (!syncUtilUserCommit(pMsg->msgType)) goto _exit;

L
Liu Jicong 已提交
199
  if (pMsg->msgType == TDMT_VND_STREAM_RECOVER_BLOCKING_STAGE || pMsg->msgType == TDMT_STREAM_TASK_CHECK_RSP) {
L
Liu Jicong 已提交
200 201 202
    if (tqCheckLogInWal(pVnode->pTq, version)) return 0;
  }

H
Hongze Cheng 已提交
203 204 205
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
B
Benguang Zhao 已提交
206
  bool needCommit = false;
H
Hongze Cheng 已提交
207 208

  switch (pMsg->msgType) {
H
Hongze Cheng 已提交
209
    /* META */
H
Hongze Cheng 已提交
210
    case TDMT_VND_CREATE_STB:
H
Hongze Cheng 已提交
211
      if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
212
      break;
H
Hongze Cheng 已提交
213
    case TDMT_VND_ALTER_STB:
H
Hongze Cheng 已提交
214
      if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
215
      break;
H
Hongze Cheng 已提交
216
    case TDMT_VND_DROP_STB:
H
Hongze Cheng 已提交
217
      if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
218
      break;
H
Hongze Cheng 已提交
219
    case TDMT_VND_CREATE_TABLE:
H
Hongze Cheng 已提交
220
      if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
221 222
      break;
    case TDMT_VND_ALTER_TABLE:
H
Hongze Cheng 已提交
223
      if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
224
      break;
H
Hongze Cheng 已提交
225
    case TDMT_VND_DROP_TABLE:
H
Hongze Cheng 已提交
226
      if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
227
      break;
228
    case TDMT_VND_DROP_TTL_TABLE:
229
      if (vnodeProcessDropTtlTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
230
      break;
S
Shengliang Guan 已提交
231 232 233 234
    case TDMT_VND_TRIM:
      if (vnodeProcessTrimReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
      break;
    case TDMT_VND_CREATE_SMA:
235
      if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
S
Shengliang Guan 已提交
236
      break;
H
Hongze Cheng 已提交
237
    /* TSDB */
H
Hongze Cheng 已提交
238
    case TDMT_VND_SUBMIT:
H
Hongze Cheng 已提交
239
      if (vnodeProcessSubmitReq(pVnode, version, pMsg->pCont, pMsg->contLen, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
240
      break;
D
dapan1121 已提交
241
    case TDMT_VND_DELETE:
H
Hongze Cheng 已提交
242
      if (vnodeProcessDeleteReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
D
dapan1121 已提交
243
      break;
244 245 246
    case TDMT_VND_BATCH_DEL:
      if (vnodeProcessBatchDeleteReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
      break;
H
Hongze Cheng 已提交
247
    /* TQ */
L
Liu Jicong 已提交
248
    case TDMT_VND_TMQ_SUBSCRIBE:
249
      if (tqProcessSubscribeReq(pVnode->pTq, version, pReq, len) < 0) {
250
        goto _err;
L
Liu Jicong 已提交
251 252
      }
      break;
L
Liu Jicong 已提交
253 254
    case TDMT_VND_TMQ_DELETE_SUB:
      if (tqProcessDeleteSubReq(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
255 256 257
        goto _err;
      }
      break;
L
Liu Jicong 已提交
258
    case TDMT_VND_TMQ_COMMIT_OFFSET:
259
      if (tqProcessOffsetCommitReq(pVnode->pTq, version, pReq, pMsg->contLen - sizeof(SMsgHead)) < 0) {
260
        goto _err;
L
Liu Jicong 已提交
261 262
      }
      break;
L
Liu Jicong 已提交
263
    case TDMT_VND_TMQ_ADD_CHECKINFO:
264
      if (tqProcessAddCheckInfoReq(pVnode->pTq, version, pReq, len) < 0) {
265 266 267
        goto _err;
      }
      break;
L
Liu Jicong 已提交
268
    case TDMT_VND_TMQ_DEL_CHECKINFO:
269
      if (tqProcessDelCheckInfoReq(pVnode->pTq, version, pReq, len) < 0) {
L
Liu Jicong 已提交
270 271 272
        goto _err;
      }
      break;
273
    case TDMT_STREAM_TASK_DEPLOY: {
274
      if (tqProcessTaskDeployReq(pVnode->pTq, version, pReq, len) < 0) {
275
        goto _err;
H
Hongze Cheng 已提交
276 277
      }
    } break;
L
Liu Jicong 已提交
278
    case TDMT_STREAM_TASK_DROP: {
279
      if (tqProcessTaskDropReq(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
L
Liu Jicong 已提交
280 281 282
        goto _err;
      }
    } break;
L
Liu Jicong 已提交
283
    case TDMT_VND_STREAM_RECOVER_BLOCKING_STAGE: {
L
Liu Jicong 已提交
284 285 286 287
      if (tqProcessTaskRecover2Req(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
        goto _err;
      }
    } break;
288 289 290 291 292
    case TDMT_STREAM_TASK_CHECK_RSP: {
      if (tqProcessStreamTaskCheckRsp(pVnode->pTq, version, pReq, len) < 0) {
        goto _err;
      }
    } break;
293 294 295
    case TDMT_VND_ALTER_CONFIRM:
      vnodeProcessAlterConfirmReq(pVnode, version, pReq, len, pRsp);
      break;
S
Shengliang Guan 已提交
296
    case TDMT_VND_ALTER_HASHRANGE:
297
      vnodeProcessAlterHashRangeReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
298
      break;
299
    case TDMT_VND_ALTER_CONFIG:
300
      vnodeProcessAlterConfigReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
301
      break;
H
Hongze Cheng 已提交
302
    case TDMT_VND_COMMIT:
B
Benguang Zhao 已提交
303 304
      needCommit = true;
      break;
H
Hongze Cheng 已提交
305
    default:
L
Liu Jicong 已提交
306 307
      vError("vgId:%d, unprocessed msg, %d", TD_VID(pVnode), pMsg->msgType);
      return -1;
H
Hongze Cheng 已提交
308 309
  }

S
Shengliang Guan 已提交
310 311
  vTrace("vgId:%d, process %s request, code:0x%x index:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType), pRsp->code,
         version);
H
Hongze Cheng 已提交
312

313 314
  walApplyVer(pVnode->pWal, version);

L
Liu Jicong 已提交
315
  vInfo("vgId:%d, push msg begin", pVnode->config.vgId);
316
  if (tqPushMsg(pVnode->pTq, pMsg->pCont, pMsg->contLen, pMsg->msgType, version) < 0) {
S
Shengliang Guan 已提交
317
    vError("vgId:%d, failed to push msg to TQ since %s", TD_VID(pVnode), tstrerror(terrno));
L
Liu Jicong 已提交
318
    vInfo("vgId:%d, push msg end", pVnode->config.vgId);
319 320
    return -1;
  }
L
Liu Jicong 已提交
321
  vInfo("vgId:%d, push msg end", pVnode->config.vgId);
322

H
Hongze Cheng 已提交
323
  // commit if need
B
Benguang Zhao 已提交
324
  if (needCommit) {
S
Shengliang Guan 已提交
325
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
326
    vnodeAsyncCommit(pVnode);
H
Hongze Cheng 已提交
327 328

    // start a new one
329
    if (vnodeBegin(pVnode) < 0) {
H
Hongze Cheng 已提交
330 331
      vError("vgId:%d, failed to begin vnode since %s.", TD_VID(pVnode), tstrerror(terrno));
      goto _err;
332
    }
H
Hongze Cheng 已提交
333 334
  }

H
Hongze Cheng 已提交
335
_exit:
H
Hongze Cheng 已提交
336
  return 0;
H
Hongze Cheng 已提交
337 338

_err:
S
Shengliang Guan 已提交
339
  vError("vgId:%d, process %s request failed since %s, version:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
340 341
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
342 343
}

344
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
345
  if (TDMT_SCH_QUERY != pMsg->msgType && TDMT_SCH_MERGE_QUERY != pMsg->msgType) {
D
dapan1121 已提交
346 347 348
    return 0;
  }

349
  return qWorkerPreprocessQueryMsg(pVnode->pQuery, pMsg, TDMT_SCH_QUERY == pMsg->msgType);
D
dapan1121 已提交
350 351 352
}

int32_t vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
353
  vTrace("message in vnode query queue is processing");
354 355
  // if ((pMsg->msgType == TDMT_SCH_QUERY) && !vnodeIsLeader(pVnode)) {
  if ((pMsg->msgType == TDMT_SCH_QUERY) && !syncIsReadyForRead(pVnode->sync)) {
356
    vnodeRedirectRpcMsg(pVnode, pMsg, terrno);
357 358 359
    return 0;
  }

360
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
361
  switch (pMsg->msgType) {
D
dapan1121 已提交
362
    case TDMT_SCH_QUERY:
D
dapan1121 已提交
363
    case TDMT_SCH_MERGE_QUERY:
D
dapan1121 已提交
364
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
365
    case TDMT_SCH_QUERY_CONTINUE:
D
dapan1121 已提交
366
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
367 368
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
369
      return TSDB_CODE_APP_ERROR;
H
Hongze Cheng 已提交
370 371 372
  }
}

373
int32_t vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
S
Shengliang Guan 已提交
374
  vTrace("vgId:%d, msg:%p in fetch queue is processing", pVnode->config.vgId, pMsg);
L
Liu Jicong 已提交
375 376
  if ((pMsg->msgType == TDMT_SCH_FETCH || pMsg->msgType == TDMT_VND_TABLE_META || pMsg->msgType == TDMT_VND_TABLE_CFG ||
       pMsg->msgType == TDMT_VND_BATCH_META) &&
377 378
      !syncIsReadyForRead(pVnode->sync)) {
    //      !vnodeIsLeader(pVnode)) {
379
    vnodeRedirectRpcMsg(pVnode, pMsg, terrno);
380 381 382
    return 0;
  }

L
Liu Jicong 已提交
383
  if (pMsg->msgType == TDMT_VND_TMQ_CONSUME && !pVnode->restored) {
384
    vnodeRedirectRpcMsg(pVnode, pMsg, TSDB_CODE_SYN_RESTORING);
L
Liu Jicong 已提交
385 386 387
    return 0;
  }

H
Hongze Cheng 已提交
388
  switch (pMsg->msgType) {
D
dapan1121 已提交
389
    case TDMT_SCH_FETCH:
D
dapan1121 已提交
390
    case TDMT_SCH_MERGE_FETCH:
D
dapan1121 已提交
391
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
392
    case TDMT_SCH_FETCH_RSP:
D
dapan1121 已提交
393
      return qWorkerProcessRspMsg(pVnode, pVnode->pQuery, pMsg, 0);
L
Liu Jicong 已提交
394 395
    // case TDMT_SCH_CANCEL_TASK:
    //   return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
396
    case TDMT_SCH_DROP_TASK:
D
dapan1121 已提交
397
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
398
    case TDMT_SCH_QUERY_HEARTBEAT:
D
dapan1121 已提交
399
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
400
    case TDMT_VND_TABLE_META:
D
dapan1121 已提交
401
      return vnodeGetTableMeta(pVnode, pMsg, true);
D
dapan1121 已提交
402
    case TDMT_VND_TABLE_CFG:
D
dapan1121 已提交
403 404 405
      return vnodeGetTableCfg(pVnode, pMsg, true);
    case TDMT_VND_BATCH_META:
      return vnodeGetBatchMeta(pVnode, pMsg);
L
Liu Jicong 已提交
406
    case TDMT_VND_TMQ_CONSUME:
L
Liu Jicong 已提交
407
      return tqProcessPollReq(pVnode->pTq, pMsg);
408 409
    case TDMT_STREAM_TASK_RUN:
      return tqProcessTaskRunReq(pVnode->pTq, pMsg);
410
#if 1
411
    case TDMT_STREAM_TASK_DISPATCH:
L
Liu Jicong 已提交
412
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg, true);
L
Liu Jicong 已提交
413
#endif
414 415
    case TDMT_STREAM_TASK_CHECK:
      return tqProcessStreamTaskCheckReq(pVnode->pTq, pMsg);
416
    case TDMT_STREAM_TASK_DISPATCH_RSP:
L
Liu Jicong 已提交
417
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
418 419
    case TDMT_STREAM_RETRIEVE:
      return tqProcessTaskRetrieveReq(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
420 421
    case TDMT_STREAM_RETRIEVE_RSP:
      return tqProcessTaskRetrieveRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
422
    case TDMT_VND_STREAM_RECOVER_NONBLOCKING_STAGE:
L
Liu Jicong 已提交
423 424 425 426 427
      return tqProcessTaskRecover1Req(pVnode->pTq, pMsg);
    case TDMT_STREAM_RECOVER_FINISH:
      return tqProcessTaskRecoverFinishReq(pVnode->pTq, pMsg);
    case TDMT_STREAM_RECOVER_FINISH_RSP:
      return tqProcessTaskRecoverFinishRsp(pVnode->pTq, pMsg);
H
Hongze Cheng 已提交
428 429
    default:
      vError("unknown msg type:%d in fetch queue", pMsg->msgType);
430
      return TSDB_CODE_APP_ERROR;
H
Hongze Cheng 已提交
431 432 433 434 435 436
  }
}

// TODO: remove the function
void smaHandleRes(void *pVnode, int64_t smaId, const SArray *data) {
  // TODO
C
Cary Xu 已提交
437
  // blockDebugShowDataBlocks(data, __func__);
438
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
439 440
}

D
dapan1121 已提交
441
void vnodeUpdateMetaRsp(SVnode *pVnode, STableMetaRsp *pMetaRsp) {
442 443 444
  if (NULL == pMetaRsp) {
    return;
  }
L
Liu Jicong 已提交
445

D
dapan1121 已提交
446 447 448 449 450 451
  strcpy(pMetaRsp->dbFName, pVnode->config.dbname);
  pMetaRsp->dbId = pVnode->config.dbId;
  pMetaRsp->vgId = TD_VID(pVnode);
  pMetaRsp->precision = pVnode->config.tsdbCfg.precision;
}

S
Shengliang Guan 已提交
452
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
453
  int32_t     code = 0;
S
Shengliang Guan 已提交
454 455
  SVTrimDbReq trimReq = {0};

H
Hongze Cheng 已提交
456 457 458 459
  // decode
  if (tDeserializeSVTrimDbReq(pReq, len, &trimReq) != 0) {
    code = TSDB_CODE_INVALID_MSG;
    goto _exit;
S
Shengliang Guan 已提交
460 461
  }

C
Cary Xu 已提交
462 463
  vInfo("vgId:%d, trim vnode request will be processed, time:%d", pVnode->config.vgId, trimReq.timestamp);

H
Hongze Cheng 已提交
464 465 466 467
  // process
  code = tsdbDoRetention(pVnode->pTsdb, trimReq.timestamp);
  if (code) goto _exit;

C
Cary Xu 已提交
468 469 470
  code = smaDoRetention(pVnode->pSma, trimReq.timestamp);
  if (code) goto _exit;

H
Hongze Cheng 已提交
471 472
_exit:
  return code;
S
Shengliang Guan 已提交
473 474
}

L
Liu Jicong 已提交
475
static int32_t vnodeProcessDropTtlTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
476 477 478
  SArray *tbUids = taosArrayInit(8, sizeof(int64_t));
  if (tbUids == NULL) return TSDB_CODE_OUT_OF_MEMORY;

S
Shengliang Guan 已提交
479 480 481 482 483 484
  SVDropTtlTableReq ttlReq = {0};
  if (tDeserializeSVDropTtlTableReq(pReq, len, &ttlReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto end;
  }

485
  vDebug("vgId:%d, drop ttl table req will be processed, time:%d", pVnode->config.vgId, ttlReq.timestamp);
S
Shengliang Guan 已提交
486
  int32_t ret = metaTtlDropTable(pVnode->pMeta, ttlReq.timestamp, tbUids);
L
Liu Jicong 已提交
487
  if (ret != 0) {
488 489
    goto end;
  }
L
Liu Jicong 已提交
490
  if (taosArrayGetSize(tbUids) > 0) {
wmmhello's avatar
wmmhello 已提交
491 492
    tqUpdateTbUidList(pVnode->pTq, tbUids, false);
  }
493 494 495 496 497 498

end:
  taosArrayDestroy(tbUids);
  return ret;
}

499
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
500
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
501
  SDecoder       coder;
H
Hongze Cheng 已提交
502

H
Hongze Cheng 已提交
503 504 505 506 507 508
  pRsp->msgType = TDMT_VND_CREATE_STB_RSP;
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;

  // decode and process req
H
Hongze Cheng 已提交
509
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
510 511

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
512 513
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
514 515
  }

H
Hongze Cheng 已提交
516
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
517 518
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
519 520
  }

521
  if (tdProcessRSmaCreate(pVnode->pSma, &req) < 0) {
522 523 524
    pRsp->code = terrno;
    goto _err;
  }
C
Cary Xu 已提交
525

H
Hongze Cheng 已提交
526
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
527
  return 0;
H
Hongze Cheng 已提交
528 529

_err:
H
Hongze Cheng 已提交
530
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
531
  return -1;
H
Hongze Cheng 已提交
532 533
}

534
static int32_t vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
535
  SDecoder           decoder = {0};
536
  SEncoder           encoder = {0};
537
  int32_t            rcode = 0;
H
Hongze Cheng 已提交
538 539
  SVCreateTbBatchReq req = {0};
  SVCreateTbReq     *pCreateReq;
H
Hongze Cheng 已提交
540 541 542
  SVCreateTbBatchRsp rsp = {0};
  SVCreateTbRsp      cRsp = {0};
  char               tbName[TSDB_TABLE_FNAME_LEN];
C
Cary Xu 已提交
543
  STbUidStore       *pStore = NULL;
544
  SArray            *tbUids = NULL;
H
Hongze Cheng 已提交
545 546

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
547 548 549
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
550

H
Hongze Cheng 已提交
551
  // decode
H
Hongze Cheng 已提交
552 553
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
554 555 556 557
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
558

H
Hongze Cheng 已提交
559
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
560 561
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
562 563 564 565 566
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
567
  // loop to create table
568
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
569
    pCreateReq = req.pReqs + iReq;
D
dapan1121 已提交
570
    memset(&cRsp, 0, sizeof(cRsp));
H
Hongze Cheng 已提交
571

C
Cary Xu 已提交
572 573 574 575
    if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
      rcode = -1;
      goto _exit;
    }
L
Liu Jicong 已提交
576

wafwerar's avatar
wafwerar 已提交
577 578 579 580 581
    if ((terrno = grantCheck(TSDB_GRANT_TABLE)) < 0) {
      rcode = -1;
      goto _exit;
    }

H
Hongze Cheng 已提交
582 583 584 585 586 587 588 589 590
    // validate hash
    sprintf(tbName, "%s.%s", pVnode->config.dbname, pCreateReq->name);
    if (vnodeValidateTableHash(pVnode, tbName) < 0) {
      cRsp.code = TSDB_CODE_VND_HASH_MISMATCH;
      taosArrayPush(rsp.pArray, &cRsp);
      continue;
    }

    // do create table
591
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq, &cRsp.pMeta) < 0) {
H
Hongze Cheng 已提交
592 593 594 595 596
      if (pCreateReq->flags & TD_CREATE_IF_NOT_EXISTS && terrno == TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
        cRsp.code = TSDB_CODE_SUCCESS;
      } else {
        cRsp.code = terrno;
      }
H
Hongze Cheng 已提交
597
    } else {
H
Hongze Cheng 已提交
598
      cRsp.code = TSDB_CODE_SUCCESS;
599
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
600
      taosArrayPush(tbUids, &pCreateReq->uid);
L
Liu Jicong 已提交
601
      vnodeUpdateMetaRsp(pVnode, cRsp.pMeta);
H
Hongze Cheng 已提交
602
    }
H
Hongze Cheng 已提交
603 604

    taosArrayPush(rsp.pArray, &cRsp);
H
Hongze Cheng 已提交
605 606
  }

H
Haojun Liao 已提交
607
  vDebug("vgId:%d, add %d new created tables into query table list", TD_VID(pVnode), (int32_t)taosArrayGetSize(tbUids));
608
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
609
  if (tdUpdateTbUidList(pVnode->pSma, pStore, true) < 0) {
C
Cary Xu 已提交
610 611
    goto _exit;
  }
612
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
613

H
Hongze Cheng 已提交
614
  // prepare rsp
615
  int32_t ret = 0;
wafwerar's avatar
wafwerar 已提交
616
  tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
H
Hongze Cheng 已提交
617 618 619 620 621 622
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  if (pRsp->pCont == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    rcode = -1;
    goto _exit;
  }
H
Hongze Cheng 已提交
623 624
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
H
Hongze Cheng 已提交
625

H
Hongze Cheng 已提交
626
_exit:
wmmhello's avatar
wmmhello 已提交
627 628
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
    pCreateReq = req.pReqs + iReq;
629
    taosMemoryFree(pCreateReq->comment);
wmmhello's avatar
wmmhello 已提交
630 631
    taosArrayDestroy(pCreateReq->ctb.tagName);
  }
632
  taosArrayDestroyEx(rsp.pArray, tFreeSVCreateTbRsp);
633
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
634 635
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
636
  return rcode;
H
Hongze Cheng 已提交
637 638
}

639
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
  SVCreateStbReq req = {0};
  SDecoder       dc = {0};

  pRsp->msgType = TDMT_VND_ALTER_STB_RSP;
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;

  tDecoderInit(&dc, pReq, len);

  // decode req
  if (tDecodeSVCreateStbReq(&dc, &req) < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    tDecoderClear(&dc);
    return -1;
H
Hongze Cheng 已提交
655
  }
H
Hongze Cheng 已提交
656 657 658 659 660 661 662 663 664

  if (metaAlterSTable(pVnode->pMeta, version, &req) < 0) {
    pRsp->code = terrno;
    tDecoderClear(&dc);
    return -1;
  }

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
665 666 667
  return 0;
}

668
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
669
  SVDropStbReq req = {0};
670
  int32_t      rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
671
  SDecoder     decoder = {0};
672
  SArray      *tbUidList = NULL;
H
Hongze Cheng 已提交
673 674 675 676 677 678

  pRsp->msgType = TDMT_VND_CREATE_STB_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;

  // decode request
H
Hongze Cheng 已提交
679 680
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
681 682 683 684 685
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
686 687 688 689 690 691 692 693
  tbUidList = taosArrayInit(8, sizeof(int64_t));
  if (tbUidList == NULL) goto _exit;
  if (metaDropSTable(pVnode->pMeta, version, &req, tbUidList) < 0) {
    rcode = terrno;
    goto _exit;
  }

  if (tqUpdateTbUidList(pVnode->pTq, tbUidList, false) < 0) {
694 695 696
    rcode = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
697

698 699 700 701 702
  if (tdProcessRSmaDrop(pVnode->pSma, &req) < 0) {
    rcode = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
703 704
  // return rsp
_exit:
705
  if (tbUidList) taosArrayDestroy(tbUidList);
H
Hongze Cheng 已提交
706
  pRsp->code = rcode;
H
Hongze Cheng 已提交
707
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
708 709 710
  return 0;
}

711
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
D
dapan1121 已提交
712 713 714
  SVAlterTbReq  vAlterTbReq = {0};
  SVAlterTbRsp  vAlterTbRsp = {0};
  SDecoder      dc = {0};
715 716
  int32_t       rcode = 0;
  int32_t       ret;
D
dapan1121 已提交
717 718
  SEncoder      ec = {0};
  STableMetaRsp vMetaRsp = {0};
H
Hongze Cheng 已提交
719 720 721 722 723 724 725 726 727 728

  pRsp->msgType = TDMT_VND_ALTER_TABLE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

  tDecoderInit(&dc, pReq, len);

  // decode
  if (tDecodeSVAlterTbReq(&dc, &vAlterTbReq) < 0) {
H
Hongze Cheng 已提交
729
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
730
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
731 732
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
733 734 735
  }

  // process
D
dapan1121 已提交
736
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq, &vMetaRsp) < 0) {
wmmhello's avatar
wmmhello 已提交
737
    vAlterTbRsp.code = terrno;
H
Hongze Cheng 已提交
738
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
739 740
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
741 742
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
743

D
dapan1121 已提交
744 745 746 747 748
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
749 750 751 752 753 754
_exit:
  tEncodeSize(tEncodeSVAlterTbRsp, &vAlterTbRsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVAlterTbRsp(&ec, &vAlterTbRsp);
  tEncoderClear(&ec);
M
Minglei Jin 已提交
755 756 757
  if (vMetaRsp.pSchemas) {
    taosMemoryFree(vMetaRsp.pSchemas);
  }
H
Hongze Cheng 已提交
758 759 760
  return 0;
}

761
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
762 763
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
764
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
765
  SEncoder         encoder = {0};
766
  int32_t          ret;
767
  SArray          *tbUids = NULL;
768
  STbUidStore     *pStore = NULL;
H
Hongze Cheng 已提交
769

H
Hongze Cheng 已提交
770
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
771 772 773
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
774 775

  // decode req
H
Hongze Cheng 已提交
776 777
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
778 779 780 781 782
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
783 784

  // process req
785
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
786
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
787 788
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

789
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
790 791
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
792
    tb_uid_t     tbUid = 0;
H
Hongze Cheng 已提交
793

H
Hongze Cheng 已提交
794
    /* code */
795
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids, &tbUid);
H
Hongze Cheng 已提交
796
    if (ret < 0) {
797
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_TDB_TABLE_NOT_EXIST) {
H
Hongze Cheng 已提交
798 799 800 801
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
802
    } else {
H
Hongze Cheng 已提交
803
      dropTbRsp.code = TSDB_CODE_SUCCESS;
804
      if (tbUid > 0) tdFetchTbUidList(pVnode->pSma, &pStore, pDropTbReq->suid, tbUid);
H
Hongze Cheng 已提交
805 806 807 808 809
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

810
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);
811
  tdUpdateTbUidList(pVnode->pSma, pStore, false);
812

H
Hongze Cheng 已提交
813
_exit:
814
  taosArrayDestroy(tbUids);
815
  tdUidStoreFree(pStore);
H
Hongze Cheng 已提交
816
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
817 818 819 820 821
  tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVDropTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
822
  taosArrayDestroy(rsp.pArray);
H
Hongze Cheng 已提交
823 824 825
  return 0;
}

826 827
static int32_t vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter,
                                              const char *tags) {
D
dapan 已提交
828 829 830 831
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
832
  int32_t        rv = -1;
D
dapan 已提交
833 834 835

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
836 837 838 839 840

  pSchema = metaGetTbTSchema(pMeta, msgIter->suid, TD_ROW_SVER(blkIter.row), 1);  // TODO: use the real schema
  if (pSchema) {
    suid = msgIter->suid;
    rv = TD_ROW_SVER(blkIter.row);
D
dapan 已提交
841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856
  }
  if (!pSchema) {
    printf("%s:%d no valid schema\n", tags, __LINE__);
    return -1;
  }
  char __tags[128] = {0};
  snprintf(__tags, 128, "%s: uid %" PRIi64 " ", tags, msgIter->uid);
  while ((row = tGetSubmitBlkNext(&blkIter))) {
    tdSRowPrint(row, pSchema, __tags);
  }

  taosMemoryFreeClear(pSchema);

  return TSDB_CODE_SUCCESS;
}

857
static int32_t vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
858
  ASSERT(pMsg != NULL);
C
Cary Xu 已提交
859 860 861 862 863 864 865 866
  SSubmitMsgIter msgIter = {0};
  SMeta         *pMeta = pVnode->pMeta;
  SSubmitBlk    *pBlock = NULL;

  if (tInitSubmitMsgIter(pMsg, &msgIter) < 0) return -1;
  while (true) {
    if (tGetSubmitMsgNext(&msgIter, &pBlock) < 0) return -1;
    if (pBlock == NULL) break;
H
Hongze Cheng 已提交
867

D
dapan 已提交
868 869
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
870 871 872 873

  return 0;
}

874
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
875
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
876
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
877
  SSubmitMsgIter msgIter = {0};
C
Cary Xu 已提交
878
  SSubmitBlk    *pBlock = NULL;
H
Hongze Cheng 已提交
879
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
880
  SDecoder       decoder = {0};
C
Cary Xu 已提交
881
  int32_t        nRows = 0;
H
Hongze Cheng 已提交
882 883
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
884
  SArray        *newTbUids = NULL;
885
  SVStatis       statis = {0};
D
dapan1121 已提交
886
  bool           tbCreated = false;
887
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
888

H
Hongze Cheng 已提交
889 890
  vDebug("vgId:%d %s start, version:%" PRId64 ", len:%d", TD_VID(pVnode), __func__, version);

H
Hongze Cheng 已提交
891
  pRsp->code = 0;
C
Cary Xu 已提交
892
  pSubmitReq->version = version;
893
  statis.nBatchInsert = 1;
C
Cary Xu 已提交
894

C
Cary Xu 已提交
895 896 897 898
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
899 900 901 902 903
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
904
  // handle the request
H
Hongze Cheng 已提交
905 906 907
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
908 909
  }

C
Cary Xu 已提交
910 911
  submitRsp.pArray = taosArrayInit(msgIter.numOfBlocks, sizeof(SSubmitBlkRsp));
  newTbUids = taosArrayInit(msgIter.numOfBlocks, sizeof(int64_t));
C
Cary Xu 已提交
912
  if (!submitRsp.pArray || !newTbUids) {
913 914 915 916
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
917
  for (;;) {
H
Hongze Cheng 已提交
918 919 920
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
921
    SSubmitBlkRsp submitBlkRsp = {0};
D
dapan1121 已提交
922
    tbCreated = false;
H
Hongze Cheng 已提交
923

H
Hongze Cheng 已提交
924 925
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
926 927
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
928
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
929
        tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
930
        taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
931 932 933
        goto _exit;
      }

C
Cary Xu 已提交
934 935 936 937 938 939 940
      if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
        pRsp->code = terrno;
        tDecoderClear(&decoder);
        taosArrayDestroy(createTbReq.ctb.tagName);
        goto _exit;
      }

wafwerar's avatar
wafwerar 已提交
941 942 943 944 945 946 947
      if ((terrno = grantCheck(TSDB_GRANT_TABLE)) < 0) {
        pRsp->code = terrno;
        tDecoderClear(&decoder);
        taosArrayDestroy(createTbReq.ctb.tagName);
        goto _exit;
      }

948
      if (metaCreateTable(pVnode->pMeta, version, &createTbReq, &submitBlkRsp.pMeta) < 0) {
H
Hongze Cheng 已提交
949
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
950
          submitBlkRsp.code = terrno;
C
Cary Xu 已提交
951
          pRsp->code = terrno;
H
Hongze Cheng 已提交
952
          tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
953
          taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
954 955
          goto _exit;
        }
956 957 958 959
      } else {
        if (NULL != submitBlkRsp.pMeta) {
          vnodeUpdateMetaRsp(pVnode, submitBlkRsp.pMeta);
        }
960 961

        taosArrayPush(newTbUids, &createTbReq.uid);
H
Hongze Cheng 已提交
962

D
dapan1121 已提交
963 964 965 966 967
        submitBlkRsp.uid = createTbReq.uid;
        submitBlkRsp.tblFName = taosMemoryMalloc(strlen(pVnode->config.dbname) + strlen(createTbReq.name) + 2);
        sprintf(submitBlkRsp.tblFName, "%s.%s", pVnode->config.dbname, createTbReq.name);
        tbCreated = true;
      }
968

H
Hongze Cheng 已提交
969 970 971 972 973 974 975
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

wmmhello's avatar
wmmhello 已提交
976
#ifdef TD_DEBUG_PRINT_ROW
D
dapan 已提交
977
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
wmmhello's avatar
wmmhello 已提交
978
#endif
H
Hongze Cheng 已提交
979
      tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
980
      taosArrayDestroy(createTbReq.ctb.tagName);
981
    }
H
Hongze Cheng 已提交
982

H
Hongze Cheng 已提交
983
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
984
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
985 986
    }

H
Hongze Cheng 已提交
987 988
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
D
dapan1121 已提交
989 990 991
    if (tbCreated || submitBlkRsp.code) {
      taosArrayPush(submitRsp.pArray, &submitBlkRsp);
    }
H
Hongze Cheng 已提交
992
  }
H
Haojun Liao 已提交
993 994

  if (taosArrayGetSize(newTbUids) > 0) {
L
Liu Jicong 已提交
995 996
    vDebug("vgId:%d, add %d table into query table list in handling submit", TD_VID(pVnode),
           (int32_t)taosArrayGetSize(newTbUids));
H
Haojun Liao 已提交
997 998
  }

999
  tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
1000

H
Hongze Cheng 已提交
1001
_exit:
1002
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1003 1004 1005 1006 1007 1008 1009
  tEncodeSize(tEncodeSSubmitRsp, &submitRsp, tsize, ret);
  pRsp->pCont = rpcMallocCont(tsize);
  pRsp->contLen = tsize;
  tEncoderInit(&encoder, pRsp->pCont, tsize);
  tEncodeSSubmitRsp(&encoder, &submitRsp);
  tEncoderClear(&encoder);

1010
  taosArrayDestroyEx(submitRsp.pArray, tFreeSSubmitBlkRsp);
H
Hongze Cheng 已提交
1011

1012
  // TODO: the partial success scenario and the error case
H
Hongze Cheng 已提交
1013 1014
  // => If partial success, extract the success submitted rows and reconstruct a new submit msg, and push to level
  // 1/level 2.
1015
  // TODO: refactor
C
Cary Xu 已提交
1016
  if ((terrno == TSDB_CODE_SUCCESS) && (pRsp->code == TSDB_CODE_SUCCESS)) {
1017
    statis.nBatchInsertSuccess = 1;
L
Liu Jicong 已提交
1018
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_INPUT__DATA_SUBMIT);
1019
  }
C
Cary Xu 已提交
1020

1021
  // N.B. not strict as the following procedure is not atomic
1022 1023 1024
  atomic_add_fetch_64(&pVnode->statis.nInsert, submitRsp.numOfRows);
  atomic_add_fetch_64(&pVnode->statis.nInsertSuccess, submitRsp.affectedRows);
  atomic_add_fetch_64(&pVnode->statis.nBatchInsert, statis.nBatchInsert);
1025
  atomic_add_fetch_64(&pVnode->statis.nBatchInsertSuccess, statis.nBatchInsertSuccess);
1026

H
Hongze Cheng 已提交
1027
  vDebug("vgId:%d %s done, index:%" PRId64, TD_VID(pVnode), __func__, version);
H
Hongze Cheng 已提交
1028
  return 0;
L
Liu Jicong 已提交
1029
}
1030

1031
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
1032
  SVCreateTSmaReq req = {0};
C
Cary Xu 已提交
1033
  SDecoder        coder = {0};
1034

C
Cary Xu 已提交
1035 1036 1037 1038 1039 1040
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
1041 1042 1043 1044 1045

  // decode and process req
  tDecoderInit(&coder, pReq, len);

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
1046 1047
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
1048
    goto _err;
1049
  }
C
Cary Xu 已提交
1050

C
Cary Xu 已提交
1051
  if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
C
Cary Xu 已提交
1052
    if (pRsp) pRsp->code = terrno;
1053
    goto _err;
1054
  }
C
Cary Xu 已提交
1055

1056
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
1057
  vDebug("vgId:%d, success to create tsma %s:%" PRIi64 " version %" PRIi64 " for table %" PRIi64, TD_VID(pVnode),
C
Cary Xu 已提交
1058
         req.indexName, req.indexUid, version, req.tableUid);
H
Hongze Cheng 已提交
1059
  return 0;
1060 1061 1062

_err:
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
1063
  vError("vgId:%d, failed to create tsma %s:%" PRIi64 " version %" PRIi64 "for table %" PRIi64 " since %s",
C
Cary Xu 已提交
1064
         TD_VID(pVnode), req.indexName, req.indexUid, version, req.tableUid, terrstr());
1065
  return -1;
L
Liu Jicong 已提交
1066
}
C
Cary Xu 已提交
1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078

/**
 * @brief specific for smaDstVnode
 *
 * @param pVnode
 * @param pCont
 * @param contLen
 * @return int32_t
 */
int32_t vnodeProcessCreateTSma(SVnode *pVnode, void *pCont, uint32_t contLen) {
  return vnodeProcessCreateTSmaReq(pVnode, 1, pCont, contLen, NULL);
}
1079 1080 1081 1082 1083 1084 1085 1086 1087

static int32_t vnodeProcessAlterConfirmReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  vInfo("vgId:%d, alter replica confim msg is processed", TD_VID(pVnode));
  pRsp->msgType = TDMT_VND_ALTER_CONFIRM_RSP;
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;

  return 0;
1088
}
S
Shengliang Guan 已提交
1089

1090
static int32_t vnodeProcessAlterHashRangeReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
S
Shengliang Guan 已提交
1091 1092
  vInfo("vgId:%d, alter hashrange msg will be processed", TD_VID(pVnode));

1093 1094
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
1095 1096 1097
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
1098
}
H
Hongze Cheng 已提交
1099

1100
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
S
Shengliang Guan 已提交
1101 1102
  bool walChanged = false;
  bool tsdbChanged = false;
1103

S
Shengliang Guan 已提交
1104 1105
  SAlterVnodeConfigReq req = {0};
  if (tDeserializeSAlterVnodeConfigReq(pReq, len, &req) != 0) {
1106 1107 1108 1109
    terrno = TSDB_CODE_INVALID_MSG;
    return TSDB_CODE_INVALID_MSG;
  }

1110
  vInfo("vgId:%d, start to alter vnode config, page:%d pageSize:%d buffer:%d szPage:%d szBuf:%" PRIu64
S
Shengliang Guan 已提交
1111
        " cacheLast:%d cacheLastSize:%d days:%d keep0:%d keep1:%d keep2:%d fsync:%d level:%d",
1112 1113
        TD_VID(pVnode), req.pages, req.pageSize, req.buffer, req.pageSize * 1024, (uint64_t)req.buffer * 1024 * 1024,
        req.cacheLast, req.cacheLastSize, req.daysPerFile, req.daysToKeep0, req.daysToKeep1, req.daysToKeep2,
S
Shengliang Guan 已提交
1114
        req.walFsyncPeriod, req.walLevel);
1115 1116 1117

  if (pVnode->config.cacheLastSize != req.cacheLastSize) {
    pVnode->config.cacheLastSize = req.cacheLastSize;
1118 1119
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
1120

1121
  if (pVnode->config.szBuf != req.buffer * 1024LL * 1024LL) {
S
Shengliang Guan 已提交
1122
    vInfo("vgId:%d, vnode buffer is changed from %" PRId64 " to %" PRId64, TD_VID(pVnode), pVnode->config.szBuf,
1123
          (uint64_t)(req.buffer * 1024LL * 1024LL));
1124
    pVnode->config.szBuf = req.buffer * 1024LL * 1024LL;
H
Hongze Cheng 已提交
1125 1126
  }

1127 1128
  if (pVnode->config.szCache != req.pages) {
    if (metaAlterCache(pVnode->pMeta, req.pages) < 0) {
S
Shengliang Guan 已提交
1129
      vError("vgId:%d, failed to change vnode pages from %d to %d failed since %s", TD_VID(pVnode),
1130
             pVnode->config.szCache, req.pages, tstrerror(errno));
H
Hongze Cheng 已提交
1131 1132
      return errno;
    } else {
S
Shengliang Guan 已提交
1133
      vInfo("vgId:%d, vnode pages is changed from %d to %d", TD_VID(pVnode), pVnode->config.szCache, req.pages);
1134
      pVnode->config.szCache = req.pages;
H
Hongze Cheng 已提交
1135
    }
H
Hongze Cheng 已提交
1136 1137
  }

1138 1139
  if (pVnode->config.cacheLast != req.cacheLast) {
    pVnode->config.cacheLast = req.cacheLast;
1140 1141
  }

1142 1143
  if (pVnode->config.walCfg.fsyncPeriod != req.walFsyncPeriod) {
    pVnode->config.walCfg.fsyncPeriod = req.walFsyncPeriod;
1144 1145 1146 1147

    walChanged = true;
  }

1148 1149
  if (pVnode->config.walCfg.level != req.walLevel) {
    pVnode->config.walCfg.level = req.walLevel;
1150 1151 1152 1153

    walChanged = true;
  }

1154 1155
  if (pVnode->config.tsdbCfg.keep0 != req.daysToKeep0) {
    pVnode->config.tsdbCfg.keep0 = req.daysToKeep0;
1156
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1157
      tsdbChanged = true;
1158 1159 1160
    }
  }

1161 1162
  if (pVnode->config.tsdbCfg.keep1 != req.daysToKeep1) {
    pVnode->config.tsdbCfg.keep1 = req.daysToKeep1;
1163
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1164
      tsdbChanged = true;
1165 1166 1167
    }
  }

1168 1169
  if (pVnode->config.tsdbCfg.keep2 != req.daysToKeep2) {
    pVnode->config.tsdbCfg.keep2 = req.daysToKeep2;
1170
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1171
      tsdbChanged = true;
1172 1173 1174 1175
    }
  }

  if (walChanged) {
M
Minglei Jin 已提交
1176 1177 1178 1179 1180
    walAlter(pVnode->pWal, &pVnode->config.walCfg);
  }

  if (tsdbChanged) {
    tsdbSetKeepCfg(pVnode->pTsdb, &pVnode->config.tsdbCfg);
1181 1182
  }

1183 1184 1185
  return 0;
}

1186 1187 1188 1189 1190 1191
static int32_t vnodeProcessBatchDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  SBatchDeleteReq deleteReq;
  SDecoder        decoder;
  tDecoderInit(&decoder, pReq, len);
  tDecodeSBatchDeleteReq(&decoder, &deleteReq);

1192 1193 1194
  SMetaReader mr = {0};
  metaReaderInit(&mr, pVnode->pMeta, 0);

1195 1196 1197
  int32_t sz = taosArrayGetSize(deleteReq.deleteReqs);
  for (int32_t i = 0; i < sz; i++) {
    SSingleDeleteReq *pOneReq = taosArrayGet(deleteReq.deleteReqs, i);
1198 1199
    char             *name = pOneReq->tbname;
    if (metaGetTableEntryByName(&mr, name) < 0) {
S
Shengliang Guan 已提交
1200
      vDebug("vgId:%d, stream delete msg, skip since no table: %s", pVnode->config.vgId, name);
1201 1202 1203 1204 1205
      continue;
    }

    int64_t uid = mr.me.uid;

L
Liu Jicong 已提交
1206
    int32_t code = tsdbDeleteTableData(pVnode->pTsdb, version, deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1207 1208 1209
    if (code < 0) {
      terrno = code;
      vError("vgId:%d, delete error since %s, suid:%" PRId64 ", uid:%" PRId64 ", start ts:%" PRId64 ", end ts:%" PRId64,
L
Liu Jicong 已提交
1210
             TD_VID(pVnode), terrstr(), deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1211
    }
1212 1213

    tDecoderClear(&mr.coder);
1214
  }
1215
  metaReaderClear(&mr);
1216
  taosArrayDestroy(deleteReq.deleteReqs);
1217 1218 1219
  return 0;
}

H
Hongze Cheng 已提交
1220 1221 1222 1223 1224
static int32_t vnodeProcessDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  int32_t     code = 0;
  SDecoder   *pCoder = &(SDecoder){0};
  SDeleteRes *pRes = &(SDeleteRes){0};

wmmhello's avatar
wmmhello 已提交
1225 1226 1227 1228 1229
  pRsp->msgType = TDMT_VND_DELETE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

H
Hongze Cheng 已提交
1230 1231 1232 1233 1234 1235 1236 1237
  pRes->uidList = taosArrayInit(0, sizeof(tb_uid_t));
  if (pRes->uidList == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }

  tDecoderInit(pCoder, pReq, len);
  tDecodeDeleteRes(pCoder, pRes);
1238
  ASSERT(taosArrayGetSize(pRes->uidList) == 0 || (pRes->skey != 0 && pRes->ekey != 0));
H
Hongze Cheng 已提交
1239 1240 1241 1242 1243 1244 1245 1246 1247

  for (int32_t iUid = 0; iUid < taosArrayGetSize(pRes->uidList); iUid++) {
    code = tsdbDeleteTableData(pVnode->pTsdb, version, pRes->suid, *(uint64_t *)taosArrayGet(pRes->uidList, iUid),
                               pRes->skey, pRes->ekey);
    if (code) goto _err;
  }

  tDecoderClear(pCoder);
  taosArrayDestroy(pRes->uidList);
wmmhello's avatar
wmmhello 已提交
1248 1249

  SVDeleteRsp rsp = {.affectedRows = pRes->affectedRows};
L
Liu Jicong 已提交
1250
  int32_t     ret = 0;
wmmhello's avatar
wmmhello 已提交
1251 1252
  tEncodeSize(tEncodeSVDeleteRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
L
Liu Jicong 已提交
1253
  SEncoder ec = {0};
wmmhello's avatar
wmmhello 已提交
1254 1255 1256
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVDeleteRsp(&ec, &rsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
1257 1258 1259 1260
  return code;

_err:
  return code;
M
Minglei Jin 已提交
1261
}