vnodeSvr.c 30.8 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 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
    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);
      if (code) goto _err;

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

      ((SMsgHead *)pCont)->contLen = htonl(size + sizeof(SMsgHead));
      ((SMsgHead *)pCont)->vgId = htonl(TD_VID(pVnode));

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

S
Shengliang Guan 已提交
132
  return code;
H
Hongze Cheng 已提交
133 134 135 136

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

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

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

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

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

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

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

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

H
Hongze Cheng 已提交
243
  // commit if need
H
Hongze Cheng 已提交
244
  if (vnodeShouldCommit(pVnode)) {
H
Hongze Cheng 已提交
245
  _do_commit:
S
Shengliang Guan 已提交
246
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
247 248 249 250 251
    // commit current change
    vnodeCommit(pVnode);

    // start a new one
    vnodeBegin(pVnode);
H
Hongze Cheng 已提交
252 253 254
  }

  return 0;
H
Hongze Cheng 已提交
255 256

_err:
S
Shengliang Guan 已提交
257
  vError("vgId:%d, process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
258 259
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
260 261
}

262
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
263
  if (TDMT_SCH_QUERY != pMsg->msgType && TDMT_SCH_MERGE_QUERY != pMsg->msgType) {
D
dapan1121 已提交
264 265 266 267 268 269 270
    return 0;
  }

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

int32_t vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
271
  vTrace("message in vnode query queue is processing");
D
dapan1121 已提交
272
  if ((pMsg->msgType == TDMT_SCH_QUERY) && !vnodeIsLeader(pVnode)) {
273 274 275 276
    vnodeRedirectRpcMsg(pVnode, pMsg);
    return 0;
  }

277
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
278
  switch (pMsg->msgType) {
D
dapan1121 已提交
279
    case TDMT_SCH_QUERY:
D
dapan1121 已提交
280
    case TDMT_SCH_MERGE_QUERY:
D
dapan1121 已提交
281
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
282
    case TDMT_SCH_QUERY_CONTINUE:
D
dapan1121 已提交
283
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
284 285 286 287 288 289
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

290
int32_t vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
H
Hongze Cheng 已提交
291
  vTrace("message in fetch queue is processing");
M
Minglei Jin 已提交
292 293 294
  if ((pMsg->msgType == TDMT_SCH_FETCH || pMsg->msgType == TDMT_VND_TABLE_META ||
       pMsg->msgType == TDMT_VND_TABLE_CFG) &&
      !vnodeIsLeader(pVnode)) {
295 296 297 298
    vnodeRedirectRpcMsg(pVnode, pMsg);
    return 0;
  }

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

H
Hongze Cheng 已提交
302
  switch (pMsg->msgType) {
D
dapan1121 已提交
303
    case TDMT_SCH_FETCH:
D
dapan1121 已提交
304
    case TDMT_SCH_MERGE_FETCH:
D
dapan1121 已提交
305
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
306
    case TDMT_SCH_FETCH_RSP:
D
dapan1121 已提交
307
      return qWorkerProcessRspMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
308
    case TDMT_SCH_CANCEL_TASK:
D
dapan1121 已提交
309
      return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
310
    case TDMT_SCH_DROP_TASK:
D
dapan1121 已提交
311
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
D
dapan1121 已提交
312
    case TDMT_SCH_QUERY_HEARTBEAT:
D
dapan1121 已提交
313
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
314 315
    case TDMT_VND_TABLE_META:
      return vnodeGetTableMeta(pVnode, pMsg);
D
dapan1121 已提交
316 317
    case TDMT_VND_TABLE_CFG:
      return vnodeGetTableCfg(pVnode, pMsg);
H
Hongze Cheng 已提交
318
    case TDMT_VND_CONSUME:
L
Liu Jicong 已提交
319
      return tqProcessPollReq(pVnode->pTq, pMsg);
320 321 322
    case TDMT_STREAM_TASK_RUN:
      return tqProcessTaskRunReq(pVnode->pTq, pMsg);
    case TDMT_STREAM_TASK_DISPATCH:
L
Liu Jicong 已提交
323
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg);
324
    case TDMT_STREAM_TASK_RECOVER:
L
Liu Jicong 已提交
325
      return tqProcessTaskRecoverReq(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
326 327
    case TDMT_STREAM_RETRIEVE:
      return tqProcessTaskRetrieveReq(pVnode->pTq, pMsg);
328
    case TDMT_STREAM_TASK_DISPATCH_RSP:
L
Liu Jicong 已提交
329
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
330
    case TDMT_STREAM_TASK_RECOVER_RSP:
L
Liu Jicong 已提交
331
      return tqProcessTaskRecoverRsp(pVnode->pTq, pMsg);
L
Liu Jicong 已提交
332 333
    case TDMT_STREAM_RETRIEVE_RSP:
      return tqProcessTaskRetrieveRsp(pVnode->pTq, pMsg);
H
Hongze Cheng 已提交
334 335 336 337 338 339 340 341 342 343
    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 已提交
344
  blockDebugShowDataBlocks(data, __func__);
345
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
346 347
}

D
dapan1121 已提交
348 349 350 351 352 353 354
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 已提交
355
static int32_t vnodeProcessTrimReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
356
  int32_t     code = 0;
S
Shengliang Guan 已提交
357 358 359
  SVTrimDbReq trimReq = {0};

  vInfo("vgId:%d, trim vnode request will be processed, time:%d", pVnode->config.vgId, trimReq.timestamp);
H
Hongze Cheng 已提交
360 361 362 363 364

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

H
Hongze Cheng 已提交
367 368 369 370 371 372
  // process
  code = tsdbDoRetention(pVnode->pTsdb, trimReq.timestamp);
  if (code) goto _exit;

_exit:
  return code;
S
Shengliang Guan 已提交
373 374
}

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

