vnodeSvr.c 43.7 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 88 89
      uint64_t nSubmitTbData;
      if (tDecodeU64v(&dc, &nSubmitTbData) < 0) {
        code = TSDB_CODE_INVALID_MSG;
        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 102 103
        int32_t flags;
        if (tDecodeI32v(&dc, &flags) < 0) {
          code = TSDB_CODE_INVALID_MSG;
          goto _err;
        }

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

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

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

          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
          *(int64_t *)(dc.data + dc.pos + 8) = ctime;
H
Hongze Cheng 已提交
128

H
Hongze Cheng 已提交
129
          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
  ASSERT(pVnode->state.applyTerm <= pMsg->info.conn.applyTerm);
213
  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;

H
Hongze Cheng 已提交
220 221 222
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
223 224

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

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

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

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

H
Hongze Cheng 已提交
337
  // commit if need
H
Hongze Cheng 已提交
338
  if (vnodeShouldCommit(pVnode)) {
S
Shengliang Guan 已提交
339
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
340
    vnodeAsyncCommit(pVnode);
H
Hongze Cheng 已提交
341 342

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

H
Hongze Cheng 已提交
349
_exit:
H
Hongze Cheng 已提交
350
  return 0;
H
Hongze Cheng 已提交
351 352

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

end:
  taosArrayDestroy(tbUids);
  return ret;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
679 680 681
  return 0;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

824
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);
825
  tdUpdateTbUidList(pVnode->pSma, pStore, false);
826

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

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

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

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

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

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

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

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

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

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

    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 已提交
951 952
  }

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

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

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

        taosArrayPush(newTbUids, &pSubmitTbData->uid);

        if (pCreateTbRsp->pMeta) {
D
dapan1121 已提交
991
          vnodeUpdateMetaRsp(pVnode, pCreateTbRsp->pMeta);
H
Hongze Cheng 已提交
992 993 994 995 996 997
        }
      } else {  // create table failed
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
          code = terrno;
          goto _exit;
        }
H
Hongze Cheng 已提交
998
      }
H
Hongze Cheng 已提交
999 1000 1001
    }

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

    pSubmitRsp->affectedRows += affectedRows;
H
Hongze Cheng 已提交
1007
  }
H
Hongze Cheng 已提交
1008

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

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

H
Hongze Cheng 已提交
1034 1035
  // clear
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
1036
  tDestroySSubmitReq2(pSubmitReq, TSDB_MSG_FLG_DECODE);
H
Hongze Cheng 已提交
1037
  tDestroySSubmitRsp2(pSubmitRsp, TSDB_MSG_FLG_ENCODE);
H
Hongze Cheng 已提交
1038

D
dapan1121 已提交
1039 1040
  if (code) terrno = code;

H
Hongze Cheng 已提交
1041 1042
  return code;

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

  pRsp->code = 0;
C
Cary Xu 已提交
1055
  pSubmitReq->version = version;
1056
  statis.nBatchInsert = 1;
C
Cary Xu 已提交
1057

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

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

H
Hongze Cheng 已提交
1070
  for (;;) {
H
Hongze Cheng 已提交
1071 1072 1073
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
1074
    SSubmitBlkRsp submitBlkRsp = {0};
D
dapan1121 已提交
1075
    tbCreated = false;
H
Hongze Cheng 已提交
1076

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

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

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

D
dapan1121 已提交
1116 1117 1118 1119 1120
        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;
      }
1121

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

H
Hongze Cheng 已提交
1129 1130
      // tDecoderClear(&decoder);
      // taosArrayDestroy(createTbReq.ctb.tagName);
1131
    }
H
Hongze Cheng 已提交
1132

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

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

H
Hongze Cheng 已提交
1144 1145 1146 1147
  // 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 已提交
1148

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

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

1160
  taosArrayDestroyEx(submitRsp.pArray, tFreeSSubmitBlkRsp);
H
Hongze Cheng 已提交
1161

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

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

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

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

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

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

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

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

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

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

