vnodeSvr.c 44.0 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
        tEndDecode(&dc);
      }

      tEndDecode(&dc);
      tDecoderClear(&dc);
    } break;
    case TDMT_VND_SUBMIT: {
H
Hongze Cheng 已提交
80 81
      int64_t ctime = taosGetTimestampMs();

H
Hongze Cheng 已提交
82 83
      tDecoderInit(&dc, (uint8_t *)pMsg->pCont + sizeof(SMsgHead), pMsg->contLen - sizeof(SMsgHead));
      tStartDecode(&dc);
H
Hongze Cheng 已提交
84

H
Hongze Cheng 已提交
85 86 87
      uint64_t nSubmitTbData;
      if (tDecodeU64v(&dc, &nSubmitTbData) < 0) {
        code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
88 89
        goto _err;
      }
H
Hongze Cheng 已提交
90

H
Hongze Cheng 已提交
91 92 93 94 95
      for (int32_t i = 0; i < nSubmitTbData; i++) {
        if (tStartDecode(&dc) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          goto _err;
        }
H
Hongze Cheng 已提交
96

H
Hongze Cheng 已提交
97 98 99 100 101
        int32_t flags;
        if (tDecodeI32v(&dc, &flags) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          goto _err;
        }
H
Hongze Cheng 已提交
102

H
Hongze Cheng 已提交
103
        if (flags & SUBMIT_REQ_AUTO_CREATE_TABLE) {
H
Hongze Cheng 已提交
104
          // SVCreateTbReq
H
Hongze Cheng 已提交
105 106
          if (tStartDecode(&dc) < 0) {
            code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
107
            goto _err;
H
Hongze Cheng 已提交
108
          }
H
Hongze Cheng 已提交
109

H
Hongze Cheng 已提交
110 111
          if (tDecodeI32v(&dc, NULL) < 0) {
            code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
112
            goto _err;
H
Hongze Cheng 已提交
113
          }
H
Hongze Cheng 已提交
114

H
Hongze Cheng 已提交
115
          char *name = NULL;
H
Hongze Cheng 已提交
116 117
          if (tDecodeCStr(&dc, &name) < 0) {
            code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
118
            goto _err;
H
Hongze Cheng 已提交
119
          }
H
Hongze Cheng 已提交
120

H
Hongze Cheng 已提交
121
          int64_t uid = metaGetTableEntryUidByName(pVnode->pMeta, name);
H
Hongze Cheng 已提交
122 123 124
          if (uid == 0) {
            uid = tGenIdPI64();
          }
H
Hongze Cheng 已提交
125

H
Hongze Cheng 已提交
126
          *(int64_t *)(dc.data + dc.pos) = uid;
H
Hongze Cheng 已提交
127 128 129
          *(int64_t *)(dc.data + dc.pos + 8) = ctime;

          tEndDecode(&dc);
H
Hongze Cheng 已提交
130 131 132 133 134 135 136 137 138

          // SSubmitTbData
          int64_t suid;
          if (tDecodeI64(&dc, &suid) < 0) {
            code = TSDB_CODE_INVALID_MSG;
            goto _err;
          }

          *(int64_t *)(dc.data + dc.pos) = uid;
H
Hongze Cheng 已提交
139
        }
H
Hongze Cheng 已提交
140 141

        tEndDecode(&dc);
H
Hongze Cheng 已提交
142 143
      }

H
Hongze Cheng 已提交
144
      tEndDecode(&dc);
H
Hongze Cheng 已提交
145
      tDecoderClear(&dc);
H
Hongze Cheng 已提交
146
    } break;
H
Hongze Cheng 已提交
147 148 149 150 151 152 153 154 155 156
    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 已提交
157 158 159
      if (code) {
        goto _err;
      }
H
Hongze Cheng 已提交
160 161 162 163 164

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

165 166
      ((SMsgHead *)pCont)->contLen = size + sizeof(SMsgHead);
      ((SMsgHead *)pCont)->vgId = TD_VID(pVnode);
H
Hongze Cheng 已提交
167 168 169 170 171 172 173 174 175 176 177

      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 已提交
178 179 180
    default:
      break;
  }
H
Hongze Cheng 已提交
181

S
Shengliang Guan 已提交
182
  return code;
H
Hongze Cheng 已提交
183 184 185 186

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

