vnodeSvr.c 28.3 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);
S
Shengliang Guan 已提交
27
static int32_t vnodeProcessAlterHasnRangeReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
D
dapan1121 已提交
28
static int32_t vnodeProcessWriteMsg(SVnode *pVnode, int64_t version, SRpcMsg *pMsg, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
29

H
Hongze Cheng 已提交
30
int32_t vnodePreprocessReq(SVnode *pVnode, SRpcMsg *pMsg) {
S
Shengliang Guan 已提交
31
  int32_t  code = 0;
H
Hongze Cheng 已提交
32
  SDecoder dc = {0};
H
Hongze Cheng 已提交
33

H
Hongze Cheng 已提交
34 35 36 37 38 39 40 41 42 43 44
  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 已提交
45
        char    *name = NULL;
H
Hongze Cheng 已提交
46 47 48
        tStartDecode(&dc);

        tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
49
        tDecodeCStr(&dc, &name);
H
Hongze Cheng 已提交
50 51 52 53 54 55 56 57 58 59 60 61 62 63
        *(int64_t *)(dc.data + dc.pos) = uid;
        *(int64_t *)(dc.data + dc.pos + 8) = ctime;

        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 已提交
64
      tb_uid_t       uid;
H
Hongze Cheng 已提交
65 66 67 68 69 70 71 72

      tInitSubmitMsgIter(pSubmitReq, &msgIter);

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

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

H
Hongze Cheng 已提交
75 76 77 78
          tDecoderInit(&dc, pBlock->data, msgIter.schemaLen);
          tStartDecode(&dc);

          tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
79 80 81 82 83 84
          tDecodeCStr(&dc, &name);

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

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

    } break;
    default:
      break;
  }
H
Hongze Cheng 已提交
98

S
Shengliang Guan 已提交
99
  return code;
H
Hongze Cheng 已提交
100 101
}

