vnodeSvr.c 43.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
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
28
static int32_t vnodeProcessDropTtlTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
S
Shengliang Guan 已提交
29
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
30
static int32_t vnodeProcessDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
31
static int32_t vnodeProcessBatchDeleteReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
32

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

H
Hongze Cheng 已提交
37 38 39 40 41 42
  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 已提交
43 44 45 46
      if (tStartDecode(&dc) < 0) {
        code = TSDB_CODE_INVALID_MSG;
        return code;
      }
H
Hongze Cheng 已提交
47

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

H
Hongze Cheng 已提交
60 61 62 63 64 65 66 67
        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 已提交
68 69 70
        *(int64_t *)(dc.data + dc.pos) = uid;
        *(int64_t *)(dc.data + dc.pos + 8) = ctime;

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

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

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

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

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

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

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

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

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

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

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

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

          // 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 已提交
138
        }
H
Hongze Cheng 已提交
139 140

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

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

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

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

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

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

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

188
int32_t vnodeProcessWriteMsg(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
189 190 191 192
  void   *ptr = NULL;
  void   *pReq;
  int32_t len;
  int32_t ret;
193
  /*
194
  if (!pVnode->inUse) {
195
    terrno = TSDB_CODE_VND_NO_AVAIL_BUFPOOL;
H
Hongze Cheng 已提交
196 197
    vError("vgId:%d, not ready to write since %s", TD_VID(pVnode), terrstr());
    return -1;
198
  }
199
  */
200 201 202
  if (version <= pVnode->state.applied) {
    vError("vgId:%d, duplicate write request. version: %" PRId64 ", applied: %" PRId64 "", TD_VID(pVnode), version,
           pVnode->state.applied);
203
    terrno = TSDB_CODE_VND_DUP_REQUEST;
204 205 206
    return -1;
  }

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

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

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

216 217
  if (!syncUtilUserCommit(pMsg->msgType)) goto _exit;

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

H
Hongze Cheng 已提交
222 223 224
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
B
Benguang Zhao 已提交
225
  bool needCommit = false;
H
Hongze Cheng 已提交
226 227

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

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

329 330
  walApplyVer(pVnode->pWal, version);

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

H
Hongze Cheng 已提交
336
  // commit if need
B
Benguang Zhao 已提交
337
  if (needCommit) {
S
Shengliang Guan 已提交
338
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
339 340 341 342
    if (vnodeAsyncCommit(pVnode) < 0) {
      vError("vgId:%d, failed to vnode async commit since %s.", TD_VID(pVnode), tstrerror(terrno));
      goto _err;
    }
H
Hongze Cheng 已提交
343 344

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

H
Hongze Cheng 已提交
351
_exit:
H
Hongze Cheng 已提交
352
  return 0;
H
Hongze Cheng 已提交
353 354

_err:
S
Shengliang Guan 已提交
355
  vError("vgId:%d, process %s request failed since %s, version:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
356 357
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
358 359
}

360
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
361
  if (TDMT_SCH_QUERY != pMsg->msgType && TDMT_SCH_MERGE_QUERY != pMsg->msgType) {
D
dapan1121 已提交
362 363 364
    return 0;
  }

365
  return qWorkerPreprocessQueryMsg(pVnode->pQuery, pMsg, TDMT_SCH_QUERY == pMsg->msgType);
D
dapan1121 已提交
366 367 368
}

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

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

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

L
Liu Jicong 已提交
399
  if (pMsg->msgType == TDMT_VND_TMQ_CONSUME && !pVnode->restored) {
400
    vnodeRedirectRpcMsg(pVnode, pMsg, TSDB_CODE_SYN_RESTORING);
L
Liu Jicong 已提交
401 402 403
    return 0;
  }

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

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

D
dapan1121 已提交
457
void vnodeUpdateMetaRsp(SVnode *pVnode, STableMetaRsp *pMetaRsp) {
458 459 460
  if (NULL == pMetaRsp) {
    return;
  }
L
Liu Jicong 已提交
461

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

H
Hongze Cheng 已提交
472 473 474 475
  // decode
  if (tDeserializeSVTrimDbReq(pReq, len, &trimReq) != 0) {
    code = TSDB_CODE_INVALID_MSG;
    goto _exit;
S
Shengliang Guan 已提交
476 477
  }

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

H
Hongze Cheng 已提交
480 481 482 483
  // process
  code = tsdbDoRetention(pVnode->pTsdb, trimReq.timestamp);
  if (code) goto _exit;

C
Cary Xu 已提交
484 485 486
  code = smaDoRetention(pVnode->pSma, trimReq.timestamp);
  if (code) goto _exit;

H
Hongze Cheng 已提交
487 488
_exit:
  return code;
S
Shengliang Guan 已提交
489 490
}

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

S
Shengliang Guan 已提交
495 496 497 498 499 500
  SVDropTtlTableReq ttlReq = {0};
  if (tDeserializeSVDropTtlTableReq(pReq, len, &ttlReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto end;
  }

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

end:
  taosArrayDestroy(tbUids);
  return ret;
}

515
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
516
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
517
  SDecoder       coder;
H
Hongze Cheng 已提交
518

H
Hongze Cheng 已提交
519 520 521 522 523 524
  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 已提交
525
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
526 527

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
528 529
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
530 531
  }

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