189
int32_t vnodeProcessWriteMsg(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
190 191 192 193
  void   *ptr = NULL;
  void   *pReq;
  int32_t len;
  int32_t ret;
H
Hongze Cheng 已提交
194

195
  if (!pVnode->inUse) {
196
    terrno = TSDB_CODE_VND_NO_AVAIL_BUFPOOL;
H
Hongze Cheng 已提交
197 198
    vError("vgId:%d, not ready to write since %s", TD_VID(pVnode), terrstr());
    return -1;
199 200
  }

201 202 203
  if (version <= pVnode->state.applied) {
    vError("vgId:%d, duplicate write request. version: %" PRId64 ", applied: %" PRId64 "", TD_VID(pVnode), version,
           pVnode->state.applied);
204
    terrno = TSDB_CODE_VND_DUP_REQUEST;
205 206 207 208
    pRsp->info.handle = NULL;
    return -1;
  }

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

212 213
  ASSERT(pVnode->state.applyTerm <= pMsg->info.conn.applyTerm);
  ASSERT(pVnode->state.applied + 1 == version);
214

H
Hongze Cheng 已提交
215
  pVnode->state.applied = version;
H
Hongze Cheng 已提交
216
  pVnode->state.applyTerm = pMsg->info.conn.applyTerm;
H
Hongze Cheng 已提交
217

218 219
  if (!syncUtilUserCommit(pMsg->msgType)) goto _exit;

L
Liu Jicong 已提交
220
  if (pMsg->msgType == TDMT_VND_STREAM_RECOVER_BLOCKING_STAGE || pMsg->msgType == TDMT_STREAM_TASK_CHECK_RSP) {
L
Liu Jicong 已提交
221 222 223
    if (tqCheckLogInWal(pVnode->pTq, version)) return 0;
  }

H
Hongze Cheng 已提交
224 225 226
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
227 228

  switch (pMsg->msgType) {
H
Hongze Cheng 已提交
229
    /* META */
H
Hongze Cheng 已提交
230
    case TDMT_VND_CREATE_STB:
H
Hongze Cheng 已提交
231
      if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
232
      break;
H
Hongze Cheng 已提交
233
    case TDMT_VND_ALTER_STB:
H
Hongze Cheng 已提交
234
      if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
235
      break;
H
Hongze Cheng 已提交
236
    case TDMT_VND_DROP_STB:
H
Hongze Cheng 已提交
237
      if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
238
      break;
H
Hongze Cheng 已提交
239
    case TDMT_VND_CREATE_TABLE:
H
Hongze Cheng 已提交
240
      if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
241 242
      break;
    case TDMT_VND_ALTER_TABLE:
H
Hongze Cheng 已提交
243
      if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
244
      break;
H
Hongze Cheng 已提交
245
    case TDMT_VND_DROP_TABLE:
H
Hongze Cheng 已提交
246
      if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
247
      break;
248
    case TDMT_VND_DROP_TTL_TABLE:
249
      if (vnodeProcessDropTtlTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
250
      break;
S
Shengliang Guan 已提交
251 252 253 254
    case TDMT_VND_TRIM:
      if (vnodeProcessTrimReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
      break;
    case TDMT_VND_CREATE_SMA:
255
      if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
S
Shengliang Guan 已提交
256
      break;
H
Hongze Cheng 已提交
257
    /* TSDB */
H
Hongze Cheng 已提交
258
    case TDMT_VND_SUBMIT:
259
      if (vnodeProcessSubmitReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
260
      break;
D
dapan1121 已提交
261
    case TDMT_VND_DELETE:
H
Hongze Cheng 已提交
262
      if (vnodeProcessDeleteReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
D
dapan1121 已提交
263
      break;
264 265 266
    case TDMT_VND_BATCH_DEL:
      if (vnodeProcessBatchDeleteReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
      break;
H
Hongze Cheng 已提交
267
    /* TQ */
L
Liu Jicong 已提交
268
    case TDMT_VND_TMQ_SUBSCRIBE:
269
      if (tqProcessSubscribeReq(pVnode->pTq, version, pReq, len) < 0) {
270
        goto _err;
L
Liu Jicong 已提交
271 272
      }
      break;
L
Liu Jicong 已提交
273 274
    case TDMT_VND_TMQ_DELETE_SUB:
      if (tqProcessDeleteSubReq(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
275 276 277
        goto _err;
      }
      break;
L
Liu Jicong 已提交
278
    case TDMT_VND_TMQ_COMMIT_OFFSET:
279
      if (tqProcessOffsetCommitReq(pVnode->pTq, version, pReq, pMsg->contLen - sizeof(SMsgHead)) < 0) {
280
        goto _err;
L
Liu Jicong 已提交
281 282
      }
      break;
L
Liu Jicong 已提交
283
    case TDMT_VND_TMQ_ADD_CHECKINFO:
284
      if (tqProcessAddCheckInfoReq(pVnode->pTq, version, pReq, len) < 0) {
285 286 287
        goto _err;
      }
      break;
L
Liu Jicong 已提交
288
    case TDMT_VND_TMQ_DEL_CHECKINFO:
289
      if (tqProcessDelCheckInfoReq(pVnode->pTq, version, pReq, len) < 0) {
L
Liu Jicong 已提交
290 291 292
        goto _err;
      }
      break;
293
    case TDMT_STREAM_TASK_DEPLOY: {
294
      if (tqProcessTaskDeployReq(pVnode->pTq, version, pReq, len) < 0) {
295
        goto _err;
H
Hongze Cheng 已提交
296 297
      }
    } break;
L
Liu Jicong 已提交
298
    case TDMT_STREAM_TASK_DROP: {
299
      if (tqProcessTaskDropReq(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
L
Liu Jicong 已提交
300 301 302
        goto _err;
      }
    } break;
L
Liu Jicong 已提交
303
    case TDMT_VND_STREAM_RECOVER_BLOCKING_STAGE: {
L
Liu Jicong 已提交
304 305 306 307
      if (tqProcessTaskRecover2Req(pVnode->pTq, version, pMsg->pCont, pMsg->contLen) < 0) {
        goto _err;
      }
    } break;
308 309 310 311 312
    case TDMT_STREAM_TASK_CHECK_RSP: {
      if (tqProcessStreamTaskCheckRsp(pVnode->pTq, version, pReq, len) < 0) {
        goto _err;
      }
    } break;
313 314 315
    case TDMT_VND_ALTER_CONFIRM:
      vnodeProcessAlterConfirmReq(pVnode, version, pReq, len, pRsp);
      break;
S
Shengliang Guan 已提交
316
    case TDMT_VND_ALTER_HASHRANGE:
317
      vnodeProcessAlterHashRangeReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
318
      break;
319
    case TDMT_VND_ALTER_CONFIG:
320
      vnodeProcessAlterConfigReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
321
      break;
H
Hongze Cheng 已提交
322
    case TDMT_VND_COMMIT:
H
Hongze Cheng 已提交
323 324 325
      vnodeSyncCommit(pVnode);
      vnodeBegin(pVnode);
      goto _exit;
H
Hongze Cheng 已提交
326
    default:
L
Liu Jicong 已提交
327 328
      vError("vgId:%d, unprocessed msg, %d", TD_VID(pVnode), pMsg->msgType);
      return -1;
H
Hongze Cheng 已提交
329 330
  }

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

334 335
  walApplyVer(pVnode->pWal, version);

336
  if (tqPushMsg(pVnode->pTq, pMsg->pCont, pMsg->contLen, pMsg->msgType, version) < 0) {
S
Shengliang Guan 已提交
337
    vError("vgId:%d, failed to push msg to TQ since %s", TD_VID(pVnode), tstrerror(terrno));
338 339 340
    return -1;
  }

H
Hongze Cheng 已提交
341
  // commit if need
H
Hongze Cheng 已提交
342
  if (vnodeShouldCommit(pVnode)) {
S
Shengliang Guan 已提交
343
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
344 345 346 347
    if (vnodeAsyncCommit(pVnode) < 0) {
      vError("vgId:%d, failed to vnode async commit since %s.", TD_VID(pVnode), tstrerror(terrno));
      goto _err;
    }
H
Hongze Cheng 已提交
348 349

    // start a new one
350
    if (vnodeBegin(pVnode) < 0) {
H
Hongze Cheng 已提交
351 352
      vError("vgId:%d, failed to begin vnode since %s.", TD_VID(pVnode), tstrerror(terrno));
      goto _err;
353
    }
H
Hongze Cheng 已提交
354 355
  }

H
Hongze Cheng 已提交
356
_exit:
H
Hongze Cheng 已提交
357
  return 0;
H
Hongze Cheng 已提交
358 359

_err:
S
Shengliang Guan 已提交
360
  vError("vgId:%d, process %s request failed since %s, version:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
361 362
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
363 364
}

365
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
366
  if (TDMT_SCH_QUERY != pMsg->msgType && TDMT_SCH_MERGE_QUERY != pMsg->msgType) {
D
dapan1121 已提交
367 368 369
    return 0;
  }

370
  return qWorkerPreprocessQueryMsg(pVnode->pQuery, pMsg, TDMT_SCH_QUERY == pMsg->msgType);
D
dapan1121 已提交
371 372 373
}

int32_t vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
374
  vTrace("message in vnode query queue is processing");
375 376
  // if ((pMsg->msgType == TDMT_SCH_QUERY) && !vnodeIsLeader(pVnode)) {
  if ((pMsg->msgType == TDMT_SCH_QUERY) && !syncIsReadyForRead(pVnode->sync)) {
377
    vnodeRedirectRpcMsg(pVnode, pMsg, terrno);
378 379 380
    return 0;
  }

381
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
382
  switch (pMsg->msgType) {
D
dapan1121 已提交
383
    case TDMT_SCH_QUERY:
D
dapan1121 已提交
384
    case TDMT_SCH_MERGE_QUERY:
D
dapan1121 已提交
385
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
386
    case TDMT_SCH_QUERY_CONTINUE:
D
dapan1121 已提交
387
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
388 389
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
390
      return TSDB_CODE_APP_ERROR;
H
Hongze Cheng 已提交
391 392 393
  }
}

394
int32_t vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
S
Shengliang Guan 已提交
395
  vTrace("vgId:%d, msg:%p in fetch queue is processing", pVnode->config.vgId, pMsg);
L
Liu Jicong 已提交
396 397
  if ((pMsg->msgType == TDMT_SCH_FETCH || pMsg->msgType == TDMT_VND_TABLE_META || pMsg->msgType == TDMT_VND_TABLE_CFG ||
       pMsg->msgType == TDMT_VND_BATCH_META) &&
398 399
      !syncIsReadyForRead(pVnode->sync)) {
    //      !vnodeIsLeader(pVnode)) {
400
    vnodeRedirectRpcMsg(pVnode, pMsg, terrno);
401 402 403
    return 0;
  }

L
Liu Jicong 已提交
404
  if (pMsg->msgType == TDMT_VND_TMQ_CONSUME && !pVnode->restored) {
405
    vnodeRedirectRpcMsg(pVnode, pMsg, TSDB_CODE_SYN_RESTORING);
L
Liu Jicong 已提交
406 407 408
    return 0;
  }

H
Hongze Cheng 已提交
409
  switch (pMsg->msgType) {
D
dapan1121 已提交
410
    case TDMT_SCH_FETCH:
D
dapan1121 已提交
411
    case TDMT_SCH_MERGE_FETCH:
D
dapan1121 已提交
412
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
413
    case TDMT_SCH_FETCH_RSP:
D
dapan1121 已提交
414
      return qWorkerProcessRspMsg(pVnode, pVnode->pQuery, pMsg, 0);
L
Liu Jicong 已提交
415 416
    // case TDMT_SCH_CANCEL_TASK:
    //   return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
417
    case TDMT_SCH_DROP_TASK:
D
dapan1121 已提交
418
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
419
    case TDMT_SCH_QUERY_HEARTBEAT:
D
dapan1121 已提交
420
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
421
    case TDMT_VND_TABLE_META:
D
dapan1121 已提交
422
      return vnodeGetTableMeta(pVnode, pMsg, true);
D
dapan1121 已提交
423
    case TDMT_VND_TABLE_CFG:
D
dapan1121 已提交
424 425 426
      return vnodeGetTableCfg(pVnode, pMsg, true);
    case TDMT_VND_BATCH_META:
      return vnodeGetBatchMeta(pVnode, pMsg);
L
Liu Jicong 已提交
427
    case TDMT_VND_TMQ_CONSUME:
L
Liu Jicong 已提交
428
      return tqProcessPollReq(pVnode->pTq, pMsg);
429 430
    case TDMT_STREAM_TASK_RUN:
      return tqProcessTaskRunReq(pVnode->pTq, pMsg);
431
#if 1
432
    case TDMT_STREAM_TASK_DISPATCH:
L
Liu Jicong 已提交
433
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg, true);
L
Liu Jicong 已提交
434
#endif
435 436
    case TDMT_STREAM_TASK_CHECK:
      return tqProcessStreamTaskCheckReq(pVnode->pTq, pMsg);
437
    case TDMT_STREAM_TASK_DISPATCH_RSP:
L
Liu Jicong 已提交
438
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
439 440
    case TDMT_STREAM_RETRIEVE:
      return tqProcessTaskRetrieveReq(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
441 442
    case TDMT_STREAM_RETRIEVE_RSP:
      return tqProcessTaskRetrieveRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
443
    case TDMT_VND_STREAM_RECOVER_NONBLOCKING_STAGE:
L
Liu Jicong 已提交
444 445 446 447 448
      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 已提交
449 450
    default:
      vError("unknown msg type:%d in fetch queue", pMsg->msgType);
451
      return TSDB_CODE_APP_ERROR;
H
Hongze Cheng 已提交
452 453 454 455 456 457
  }
}

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

D
dapan1121 已提交
462
void vnodeUpdateMetaRsp(SVnode *pVnode, STableMetaRsp *pMetaRsp) {
463 464 465
  if (NULL == pMetaRsp) {
    return;
  }
L
Liu Jicong 已提交
466

D
dapan1121 已提交
467 468 469 470 471 472
  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 已提交
473
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
474
  int32_t     code = 0;
S
Shengliang Guan 已提交
475 476
  SVTrimDbReq trimReq = {0};

H
Hongze Cheng 已提交
477 478 479 480
  // decode
  if (tDeserializeSVTrimDbReq(pReq, len, &trimReq) != 0) {
    code = TSDB_CODE_INVALID_MSG;
    goto _exit;
S
Shengliang Guan 已提交
481 482
  }

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

H
Hongze Cheng 已提交
485 486 487 488
  // process
  code = tsdbDoRetention(pVnode->pTsdb, trimReq.timestamp);
  if (code) goto _exit;

C
Cary Xu 已提交
489 490 491
  code = smaDoRetention(pVnode->pSma, trimReq.timestamp);
  if (code) goto _exit;

H
Hongze Cheng 已提交
492 493
_exit:
  return code;
S
Shengliang Guan 已提交
494 495
}

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

S
Shengliang Guan 已提交
500 501 502 503 504 505
  SVDropTtlTableReq ttlReq = {0};
  if (tDeserializeSVDropTtlTableReq(pReq, len, &ttlReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto end;
  }

506
  vDebug("vgId:%d, drop ttl table req will be processed, time:%d", pVnode->config.vgId, ttlReq.timestamp);
S
Shengliang Guan 已提交
507
  int32_t ret = metaTtlDropTable(pVnode->pMeta, ttlReq.timestamp, tbUids);
L
Liu Jicong 已提交
508
  if (ret != 0) {
509 510
    goto end;
  }
L
Liu Jicong 已提交
511
  if (taosArrayGetSize(tbUids) > 0) {
wmmhello's avatar
wmmhello 已提交
512 513
    tqUpdateTbUidList(pVnode->pTq, tbUids, false);
  }
514 515 516 517 518 519

end:
  taosArrayDestroy(tbUids);
  return ret;
}

520
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
521
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
522
  SDecoder       coder;
H
Hongze Cheng 已提交
523

H
Hongze Cheng 已提交
524 525 526 527 528 529
  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 已提交
530
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
531 532

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
533 534
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
535 536
  }

H
Hongze Cheng 已提交
537
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
538 539
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
540 541
  }

542
  if (tdProcessRSmaCreate(pVnode->pSma, &req) < 0) {
543 544 545
    pRsp->code = terrno;
    goto _err;
  }
C
Cary Xu 已提交
546

H
Hongze Cheng 已提交
547
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
548
  return 0;
H
Hongze Cheng 已提交
549 550

_err:
H
Hongze Cheng 已提交
551
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
552
  return -1;
H
Hongze Cheng 已提交
553 554
}

555
static int32_t vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
556
  SDecoder           decoder = {0};
557
  SEncoder           encoder = {0};
558
  int32_t            rcode = 0;
H
Hongze Cheng 已提交
559 560
  SVCreateTbBatchReq req = {0};
  SVCreateTbReq     *pCreateReq;
H
Hongze Cheng 已提交
561 562 563
  SVCreateTbBatchRsp rsp = {0};
  SVCreateTbRsp      cRsp = {0};
  char               tbName[TSDB_TABLE_FNAME_LEN];
C
Cary Xu 已提交
564
  STbUidStore       *pStore = NULL;
565
  SArray            *tbUids = NULL;
H
Hongze Cheng 已提交
566 567

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
568 569 570
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
571

H
Hongze Cheng 已提交
572
  // decode
H
Hongze Cheng 已提交
573 574
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
575 576 577 578
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
579

H
Hongze Cheng 已提交
580
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
581 582
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
583 584 585 586 587
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
588
  // loop to create table
589
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
590
    pCreateReq = req.pReqs + iReq;
D
dapan1121 已提交
591
    memset(&cRsp, 0, sizeof(cRsp));
H
Hongze Cheng 已提交
592

C
Cary Xu 已提交
593 594 595 596
    if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
      rcode = -1;
      goto _exit;
    }
L
Liu Jicong 已提交
597

wafwerar's avatar
wafwerar 已提交
598 599 600 601 602
    if ((terrno = grantCheck(TSDB_GRANT_TABLE)) < 0) {
      rcode = -1;
      goto _exit;
    }

H
Hongze Cheng 已提交
603 604 605 606 607 608 609 610 611
    // 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
612
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq, &cRsp.pMeta) < 0) {
H
Hongze Cheng 已提交
613 614 615 616 617
      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 已提交
618
    } else {
H
Hongze Cheng 已提交
619
      cRsp.code = TSDB_CODE_SUCCESS;
620
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
621
      taosArrayPush(tbUids, &pCreateReq->uid);
L
Liu Jicong 已提交
622
      vnodeUpdateMetaRsp(pVnode, cRsp.pMeta);
H
Hongze Cheng 已提交
623
    }