102 103 104 105 106
int32_t vnodeProcessWriteReq(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
  void   *ptr = NULL;
  void   *pReq;
  int32_t len;
  int32_t ret;
H
Hongze Cheng 已提交
107

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

H
Hongze Cheng 已提交
111 112
  pVnode->state.applied = version;

H
Hongze Cheng 已提交
113 114 115
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
116 117

  switch (pMsg->msgType) {
H
Hongze Cheng 已提交
118
    /* META */
H
Hongze Cheng 已提交
119
    case TDMT_VND_CREATE_STB:
H
Hongze Cheng 已提交
120
      if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
121
      break;
H
Hongze Cheng 已提交
122
    case TDMT_VND_ALTER_STB:
H
Hongze Cheng 已提交
123
      if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
124
      break;
H
Hongze Cheng 已提交
125
    case TDMT_VND_DROP_STB:
H
Hongze Cheng 已提交
126
      if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
127
      break;
H
Hongze Cheng 已提交
128
    case TDMT_VND_CREATE_TABLE:
H
Hongze Cheng 已提交
129
      if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
130 131
      break;
    case TDMT_VND_ALTER_TABLE:
H
Hongze Cheng 已提交
132
      if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
133
      break;
H
Hongze Cheng 已提交
134
    case TDMT_VND_DROP_TABLE:
H
Hongze Cheng 已提交
135
      if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
136
      break;
137 138
    case TDMT_VND_CREATE_SMA: {
      if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
139 140
    } break;
    /* TSDB */
H
Hongze Cheng 已提交
141
    case TDMT_VND_SUBMIT:
H
Hongze Cheng 已提交
142
      if (vnodeProcessSubmitReq(pVnode, version, pMsg->pCont, pMsg->contLen, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
143
      break;
D
dapan1121 已提交
144
    case TDMT_VND_DELETE:
D
dapan1121 已提交
145
      if (vnodeProcessWriteMsg(pVnode, version, pMsg, pRsp) < 0) goto _err;
D
dapan1121 已提交
146
      break;
H
Hongze Cheng 已提交
147
    /* TQ */
L
Liu Jicong 已提交
148 149 150
    case TDMT_VND_MQ_VG_CHANGE:
      if (tqProcessVgChangeReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                               pMsg->contLen - sizeof(SMsgHead)) < 0) {
151
        goto _err;
L
Liu Jicong 已提交
152 153
      }
      break;
L
Liu Jicong 已提交
154 155
    case TDMT_VND_MQ_VG_DELETE:
      if (tqProcessVgDeleteReq(pVnode->pTq, pMsg->pCont, pMsg->contLen) < 0) {
156 157 158 159 160 161 162
        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 已提交
163 164
      }
      break;
165
    case TDMT_STREAM_TASK_DEPLOY: {
H
Hongze Cheng 已提交
166 167
      if (tqProcessTaskDeploy(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                              pMsg->contLen - sizeof(SMsgHead)) < 0) {
168
        goto _err;
H
Hongze Cheng 已提交
169 170
      }
    } break;
171 172 173
    case TDMT_VND_ALTER_CONFIRM:
      vnodeProcessAlterConfirmReq(pVnode, version, pReq, len, pRsp);
      break;
S
Shengliang Guan 已提交
174 175 176
    case TDMT_VND_ALTER_HASHRANGE:
      vnodeProcessAlterHasnRangeReq(pVnode, version, pReq, len, pRsp);
      break;
177
    case TDMT_VND_ALTER_CONFIG:
S
Shengliang Guan 已提交
178
      break;
H
Hongze Cheng 已提交
179 180 181 182 183
    default:
      ASSERT(0);
      break;
  }

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

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

H
Hongze Cheng 已提交
191
  // commit if need
H
Hongze Cheng 已提交
192
  if (vnodeShouldCommit(pVnode)) {
S
Shengliang Guan 已提交
193
    vInfo("vgId:%d, commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
194 195 196 197 198
    // commit current change
    vnodeCommit(pVnode);

    // start a new one
    vnodeBegin(pVnode);
H
Hongze Cheng 已提交
199 200 201
  }

  return 0;
H
Hongze Cheng 已提交
202 203

_err:
S
Shengliang Guan 已提交
204
  vError("vgId:%d, process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
205 206
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
207 208
}

209
int32_t vnodePreprocessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan1121 已提交
210 211 212 213 214 215 216 217
  if (TDMT_VND_QUERY != pMsg->msgType) {
    return 0;
  }

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

int32_t vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
218
  vTrace("message in vnode query queue is processing");
219
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
220 221
  switch (pMsg->msgType) {
    case TDMT_VND_QUERY:
D
dapan1121 已提交
222
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
223
    case TDMT_VND_QUERY_CONTINUE:
D
dapan1121 已提交
224
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
225 226 227 228 229 230
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

231
int32_t vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
H
Hongze Cheng 已提交
232 233 234
  vTrace("message in fetch queue is processing");
  char   *msgstr = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  int32_t msgLen = pMsg->contLen - sizeof(SMsgHead);
C
Cary Xu 已提交
235

H
Hongze Cheng 已提交
236 237
  switch (pMsg->msgType) {
    case TDMT_VND_FETCH:
D
dapan1121 已提交
238
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
239
    case TDMT_VND_FETCH_RSP:
D
dapan1121 已提交
240
      return qWorkerProcessFetchRsp(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
241
    case TDMT_VND_CANCEL_TASK:
D
dapan1121 已提交
242
      return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
243
    case TDMT_VND_DROP_TASK:
D
dapan1121 已提交
244 245 246
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
    case TDMT_VND_QUERY_HEARTBEAT:
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
247 248 249 250
    case TDMT_VND_TABLE_META:
      return vnodeGetTableMeta(pVnode, pMsg);
    case TDMT_VND_CONSUME:
      return tqProcessPollReq(pVnode->pTq, pMsg, pInfo->workerId);
251 252 253
    case TDMT_STREAM_TASK_RUN:
      return tqProcessTaskRunReq(pVnode->pTq, pMsg);
    case TDMT_STREAM_TASK_DISPATCH:
L
Liu Jicong 已提交
254
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg);
255
    case TDMT_STREAM_TASK_RECOVER:
L
Liu Jicong 已提交
256
      return tqProcessTaskRecoverReq(pVnode->pTq, pMsg);
257
    case TDMT_STREAM_TASK_DISPATCH_RSP:
L
Liu Jicong 已提交
258
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
259
    case TDMT_STREAM_TASK_RECOVER_RSP:
L
Liu Jicong 已提交
260
      return tqProcessTaskRecoverRsp(pVnode->pTq, pMsg);
H
Hongze Cheng 已提交
261 262 263 264 265 266
    default:
      vError("unknown msg type:%d in fetch queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

D
dapan1121 已提交
267
int32_t vnodeProcessWriteMsg(SVnode *pVnode, int64_t version, SRpcMsg *pMsg, SRpcMsg *pRsp) {
D
dapan1121 已提交
268
  vTrace("message in write queue is processing");
269 270 271
  char       *msgstr = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  int32_t     msgLen = pMsg->contLen - sizeof(SMsgHead);
  SDeleteRes  res = {0};
D
dapan1121 已提交
272
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
273

D
dapan1121 已提交
274 275
  switch (pMsg->msgType) {
    case TDMT_VND_DELETE:
D
dapan1121 已提交
276
      return qWorkerProcessDeleteMsg(&handle, pVnode->pQuery, pMsg, pRsp, &res);
D
dapan1121 已提交
277 278 279 280 281 282
    default:
      vError("unknown msg type:%d in write queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

H
Hongze Cheng 已提交
283 284 285 286
// TODO: remove the function
void smaHandleRes(void *pVnode, int64_t smaId, const SArray *data) {
  // TODO

C
Cary Xu 已提交
287
  blockDebugShowData(data, __func__);
288
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
289 290
}

D
dapan1121 已提交
291 292 293 294 295 296 297
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;
}

298
int32_t vnodeProcessSyncReq(SVnode *pVnode, SRpcMsg *pMsg, SRpcMsg **pRsp) {
M
Minghao Li 已提交
299
  int32_t ret = 0;
H
Hongze Cheng 已提交
300

301 302 303
  if (syncEnvIsStart()) {
    SSyncNode *pSyncNode = syncNodeAcquire(pVnode->sync);
    assert(pSyncNode != NULL);
M
Minghao Li 已提交
304

305 306
    ESyncState state = syncGetMyRole(pVnode->sync);
    SyncTerm   currentTerm = syncGetMyTerm(pVnode->sync);
M
Minghao Li 已提交
307

308
    SMsgHead *pHead = pMsg->pCont;
M
Minghao Li 已提交
309

310
    char  logBuf[512] = {0};
311 312
    char *syncNodeStr = sync2SimpleStr(pVnode->sync);
    snprintf(logBuf, sizeof(logBuf), "==vnodeProcessSyncReq== msgType:%d, syncNode: %s", pMsg->msgType, syncNodeStr);
M
Minghao Li 已提交
313 314 315 316
    static int64_t vndTick = 0;
    if (++vndTick % 10 == 1) {
      vTrace("sync trace msg:%s, %s", TMSG_INFO(pMsg->msgType), syncNodeStr);
    }
317 318
    syncRpcMsgLog2(logBuf, pMsg);
    taosMemoryFree(syncNodeStr);
M
Minghao Li 已提交
319

320
    SRpcMsg *pRpcMsg = pMsg;
M
Minghao Li 已提交
321

322
    if (pRpcMsg->msgType == TDMT_SYNC_TIMEOUT) {
323 324
      SyncTimeout *pSyncMsg = syncTimeoutFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
325

326
      ret = syncNodeOnTimeoutCb(pSyncNode, pSyncMsg);
327
      syncTimeoutDestroy(pSyncMsg);
M
Minghao Li 已提交
328

329
    } else if (pRpcMsg->msgType == TDMT_SYNC_PING) {
330 331
      SyncPing *pSyncMsg = syncPingFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
332

333
      ret = syncNodeOnPingCb(pSyncNode, pSyncMsg);
334
      syncPingDestroy(pSyncMsg);
M
Minghao Li 已提交
335

336
    } else if (pRpcMsg->msgType == TDMT_SYNC_PING_REPLY) {
337 338
      SyncPingReply *pSyncMsg = syncPingReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
339

340
      ret = syncNodeOnPingReplyCb(pSyncNode, pSyncMsg);
341
      syncPingReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
342

343
    } else if (pRpcMsg->msgType == TDMT_SYNC_CLIENT_REQUEST) {
344 345
      SyncClientRequest *pSyncMsg = syncClientRequestFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
346

347
      ret = syncNodeOnClientRequestCb(pSyncNode, pSyncMsg);
348
      syncClientRequestDestroy(pSyncMsg);
M
Minghao Li 已提交
349

350
    } else if (pRpcMsg->msgType == TDMT_SYNC_REQUEST_VOTE) {
351 352
      SyncRequestVote *pSyncMsg = syncRequestVoteFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
353

354
      ret = syncNodeOnRequestVoteCb(pSyncNode, pSyncMsg);
355
      syncRequestVoteDestroy(pSyncMsg);
M
Minghao Li 已提交
356

357
    } else if (pRpcMsg->msgType == TDMT_SYNC_REQUEST_VOTE_REPLY) {
358 359
      SyncRequestVoteReply *pSyncMsg = syncRequestVoteReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
360

361
      ret = syncNodeOnRequestVoteReplyCb(pSyncNode, pSyncMsg);
362
      syncRequestVoteReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
363

364
    } else if (pRpcMsg->msgType == TDMT_SYNC_APPEND_ENTRIES) {
365 366
      SyncAppendEntries *pSyncMsg = syncAppendEntriesFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
367

368
      ret = syncNodeOnAppendEntriesCb(pSyncNode, pSyncMsg);
369
      syncAppendEntriesDestroy(pSyncMsg);
M
Minghao Li 已提交
370

371
    } else if (pRpcMsg->msgType == TDMT_SYNC_APPEND_ENTRIES_REPLY) {
372 373 374
      SyncAppendEntriesReply *pSyncMsg = syncAppendEntriesReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);

375
      ret = syncNodeOnAppendEntriesReplyCb(pSyncNode, pSyncMsg);
376
      syncAppendEntriesReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
377

378 379 380 381
    } else if (pRpcMsg->msgType == TDMT_SYNC_SET_MNODE_STANDBY) {
      ret = syncSetStandby(pVnode->sync);
      SRpcMsg rsp = {.code = ret, .info = pMsg->info};
      tmsgSendRsp(&rsp);
382 383
    } else {
      vError("==vnodeProcessSyncReq== error msg type:%d", pRpcMsg->msgType);
M
Minghao Li 已提交
384
      ret = -1;
385
    }
M
Minghao Li 已提交
386

387 388 389
    syncNodeRelease(pSyncNode);
  } else {
    vError("==vnodeProcessSyncReq== error syncEnv stop");
M
Minghao Li 已提交
390
    ret = -1;
391
  }
392

M
Minghao Li 已提交
393 394 395
  if (ret != 0) {
    terrno = TSDB_CODE_SYN_INTERNAL_ERROR;
  }
396
  return ret;
H
Hongze Cheng 已提交
397 398
}

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
  }

5
54liuyao 已提交
421
  tdProcessRSmaCreate(pVnode, &req);
C
Cary Xu 已提交
422

H
Hongze Cheng 已提交
423
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
424
  return 0;
H
Hongze Cheng 已提交
425 426

_err:
H
Hongze Cheng 已提交
427
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
428
  return -1;
H
Hongze Cheng 已提交
429 430
}

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

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
443 444 445
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
446

H
Hongze Cheng 已提交
447
  // decode
H
Hongze Cheng 已提交
448 449
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
450 451 452 453
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
454

H
Hongze Cheng 已提交
455
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
456 457
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
458 459 460 461 462
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
463
  // loop to create table
464
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
465
    pCreateReq = req.pReqs + iReq;
H
Hongze Cheng 已提交
466 467 468 469 470 471 472 473 474 475

    // 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 已提交
476
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq) < 0) {
H
Hongze Cheng 已提交
477 478 479 480 481
      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 已提交
482
    } else {
H
Hongze Cheng 已提交
483
      cRsp.code = TSDB_CODE_SUCCESS;
484
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
485
      taosArrayPush(tbUids, &pCreateReq->uid);
H
Hongze Cheng 已提交
486
    }
H
Hongze Cheng 已提交
487 488

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

H
Hongze Cheng 已提交
491
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
492

493
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
494 495
  tdUpdateTbUidList(pVnode->pSma, pStore);
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
496

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

H
Hongze Cheng 已提交
511
_exit:
H
Hongze Cheng 已提交
512
  taosArrayDestroy(rsp.pArray);
513
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
514 515
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
516
  return rcode;
H
Hongze Cheng 已提交
517 518
}

519
static int32_t vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
520 521 522 523 524 525 526 527 528 529 530 531 532 533 534
  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 已提交
535
  }
H
Hongze Cheng 已提交
536 537 538 539 540 541 542 543 544

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
545 546 547
  return 0;
}