S
Shengliang Guan 已提交
379 380 381 382 383 384
  SVDropTtlTableReq ttlReq = {0};
  if (tDeserializeSVDropTtlTableReq(pReq, len, &ttlReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto end;
  }

385
  vDebug("vgId:%d, drop ttl table req will be processed, time:%d", pVnode->config.vgId, ttlReq.timestamp);
S
Shengliang Guan 已提交
386
  int32_t ret = metaTtlDropTable(pVnode->pMeta, ttlReq.timestamp, tbUids);
L
Liu Jicong 已提交
387
  if (ret != 0) {
388 389
    goto end;
  }
L
Liu Jicong 已提交
390
  if (taosArrayGetSize(tbUids) > 0) {
wmmhello's avatar
wmmhello 已提交
391 392
    tqUpdateTbUidList(pVnode->pTq, tbUids, false);
  }
393 394 395 396 397 398

end:
  taosArrayDestroy(tbUids);
  return ret;
}

399
static int32_t vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
400
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
401
  SDecoder       coder;
H
Hongze Cheng 已提交
402

H
Hongze Cheng 已提交
403 404 405 406 407 408
  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 已提交
409
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
410 411

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
412 413
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
414 415
  }

H
Hongze Cheng 已提交
416
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
417 418
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
419 420
  }

421
  if (tdProcessRSmaCreate(pVnode->pSma, &req) < 0) {
422 423 424
    pRsp->code = terrno;
    goto _err;
  }
C
Cary Xu 已提交
425

M
Minglei Jin 已提交
426 427
  // taosMemoryFree(req.schemaRow.pSchema);
  // taosMemoryFree(req.schemaTag.pSchema);
H
Hongze Cheng 已提交
428
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
429
  return 0;
H
Hongze Cheng 已提交
430 431

_err:
M
Minglei Jin 已提交
432 433
  taosMemoryFree(req.schemaRow.pSchema);
  taosMemoryFree(req.schemaTag.pSchema);
H
Hongze Cheng 已提交
434
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
435
  return -1;
H
Hongze Cheng 已提交
436 437
}

438
static int32_t vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
439
  SDecoder           decoder = {0};
440
  int32_t            rcode = 0;
H
Hongze Cheng 已提交
441 442
  SVCreateTbBatchReq req = {0};
  SVCreateTbReq     *pCreateReq;