537
  if (tdProcessRSmaCreate(pVnode->pSma, &req) < 0) {
538 539 540
    pRsp->code = terrno;
    goto _err;
  }
C
Cary Xu 已提交
541

H
Hongze Cheng 已提交
542
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
543
  return 0;
H
Hongze Cheng 已提交
544 545

_err:
H
Hongze Cheng 已提交
546
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
547
  return -1;
H
Hongze Cheng 已提交
548 549
}

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

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
563 564 565
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
566

H
Hongze Cheng 已提交
567
  // decode
H
Hongze Cheng 已提交
568 569
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
570 571 572 573
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
574

H
Hongze Cheng 已提交
575
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
576 577
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
578 579 580 581 582
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
583
  // loop to create table
584
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
585
    pCreateReq = req.pReqs + iReq;
D
dapan1121 已提交
586
    memset(&cRsp, 0, sizeof(cRsp));
H
Hongze Cheng 已提交
587

C
Cary Xu 已提交
588 589 590 591
    if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
      rcode = -1;
      goto _exit;
    }
L
Liu Jicong 已提交
592

wafwerar's avatar
wafwerar 已提交
593 594 595 596 597
    if ((terrno = grantCheck(TSDB_GRANT_TABLE)) < 0) {
      rcode = -1;
      goto _exit;
    }

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

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

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

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

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

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

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
681 682 683
  return 0;
}

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

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

  // decode request
H
Hongze Cheng 已提交
695 696
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
697 698 699 700 701
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
702 703 704 705 706 707 708 709
  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) {
710 711 712
    rcode = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
713

714 715 716 717 718
  if (tdProcessRSmaDrop(pVnode->pSma, &req) < 0) {
    rcode = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
719 720
  // return rsp
_exit:
721
  if (tbUidList) taosArrayDestroy(tbUidList);
H
Hongze Cheng 已提交
722
  pRsp->code = rcode;
H
Hongze Cheng 已提交
723
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
724 725 726
  return 0;
}

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

  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 已提交
745
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
746
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
747 748
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
749 750 751
  }

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

D
dapan1121 已提交
760 761 762 763 764
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
765 766 767 768 769 770
_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 已提交
771 772 773
  if (vMetaRsp.pSchemas) {
    taosMemoryFree(vMetaRsp.pSchemas);
  }
H
Hongze Cheng 已提交
774 775 776
  return 0;
}

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

H
Hongze Cheng 已提交
786
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
787 788 789
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
790 791

  // decode req
H
Hongze Cheng 已提交
792 793
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
794 795 796 797 798
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
799 800

  // process req
801
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
802
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
803 804
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

805
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
806 807
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
808
    tb_uid_t     tbUid = 0;
H
Hongze Cheng 已提交
809

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

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

826
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);
827
  tdUpdateTbUidList(pVnode->pSma, pStore, false);
828

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

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

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
852 853 854 855 856

  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 已提交
857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872
  }
  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;
}

873
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
874
#if 1
H
Hongze Cheng 已提交
875
  int32_t code = 0;
D
dapan1121 已提交
876
  terrno = 0;
C
Cary Xu 已提交
877

H
Hongze Cheng 已提交
878
  SSubmitReq2 *pSubmitReq = &(SSubmitReq2){0};