H
Hongze Cheng 已提交
624 625

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

H
Haojun Liao 已提交
628
  vDebug("vgId:%d, add %d new created tables into query table list", TD_VID(pVnode), (int32_t)taosArrayGetSize(tbUids));
629
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
630
  if (tdUpdateTbUidList(pVnode->pSma, pStore, true) < 0) {
C
Cary Xu 已提交
631 632
    goto _exit;
  }
633
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
634

H
Hongze Cheng 已提交
635
  // prepare rsp
636
  int32_t ret = 0;
wafwerar's avatar
wafwerar 已提交
637
  tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
H
Hongze Cheng 已提交
638 639 640 641 642 643
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  if (pRsp->pCont == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    rcode = -1;
    goto _exit;
  }
H
Hongze Cheng 已提交
644 645
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
H
Hongze Cheng 已提交
646

H
Hongze Cheng 已提交
647
_exit:
wmmhello's avatar
wmmhello 已提交
648 649
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
    pCreateReq = req.pReqs + iReq;
650
    taosMemoryFree(pCreateReq->comment);
wmmhello's avatar
wmmhello 已提交
651 652
    taosArrayDestroy(pCreateReq->ctb.tagName);
  }
653
  taosArrayDestroyEx(rsp.pArray, tFreeSVCreateTbRsp);