H
Hongze Cheng 已提交
443 444 445
  SVCreateTbBatchRsp rsp = {0};
  SVCreateTbRsp      cRsp = {0};
  char               tbName[TSDB_TABLE_FNAME_LEN];
C
Cary Xu 已提交
446
  STbUidStore       *pStore = NULL;
447
  SArray            *tbUids = NULL;
H
Hongze Cheng 已提交
448 449

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
450 451 452
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
453

H
Hongze Cheng 已提交
454
  // decode
H
Hongze Cheng 已提交
455 456
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
457 458 459 460
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
461

H
Hongze Cheng 已提交
462
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
463 464
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
465 466 467 468 469
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
470
  // loop to create table
471
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
472
    pCreateReq = req.pReqs + iReq;
H
Hongze Cheng 已提交
473 474 475 476 477 478 479 480 481 482

    // 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 已提交
483
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq) < 0) {
H
Hongze Cheng 已提交
484 485 486 487 488
      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 已提交
489
    } else {
H
Hongze Cheng 已提交
490
      cRsp.code = TSDB_CODE_SUCCESS;
491
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
492
      taosArrayPush(tbUids, &pCreateReq->uid);
H
Hongze Cheng 已提交
493
    }
H
Hongze Cheng 已提交
494 495

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

498
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
499 500
  tdUpdateTbUidList(pVnode->pSma, pStore);
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
501

H
Hongze Cheng 已提交
502
  // prepare rsp
H
Hongze Cheng 已提交
503 504
  SEncoder encoder = {0};
  int32_t  ret = 0;
wafwerar's avatar
wafwerar 已提交
505
  tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
H
Hongze Cheng 已提交
506 507 508 509 510 511
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  if (pRsp->pCont == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    rcode = -1;
    goto _exit;
  }
H
Hongze Cheng 已提交
512 513
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
H
Hongze Cheng 已提交
514

H
Hongze Cheng 已提交
515
_exit:
wmmhello's avatar
wmmhello 已提交
516 517 518 519
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
    pCreateReq = req.pReqs + iReq;
    taosArrayDestroy(pCreateReq->ctb.tagName);
  }
H
Hongze Cheng 已提交
520
  taosArrayDestroy(rsp.pArray);
521
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
522 523
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
524
  return rcode;
H
Hongze Cheng 已提交
525 526
}

527
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542
  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 已提交
543
  }
H
Hongze Cheng 已提交
544 545 546 547 548 549 550 551 552

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
553 554 555
  return 0;
}

556
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
557
  SVDropStbReq req = {0};
558
  int32_t      rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
559
  SDecoder     decoder = {0};
560
  SArray      *tbUidList = NULL;
H
Hongze Cheng 已提交
561 562 563 564 565 566

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

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

  // process request
574 575 576 577 578 579 580 581
  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) {
582 583 584
    rcode = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
585

586 587 588 589 590
  if (tdProcessRSmaDrop(pVnode->pSma, &req) < 0) {
    rcode = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
591 592
  // return rsp
_exit:
593
  if (tbUidList) taosArrayDestroy(tbUidList);
H
Hongze Cheng 已提交
594
  pRsp->code = rcode;
H
Hongze Cheng 已提交
595
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
596 597 598
  return 0;
}

599
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
D
dapan1121 已提交
600 601 602
  SVAlterTbReq  vAlterTbReq = {0};
  SVAlterTbRsp  vAlterTbRsp = {0};
  SDecoder      dc = {0};
603 604
  int32_t       rcode = 0;
  int32_t       ret;
D
dapan1121 已提交
605 606
  SEncoder      ec = {0};
  STableMetaRsp vMetaRsp = {0};
H
Hongze Cheng 已提交
607 608 609 610 611 612 613 614 615 616

  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 已提交
617
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
618
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
619 620
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
621 622 623
  }

  // process
