vnodeSvr.c 24.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

H
Hongze Cheng 已提交
18
static int vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
19
static int vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
20
static int vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
21
static int vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
22
static int vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
23
static int vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
24
static int vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
25
static int vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
H
Hongze Cheng 已提交
26

H
Hongze Cheng 已提交
27 28
int32_t vnodePreprocessReq(SVnode *pVnode, SRpcMsg *pMsg) {
  SDecoder dc = {0};
H
Hongze Cheng 已提交
29

H
Hongze Cheng 已提交
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
  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();
        tStartDecode(&dc);

        tDecodeI32v(&dc, NULL);
        *(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 已提交
58
      tb_uid_t       uid;
H
Hongze Cheng 已提交
59 60 61 62 63 64 65 66

      tInitSubmitMsgIter(pSubmitReq, &msgIter);

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

        if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
67 68
          uid = tGenIdPI64();

H
Hongze Cheng 已提交
69 70 71 72
          tDecoderInit(&dc, pBlock->data, msgIter.schemaLen);
          tStartDecode(&dc);

          tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
73
          *(int64_t *)(dc.data + dc.pos) = uid;
H
Hongze Cheng 已提交
74
          *(int64_t *)(dc.data + dc.pos + 8) = ctime;
H
Hongze Cheng 已提交
75
          pBlock->uid = htobe64(uid);
H
Hongze Cheng 已提交
76 77 78 79 80 81 82 83 84 85

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

    } break;
    default:
      break;
  }
H
Hongze Cheng 已提交
86 87

  return 0;
H
Hongze Cheng 已提交
88 89
}

H
Hongze Cheng 已提交
90
int vnodeProcessWriteReq(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
91
  void *ptr = NULL;
H
Hongze Cheng 已提交
92 93
  void *pReq;
  int   len;
H
Hongze Cheng 已提交
94
  int   ret;
H
Hongze Cheng 已提交
95

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

H
Hongze Cheng 已提交
99 100
  pVnode->state.applied = version;

H
Hongze Cheng 已提交
101 102 103
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
104 105

  switch (pMsg->msgType) {
H
Hongze Cheng 已提交
106
    /* META */
H
Hongze Cheng 已提交
107
    case TDMT_VND_CREATE_STB:
H
Hongze Cheng 已提交
108
      if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
109
      break;
H
Hongze Cheng 已提交
110
    case TDMT_VND_ALTER_STB:
H
Hongze Cheng 已提交
111
      if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
112
      break;
H
Hongze Cheng 已提交
113
    case TDMT_VND_DROP_STB:
H
Hongze Cheng 已提交
114
      if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
115
      break;
H
Hongze Cheng 已提交
116
    case TDMT_VND_CREATE_TABLE:
H
Hongze Cheng 已提交
117
      if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
118 119
      break;
    case TDMT_VND_ALTER_TABLE:
H
Hongze Cheng 已提交
120
      if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
121
      break;
H
Hongze Cheng 已提交
122
    case TDMT_VND_DROP_TABLE:
H
Hongze Cheng 已提交
123
      if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
124
      break;
125 126
    case TDMT_VND_CREATE_SMA: {
      if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
127 128
    } break;
    /* TSDB */
H
Hongze Cheng 已提交
129
    case TDMT_VND_SUBMIT:
H
Hongze Cheng 已提交
130
      if (vnodeProcessSubmitReq(pVnode, version, pMsg->pCont, pMsg->contLen, pRsp) < 0) goto _err;
H
Hongze Cheng 已提交
131
      break;
H
Hongze Cheng 已提交
132
    /* TQ */
L
Liu Jicong 已提交
133 134 135 136 137 138
    case TDMT_VND_MQ_VG_CHANGE:
      if (tqProcessVgChangeReq(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                               pMsg->contLen - sizeof(SMsgHead)) < 0) {
        // TODO: handle error
      }
      break;
L
Liu Jicong 已提交
139 140 141 142 143
    case TDMT_VND_MQ_VG_DELETE:
      if (tqProcessVgDeleteReq(pVnode->pTq, pMsg->pCont, pMsg->contLen) < 0) {
        // TODO: handle error
      }
      break;
H
Hongze Cheng 已提交
144 145 146 147 148
    case TDMT_VND_TASK_DEPLOY: {
      if (tqProcessTaskDeploy(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
                              pMsg->contLen - sizeof(SMsgHead)) < 0) {
      }
    } break;
S
Shengliang Guan 已提交
149 150
    case TDMT_VND_ALTER_VNODE:
      break;
H
Hongze Cheng 已提交
151 152 153 154 155
    default:
      ASSERT(0);
      break;
  }

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

158 159 160 161 162
  if (tqPushMsg(pVnode->pTq, pMsg->pCont, pMsg->contLen, pMsg->msgType, version) < 0) {
    vError("vgId:%d failed to push msg to TQ since %s", TD_VID(pVnode), tstrerror(terrno));
    return -1;
  }

H
Hongze Cheng 已提交
163
  // commit if need
H
Hongze Cheng 已提交
164
  if (vnodeShouldCommit(pVnode)) {
H
Hongze Cheng 已提交
165
    vInfo("vgId:%d commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
166 167 168 169 170
    // commit current change
    vnodeCommit(pVnode);

    // start a new one
    vnodeBegin(pVnode);
H
Hongze Cheng 已提交
171 172 173
  }

  return 0;
H
Hongze Cheng 已提交
174 175

_err:
H
Hongze Cheng 已提交
176
  vDebug("vgId:%d process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
177 178
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
179 180 181
}

int vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
182
  vTrace("message in vnode query queue is processing");
183
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
  switch (pMsg->msgType) {
    case TDMT_VND_QUERY:
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg);
    case TDMT_VND_QUERY_CONTINUE:
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg);
    default:
      vError("unknown msg type:%d in query queue", pMsg->msgType);
      return TSDB_CODE_VND_APP_ERROR;
  }
}