654
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
655 656
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
657
  return rcode;
H
Hongze Cheng 已提交
658 659
}

660
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
  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 已提交
676
  }
H
Hongze Cheng 已提交
677 678 679 680 681 682 683 684 685

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
686 687 688
  return 0;
}

689
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
690
  SVDropStbReq req = {0};
691
  int32_t      rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
692
  SDecoder     decoder = {0};
693
  SArray      *tbUidList = NULL;
H
Hongze Cheng 已提交
694 695 696 697 698 699

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

  // decode request
H
Hongze Cheng 已提交
700 701
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
702 703 704 705 706
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
707 708 709 710 711 712 713 714
  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) {
715 716 717
    rcode = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
718

719 720 721 722 723
  if (tdProcessRSmaDrop(pVnode->pSma, &req) < 0) {
    rcode = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
724 725
  // return rsp
_exit:
726
  if (tbUidList) taosArrayDestroy(tbUidList);
H
Hongze Cheng 已提交
727
  pRsp->code = rcode;
H
Hongze Cheng 已提交
728
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
729 730 731
  return 0;
}

732
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
D
dapan1121 已提交
733 734 735
  SVAlterTbReq  vAlterTbReq = {0};
  SVAlterTbRsp  vAlterTbRsp = {0};
  SDecoder      dc = {0};
736 737
  int32_t       rcode = 0;
  int32_t       ret;
D
dapan1121 已提交
738 739
  SEncoder      ec = {0};
  STableMetaRsp vMetaRsp = {0};
H
Hongze Cheng 已提交
740 741 742 743 744 745 746 747 748 749

  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 已提交
750
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
751
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
752 753
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
754 755 756
  }

  // process