548
static int32_t vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
549
  SVDropStbReq req = {0};
550
  int32_t      rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
551
  SDecoder     decoder = {0};
H
Hongze Cheng 已提交
552 553 554 555 556 557

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

  // decode request
H
Hongze Cheng 已提交
558 559
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
560 561 562 563 564
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
H
Hongze Cheng 已提交
565 566 567 568
  // if (metaDropSTable(pVnode->pMeta, version, &req) < 0) {
  //   rcode = terrno;
  //   goto _exit;
  // }
H
Hongze Cheng 已提交
569 570 571 572

  // return rsp
_exit:
  pRsp->code = rcode;
H
Hongze Cheng 已提交
573
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
574 575 576
  return 0;
}

577
static int32_t vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
D
dapan1121 已提交
578 579 580
  SVAlterTbReq  vAlterTbReq = {0};
  SVAlterTbRsp  vAlterTbRsp = {0};
  SDecoder      dc = {0};
581 582
  int32_t       rcode = 0;
  int32_t       ret;
D
dapan1121 已提交
583 584
  SEncoder      ec = {0};
  STableMetaRsp vMetaRsp = {0};
H
Hongze Cheng 已提交
585 586 587 588 589 590 591 592 593 594

  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 已提交
595
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
596
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
597 598
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
599 600 601
  }

  // process
