vnodeSvr.c 31.9 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);
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 43 44 45 46 47
  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));
      tStartDecode(&dc);

      tDecodeI32v(&dc, &nReqs);
      for (int32_t iReq = 0; iReq < nReqs; iReq++) {
        tb_uid_t uid = tGenIdPI64();
H
Hongze Cheng 已提交
48
        char    *name = NULL;
H
Hongze Cheng 已提交
49 50 51
        tStartDecode(&dc);

        tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
52
        tDecodeCStr(&dc, &name);
H
Hongze Cheng 已提交
53 54 55
        *(int64_t *)(dc.data + dc.pos) = uid;
        *(int64_t *)(dc.data + dc.pos + 8) = ctime;

56
        vTrace("vgId:%d, table:%s uid:%" PRId64 " is generated", pVnode->config.vgId, name, uid);
H
Hongze Cheng 已提交
57 58 59 60 61 62 63 64 65 66 67
        tEndDecode(&dc);
      }

      tEndDecode(&dc);
      tDecoderClear(&dc);
    } break;
    case TDMT_VND_SUBMIT: {
      SSubmitMsgIter msgIter = {0};
      SSubmitReq    *pSubmitReq = (SSubmitReq *)pMsg->pCont;
      SSubmitBlk    *pBlock = NULL;
      int64_t        ctime = taosGetTimestampMs();
H
Hongze Cheng 已提交
68
      tb_uid_t       uid;
H
Hongze Cheng 已提交
69 70 71 72 73 74 75 76

      tInitSubmitMsgIter(pSubmitReq, &msgIter);

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

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

H
Hongze Cheng 已提交
79 80 81 82
          tDecoderInit(&dc, pBlock->data, msgIter.schemaLen);
          tStartDecode(&dc);

          tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
83 84 85 86 87 88
          tDecodeCStr(&dc, &name);

          uid = metaGetTableEntryUidByName(pVnode->pMeta, name);
          if (uid == 0) {
            uid = tGenIdPI64();
          }
H
Hongze Cheng 已提交
89
          *(int64_t *)(dc.data + dc.pos) = uid;
H
Hongze Cheng 已提交
90
          *(int64_t *)(dc.data + dc.pos + 8) = ctime;
H
Hongze Cheng 已提交
91
          pBlock->uid = htobe64(uid);
H
Hongze Cheng 已提交
92 93 94 95 96 97 98

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

    } break;
H
Hongze Cheng 已提交
99 100 101 102 103 104 105 106 107 108
    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 已提交
109 110 111
      if (code) {
        goto _err;
      }
H
Hongze Cheng 已提交
112 113 114 115 116

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

117 118
      ((SMsgHead *)pCont)->contLen = size + sizeof(SMsgHead);
      ((SMsgHead *)pCont)->vgId = TD_VID(pVnode);
H
Hongze Cheng 已提交
119 120 121 122 123 124 125 126 127 128 129

      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 已提交
130 131 132
    default:
      break;
  }
H
Hongze Cheng 已提交
133

S
Shengliang Guan 已提交
134
  return code;
H
Hongze Cheng 已提交
135 136 137 138

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