D
dapan1121 已提交
757
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq, &vMetaRsp) < 0) {
wmmhello's avatar
wmmhello 已提交
758
    vAlterTbRsp.code = terrno;
H
Hongze Cheng 已提交
759
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
760 761
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
762 763
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
764

D
dapan1121 已提交
765 766 767 768 769
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
770 771 772 773 774 775
_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 已提交
776 777 778
  if (vMetaRsp.pSchemas) {
    taosMemoryFree(vMetaRsp.pSchemas);
  }
H
Hongze Cheng 已提交
779 780 781
  return 0;
}

782
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
783 784
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
785
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
786
  SEncoder         encoder = {0};
787
  int32_t          ret;
788
  SArray          *tbUids = NULL;
789
  STbUidStore     *pStore = NULL;
H
Hongze Cheng 已提交
790

H
Hongze Cheng 已提交
791
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
792 793 794
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
795 796

  // decode req
H
Hongze Cheng 已提交
797 798
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
799 800 801 802 803
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
804 805

  // process req
806
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
807
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
808 809
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

810
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
811 812
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
813
    tb_uid_t     tbUid = 0;
H
Hongze Cheng 已提交
814

H
Hongze Cheng 已提交
815
    /* code */
816
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids, &tbUid);
H
Hongze Cheng 已提交
817
    if (ret < 0) {
818
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_TDB_TABLE_NOT_EXIST) {
H
Hongze Cheng 已提交
819 820 821 822
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
823
    } else {
H
Hongze Cheng 已提交
824
      dropTbRsp.code = TSDB_CODE_SUCCESS;
825
      if (tbUid > 0) tdFetchTbUidList(pVnode->pSma, &pStore, pDropTbReq->suid, tbUid);
H
Hongze Cheng 已提交
826 827 828 829 830
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

831
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);
832
  tdUpdateTbUidList(pVnode->pSma, pStore, false);
833

H
Hongze Cheng 已提交
834
_exit:
835
  taosArrayDestroy(tbUids);
836
  tdUidStoreFree(pStore);
H
Hongze Cheng 已提交
837
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
838 839 840 841 842
  tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVDropTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
843
  taosArrayDestroy(rsp.pArray);
H
Hongze Cheng 已提交
844 845 846
  return 0;
}

847 848
static int32_t vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter,
                                              const char *tags) {
D
dapan 已提交
849 850 851 852
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
853
  int32_t        rv = -1;
D
dapan 已提交
854 855 856

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
857 858 859 860 861

  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 已提交
862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877
  }
  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;
}

878
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
879
#if 1
H
Hongze Cheng 已提交
880
  int32_t code = 0;
D
dapan1121 已提交
881
  terrno = 0;
C
Cary Xu 已提交
882

H
Hongze Cheng 已提交
883
  SSubmitReq2 *pSubmitReq = &(SSubmitReq2){0};
H
Hongze Cheng 已提交
884
  SSubmitRsp2 *pSubmitRsp = &(SSubmitRsp2){0};
H
Hongze Cheng 已提交
885
  SArray      *newTbUids = NULL;
H
Hongze Cheng 已提交
886 887 888 889
  int32_t      ret;
  SEncoder     ec = {0};

  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
890

H
Hongze Cheng 已提交
891 892
  // decode
  SDecoder dc = {0};
893
  tDecoderInit(&dc, pReq, len);
H
Hongze Cheng 已提交
894
  if (tDecodeSSubmitReq2(&dc, pSubmitReq) < 0) {
H
Hongze Cheng 已提交
895 896
    code = TSDB_CODE_INVALID_MSG;
    goto _exit;
D
dapan 已提交
897
  }
H
Hongze Cheng 已提交
898
  tDecoderClear(&dc);
C
Cary Xu 已提交
899

H
Hongze Cheng 已提交
900
  // check
901 902 903 904
  code = tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq);
  if (code) {
    goto _exit;
  }
C
Cary Xu 已提交
905

H
Hongze Cheng 已提交
906
  for (int32_t i = 0; i < TARRAY_SIZE(pSubmitReq->aSubmitTbData); ++i) {
H
Hongze Cheng 已提交
907 908 909 910 911 912 913 914 915 916
    SSubmitTbData *pSubmitTbData = taosArrayGet(pSubmitReq->aSubmitTbData, i);

    if (pSubmitTbData->pCreateTbReq) {
      pSubmitTbData->uid = pSubmitTbData->pCreateTbReq->uid;
    } else {
      SMetaInfo info = {0};

      code = metaGetInfo(pVnode->pMeta, pSubmitTbData->uid, &info, NULL);
      if (code) {
        code = TSDB_CODE_TDB_TABLE_NOT_EXIST;
D
dapan1121 已提交
917
        vWarn("vgId:%d, table uid:%" PRId64 " not exists", TD_VID(pVnode), pSubmitTbData->uid);
H
Hongze Cheng 已提交
918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934
        goto _exit;
      }

      if (info.suid != pSubmitTbData->suid) {
        code = TSDB_CODE_INVALID_MSG;
        goto _exit;
      }

      if (info.suid) {
        metaGetInfo(pVnode->pMeta, info.suid, &info, NULL);
      }

      if (pSubmitTbData->sver != info.skmVer) {
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
        goto _exit;
      }
    }
H
Hongze Cheng 已提交
935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957

    if (pSubmitTbData->flags & SUBMIT_REQ_COLUMN_DATA_FORMAT) {
      int32_t   nColData = TARRAY_SIZE(pSubmitTbData->aCol);
      SColData *aColData = (SColData *)TARRAY_DATA(pSubmitTbData->aCol);

      if (nColData <= 0) {
        code = TSDB_CODE_INVALID_MSG;
        goto _exit;
      }

      if (aColData[0].cid != PRIMARYKEY_TIMESTAMP_COL_ID || aColData[0].type != TSDB_DATA_TYPE_TIMESTAMP ||
          aColData[0].nVal <= 0) {
        code = TSDB_CODE_INVALID_MSG;
        goto _exit;
      }

      for (int32_t i = 1; i < nColData; i++) {
        if (aColData[i].nVal != aColData[0].nVal) {
          code = TSDB_CODE_INVALID_MSG;
          goto _exit;
        }
      }
    }
H
Hongze Cheng 已提交
958 959
  }