D
dapan1121 已提交
602
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq, &vMetaRsp) < 0) {
H
Hongze Cheng 已提交
603
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
604
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
605 606
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
607 608
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
609

D
dapan1121 已提交
610 611 612 613 614
  if (NULL != vMetaRsp.pSchemas) {
    vnodeUpdateMetaRsp(pVnode, &vMetaRsp);
    vAlterTbRsp.pMeta = &vMetaRsp;
  }

H
Hongze Cheng 已提交
615 616 617 618 619 620
_exit:
  tEncodeSize(tEncodeSVAlterTbRsp, &vAlterTbRsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
  tEncodeSVAlterTbRsp(&ec, &vAlterTbRsp);
  tEncoderClear(&ec);
H
Hongze Cheng 已提交
621 622 623
  return 0;
}

624
static int32_t vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
625 626
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
627
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
628
  SEncoder         encoder = {0};
629
  int32_t          ret;
630
  SArray          *tbUids = NULL;
H
Hongze Cheng 已提交
631

H
Hongze Cheng 已提交
632
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
633 634 635
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
636 637

  // decode req
H
Hongze Cheng 已提交
638 639
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
640 641 642 643 644
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
645 646

  // process req
647
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
648
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
649 650
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

651
  for (int32_t iReq = 0; iReq < req.nReqs; iReq++) {
H
Hongze Cheng 已提交
652 653
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
H
Hongze Cheng 已提交
654

H
Hongze Cheng 已提交
655
    /* code */
656
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids);
H
Hongze Cheng 已提交
657
    if (ret < 0) {
H
Hongze Cheng 已提交
658 659 660 661 662
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_VND_TABLE_NOT_EXIST) {
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
663
    } else {
H
Hongze Cheng 已提交
664
      dropTbRsp.code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
665 666 667 668 669
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

670 671
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);

H
Hongze Cheng 已提交
672
_exit:
673
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
674
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
675 676 677 678 679
  tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVDropTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
680 681 682
  return 0;
}