141
int32_t vnodeProcessWriteMsg(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
142 143 144 145
  void   *ptr = NULL;
  void   *pReq;
  int32_t len;
  int32_t ret;
H
Hongze Cheng 已提交
146

147
  vTrace("vgId:%d, start to process write request %s, index:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
148
         version);
H
Hongze Cheng 已提交
149

H
Hongze Cheng 已提交
150
  pVnode->state.applied = version;
H
Hongze Cheng 已提交
151
  pVnode->state.applyTerm = pMsg->info.conn.applyTerm;
H
Hongze Cheng 已提交
152

H
Hongze Cheng 已提交
153 154 155
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
156 157

  switch (pMsg->msgType) {
H
Hongze Cheng 已提交
158
    /* META */
H
Hongze Cheng 已提交
159
    case TDMT_VND_CREATE_STB:
H
Hongze Cheng 已提交
160
      if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
161
      break;
H
Hongze Cheng 已提交
162
    case TDMT_VND_ALTER_STB:
H
Hongze Cheng 已提交
163
      if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
164
      break;
H
Hongze Cheng 已提交
165
    case TDMT_VND_DROP_STB:
H
Hongze Cheng 已提交
166
      if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
167
      break;
H
Hongze Cheng 已提交
168
    case TDMT_VND_CREATE_TABLE:
H
Hongze Cheng 已提交
169
      if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
170 171
      break;
    case TDMT_VND_ALTER_TABLE:
H
Hongze Cheng 已提交
172
      if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
173
      break;
H
Hongze Cheng 已提交
174
    case TDMT_VND_DROP_TABLE:
H
Hongze Cheng 已提交
175
      if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
176
      break;
177
    case TDMT_VND_DROP_TTL_TABLE:
178
      if (vnodeProcessDropTtlTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
179
      break;
S
Shengliang Guan 已提交
180 181 182 183
    case TDMT_VND_TRIM:
      if (vnodeProcessTrimReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
      break;
    case TDMT_VND_CREATE_SMA:
184
      if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
S
Shengliang Guan 已提交
185
      break;
H
Hongze Cheng 已提交
186
    /* TSDB */
H
Hongze Cheng 已提交
187
    case TDMT_VND_SUBMIT:
H
Hongze Cheng 已提交
188
      if (vnodeProcessSubmitReq(pVnode, version, pMsg->pCont, pMsg->contLen, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
189
      break;
D
dapan1121 已提交
190
    case TDMT_VND_DELETE:
H
Hongze Cheng 已提交
191
      if (vnodeProcessDeleteReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
D
dapan1121 已提交
192
      break;
H
Hongze Cheng 已提交
193
    /* TQ */
L
Liu Jicong 已提交
194 195 196
    case TDMT_VND_MQ_VG_CHANGE:
      if (tqProcessVgChangeReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                               pMsg->contLen - sizeof(SMsgHead)) < 0) {
197
        goto _err;
L
Liu Jicong 已提交
198 199
      }
      break;
L
Liu Jicong 已提交
200 201
    case TDMT_VND_MQ_VG_DELETE:
      if (tqProcessVgDeleteReq(pVnode->pTq, pMsg->pCont, pMsg->contLen) < 0) {
202 203 204 205 206 207 208
        goto _err;
      }
      break;
    case TDMT_VND_MQ_COMMIT_OFFSET:
      if (tqProcessOffsetCommitReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                                   pMsg->contLen - sizeof(SMsgHead)) < 0) {
        goto _err;
L
Liu Jicong 已提交
209 210
      }
      break;
L
Liu Jicong 已提交
211 212 213 214 215 216
    case TDMT_VND_CHECK_ALTER_INFO:
      if (tqProcessCheckAlterInfoReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                                     pMsg->contLen - sizeof(SMsgHead)) < 0) {
        goto _err;
      }
      break;
217
    case TDMT_STREAM_TASK_DEPLOY: {
L
Liu Jicong 已提交
218 219
      if (tqProcessTaskDeployReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                                 pMsg->contLen - sizeof(SMsgHead)) < 0) {
220
        goto _err;
H
Hongze Cheng 已提交
221 222
      }
    } break;
L
Liu Jicong 已提交
223
    case TDMT_STREAM_TASK_DROP: {
L
Liu Jicong 已提交
224 225 226 227
      if (tqProcessTaskDropReq(pVnode->pTq, pMsg->pCont, pMsg->contLen) < 0) {
        goto _err;
      }
    } break;
228 229 230
    case TDMT_VND_ALTER_CONFIRM:
      vnodeProcessAlterConfirmReq(pVnode, version, pReq, len, pRsp);
      break;
S
Shengliang Guan 已提交
231
    case TDMT_VND_ALTER_HASHRANGE:
232
      vnodeProcessAlterHashRangeReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
233
      break;
234
    case TDMT_VND_ALTER_CONFIG:
235
      vnodeProcessAlterConfigReq(pVnode, version, pReq, len, pRsp);
S
Shengliang Guan 已提交
236
      break;
H
Hongze Cheng 已提交
237 238
    case TDMT_VND_COMMIT:
      goto _do_commit;
H
Hongze Cheng 已提交
239 240 241 242 243
    default:
      ASSERT(0);
      break;
  }

244
  vTrace("vgId:%d, process %s request success, index:%" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType), version);
H
Hongze Cheng 已提交
245

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

H
Hongze Cheng 已提交
251
  // commit if need
H
Hongze Cheng 已提交
252
  if (vnodeShouldCommit(pVnode)) {
H
Hongze Cheng 已提交
253
  _do_commit:
S
Shengliang Guan 已提交
254
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
255 256 257 258 259
    // commit current change
    vnodeCommit(pVnode);

    // start a new one
    vnodeBegin(pVnode);
H
Hongze Cheng 已提交
260 261 262
  }

  return 0;
H
Hongze Cheng 已提交
263 264

_err:
S
Shengliang Guan 已提交
265
  vError("vgId:%d, process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
266 267
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
268 269
}

270
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
271
  if (TDMT_SCH_QUERY != pMsg->msgType && TDMT_SCH_MERGE_QUERY != pMsg->msgType) {
D
dapan1121 已提交
272 273 274 275 276 277 278
    return 0;
  }

  return qWorkerPreprocessQueryMsg(pVnode->pQuery, pMsg);
}