L
Liu Jicong 已提交
960 961
  vDebug("vgId:%d, submit block size %d", TD_VID(pVnode), (int32_t)taosArrayGetSize(pSubmitReq->aSubmitTbData));

H
Hongze Cheng 已提交
962
  // loop to handle
H
Hongze Cheng 已提交
963
  for (int32_t i = 0; i < TARRAY_SIZE(pSubmitReq->aSubmitTbData); ++i) {
H
Hongze Cheng 已提交
964 965 966 967
    SSubmitTbData *pSubmitTbData = taosArrayGet(pSubmitReq->aSubmitTbData, i);

    // create table
    if (pSubmitTbData->pCreateTbReq) {
H
Hongze Cheng 已提交
968 969 970 971 972 973 974 975 976 977 978
      // check (TODO: move check to create table)
      code = grantCheck(TSDB_GRANT_TIMESERIES);
      if (code) goto _exit;

      code = grantCheck(TSDB_GRANT_TABLE);
      if (code) goto _exit;

      // alloc if need
      if (pSubmitRsp->aCreateTbRsp == NULL &&
          (pSubmitRsp->aCreateTbRsp = taosArrayInit(TARRAY_SIZE(pSubmitReq->aSubmitTbData), sizeof(SVCreateTbRsp))) ==
              NULL) {
H
Hongze Cheng 已提交
979
        code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
980 981 982 983 984 985 986 987 988 989 990
        goto _exit;
      }

      SVCreateTbRsp *pCreateTbRsp = taosArrayReserve(pSubmitRsp->aCreateTbRsp, 1);

      // create table
      if (metaCreateTable(pVnode->pMeta, version, pSubmitTbData->pCreateTbReq, &pCreateTbRsp->pMeta) ==
          0) {  // create table success

        if (newTbUids == NULL &&
            (newTbUids = taosArrayInit(TARRAY_SIZE(pSubmitReq->aSubmitTbData), sizeof(int64_t))) == NULL) {
H
Hongze Cheng 已提交
991
          code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
992 993 994 995 996 997
          goto _exit;
        }

        taosArrayPush(newTbUids, &pSubmitTbData->uid);

        if (pCreateTbRsp->pMeta) {
D
dapan1121 已提交
998
          vnodeUpdateMetaRsp(pVnode, pCreateTbRsp->pMeta);
H
Hongze Cheng 已提交
999 1000 1001 1002 1003 1004
        }
      } else {  // create table failed
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
          code = terrno;
          goto _exit;
        }
H
Hongze Cheng 已提交
1005
      }
H
Hongze Cheng 已提交
1006 1007 1008
    }

    // insert data
H
Hongze Cheng 已提交
1009 1010 1011 1012 1013
    int32_t affectedRows;
    code = tsdbInsertTableData(pVnode->pTsdb, version, pSubmitTbData, &affectedRows);
    if (code) goto _exit;

    pSubmitRsp->affectedRows += affectedRows;
H
Hongze Cheng 已提交
1014
  }
H
Hongze Cheng 已提交
1015

H
Hongze Cheng 已提交
1016 1017 1018 1019 1020
  // update table uid list
  if (taosArrayGetSize(newTbUids) > 0) {
    vDebug("vgId:%d, add %d table into query table list in handling submit", TD_VID(pVnode),
           (int32_t)taosArrayGetSize(newTbUids));
    tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
H
Hongze Cheng 已提交
1021
  }
H
Hongze Cheng 已提交
1022 1023 1024 1025 1026 1027 1028 1029 1030 1031

_exit:
  // message
  pRsp->code = code;
  tEncodeSize(tEncodeSSubmitRsp2, pSubmitRsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSSubmitRsp2(&ec, pSubmitRsp);
  tEncoderClear(&ec);

H
Hongze Cheng 已提交
1032 1033 1034 1035 1036 1037
  // update statistics
  atomic_add_fetch_64(&pVnode->statis.nInsert, pSubmitRsp->affectedRows);
  atomic_add_fetch_64(&pVnode->statis.nInsertSuccess, pSubmitRsp->affectedRows);
  atomic_add_fetch_64(&pVnode->statis.nBatchInsert, 1);
  if (code == 0) {
    atomic_add_fetch_64(&pVnode->statis.nBatchInsertSuccess, 1);
K
kailixu 已提交
1038
    tdProcessRSmaSubmit(pVnode->pSma, version, pSubmitReq, pReq, len, STREAM_INPUT__DATA_SUBMIT);
H
Hongze Cheng 已提交
1039 1040
  }

H
Hongze Cheng 已提交
1041 1042
  // clear
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1043
  tDestroySSubmitReq2(pSubmitReq, TSDB_MSG_FLG_DECODE);
H
Hongze Cheng 已提交
1044
  tDestroySSubmitRsp2(pSubmitRsp, TSDB_MSG_FLG_ENCODE);
H
Hongze Cheng 已提交
1045

D
dapan1121 已提交
1046 1047
  if (code) terrno = code;

H
Hongze Cheng 已提交
1048 1049
  return code;

H
Hongze Cheng 已提交
1050
#else
H
Hongze Cheng 已提交
1051
  SSubmitReq *pSubmitReq = (SSubmitReq *)pReq;
L
Liu Jicong 已提交
1052 1053 1054 1055 1056 1057 1058
  SSubmitRsp  submitRsp = {0};
  int32_t     nRows = 0;
  int32_t     tsize, ret;
  SEncoder    encoder = {0};
  SArray     *newTbUids = NULL;
  SVStatis    statis = {0};
  bool        tbCreated = false;
1059
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
1060 1061

  pRsp->code = 0;
C
Cary Xu 已提交
1062
  pSubmitReq->version = version;
1063
  statis.nBatchInsert = 1;
C
Cary Xu 已提交
1064

C
Cary Xu 已提交
1065 1066 1067 1068 1069
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

C
Cary Xu 已提交
1070 1071
  submitRsp.pArray = taosArrayInit(msgIter.numOfBlocks, sizeof(SSubmitBlkRsp));
  newTbUids = taosArrayInit(msgIter.numOfBlocks, sizeof(int64_t));
C
Cary Xu 已提交
1072
  if (!submitRsp.pArray || !newTbUids) {
1073 1074 1075 1076
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
1077
  for (;;) {
H
Hongze Cheng 已提交
1078 1079 1080
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
1081
    SSubmitBlkRsp submitBlkRsp = {0};
D
dapan1121 已提交
1082
    tbCreated = false;
H
Hongze Cheng 已提交
1083

H
Hongze Cheng 已提交
1084 1085
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106
      // tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      // if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
      //   pRsp->code = TSDB_CODE_INVALID_MSG;
      //   tDecoderClear(&decoder);
      //   taosArrayDestroy(createTbReq.ctb.tagName);
      //   goto _exit;
      // }

      // if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
      //   pRsp->code = terrno;
      //   tDecoderClear(&decoder);
      //   taosArrayDestroy(createTbReq.ctb.tagName);
      //   goto _exit;
      // }

      // if ((terrno = grantCheck(TSDB_GRANT_TABLE)) < 0) {
      //   pRsp->code = terrno;
      //   tDecoderClear(&decoder);
      //   taosArrayDestroy(createTbReq.ctb.tagName);
      //   goto _exit;
      // }
wafwerar's avatar
wafwerar 已提交
1107

1108
      if (metaCreateTable(pVnode->pMeta, version, &createTbReq, &submitBlkRsp.pMeta) < 0) {
H
Hongze Cheng 已提交
1109 1110 1111 1112 1113 1114 1115
        // if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
        //   submitBlkRsp.code = terrno;
        //   pRsp->code = terrno;
        //   tDecoderClear(&decoder);
        //   taosArrayDestroy(createTbReq.ctb.tagName);
        //   goto _exit;
        // }
1116 1117 1118 1119
      } else {
        if (NULL != submitBlkRsp.pMeta) {
          vnodeUpdateMetaRsp(pVnode, submitBlkRsp.pMeta);
        }
1120

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

D
dapan1121 已提交
1123 1124 1125 1126 1127
        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;
      }
1128

H
Hongze Cheng 已提交
1129 1130 1131 1132 1133 1134
      // msgIter.uid = createTbReq.uid;
      // if (createTbReq.type == TSDB_CHILD_TABLE) {
      //   msgIter.suid = createTbReq.ctb.suid;
      // } else {
      //   msgIter.suid = 0;
      // }
H
Hongze Cheng 已提交
1135

H
Hongze Cheng 已提交
1136 1137
      // tDecoderClear(&decoder);
      // taosArrayDestroy(createTbReq.ctb.tagName);
1138
    }
H
Hongze Cheng 已提交
1139

H
Hongze Cheng 已提交
1140
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
1141
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
1142 1143
    }

H
Hongze Cheng 已提交
1144 1145
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
D
dapan1121 已提交
1146 1147 1148
    if (tbCreated || submitBlkRsp.code) {
      taosArrayPush(submitRsp.pArray, &submitBlkRsp);
    }
H
Hongze Cheng 已提交
1149
  }