683 684
static int32_t vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter,
                                              const char *tags) {
D
dapan 已提交
685 686 687 688
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
689
  int32_t        rv = -1;
D
dapan 已提交
690 691 692

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
C
Cary Xu 已提交
693
  if (!pSchema || (suid != msgIter->suid) || rv != TD_ROW_SVER(blkIter.row)) {
D
dapan 已提交
694 695 696
    if (pSchema) {
      taosMemoryFreeClear(pSchema);
    }
C
Cary Xu 已提交
697
    pSchema = metaGetTbTSchema(pMeta, msgIter->suid, TD_ROW_SVER(blkIter.row));  // TODO: use the real schema
D
dapan 已提交
698 699
    if (pSchema) {
      suid = msgIter->suid;
C
Cary Xu 已提交
700
      rv = TD_ROW_SVER(blkIter.row);
D
dapan 已提交
701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
    }
  }
  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;
}

718
static int32_t vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
C
Cary Xu 已提交
719 720 721 722 723 724 725 726 727
  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 已提交
728

D
dapan 已提交
729 730
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
731 732 733 734

  return 0;
}

735
static int32_t vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
736
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
737
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
738 739 740 741
  SSubmitMsgIter msgIter = {0};
  SSubmitBlk    *pBlock;
  SSubmitRsp     rsp = {0};
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
742
  SDecoder       decoder = {0};