H
Hongze Cheng 已提交
879
  SSubmitRsp2 *pSubmitRsp = &(SSubmitRsp2){0};
H
Hongze Cheng 已提交
880
  SArray      *newTbUids = NULL;
H
Hongze Cheng 已提交
881 882 883 884
  int32_t      ret;
  SEncoder     ec = {0};

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

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

H
Hongze Cheng 已提交
895
  // check
896 897 898 899
  code = tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq);
  if (code) {
    goto _exit;
  }
C
Cary Xu 已提交
900

H
Hongze Cheng 已提交
901
  for (int32_t i = 0; i < TARRAY_SIZE(pSubmitReq->aSubmitTbData); ++i) {
H
Hongze Cheng 已提交
902 903 904 905 906 907 908 909 910 911
    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 已提交
912
        vWarn("vgId:%d, table uid:%" PRId64 " not exists", TD_VID(pVnode), pSubmitTbData->uid);
H
Hongze Cheng 已提交
913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929
        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 已提交
930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952

    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 已提交
953 954
  }

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

H
Hongze Cheng 已提交
957
  // loop to handle
H
Hongze Cheng 已提交
958
  for (int32_t i = 0; i < TARRAY_SIZE(pSubmitReq->aSubmitTbData); ++i) {
H
Hongze Cheng 已提交
959 960 961 962
    SSubmitTbData *pSubmitTbData = taosArrayGet(pSubmitReq->aSubmitTbData, i);

    // create table
    if (pSubmitTbData->pCreateTbReq) {
H
Hongze Cheng 已提交
963 964 965 966 967 968 969 970 971 972 973
      // 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 已提交
974
        code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
975 976 977 978 979 980 981 982 983 984 985
        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 已提交
986
          code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
987 988 989 990 991 992
          goto _exit;
        }

        taosArrayPush(newTbUids, &pSubmitTbData->uid);

        if (pCreateTbRsp->pMeta) {
D
dapan1121 已提交
993
          vnodeUpdateMetaRsp(pVnode, pCreateTbRsp->pMeta);
H
Hongze Cheng 已提交
994 995 996 997 998 999
        }
      } else {  // create table failed
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
          code = terrno;
          goto _exit;
        }
1000
        pSubmitTbData->uid = pSubmitTbData->pCreateTbReq->uid;  // update uid if table exist for using below
H
Hongze Cheng 已提交
1001
      }
H
Hongze Cheng 已提交
1002 1003 1004
    }

    // insert data
H
Hongze Cheng 已提交
1005 1006 1007 1008 1009
    int32_t affectedRows;
    code = tsdbInsertTableData(pVnode->pTsdb, version, pSubmitTbData, &affectedRows);
    if (code) goto _exit;

    pSubmitRsp->affectedRows += affectedRows;
H
Hongze Cheng 已提交
1010
  }
H
Hongze Cheng 已提交
1011

H
Hongze Cheng 已提交
1012 1013 1014 1015 1016
  // 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 已提交
1017
  }
H
Hongze Cheng 已提交
1018 1019 1020 1021 1022 1023 1024 1025 1026 1027

_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 已提交
1028 1029 1030 1031 1032 1033
  // 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 已提交
1034
    tdProcessRSmaSubmit(pVnode->pSma, version, pSubmitReq, pReq, len, STREAM_INPUT__DATA_SUBMIT);
H
Hongze Cheng 已提交
1035 1036
  }

H
Hongze Cheng 已提交
1037 1038
  // clear
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1039
  tDestroySSubmitReq2(pSubmitReq, TSDB_MSG_FLG_DECODE);
H
Hongze Cheng 已提交
1040
  tDestroySSubmitRsp2(pSubmitRsp, TSDB_MSG_FLG_ENCODE);
H
Hongze Cheng 已提交
1041

D
dapan1121 已提交
1042 1043
  if (code) terrno = code;

H
Hongze Cheng 已提交
1044 1045
  return code;

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

  pRsp->code = 0;
C
Cary Xu 已提交
1058
  pSubmitReq->version = version;
1059
  statis.nBatchInsert = 1;
C
Cary Xu 已提交
1060

C
Cary Xu 已提交
1061 1062 1063 1064 1065
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

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