int32_t vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
279
  vTrace("message in vnode query queue is processing");
D
dapan1121 已提交
280
  if ((pMsg->msgType == TDMT_SCH_QUERY) && !vnodeIsLeader(pVnode)) {
281 282 283 284
    vnodeRedirectRpcMsg(pVnode, pMsg);
    return 0;
  }

285
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
286
  switch (pMsg->msgType) {
D
dapan1121 已提交
287
    case TDMT_SCH_QUERY:
D
dapan1121 已提交
288
    case TDMT_SCH_MERGE_QUERY:
D
dapan1121 已提交
289
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
290
    case TDMT_SCH_QUERY_CONTINUE:
D
dapan1121 已提交
291
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
292 293 294 295 296 297
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

298
int32_t vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
H
Hongze Cheng 已提交
299
  vTrace("message in fetch queue is processing");
M
Minglei Jin 已提交
300
  if ((pMsg->msgType == TDMT_SCH_FETCH || pMsg->msgType == TDMT_VND_TABLE_META ||
301
       pMsg->msgType == TDMT_VND_TABLE_CFG || pMsg->msgType == TDMT_VND_BATCH_META) &&
M
Minglei Jin 已提交
302
      !vnodeIsLeader(pVnode)) {
303 304 305 306
    vnodeRedirectRpcMsg(pVnode, pMsg);
    return 0;
  }

H
Hongze Cheng 已提交
307 308
  char   *msgstr = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  int32_t msgLen = pMsg->contLen - sizeof(SMsgHead);
C
Cary Xu 已提交
309

H
Hongze Cheng 已提交
310
  switch (pMsg->msgType) {
D
dapan1121 已提交
311
    case TDMT_SCH_FETCH:
D
dapan1121 已提交
312
    case TDMT_SCH_MERGE_FETCH:
D
dapan1121 已提交
313
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
314
    case TDMT_SCH_FETCH_RSP:
D
dapan1121 已提交
315
      return qWorkerProcessRspMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
316
    case TDMT_SCH_CANCEL_TASK:
D
dapan1121 已提交
317
      return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
318
    case TDMT_SCH_DROP_TASK:
D
dapan1121 已提交
319
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
320
    case TDMT_SCH_QUERY_HEARTBEAT:
D
dapan1121 已提交
321
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
322
    case TDMT_VND_TABLE_META:
D
dapan1121 已提交
323
      return vnodeGetTableMeta(pVnode, pMsg, true);
D
dapan1121 已提交
324
    case TDMT_VND_TABLE_CFG:
D
dapan1121 已提交
325 326 327
      return vnodeGetTableCfg(pVnode, pMsg, true);
    case TDMT_VND_BATCH_META:
      return vnodeGetBatchMeta(pVnode, pMsg);
H
Hongze Cheng 已提交
328
    case TDMT_VND_CONSUME:
L
Liu Jicong 已提交
329
      return tqProcessPollReq(pVnode->pTq, pMsg);
330 331 332
    case TDMT_STREAM_TASK_RUN:
      return tqProcessTaskRunReq(pVnode->pTq, pMsg);
    case TDMT_STREAM_TASK_DISPATCH:
L
Liu Jicong 已提交
333
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg);
334
    case TDMT_STREAM_TASK_RECOVER:
L
Liu Jicong 已提交
335
      return tqProcessTaskRecoverReq(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
336 337
    case TDMT_STREAM_RETRIEVE:
      return tqProcessTaskRetrieveReq(pVnode->pTq, pMsg);
338
    case TDMT_STREAM_TASK_DISPATCH_RSP:
L
Liu Jicong 已提交
339
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
340
    case TDMT_STREAM_TASK_RECOVER_RSP:
L
Liu Jicong 已提交
341
      return tqProcessTaskRecoverRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
342 343
    case TDMT_STREAM_RETRIEVE_RSP:
      return tqProcessTaskRetrieveRsp(pVnode->pTq, pMsg);
H
Hongze Cheng 已提交
344 345 346 347 348 349 350 351 352 353
    default:
      vError("unknown msg type:%d in fetch queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

// TODO: remove the function
void smaHandleRes(void *pVnode, int64_t smaId, const SArray *data) {
  // TODO

S
slzhou 已提交
354
  blockDebugShowDataBlocks(data, __func__);
355
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
356 357
}

D
dapan1121 已提交
358 359 360 361 362 363 364
void vnodeUpdateMetaRsp(SVnode *pVnode, STableMetaRsp *pMetaRsp) {
  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 已提交
365
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
366
  int32_t     code = 0;
S
Shengliang Guan 已提交
367 368 369
  SVTrimDbReq trimReq = {0};

  vInfo("vgId:%d, trim vnode request will be processed, time:%d", pVnode->config.vgId, trimReq.timestamp);
H
Hongze Cheng 已提交
370 371 372 373 374

  // decode
  if (tDeserializeSVTrimDbReq(pReq, len, &trimReq) != 0) {
    code = TSDB_CODE_INVALID_MSG;
    goto _exit;
S
Shengliang Guan 已提交
375 376
  }

H
Hongze Cheng 已提交
377 378 379 380 381 382
  // process
  code = tsdbDoRetention(pVnode->pTsdb, trimReq.timestamp);
  if (code) goto _exit;

_exit:
  return code;
S
Shengliang Guan 已提交
383 384
}

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

S
Shengliang Guan 已提交
389 390 391 392 393 394
  SVDropTtlTableReq ttlReq = {0};
  if (tDeserializeSVDropTtlTableReq(pReq, len, &ttlReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto end;
  }

395
  vDebug("vgId:%d, drop ttl table req will be processed, time:%d", pVnode->config.vgId, ttlReq.timestamp);
S
Shengliang Guan 已提交
396
  int32_t ret = metaTtlDropTable(pVnode->pMeta, ttlReq.timestamp, tbUids);
L
Liu Jicong 已提交
397
  if (ret != 0) {
398 399
    goto end;
  }
L
Liu Jicong 已提交
400
  if (taosArrayGetSize(tbUids) > 0) {
wmmhello's avatar
wmmhello 已提交
401 402
    tqUpdateTbUidList(pVnode->pTq, tbUids, false);
  }
403 404 405 406 407 408

end:
  taosArrayDestroy(tbUids);
  return ret;
}

409
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
410
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
411
  SDecoder       coder;
H
Hongze Cheng 已提交
412

H
Hongze Cheng 已提交
413 414 415 416 417 418
  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 已提交
419
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
420 421

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
422 423
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
424 425
  }

H
Hongze Cheng 已提交
426
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
427 428
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
429 430
  }

431
  if (tdProcessRSmaCreate(pVnode->pSma, &req) < 0) {
432 433 434
    pRsp->code = terrno;
    goto _err;
  }
C
Cary Xu 已提交
435

M
Minglei Jin 已提交
436 437
  // taosMemoryFree(req.schemaRow.pSchema);
  // taosMemoryFree(req.schemaTag.pSchema);
H
Hongze Cheng 已提交
438
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
439
  return 0;
H
Hongze Cheng 已提交
440 441

_err:
M
Minglei Jin 已提交
442 443
  taosMemoryFree(req.schemaRow.pSchema);
  taosMemoryFree(req.schemaTag.pSchema);
H
Hongze Cheng 已提交
444
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
445
  return -1;
H
Hongze Cheng 已提交
446 447
}