/**
 * @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);
}
1231 1232 1233 1234 1235 1236 1237 1238 1239

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

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

1245 1246
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
1247 1248 1249
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
1250
}
H
Hongze Cheng 已提交
1251

1252
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
S
Shengliang Guan 已提交
1253 1254
  bool walChanged = false;
  bool tsdbChanged = false;
1255

S
Shengliang Guan 已提交
1256 1257
  SAlterVnodeConfigReq req = {0};
  if (tDeserializeSAlterVnodeConfigReq(pReq, len, &req) != 0) {
1258 1259 1260 1261
    terrno = TSDB_CODE_INVALID_MSG;
    return TSDB_CODE_INVALID_MSG;
  }

1262
  vInfo("vgId:%d, start to alter vnode config, page:%d pageSize:%d buffer:%d szPage:%d szBuf:%" PRIu64
S
Shengliang Guan 已提交
1263
        " cacheLast:%d cacheLastSize:%d days:%d keep0:%d keep1:%d keep2:%d fsync:%d level:%d",
1264 1265
        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 已提交
1266
        req.walFsyncPeriod, req.walLevel);
1267 1268 1269

  if (pVnode->config.cacheLastSize != req.cacheLastSize) {
    pVnode->config.cacheLastSize = req.cacheLastSize;
1270 1271
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
1272

1273
  if (pVnode->config.szBuf != req.buffer * 1024LL * 1024LL) {
S
Shengliang Guan 已提交
1274
    vInfo("vgId:%d, vnode buffer is changed from %" PRId64 " to %" PRId64, TD_VID(pVnode), pVnode->config.szBuf,
1275
          (uint64_t)(req.buffer * 1024LL * 1024LL));
1276
    pVnode->config.szBuf = req.buffer * 1024LL * 1024LL;
H
Hongze Cheng 已提交
1277 1278
  }

1279 1280
  if (pVnode->config.szCache != req.pages) {
    if (metaAlterCache(pVnode->pMeta, req.pages) < 0) {
S
Shengliang Guan 已提交
1281
      vError("vgId:%d, failed to change vnode pages from %d to %d failed since %s", TD_VID(pVnode),
1282
             pVnode->config.szCache, req.pages, tstrerror(errno));
H
Hongze Cheng 已提交
1283 1284
      return errno;
    } else {
S
Shengliang Guan 已提交
1285
      vInfo("vgId:%d, vnode pages is changed from %d to %d", TD_VID(pVnode), pVnode->config.szCache, req.pages);
1286
      pVnode->config.szCache = req.pages;
H
Hongze Cheng 已提交
1287
    }
H
Hongze Cheng 已提交
1288 1289
  }

1290 1291
  if (pVnode->config.cacheLast != req.cacheLast) {
    pVnode->config.cacheLast = req.cacheLast;
1292 1293
  }

1294 1295
  if (pVnode->config.walCfg.fsyncPeriod != req.walFsyncPeriod) {
    pVnode->config.walCfg.fsyncPeriod = req.walFsyncPeriod;
1296 1297 1298 1299

    walChanged = true;
  }

1300 1301
  if (pVnode->config.walCfg.level != req.walLevel) {
    pVnode->config.walCfg.level = req.walLevel;
1302 1303 1304 1305

    walChanged = true;
  }

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

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

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

  if (walChanged) {
M
Minglei Jin 已提交
1328 1329 1330 1331 1332
    walAlter(pVnode->pWal, &pVnode->config.walCfg);
  }

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

1335 1336 1337
  return 0;
}

1338 1339 1340 1341 1342 1343
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);

1344 1345 1346
  SMetaReader mr = {0};
  metaReaderInit(&mr, pVnode->pMeta, 0);

1347 1348 1349
  int32_t sz = taosArrayGetSize(deleteReq.deleteReqs);
  for (int32_t i = 0; i < sz; i++) {
    SSingleDeleteReq *pOneReq = taosArrayGet(deleteReq.deleteReqs, i);
1350 1351 1352 1353 1354 1355 1356 1357
    char             *name = pOneReq->tbname;
    if (metaGetTableEntryByName(&mr, name) < 0) {
      vDebug("stream delete msg, skip vgId:%d since no table: %s", pVnode->config.vgId, name);
      continue;
    }

    int64_t uid = mr.me.uid;

L
Liu Jicong 已提交
1358
    int32_t code = tsdbDeleteTableData(pVnode->pTsdb, version, deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1359 1360 1361
    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 已提交
1362
             TD_VID(pVnode), terrstr(), deleteReq.suid, uid, pOneReq->startTs, pOneReq->endTs);
1363
    }
1364 1365

    tDecoderClear(&mr.coder);
1366
  }
1367
  metaReaderClear(&mr);
1368
  taosArrayDestroy(deleteReq.deleteReqs);
1369 1370 1371
  return 0;
}

H
Hongze Cheng 已提交
1372 1373 1374 1375 1376
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 已提交
1377 1378 1379 1380 1381
  pRsp->msgType = TDMT_VND_DELETE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

H
Hongze Cheng 已提交
1382 1383 1384 1385 1386 1387 1388 1389
  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);
L
Liu Jicong 已提交
1390
  ASSERT(taosArrayGetSize(pRes->uidList) == 0 || (pRes->skey != 0 && pRes->ekey != 0));
H
Hongze Cheng 已提交
1391 1392 1393 1394 1395 1396 1397 1398 1399

  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 已提交
1400 1401

  SVDeleteRsp rsp = {.affectedRows = pRes->affectedRows};
L
Liu Jicong 已提交
1402
  int32_t     ret = 0;
wmmhello's avatar
wmmhello 已提交
1403 1404
  tEncodeSize(tEncodeSVDeleteRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
L
Liu Jicong 已提交
1405
  SEncoder ec = {0};
wmmhello's avatar
wmmhello 已提交
1406 1407 1408
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVDeleteRsp(&ec, &rsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
1409 1410 1411 1412
  return code;

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