H
Hongze Cheng 已提交
743
  int32_t        nRows;
H
Hongze Cheng 已提交
744 745
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
746
  SArray        *newTbUids = NULL;
747
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
748 749

  pRsp->code = 0;
C
Cary Xu 已提交
750

C
Cary Xu 已提交
751 752 753 754
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
755 756 757 758 759
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
760
  // handle the request
H
Hongze Cheng 已提交
761 762 763
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
764 765
  }

C
Cary Xu 已提交
766 767
  submitRsp.pArray = taosArrayInit(msgIter.numOfBlocks, sizeof(SSubmitBlkRsp));
  newTbUids = taosArrayInit(msgIter.numOfBlocks, sizeof(int64_t));
768 769 770 771 772
  if (!submitRsp.pArray) {
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
773
  for (;;) {
H
Hongze Cheng 已提交
774 775 776
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
777 778
    SSubmitBlkRsp submitBlkRsp = {0};

H
Hongze Cheng 已提交
779 780
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
781 782
      submitBlkRsp.hashMeta = 1;

H
Hongze Cheng 已提交
783 784
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
785
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
786
        tDecoderClear(&decoder);
H
Hongze Cheng 已提交
787 788 789 790 791
        goto _exit;
      }

      if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
792
          submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
793
          tDecoderClear(&decoder);
H
Hongze Cheng 已提交
794 795 796
          goto _exit;
        }
      }
797
      taosArrayPush(newTbUids, &createTbReq.uid);