448
static int32_t vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
449
  SDecoder           decoder = {0};
450
  int32_t            rcode = 0;
H
Hongze Cheng 已提交
451 452
  SVCreateTbBatchReq req = {0};
  SVCreateTbReq     *pCreateReq;
H
Hongze Cheng 已提交
453 454 455
  SVCreateTbBatchRsp rsp = {0};
  SVCreateTbRsp      cRsp = {0};
  char               tbName[TSDB_TABLE_FNAME_LEN];
C
Cary Xu 已提交
456
  STbUidStore       *pStore = NULL;
457
  SArray            *tbUids = NULL;
H
Hongze Cheng 已提交
458 459

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
460 461 462
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
463

H
Hongze Cheng 已提交
464
  // decode
H
Hongze Cheng 已提交
465 466
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
467 468 469 470
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
471

H
Hongze Cheng 已提交
472
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
473 474
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
475 476 477 478 479
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
480
  // loop to create table
481
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
482
    pCreateReq = req.pReqs + iReq;
H
Hongze Cheng 已提交
483

C
Cary Xu 已提交
484 485 486 487 488
    if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
      rcode = -1;
      goto _exit;
    }
    
H
Hongze Cheng 已提交
489 490 491 492 493 494 495 496 497
    // 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
H
Hongze Cheng 已提交
498
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq) < 0) {
H
Hongze Cheng 已提交
499 500 501 502 503
      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 已提交
504
    } else {
H
Hongze Cheng 已提交
505
      cRsp.code = TSDB_CODE_SUCCESS;
506
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
507
      taosArrayPush(tbUids, &pCreateReq->uid);
H
Hongze Cheng 已提交
508
    }