int vnodeProcessFetchMsg(SVnode *pVnode, SRpcMsg *pMsg, SQueueInfo *pInfo) {
  vTrace("message in fetch queue is processing");
  char   *msgstr = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  int32_t msgLen = pMsg->contLen - sizeof(SMsgHead);
  switch (pMsg->msgType) {
    case TDMT_VND_FETCH:
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_FETCH_RSP:
      return qWorkerProcessFetchRsp(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_RES_READY:
      return qWorkerProcessReadyMsg(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_TASKS_STATUS:
      return qWorkerProcessStatusMsg(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_CANCEL_TASK:
      return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_DROP_TASK:
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg);
    case TDMT_VND_TABLE_META:
      return vnodeGetTableMeta(pVnode, pMsg);
    case TDMT_VND_CONSUME:
      return tqProcessPollReq(pVnode->pTq, pMsg, pInfo->workerId);
L
Liu Jicong 已提交
216

L
Liu Jicong 已提交
217 218 219 220 221
    case TDMT_VND_TASK_RUN: {
      int32_t code = tqProcessTaskRunReq(pVnode->pTq, pMsg);
      pMsg->pCont = NULL;
      return code;
    }
L
Liu Jicong 已提交
222 223 224 225 226 227 228 229 230
    case TDMT_VND_TASK_DISPATCH:
      return tqProcessTaskDispatchReq(pVnode->pTq, pMsg);
    case TDMT_VND_TASK_RECOVER:
      return tqProcessTaskRecoverReq(pVnode->pTq, pMsg);
    case TDMT_VND_TASK_DISPATCH_RSP:
      return tqProcessTaskDispatchRsp(pVnode->pTq, pMsg);
    case TDMT_VND_TASK_RECOVER_RSP:
      return tqProcessTaskRecoverRsp(pVnode->pTq, pMsg);

H
Hongze Cheng 已提交
231 232 233 234 235 236 237 238 239 240 241 242 243
    case TDMT_VND_QUERY_HEARTBEAT:
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg);
    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

  // blockDebugShowData(data);
244
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
245 246 247
}

int vnodeProcessSyncReq(SVnode *pVnode, SRpcMsg *pMsg, SRpcMsg **pRsp) {
248
  int32_t ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
H
Hongze Cheng 已提交
249

250 251 252
  if (syncEnvIsStart()) {
    SSyncNode *pSyncNode = syncNodeAcquire(pVnode->sync);
    assert(pSyncNode != NULL);
M
Minghao Li 已提交
253

254 255
    ESyncState state = syncGetMyRole(pVnode->sync);
    SyncTerm   currentTerm = syncGetMyTerm(pVnode->sync);
M
Minghao Li 已提交
256

257
    SMsgHead *pHead = pMsg->pCont;
M
Minghao Li 已提交
258

259 260 261 262 263
    char  logBuf[512];
    char *syncNodeStr = sync2SimpleStr(pVnode->sync);
    snprintf(logBuf, sizeof(logBuf), "==vnodeProcessSyncReq== msgType:%d, syncNode: %s", pMsg->msgType, syncNodeStr);
    syncRpcMsgLog2(logBuf, pMsg);
    taosMemoryFree(syncNodeStr);
M
Minghao Li 已提交
264

265
    SRpcMsg *pRpcMsg = pMsg;
M
Minghao Li 已提交
266

267 268 269
    if (pRpcMsg->msgType == TDMT_VND_SYNC_TIMEOUT) {
      SyncTimeout *pSyncMsg = syncTimeoutFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
270

271
      ret = syncNodeOnTimeoutCb(pSyncNode, pSyncMsg);
272
      syncTimeoutDestroy(pSyncMsg);
M
Minghao Li 已提交
273

274 275 276
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING) {
      SyncPing *pSyncMsg = syncPingFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
277

278
      ret = syncNodeOnPingCb(pSyncNode, pSyncMsg);
279
      syncPingDestroy(pSyncMsg);
M
Minghao Li 已提交
280

281 282 283
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING_REPLY) {
      SyncPingReply *pSyncMsg = syncPingReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
284

285
      ret = syncNodeOnPingReplyCb(pSyncNode, pSyncMsg);
286
      syncPingReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
287

288 289 290
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_CLIENT_REQUEST) {
      SyncClientRequest *pSyncMsg = syncClientRequestFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
291

292
      ret = syncNodeOnClientRequestCb(pSyncNode, pSyncMsg);
293
      syncClientRequestDestroy(pSyncMsg);
M
Minghao Li 已提交
294

295 296 297
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE) {
      SyncRequestVote *pSyncMsg = syncRequestVoteFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
298

299
      ret = syncNodeOnRequestVoteCb(pSyncNode, pSyncMsg);
300
      syncRequestVoteDestroy(pSyncMsg);
M
Minghao Li 已提交
301

302 303 304
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE_REPLY) {
      SyncRequestVoteReply *pSyncMsg = syncRequestVoteReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
305

306
      ret = syncNodeOnRequestVoteReplyCb(pSyncNode, pSyncMsg);
307
      syncRequestVoteReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
308

309 310 311
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES) {
      SyncAppendEntries *pSyncMsg = syncAppendEntriesFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
312

313
      ret = syncNodeOnAppendEntriesCb(pSyncNode, pSyncMsg);
314
      syncAppendEntriesDestroy(pSyncMsg);
M
Minghao Li 已提交
315

316 317 318 319
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES_REPLY) {
      SyncAppendEntriesReply *pSyncMsg = syncAppendEntriesReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);

320
      ret = syncNodeOnAppendEntriesReplyCb(pSyncNode, pSyncMsg);
321
      syncAppendEntriesReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
322

323 324
    } else {
      vError("==vnodeProcessSyncReq== error msg type:%d", pRpcMsg->msgType);
325
      ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
326
    }
M
Minghao Li 已提交
327

328 329 330
    syncNodeRelease(pSyncNode);
  } else {
    vError("==vnodeProcessSyncReq== error syncEnv stop");
331
    ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
332
  }
333 334

  return ret;
H
Hongze Cheng 已提交
335 336
}

H
Hongze Cheng 已提交
337
static int vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
338
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
339
  SDecoder       coder;
H
Hongze Cheng 已提交
340

H
Hongze Cheng 已提交
341 342 343 344 345 346
  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 已提交
347
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
348 349

  if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
H
Hongze Cheng 已提交
350 351
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
352 353
  }