D
dapan1121 已提交
624
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq, &vMetaRsp) < 0) {
wmmhello's avatar
wmmhello 已提交
625
    vAlterTbRsp.code = terrno;
H
Hongze Cheng 已提交
626
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
627 628
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
629 630
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
631

D
dapan1121 已提交
632 633 634 635 636
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
637 638 639 640 641 642
_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 已提交
643 644 645
  if (vMetaRsp.pSchemas) {
    taosMemoryFree(vMetaRsp.pSchemas);
  }
H
Hongze Cheng 已提交
646 647 648
  return 0;
}

649
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
650 651
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
652
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
653
  SEncoder         encoder = {0};
654
  int32_t          ret;
655
  SArray          *tbUids = NULL;
H
Hongze Cheng 已提交
656

H
Hongze Cheng 已提交
657
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
658 659 660
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
661 662

  // decode req
H
Hongze Cheng 已提交
663 664
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
665 666 667 668 669
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
670 671

  // process req
672
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
673
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
674 675
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

676
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
677 678
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
H
Hongze Cheng 已提交
679

H
Hongze Cheng 已提交
680
    /* code */
681
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids);
H
Hongze Cheng 已提交
682
    if (ret < 0) {
H
Hongze Cheng 已提交
683 684 685 686 687
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_VND_TABLE_NOT_EXIST) {
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
688
    } else {
H
Hongze Cheng 已提交
689
      dropTbRsp.code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
690 691 692 693 694
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

695 696
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);

H
Hongze Cheng 已提交
697
_exit:
698
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
699
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
700 701 702 703 704
  tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVDropTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
705
  taosArrayDestroy(rsp.pArray);
H
Hongze Cheng 已提交
706 707 708
  return 0;
}

709 710
static int32_t vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter,
                                              const char *tags) {
D
dapan 已提交
711 712 713 714
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
715
  int32_t        rv = -1;
D
dapan 已提交
716 717 718

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
C
Cary Xu 已提交
719
  if (!pSchema || (suid != msgIter->suid) || rv != TD_ROW_SVER(blkIter.row)) {
D
dapan 已提交
720 721 722
    if (pSchema) {
      taosMemoryFreeClear(pSchema);
    }
C
Cary Xu 已提交
723
    pSchema = metaGetTbTSchema(pMeta, msgIter->suid, TD_ROW_SVER(blkIter.row));  // TODO: use the real schema
D
dapan 已提交
724 725
    if (pSchema) {
      suid = msgIter->suid;
C
Cary Xu 已提交
726
      rv = TD_ROW_SVER(blkIter.row);
D
dapan 已提交
727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743
    }
  }
  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;
}

744
static int32_t vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
C
Cary Xu 已提交
745 746 747 748 749 750 751 752 753
  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 已提交
754

D
dapan 已提交
755 756
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
757 758 759 760

  return 0;
}

761
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
762
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
763
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
764 765 766 767
  SSubmitMsgIter msgIter = {0};
  SSubmitBlk    *pBlock;
  SSubmitRsp     rsp = {0};
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
768
  SDecoder       decoder = {0};
H
Hongze Cheng 已提交
769
  int32_t        nRows;
H
Hongze Cheng 已提交
770 771
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
772
  SArray        *newTbUids = NULL;
773
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
774 775

  pRsp->code = 0;
C
Cary Xu 已提交
776
  pSubmitReq->version = version;
C
Cary Xu 已提交
777

C
Cary Xu 已提交
778 779 780 781
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
782 783 784 785 786
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
787
  // handle the request
H
Hongze Cheng 已提交
788 789 790
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
791 792
  }

C
Cary Xu 已提交
793 794
  submitRsp.pArray = taosArrayInit(msgIter.numOfBlocks, sizeof(SSubmitBlkRsp));
  newTbUids = taosArrayInit(msgIter.numOfBlocks, sizeof(int64_t));
C
Cary Xu 已提交
795
  if (!submitRsp.pArray || !newTbUids) {
796 797 798 799
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
800
  for (;;) {
H
Hongze Cheng 已提交
801 802 803
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
804 805
    SSubmitBlkRsp submitBlkRsp = {0};

H
Hongze Cheng 已提交
806 807
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
808 809
      submitBlkRsp.hashMeta = 1;

H
Hongze Cheng 已提交
810 811
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
812
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
813
        tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
814
        taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
815 816 817 818 819
        goto _exit;
      }

      if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
820
          submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
821
          tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
822
          taosArrayDestroy(createTbReq.ctb.tagName);
H
Hongze Cheng 已提交
823 824 825
          goto _exit;
        }
      }