H
Hongze Cheng 已提交
509 510

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

513
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
514 515
  tdUpdateTbUidList(pVnode->pSma, pStore);
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
516

H
Hongze Cheng 已提交
517
  // prepare rsp
H
Hongze Cheng 已提交
518 519
  SEncoder encoder = {0};
  int32_t  ret = 0;
wafwerar's avatar
wafwerar 已提交
520
  tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
H
Hongze Cheng 已提交
521 522 523 524 525 526
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  if (pRsp->pCont == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    rcode = -1;
    goto _exit;
  }
H
Hongze Cheng 已提交
527 528
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
H
Hongze Cheng 已提交
529

H
Hongze Cheng 已提交
530
_exit:
wmmhello's avatar
wmmhello 已提交
531 532 533 534
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
    pCreateReq = req.pReqs + iReq;
    taosArrayDestroy(pCreateReq->ctb.tagName);
  }
H
Hongze Cheng 已提交
535
  taosArrayDestroy(rsp.pArray);
536
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
537 538
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
539
  return rcode;
H
Hongze Cheng 已提交
540 541
}

542
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
  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 已提交
558
  }
H
Hongze Cheng 已提交
559 560 561 562 563 564 565 566 567

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
568 569 570
  return 0;
}

571
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
572
  SVDropStbReq req = {0};
573
  int32_t      rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
574
  SDecoder     decoder = {0};