H
Hongze Cheng 已提交
1073
  for (;;) {
H
Hongze Cheng 已提交
1074 1075 1076
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
1077
    SSubmitBlkRsp submitBlkRsp = {0};
D
dapan1121 已提交
1078
    tbCreated = false;
H
Hongze Cheng 已提交
1079

H
Hongze Cheng 已提交
1080 1081
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102
      // 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 已提交
1103

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

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

D
dapan1121 已提交
1119 1120 1121 1122 1123
        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;
      }
1124

H
Hongze Cheng 已提交
1125 1126 1127 1128 1129 1130
      // msgIter.uid = createTbReq.uid;
      // if (createTbReq.type == TSDB_CHILD_TABLE) {
      //   msgIter.suid = createTbReq.ctb.suid;
      // } else {
      //   msgIter.suid = 0;
      // }
H
Hongze Cheng 已提交
1131

H
Hongze Cheng 已提交
1132 1133
      // tDecoderClear(&decoder);
      // taosArrayDestroy(createTbReq.ctb.tagName);
1134
    }
H
Hongze Cheng 已提交
1135

H
Hongze Cheng 已提交
1136
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
1137
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
1138 1139
    }

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

H
Hongze Cheng 已提交
1147 1148 1149 1150
  // 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 已提交
1151

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

H
Hongze Cheng 已提交
1154
_exit:
1155
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1156 1157 1158 1159 1160 1161
  // 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 已提交
1162

1163
  taosArrayDestroyEx(submitRsp.pArray, tFreeSSubmitBlkRsp);
H
Hongze Cheng 已提交
1164

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

1174
  // N.B. not strict as the following procedure is not atomic
1175 1176 1177
  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);
1178
  atomic_add_fetch_64(&pVnode->statis.nBatchInsertSuccess, statis.nBatchInsertSuccess);
1179

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

1186
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
1187
  SVCreateTSmaReq req = {0};
C
Cary Xu 已提交
1188
  SDecoder        coder = {0};
1189

C
Cary Xu 已提交
1190 1191 1192 1193 1194 1195
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
1196 1197 1198 1199 1200

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
1201 1202
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
1203
    goto _err;
1204
  }
C
Cary Xu 已提交
1205

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

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

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