H
Haojun Liao 已提交
1150

H
Hongze Cheng 已提交
1151 1152 1153 1154
  // if (taosArrayGetSize(newTbUids) > 0) {
  //   vDebug("vgId:%d, add %d table into query table list in handling submit", TD_VID(pVnode),
  //          (int32_t)taosArrayGetSize(newTbUids));
  // }
H
Haojun Liao 已提交
1155

H
Hongze Cheng 已提交
1156
  // tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
1157

H
Hongze Cheng 已提交
1158
_exit:
1159
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1160 1161 1162 1163 1164 1165
  // tEncodeSize(tEncodeSSubmitRsp, &submitRsp, tsize, ret);
  // pRsp->pCont = rpcMallocCont(tsize);
  // pRsp->contLen = tsize;
  // tEncoderInit(&encoder, pRsp->pCont, tsize);
  // tEncodeSSubmitRsp(&encoder, &submitRsp);
  // tEncoderClear(&encoder);
H
Hongze Cheng 已提交
1166

1167
  taosArrayDestroyEx(submitRsp.pArray, tFreeSSubmitBlkRsp);
H
Hongze Cheng 已提交
1168

1169
  // TODO: the partial success scenario and the error case
H
Hongze Cheng 已提交
1170 1171
  // => If partial success, extract the success submitted rows and reconstruct a new submit msg, and push to level
  // 1/level 2.
1172
  // TODO: refactor
C
Cary Xu 已提交
1173
  if ((terrno == TSDB_CODE_SUCCESS) && (pRsp->code == TSDB_CODE_SUCCESS)) {
1174
    statis.nBatchInsertSuccess = 1;
L
Liu Jicong 已提交
1175
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_INPUT__DATA_SUBMIT);
1176
  }
C
Cary Xu 已提交
1177

1178
  // N.B. not strict as the following procedure is not atomic
1179 1180 1181
  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);
1182
  atomic_add_fetch_64(&pVnode->statis.nBatchInsertSuccess, statis.nBatchInsertSuccess);
1183

S
Shengliang Guan 已提交
1184
  vDebug("vgId:%d, submit success, index:%" PRId64, pVnode->config.vgId, version);
H
Hongze Cheng 已提交
1185
  return 0;
H
Hongze Cheng 已提交
1186 1187
#endif
  return 0;
L
Liu Jicong 已提交
1188
}
1189

1190
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
1191
  SVCreateTSmaReq req = {0};
C
Cary Xu 已提交
1192
  SDecoder        coder = {0};
1193

C
Cary Xu 已提交
1194 1195 1196 1197 1198 1199
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
1200 1201 1202 1203 1204

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
1205 1206
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
1207
    goto _err;
1208
  }
C
Cary Xu 已提交
1209

C
Cary Xu 已提交
1210
  if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
C
Cary Xu 已提交
1211
    if (pRsp) pRsp->code = terrno;
1212
    goto _err;
1213
  }
C
Cary Xu 已提交
1214

1215
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
1216
  vDebug("vgId:%d, success to create tsma %s:%" PRIi64 " version %" PRIi64 " for table %" PRIi64, TD_VID(pVnode),
C
Cary Xu 已提交
1217
         req.indexName, req.indexUid, version, req.tableUid);
H
Hongze Cheng 已提交
1218
  return 0;
1219 1220 1221

_err:
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
1222
  vError("vgId:%d, failed to create tsma %s:%" PRIi64 " version %" PRIi64 "for table %" PRIi64 " since %s",
C
Cary Xu 已提交
1223
         TD_VID(pVnode), req.indexName, req.indexUid, version, req.tableUid, terrstr());
1224
  return -1;
L
Liu Jicong 已提交
1225
}
C
Cary Xu 已提交
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237