575
  SArray      *tbUidList = NULL;
H
Hongze Cheng 已提交
576 577 578 579 580 581

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

  // decode request
H
Hongze Cheng 已提交
582 583
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
584 585 586 587 588
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
589 590 591 592 593 594 595 596
  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) {
597 598 599
    rcode = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
600

601 602 603 604 605
  if (tdProcessRSmaDrop(pVnode->pSma, &req) < 0) {
    rcode = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
606 607
  // return rsp
_exit:
608
  if (tbUidList) taosArrayDestroy(tbUidList);
H
Hongze Cheng 已提交
609
  pRsp->code = rcode;
H
Hongze Cheng 已提交
610
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
611 612 613
  return 0;
}

614
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
D
dapan1121 已提交
615 616 617
  SVAlterTbReq  vAlterTbReq = {0};
  SVAlterTbRsp  vAlterTbRsp = {0};
  SDecoder      dc = {0};
618 619
  int32_t       rcode = 0;
  int32_t       ret;
D
dapan1121 已提交
620 621
  SEncoder      ec = {0};
  STableMetaRsp vMetaRsp = {0};
H
Hongze Cheng 已提交
622 623 624 625 626 627 628 629 630 631

  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 已提交
632
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
633
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
634 635
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
636 637 638
  }

  // process
D
dapan1121 已提交
639
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq, &vMetaRsp) < 0) {
wmmhello's avatar
wmmhello 已提交
640
    vAlterTbRsp.code = terrno;
H
Hongze Cheng 已提交
641
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
642 643
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
644 645
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
646

D
dapan1121 已提交
647 648 649 650 651
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
652 653 654 655 656 657
_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 已提交
658 659 660
  if (vMetaRsp.pSchemas) {
    taosMemoryFree(vMetaRsp.pSchemas);
  }
H
Hongze Cheng 已提交
661 662 663
  return 0;
}

664
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
665 666
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
667
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
668
  SEncoder         encoder = {0};
669
  int32_t          ret;
670
  SArray          *tbUids = NULL;
H
Hongze Cheng 已提交
671

H
Hongze Cheng 已提交
672
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
673 674 675
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
676 677

  // decode req
H
Hongze Cheng 已提交
678 679
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
680 681 682 683 684
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
685 686

  // process req
687
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
688
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
689 690
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

691
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
692 693
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
H
Hongze Cheng 已提交
694

H
Hongze Cheng 已提交
695
    /* code */
696
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids);
H
Hongze Cheng 已提交
697
    if (ret < 0) {
H
Hongze Cheng 已提交
698 699 700 701 702
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_VND_TABLE_NOT_EXIST) {
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
703
    } else {
H
Hongze Cheng 已提交
704
      dropTbRsp.code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
705 706 707 708 709
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

710 711
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);

H
Hongze Cheng 已提交
712
_exit:
713
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
714
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
715 716 717 718 719
  tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVDropTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
720
  taosArrayDestroy(rsp.pArray);
H
Hongze Cheng 已提交
721 722 723
  return 0;
}

724 725
static int32_t vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter,
                                              const char *tags) {
D
dapan 已提交
726 727 728 729
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
730
  int32_t        rv = -1;
D
dapan 已提交
731 732 733

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
C
Cary Xu 已提交
734
  if (!pSchema || (suid != msgIter->suid) || rv != TD_ROW_SVER(blkIter.row)) {
D
dapan 已提交
735 736 737
    if (pSchema) {
      taosMemoryFreeClear(pSchema);
    }
C
Cary Xu 已提交
738
    pSchema = metaGetTbTSchema(pMeta, msgIter->suid, TD_ROW_SVER(blkIter.row));  // TODO: use the real schema
D
dapan 已提交
739 740
    if (pSchema) {
      suid = msgIter->suid;
C
Cary Xu 已提交
741
      rv = TD_ROW_SVER(blkIter.row);
D
dapan 已提交
742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
    }
  }
  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;
}