H
Hongze Cheng 已提交
798

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

H
Hongze Cheng 已提交
803 804 805 806 807 808 809
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

wmmhello's avatar
wmmhello 已提交
810
#ifdef TD_DEBUG_PRINT_ROW
D
dapan 已提交
811
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
wmmhello's avatar
wmmhello 已提交
812
#endif
H
Hongze Cheng 已提交
813
      tDecoderClear(&decoder);
D
dapan1121 已提交
814 815 816
    } else {
      submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
817 818
    }

H
Hongze Cheng 已提交
819
    if (tsdbInsertTableData(pVnode->pTsdb, version, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
820
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
821 822
    }

H
Hongze Cheng 已提交
823 824 825
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
    taosArrayPush(submitRsp.pArray, &submitBlkRsp);
H
Hongze Cheng 已提交
826
  }
827
  tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
828

H
Hongze Cheng 已提交
829
_exit:
830
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
831 832 833 834 835 836 837 838
  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 已提交
839
    taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
H
Hongze Cheng 已提交
840 841 842
  }

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

844 845
  // TODO: the partial success scenario and the error case
  // TODO: refactor
C
Cary Xu 已提交
846
  if ((terrno == TSDB_CODE_SUCCESS) && (pRsp->code == TSDB_CODE_SUCCESS)) {
847
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_DATA_TYPE_SUBMIT_BLOCK);
848
  }
C
Cary Xu 已提交
849

H
Hongze Cheng 已提交
850
  return 0;
L
Liu Jicong 已提交
851
}
852

853
static int32_t vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
854
  SVCreateTSmaReq req = {0};
H
Hongze Cheng 已提交
855
  SDecoder        coder;
856

C
Cary Xu 已提交
857 858 859 860 861 862
  if (pRsp) {
    pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
    pRsp->code = TSDB_CODE_SUCCESS;
    pRsp->pCont = NULL;
    pRsp->contLen = 0;
  }
863 864 865 866 867

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
C
Cary Xu 已提交
868 869
    terrno = TSDB_CODE_MSG_DECODE_ERROR;
    if (pRsp) pRsp->code = terrno;
870
    goto _err;
871
  }
C
Cary Xu 已提交
872

C
Cary Xu 已提交
873 874 875 876
  // record current timezone of server side
  req.timezoneInt = tsTimezone;

  if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
C
Cary Xu 已提交
877
    if (pRsp) pRsp->code = terrno;
878
    goto _err;
879
  }
C
Cary Xu 已提交
880

881
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
882
  vDebug("vgId:%d, success to create tsma %s:%" PRIi64 " version %" PRIi64 " for table %" PRIi64, TD_VID(pVnode),
C
Cary Xu 已提交
883
         req.indexName, req.indexUid, version, req.tableUid);
H
Hongze Cheng 已提交
884
  return 0;
885 886 887

_err:
  tDecoderClear(&coder);
S
Shengliang Guan 已提交
888
  vError("vgId:%d, failed to create tsma %s:%" PRIi64 " version %" PRIi64 "for table %" PRIi64 " since %s",
C
Cary Xu 已提交
889
         TD_VID(pVnode), req.indexName, req.indexUid, version, req.tableUid, terrstr(terrno));
890
  return -1;
L
Liu Jicong 已提交
891
}
C
Cary Xu 已提交
892 893 894 895 896 897 898 899 900 901 902 903

/**
 * @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);
}
904 905 906 907 908 909 910 911 912

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;
913
}
S
Shengliang Guan 已提交
914 915 916 917

static int32_t vnodeProcessAlterHasnRangeReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  vInfo("vgId:%d, alter hashrange msg will be processed", TD_VID(pVnode));

918 919
  // todo
  // 1. stop work
S
Shengliang Guan 已提交
920 921 922
  // 2. adjust hash range / compact / remove wals / rename vgroups
  // 3. reload sync
  return 0;
M
Minghao Li 已提交
923
}