/**
 * @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);
}
1238 1239 1240 1241 1242 1243 1244 1245 1246

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;
1247
}
S
Shengliang Guan 已提交
1248

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

1252 1253
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
1254 1255 1256
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
1257
}
H
Hongze Cheng 已提交
1258

1259
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
S
Shengliang Guan 已提交
1260 1261
  bool walChanged = false;
  bool tsdbChanged = false;
1262

S
Shengliang Guan 已提交
1263 1264
  SAlterVnodeConfigReq req = {0};
  if (tDeserializeSAlterVnodeConfigReq(pReq, len, &req) != 0) {
1265 1266 1267 1268
    terrno = TSDB_CODE_INVALID_MSG;
    return TSDB_CODE_INVALID_MSG;
  }

1269
  vInfo("vgId:%d, start to alter vnode config, page:%d pageSize:%d buffer:%d szPage:%d szBuf:%" PRIu64
S
Shengliang Guan 已提交
1270
        " cacheLast:%d cacheLastSize:%d days:%d keep0:%d keep1:%d keep2:%d fsync:%d level:%d",
1271 1272
        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 已提交
1273
        req.walFsyncPeriod, req.walLevel);
1274 1275 1276

  if (pVnode->config.cacheLastSize != req.cacheLastSize) {
    pVnode->config.cacheLastSize = req.cacheLastSize;
1277 1278
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
1279

1280
  if (pVnode->config.szBuf != req.buffer * 1024LL * 1024LL) {
S
Shengliang Guan 已提交
1281
    vInfo("vgId:%d, vnode buffer is changed from %" PRId64 " to %" PRId64, TD_VID(pVnode), pVnode->config.szBuf,
1282
          (uint64_t)(req.buffer * 1024LL * 1024LL));
1283
    pVnode->config.szBuf = req.buffer * 1024LL * 1024LL;
H
Hongze Cheng 已提交
1284 1285
  }

1286 1287
  if (pVnode->config.szCache != req.pages) {
    if (metaAlterCache(pVnode->pMeta, req.pages) < 0) {
S
Shengliang Guan 已提交
1288
      vError("vgId:%d, failed to change vnode pages from %d to %d failed since %s", TD_VID(pVnode),
1289
             pVnode->config.szCache, req.pages, tstrerror(errno));
H
Hongze Cheng 已提交
1290 1291
      return errno;
    } else {
S
Shengliang Guan 已提交
1292
      vInfo("vgId:%d, vnode pages is changed from %d to %d", TD_VID(pVnode), pVnode->config.szCache, req.pages);
1293
      pVnode->config.szCache = req.pages;
H
Hongze Cheng 已提交
1294
    }
H
Hongze Cheng 已提交
1295 1296
  }

1297 1298
  if (pVnode->config.cacheLast != req.cacheLast) {
    pVnode->config.cacheLast = req.cacheLast;
1299 1300
  }

1301 1302
  if (pVnode->config.walCfg.fsyncPeriod != req.walFsyncPeriod) {
    pVnode->config.walCfg.fsyncPeriod = req.walFsyncPeriod;
1303 1304 1305 1306

    walChanged = true;
  }

1307 1308
  if (pVnode->config.walCfg.level != req.walLevel) {
    pVnode->config.walCfg.level = req.walLevel;
1309 1310 1311 1312

    walChanged = true;
  }

1313 1314
  if (pVnode->config.tsdbCfg.keep0 != req.daysToKeep0) {
    pVnode->config.tsdbCfg.keep0 = req.daysToKeep0;
1315
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1316
      tsdbChanged = true;
1317 1318 1319
    }
  }

1320 1321
  if (pVnode->config.tsdbCfg.keep1 != req.daysToKeep1) {
    pVnode->config.tsdbCfg.keep1 = req.daysToKeep1;
1322
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1323
      tsdbChanged = true;
1324 1325 1326
    }
  }

1327 1328
  if (pVnode->config.tsdbCfg.keep2 != req.daysToKeep2) {
    pVnode->config.tsdbCfg.keep2 = req.daysToKeep2;
1329
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1330
      tsdbChanged = true;
1331 1332 1333 1334
    }
  }

  if (walChanged) {
M
Minglei Jin 已提交
1335 1336 1337 1338 1339
    walAlter(pVnode->pWal, &pVnode->config.walCfg);
  }

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

1342 1343 1344
  return 0;
}

1345 1346 1347 1348 1349 1350
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);

1351 1352 1353
  SMetaReader mr = {0};
  metaReaderInit(&mr, pVnode->pMeta, 0);

1354 1355 1356
  int32_t sz = taosArrayGetSize(deleteReq.deleteReqs);
  for (int32_t i = 0; i < sz; i++) {
    SSingleDeleteReq *pOneReq = taosArrayGet(deleteReq.deleteReqs, i);
1357 1358
    char             *name = pOneReq->tbname;
    if (metaGetTableEntryByName(&mr, name) < 0) {
S
Shengliang Guan 已提交
1359
      vDebug("vgId:%d, stream delete msg, skip since no table: %s", pVnode->config.vgId, name);
1360 1361 1362 1363 1364
      continue;
    }

    int64_t uid = mr.me.uid;

L
Liu Jicong 已提交
1365
    int32_t code = tsdbDeleteTableData(pVnode->pTsdb, version, deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1366 1367 1368
    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 已提交
1369
             TD_VID(pVnode), terrstr(), deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1370
    }
1371 1372

    tDecoderClear(&mr.coder);
1373
  }
1374
  metaReaderClear(&mr);
1375
  taosArrayDestroy(deleteReq.deleteReqs);
1376 1377 1378
  return 0;
}

H
Hongze Cheng 已提交
1379 1380 1381 1382 1383
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 已提交
1384 1385 1386 1387 1388
  pRsp->msgType = TDMT_VND_DELETE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

H
Hongze Cheng 已提交
1389 1390 1391 1392 1393 1394 1395 1396
  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);
1397
  ASSERT(taosArrayGetSize(pRes->uidList) == 0 || (pRes->skey != 0 && pRes->ekey != 0));
H
Hongze Cheng 已提交
1398 1399 1400 1401 1402 1403 1404 1405 1406

  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 已提交
1407 1408

  SVDeleteRsp rsp = {.affectedRows = pRes->affectedRows};
L
Liu Jicong 已提交
1409
  int32_t     ret = 0;
wmmhello's avatar
wmmhello 已提交
1410 1411
  tEncodeSize(tEncodeSVDeleteRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
L
Liu Jicong 已提交
1412
  SEncoder ec = {0};
wmmhello's avatar
wmmhello 已提交
1413 1414 1415
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVDeleteRsp(&ec, &rsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
1416 1417 1418 1419
  return code;

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