759
static int32_t vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
C
Cary Xu 已提交
760 761 762 763 764 765 766 767 768
  ASSERT(pMsg != NULL);
  SSubmitMsgIter msgIter = {0};
  SMeta         *pMeta = pVnode->pMeta;
  SSubmitBlk    *pBlock = NULL;

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

D
dapan 已提交
770 771
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
772 773 774 775

  return 0;
}

776
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
777
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
778
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
779 780 781 782
  SSubmitMsgIter msgIter = {0};
  SSubmitBlk    *pBlock;
  SSubmitRsp     rsp = {0};
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
783
  SDecoder       decoder = {0};
H
Hongze Cheng 已提交
784
  int32_t        nRows;
H
Hongze Cheng 已提交
785 786
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
787
  SArray        *newTbUids = NULL;
788
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
789 790

  pRsp->code = 0;
C
Cary Xu 已提交
791
  pSubmitReq->version = version;
C
Cary Xu 已提交
792

C
Cary Xu 已提交
793 794 795 796
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
797 798 799 800 801
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
802
  // handle the request
H
Hongze Cheng 已提交
803 804 805
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
806 807
  }

C
Cary Xu 已提交
808 809
  submitRsp.pArray = taosArrayInit(msgIter.numOfBlocks, sizeof(SSubmitBlkRsp));
  newTbUids = taosArrayInit(msgIter.numOfBlocks, sizeof(int64_t));
C
Cary Xu 已提交
810
  if (!submitRsp.pArray || !newTbUids) {
811 812 813 814
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
815
  for (;;) {
H
Hongze Cheng 已提交
816 817 818
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
819 820
    SSubmitBlkRsp submitBlkRsp = {0};

H
Hongze Cheng 已提交
821 822
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
823 824
      submitBlkRsp.hashMeta = 1;

H
Hongze Cheng 已提交
825 826
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
827
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
828
        tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
829
        taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
830 831 832
        goto _exit;
      }

C
Cary Xu 已提交
833 834 835 836 837 838 839
      if ((terrno = grantCheck(TSDB_GRANT_TIMESERIES)) < 0) {
        pRsp->code = terrno;
        tDecoderClear(&decoder);
        taosArrayDestroy(createTbReq.ctb.tagName);
        goto _exit;
      }

H
Hongze Cheng 已提交
840 841
      if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
842
          submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
843
          tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
844
          taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
845 846 847
          goto _exit;
        }
      }
848
      taosArrayPush(newTbUids, &createTbReq.uid);
H
Hongze Cheng 已提交
849

H
Hongze Cheng 已提交
850
      submitBlkRsp.uid = createTbReq.uid;
D
dapan 已提交
851
      submitBlkRsp.tblFName = taosMemoryMalloc(strlen(pVnode->config.dbname) + strlen(createTbReq.name) + 2);
D
dapan 已提交
852
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
853

H
Hongze Cheng 已提交
854 855 856 857 858 859 860
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

wmmhello's avatar
wmmhello 已提交
861
#ifdef TD_DEBUG_PRINT_ROW
D
dapan 已提交
862
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
wmmhello's avatar
wmmhello 已提交
863
#endif
H
Hongze Cheng 已提交
864
      tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
865
      taosArrayDestroy(createTbReq.ctb.tagName);
D
dapan1121 已提交
866 867 868
    } else {
      submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
869 870
    }

H
Hongze Cheng 已提交
871
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
872
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
873 874
    }

H
Hongze Cheng 已提交
875 876 877
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
    taosArrayPush(submitRsp.pArray, &submitBlkRsp);
H
Hongze Cheng 已提交
878
  }
879
  tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
880

H
Hongze Cheng 已提交
881
_exit:
882
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
883 884 885 886 887 888 889 890
  tEncodeSize(tEncodeSSubmitRsp, &submitRsp, tsize, ret);
  pRsp->pCont = rpcMallocCont(tsize);
  pRsp->contLen = tsize;
  tEncoderInit(&encoder, pRsp->pCont, tsize);
  tEncodeSSubmitRsp(&encoder, &submitRsp);
  tEncoderClear(&encoder);

  for (int32_t i = 0; i < taosArrayGetSize(submitRsp.pArray); i++) {
D
dapan 已提交
891
    taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
H
Hongze Cheng 已提交
892 893 894
  }

  taosArrayDestroy(submitRsp.pArray);