/**
 * @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);
}
1234 1235 1236 1237 1238 1239 1240 1241 1242

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

1245
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
S
Shengliang Guan 已提交
1246 1247
  bool walChanged = false;
  bool tsdbChanged = false;
1248

S
Shengliang Guan 已提交
1249 1250
  SAlterVnodeConfigReq req = {0};
  if (tDeserializeSAlterVnodeConfigReq(pReq, len, &req) != 0) {
1251 1252 1253 1254
    terrno = TSDB_CODE_INVALID_MSG;
    return TSDB_CODE_INVALID_MSG;
  }

1255
  vInfo("vgId:%d, start to alter vnode config, page:%d pageSize:%d buffer:%d szPage:%d szBuf:%" PRIu64
S
Shengliang Guan 已提交
1256
        " cacheLast:%d cacheLastSize:%d days:%d keep0:%d keep1:%d keep2:%d fsync:%d level:%d",
1257 1258
        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 已提交
1259
        req.walFsyncPeriod, req.walLevel);
1260 1261 1262

  if (pVnode->config.cacheLastSize != req.cacheLastSize) {
    pVnode->config.cacheLastSize = req.cacheLastSize;
1263 1264
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
1265

1266
  if (pVnode->config.szBuf != req.buffer * 1024LL * 1024LL) {
S
Shengliang Guan 已提交
1267
    vInfo("vgId:%d, vnode buffer is changed from %" PRId64 " to %" PRId64, TD_VID(pVnode), pVnode->config.szBuf,
1268
          (uint64_t)(req.buffer * 1024LL * 1024LL));
1269
    pVnode->config.szBuf = req.buffer * 1024LL * 1024LL;
H
Hongze Cheng 已提交
1270 1271
  }

1272 1273
  if (pVnode->config.szCache != req.pages) {
    if (metaAlterCache(pVnode->pMeta, req.pages) < 0) {
S
Shengliang Guan 已提交
1274
      vError("vgId:%d, failed to change vnode pages from %d to %d failed since %s", TD_VID(pVnode),
1275
             pVnode->config.szCache, req.pages, tstrerror(errno));
H
Hongze Cheng 已提交
1276 1277
      return errno;
    } else {
S
Shengliang Guan 已提交
1278
      vInfo("vgId:%d, vnode pages is changed from %d to %d", TD_VID(pVnode), pVnode->config.szCache, req.pages);
1279
      pVnode->config.szCache = req.pages;
H
Hongze Cheng 已提交
1280
    }
H
Hongze Cheng 已提交
1281 1282
  }

1283 1284
  if (pVnode->config.cacheLast != req.cacheLast) {
    pVnode->config.cacheLast = req.cacheLast;
1285 1286
  }

1287 1288
  if (pVnode->config.walCfg.fsyncPeriod != req.walFsyncPeriod) {
    pVnode->config.walCfg.fsyncPeriod = req.walFsyncPeriod;
1289 1290 1291 1292

    walChanged = true;
  }

1293 1294
  if (pVnode->config.walCfg.level != req.walLevel) {
    pVnode->config.walCfg.level = req.walLevel;
1295 1296 1297 1298

    walChanged = true;
  }

1299 1300
  if (pVnode->config.tsdbCfg.keep0 != req.daysToKeep0) {
    pVnode->config.tsdbCfg.keep0 = req.daysToKeep0;
1301
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1302
      tsdbChanged = true;
1303 1304 1305
    }
  }

1306 1307
  if (pVnode->config.tsdbCfg.keep1 != req.daysToKeep1) {
    pVnode->config.tsdbCfg.keep1 = req.daysToKeep1;
1308
    if (!VND_IS_RSMA(pVnode)) {
M
Minglei Jin 已提交
1309
      tsdbChanged = true;
1310 1311 1312
    }
  }

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

  if (walChanged) {
M
Minglei Jin 已提交
1321 1322 1323 1324 1325
    walAlter(pVnode->pWal, &pVnode->config.walCfg);
  }

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

1328 1329 1330
  return 0;
}

1331 1332 1333 1334 1335 1336
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);

1337 1338 1339
  SMetaReader mr = {0};
  metaReaderInit(&mr, pVnode->pMeta, 0);

1340 1341 1342
  int32_t sz = taosArrayGetSize(deleteReq.deleteReqs);
  for (int32_t i = 0; i < sz; i++) {
    SSingleDeleteReq *pOneReq = taosArrayGet(deleteReq.deleteReqs, i);
1343 1344
    char             *name = pOneReq->tbname;
    if (metaGetTableEntryByName(&mr, name) < 0) {
S
Shengliang Guan 已提交
1345
      vDebug("vgId:%d, stream delete msg, skip since no table: %s", pVnode->config.vgId, name);
1346 1347 1348 1349 1350
      continue;
    }

    int64_t uid = mr.me.uid;

L
Liu Jicong 已提交
1351
    int32_t code = tsdbDeleteTableData(pVnode->pTsdb, version, deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1352 1353 1354
    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 已提交
1355
             TD_VID(pVnode), terrstr(), deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1356
    }
1357 1358

    tDecoderClear(&mr.coder);
1359
  }
1360
  metaReaderClear(&mr);
1361
  taosArrayDestroy(deleteReq.deleteReqs);
1362 1363 1364
  return 0;
}

H
Hongze Cheng 已提交
1365 1366 1367 1368 1369
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 已提交
1370 1371 1372 1373 1374
  pRsp->msgType = TDMT_VND_DELETE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

H
Hongze Cheng 已提交
1375 1376 1377 1378 1379 1380 1381 1382
  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);
1383
  ASSERT(taosArrayGetSize(pRes->uidList) == 0 || (pRes->skey != 0 && pRes->ekey != 0));
H
Hongze Cheng 已提交
1384 1385 1386 1387 1388 1389 1390 1391 1392

  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 已提交
1393 1394

  SVDeleteRsp rsp = {.affectedRows = pRes->affectedRows};
L
Liu Jicong 已提交
1395
  int32_t     ret = 0;
wmmhello's avatar
wmmhello 已提交
1396 1397
  tEncodeSize(tEncodeSVDeleteRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
L
Liu Jicong 已提交
1398
  SEncoder ec = {0};
wmmhello's avatar
wmmhello 已提交
1399 1400 1401
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVDeleteRsp(&ec, &rsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
1402 1403 1404 1405
  return code;

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