H
Hongze Cheng 已提交
354
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
355 356
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
357 358
  }

359
  tdProcessRSmaCreate(pVnode->pSma, pVnode->pMeta, &req, &pVnode->msgCb);
C
Cary Xu 已提交
360

H
Hongze Cheng 已提交
361
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
362
  return 0;
H
Hongze Cheng 已提交
363 364

_err:
H
Hongze Cheng 已提交
365
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
366
  return -1;
H
Hongze Cheng 已提交
367 368
}

H
Hongze Cheng 已提交
369
static int vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
370
  SDecoder           decoder = {0};
H
Hongze Cheng 已提交
371 372 373
  int                rcode = 0;
  SVCreateTbBatchReq req = {0};
  SVCreateTbReq     *pCreateReq;
H
Hongze Cheng 已提交
374 375 376
  SVCreateTbBatchRsp rsp = {0};
  SVCreateTbRsp      cRsp = {0};
  char               tbName[TSDB_TABLE_FNAME_LEN];
C
Cary Xu 已提交
377
  STbUidStore       *pStore = NULL;
378
  SArray            *tbUids = NULL;
H
Hongze Cheng 已提交
379 380

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
381 382 383
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
384

H
Hongze Cheng 已提交
385
  // decode
H
Hongze Cheng 已提交
386 387
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
388 389 390 391
    rcode = -1;
    terrno = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }
H
Hongze Cheng 已提交
392

H
Hongze Cheng 已提交
393
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
394 395
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
  if (rsp.pArray == NULL || tbUids == NULL) {
H
Hongze Cheng 已提交
396 397 398 399 400
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
401 402 403
  // loop to create table
  for (int iReq = 0; iReq < req.nReqs; iReq++) {
    pCreateReq = req.pReqs + iReq;
H
Hongze Cheng 已提交
404 405 406 407 408 409 410 411 412 413

    // 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 已提交
414
    if (metaCreateTable(pVnode->pMeta, version, pCreateReq) < 0) {
H
Hongze Cheng 已提交
415 416 417 418 419
      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 已提交
420
    } else {
H
Hongze Cheng 已提交
421
      cRsp.code = TSDB_CODE_SUCCESS;
422
      tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
423
      taosArrayPush(tbUids, &pCreateReq->uid);
H
Hongze Cheng 已提交
424
    }
H
Hongze Cheng 已提交
425 426

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

H
Hongze Cheng 已提交
429
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
430

431
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
432 433
  tdUpdateTbUidList(pVnode->pSma, pStore);
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
434

H
Hongze Cheng 已提交
435
  // prepare rsp
H
Hongze Cheng 已提交
436 437
  SEncoder encoder = {0};
  int32_t  ret = 0;
wafwerar's avatar
wafwerar 已提交
438
  tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
H
Hongze Cheng 已提交
439 440 441 442 443 444
  pRsp->pCont = rpcMallocCont(pRsp->contLen);
  if (pRsp->pCont == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    rcode = -1;
    goto _exit;
  }
H
Hongze Cheng 已提交
445 446 447
  tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
  tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
448

H
Hongze Cheng 已提交
449
_exit:
H
Hongze Cheng 已提交
450
  taosArrayDestroy(rsp.pArray);
451
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
452 453
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
454
  return rcode;
H
Hongze Cheng 已提交
455 456
}

H
Hongze Cheng 已提交
457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
static int vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  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 已提交
473
  }
H
Hongze Cheng 已提交
474 475 476 477 478 479 480 481 482

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
483 484 485
  return 0;
}

H
Hongze Cheng 已提交
486
static int vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
487 488
  SVDropStbReq req = {0};
  int          rcode = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
489
  SDecoder     decoder = {0};
H
Hongze Cheng 已提交
490 491 492 493 494 495

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

  // decode request
H
Hongze Cheng 已提交
496 497
  tDecoderInit(&decoder, pReq, len);
  if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
H
Hongze Cheng 已提交
498 499 500 501 502
    rcode = TSDB_CODE_INVALID_MSG;
    goto _exit;
  }

  // process request