H
Hongze Cheng 已提交
895

896
  // TODO: the partial success scenario and the error case
H
Hongze Cheng 已提交
897 898
  // => If partial success, extract the success submitted rows and reconstruct a new submit msg, and push to level
  // 1/level 2.
899
  // TODO: refactor
C
Cary Xu 已提交
900
  if ((terrno == TSDB_CODE_SUCCESS) && (pRsp->code == TSDB_CODE_SUCCESS)) {
L
Liu Jicong 已提交
901
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_INPUT__DATA_SUBMIT);
902
  }
C
Cary Xu 已提交
903

L
Liu Jicong 已提交
904 905
  vDebug("successful submit in vg %d version %ld", pVnode->config.vgId, version);

H
Hongze Cheng 已提交
906
  return 0;
L
Liu Jicong 已提交
907
}
908

909
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
910
  SVCreateTSmaReq req = {0};
H
Hongze Cheng 已提交
911
  SDecoder        coder;
912

C
Cary Xu 已提交
913 914 915 916 917 918
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
919 920 921 922 923

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
924 925
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
926
    goto _err;
927
  }
C
Cary Xu 已提交
928

C
Cary Xu 已提交
929
  if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
C
Cary Xu 已提交
930
    if (pRsp) pRsp->code = terrno;
931
    goto _err;
932
  }
C
Cary Xu 已提交
933

934
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
935
  vDebug("vgId:%d, success to create tsma %s:%" PRIi64 " version %" PRIi64 " for table %" PRIi64, TD_VID(pVnode),
C
Cary Xu 已提交
936
         req.indexName, req.indexUid, version, req.tableUid);
H
Hongze Cheng 已提交
937
  return 0;
938 939 940

_err:
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
941
  vError("vgId:%d, failed to create tsma %s:%" PRIi64 " version %" PRIi64 "for table %" PRIi64 " since %s",
C
Cary Xu 已提交
942
         TD_VID(pVnode), req.indexName, req.indexUid, version, req.tableUid, terrstr());
943
  return -1;
L
Liu Jicong 已提交
944
}
C
Cary Xu 已提交
945 946 947 948 949 950 951 952 953 954 955 956

/**
 * @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);
}
957 958 959 960 961 962 963 964 965

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

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

971 972
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
973 974 975
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
976
}
H
Hongze Cheng 已提交
977

978 979 980 981 982 983 984 985 986
static int32_t vnodeProcessAlterConfigReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  SAlterVnodeReq alterReq = {0};
  if (tDeserializeSAlterVnodeReq(pReq, len, &alterReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    return TSDB_CODE_INVALID_MSG;
  }

  vInfo("vgId:%d, start to alter vnode config, cacheLast:%d cacheLastSize:%d", TD_VID(pVnode), alterReq.cacheLast,
        alterReq.cacheLastSize);
987 988 989 990 991
  if (pVnode->config.cacheLastSize != alterReq.cacheLastSize) {
    pVnode->config.cacheLastSize = alterReq.cacheLastSize;
    // TODO: save config
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
992 993 994
  return 0;
}

H
Hongze Cheng 已提交
995 996 997 998 999
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 已提交
1000 1001 1002 1003 1004
  pRsp->msgType = TDMT_VND_DELETE_RSP;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;

H
Hongze Cheng 已提交
1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
  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);

  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 已提交
1022 1023 1024 1025 1026 1027 1028 1029 1030

  SVDeleteRsp rsp = {.affectedRows = pRes->affectedRows};
  int32_t       ret = 0;
  tEncodeSize(tEncodeSVDeleteRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  SEncoder      ec = {0};
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVDeleteRsp(&ec, &rsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
1031 1032 1033 1034
  return code;

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