826
      taosArrayPush(newTbUids, &createTbReq.uid);
H
Hongze Cheng 已提交
827

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

H
Hongze Cheng 已提交
832 833 834 835 836 837 838
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

wmmhello's avatar
wmmhello 已提交
839
#ifdef TD_DEBUG_PRINT_ROW
D
dapan 已提交
840
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
wmmhello's avatar
wmmhello 已提交
841
#endif
H
Hongze Cheng 已提交
842
      tDecoderClear(&decoder);
wmmhello's avatar
wmmhello 已提交
843
      taosArrayDestroy(createTbReq.ctb.tagName);
D
dapan1121 已提交
844 845 846
    } else {
      submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
847 848
    }

H
Hongze Cheng 已提交
849
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
850
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
851 852
    }

H
Hongze Cheng 已提交
853 854 855
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
    taosArrayPush(submitRsp.pArray, &submitBlkRsp);
H
Hongze Cheng 已提交
856
  }
857
  tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
858

H
Hongze Cheng 已提交
859
_exit:
860
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
861 862 863 864 865 866 867 868
  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 已提交
869
    taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
H
Hongze Cheng 已提交
870 871 872
  }

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

874
  // TODO: the partial success scenario and the error case
H
Hongze Cheng 已提交
875 876
  // => If partial success, extract the success submitted rows and reconstruct a new submit msg, and push to level
  // 1/level 2.
877
  // TODO: refactor
C
Cary Xu 已提交
878
  if ((terrno == TSDB_CODE_SUCCESS) && (pRsp->code == TSDB_CODE_SUCCESS)) {
L
Liu Jicong 已提交
879
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_INPUT__DATA_SUBMIT);
880
  }
C
Cary Xu 已提交
881

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

H
Hongze Cheng 已提交
884
  return 0;
L
Liu Jicong 已提交
885
}
886

887
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
888
  SVCreateTSmaReq req = {0};
H
Hongze Cheng 已提交
889
  SDecoder        coder;
890

C
Cary Xu 已提交
891 892 893 894 895 896
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
897 898 899 900 901

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
902 903
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
904
    goto _err;
905
  }
C
Cary Xu 已提交
906

C
Cary Xu 已提交
907
  if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
C
Cary Xu 已提交
908
    if (pRsp) pRsp->code = terrno;
909
    goto _err;
910
  }
C
Cary Xu 已提交
911

912
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
913
  vDebug("vgId:%d, success to create tsma %s:%" PRIi64 " version %" PRIi64 " for table %" PRIi64, TD_VID(pVnode),
C
Cary Xu 已提交
914
         req.indexName, req.indexUid, version, req.tableUid);
H
Hongze Cheng 已提交
915
  return 0;
916 917 918

_err:
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
919
  vError("vgId:%d, failed to create tsma %s:%" PRIi64 " version %" PRIi64 "for table %" PRIi64 " since %s",
C
Cary Xu 已提交
920
         TD_VID(pVnode), req.indexName, req.indexUid, version, req.tableUid, terrstr());
921
  return -1;
L
Liu Jicong 已提交
922
}
C
Cary Xu 已提交
923 924 925 926 927 928 929 930 931 932 933 934

/**
 * @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);
}
935 936 937 938 939 940 941 942 943

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

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

949 950
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
951 952 953
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
954
}
H
Hongze Cheng 已提交
955

956 957 958 959 960 961 962 963 964
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);
965 966 967 968 969
  if (pVnode->config.cacheLastSize != alterReq.cacheLastSize) {
    pVnode->config.cacheLastSize = alterReq.cacheLastSize;
    // TODO: save config
    tsdbCacheSetCapacity(pVnode, (size_t)pVnode->config.cacheLastSize * 1024 * 1024);
  }
970 971 972
  return 0;
}

H
Hongze Cheng 已提交
973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998
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};

  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);
  return code;

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