H
Hongze Cheng 已提交
503 504 505 506
  // if (metaDropSTable(pVnode->pMeta, version, &req) < 0) {
  //   rcode = terrno;
  //   goto _exit;
  // }
H
Hongze Cheng 已提交
507 508 509 510

  // return rsp
_exit:
  pRsp->code = rcode;
H
Hongze Cheng 已提交
511
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
512 513 514
  return 0;
}

H
Hongze Cheng 已提交
515 516
static int vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
  SVAlterTbReq vAlterTbReq = {0};
H
Hongze Cheng 已提交
517
  SVAlterTbRsp vAlterTbRsp = {0};
H
Hongze Cheng 已提交
518
  SDecoder     dc = {0};
H
Hongze Cheng 已提交
519 520 521
  int          rcode = 0;
  int          ret;
  SEncoder     ec = {0};
H
Hongze Cheng 已提交
522 523 524 525 526 527 528 529 530 531

  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 已提交
532
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
533
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
534 535
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
536 537 538 539
  }

  // process
  if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq) < 0) {
H
Hongze Cheng 已提交
540
    vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
541
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
542 543
    rcode = -1;
    goto _exit;
H
Hongze Cheng 已提交
544 545
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
546 547 548 549 550 551 552

_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 已提交
553 554 555
  return 0;
}

H
Hongze Cheng 已提交
556
static int vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
557 558
  SVDropTbBatchReq req = {0};
  SVDropTbBatchRsp rsp = {0};
H
Hongze Cheng 已提交
559
  SDecoder         decoder = {0};
H
Hongze Cheng 已提交
560
  SEncoder         encoder = {0};
H
Hongze Cheng 已提交
561
  int              ret;
562
  SArray          *tbUids = NULL;
H
Hongze Cheng 已提交
563

H
Hongze Cheng 已提交
564
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
565 566 567
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
568 569

  // decode req
H
Hongze Cheng 已提交
570 571
  tDecoderInit(&decoder, pReq, len);
  ret = tDecodeSVDropTbBatchReq(&decoder, &req);
H
Hongze Cheng 已提交
572 573 574 575 576
  if (ret < 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    pRsp->code = terrno;
    goto _exit;
  }
H
Hongze Cheng 已提交
577 578

  // process req
579
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
580
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
581 582
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

H
Hongze Cheng 已提交
583 584 585
  for (int iReq = 0; iReq < req.nReqs; iReq++) {
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
H
Hongze Cheng 已提交
586

H
Hongze Cheng 已提交
587
    /* code */
588
    ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids);
H
Hongze Cheng 已提交
589
    if (ret < 0) {
H
Hongze Cheng 已提交
590 591 592 593 594
      if (pDropTbReq->igNotExists && terrno == TSDB_CODE_VND_TABLE_NOT_EXIST) {
        dropTbRsp.code = TSDB_CODE_SUCCESS;
      } else {
        dropTbRsp.code = terrno;
      }
H
Hongze Cheng 已提交
595
    } else {
H
Hongze Cheng 已提交
596
      dropTbRsp.code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
597 598 599 600 601
    }

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

602 603
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);

H
Hongze Cheng 已提交
604
_exit:
605
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
606
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
607 608 609 610 611
  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 已提交
612 613 614
  return 0;
}

H
Hongze Cheng 已提交
615
static int vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter, const char *tags) {
D
dapan 已提交
616 617 618 619 620 621 622 623 624 625 626
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
  if (!pSchema || (suid != msgIter->suid)) {
    if (pSchema) {
      taosMemoryFreeClear(pSchema);
    }
H
Hongze Cheng 已提交
627
    pSchema = metaGetTbTSchema(pMeta, msgIter->suid, 1);  // TODO: use the real schema
D
dapan 已提交
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646
    if (pSchema) {
      suid = msgIter->suid;
    }
  }
  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;
}

H
Hongze Cheng 已提交
647
static int vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
C
Cary Xu 已提交
648 649 650 651 652 653 654 655 656
  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 已提交
657

D
dapan 已提交
658 659
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
660 661 662 663

  return 0;
}

H
Hongze Cheng 已提交
664
static int vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
665
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
666
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
667 668 669 670
  SSubmitMsgIter msgIter = {0};
  SSubmitBlk    *pBlock;
  SSubmitRsp     rsp = {0};
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
671
  SDecoder       decoder = {0};
H
Hongze Cheng 已提交
672
  int32_t        nRows;
H
Hongze Cheng 已提交
673 674
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
675
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
676 677

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

C
Cary Xu 已提交
679 680 681 682
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
683 684 685 686 687
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
688
  // handle the request
H
Hongze Cheng 已提交
689 690 691
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
692 693
  }

H
Hongze Cheng 已提交
694
  submitRsp.pArray = taosArrayInit(pSubmitReq->numOfBlocks, sizeof(SSubmitBlkRsp));
695 696 697 698 699
  if (!submitRsp.pArray) {
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
700
  for (;;) {
H
Hongze Cheng 已提交
701 702 703
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
704 705
    SSubmitBlkRsp submitBlkRsp = {0};

H
Hongze Cheng 已提交
706 707
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
708 709
      submitBlkRsp.hashMeta = 1;

H
Hongze Cheng 已提交
710 711
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
712
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
713
        tDecoderClear(&decoder);
H
Hongze Cheng 已提交
714 715 716 717 718
        goto _exit;
      }

      if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
719
          submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
720
          tDecoderClear(&decoder);
H
Hongze Cheng 已提交
721 722 723 724
          goto _exit;
        }
      }

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

H
Hongze Cheng 已提交
729 730 731 732 733 734 735
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

D
dapan 已提交
736
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
H
Hongze Cheng 已提交
737
      tDecoderClear(&decoder);
D
dapan1121 已提交
738 739 740
    } else {
      submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
741 742
    }

H
Hongze Cheng 已提交
743
    if (tsdbInsertTableData(pVnode->pTsdb, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
744
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
745 746
    }

H
Hongze Cheng 已提交
747 748 749
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
    taosArrayPush(submitRsp.pArray, &submitBlkRsp);
H
Hongze Cheng 已提交
750
  }
751

H
Hongze Cheng 已提交
752
_exit:
H
Hongze Cheng 已提交
753 754 755 756 757 758 759 760
  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 已提交
761
    taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
H
Hongze Cheng 已提交
762 763 764
  }

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

766 767 768 769
  // TODO: the partial success scenario and the error case
  // TODO: refactor
  if ((terrno == TSDB_CODE_SUCCESS || terrno == TSDB_CODE_TDB_TABLE_ALREADY_EXIST) &&
      (pRsp->code == TSDB_CODE_SUCCESS)) {
770
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_DATA_TYPE_SUBMIT_BLOCK);
771
  }
C
Cary Xu 已提交
772

H
Hongze Cheng 已提交
773
  return 0;
L
Liu Jicong 已提交
774
}
775 776 777

static int vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
  SVCreateTSmaReq req = {0};
H
Hongze Cheng 已提交
778
  SDecoder        coder;
779 780 781 782 783 784 785 786 787 788 789 790

  pRsp->msgType = TDMT_VND_CREATE_SMA_RSP;
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;

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

  if (tDecodeSVCreateTSmaReq(&coder, &req) < 0) {
    pRsp->code = terrno;
    goto _err;
791
  }
C
Cary Xu 已提交
792

C
Cary Xu 已提交
793 794 795 796
  // record current timezone of server side
  req.timezoneInt = tsTimezone;

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

801
  tDecoderClear(&coder);
C
Cary Xu 已提交
802 803
  vDebug("vgId:%d success to create tsma %s:%" PRIi64 " for table %" PRIi64, TD_VID(pVnode), req.indexName,
         req.indexUid, req.tableUid);
H
Hongze Cheng 已提交
804
  return 0;
805 806 807

_err:
  tDecoderClear(&coder);
C
Cary Xu 已提交
808 809
  vError("vgId:%d failed to create tsma %s:%" PRIi64 " for table %" PRIi64 " since %s", TD_VID(pVnode), req.indexName,
         req.indexUid, req.tableUid, terrstr(terrno));
810
  return -1;
L
Liu Jicong 已提交
811
}