vnodeSvr.c 24.7 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

H
Hongze Cheng 已提交
16
#include "vnd.h"
H
Hongze Cheng 已提交
17

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
  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 已提交
41
        char    *name = NULL;
H
Hongze Cheng 已提交
42 43 44
        tStartDecode(&dc);

        tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
45
        tDecodeCStr(&dc, &name);
H
Hongze Cheng 已提交
46 47 48 49 50 51 52 53 54 55 56 57 58 59
        *(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 已提交
60
      tb_uid_t       uid;
H
Hongze Cheng 已提交
61 62 63 64 65 66 67 68

      tInitSubmitMsgIter(pSubmitReq, &msgIter);

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

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

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

          tDecodeI32v(&dc, NULL);
H
Hongze Cheng 已提交
75 76 77 78 79 80
          tDecodeCStr(&dc, &name);

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

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

    } break;
    default:
      break;
  }
H
Hongze Cheng 已提交
94 95

  return 0;
H
Hongze Cheng 已提交
96 97
}

H
Hongze Cheng 已提交
98
int vnodeProcessWriteReq(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
99
  void *ptr = NULL;
H
Hongze Cheng 已提交
100 101
  void *pReq;
  int   len;
H
Hongze Cheng 已提交
102
  int   ret;
H
Hongze Cheng 已提交
103

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

H
Hongze Cheng 已提交
107 108
  pVnode->state.applied = version;

H
Hongze Cheng 已提交
109 110 111
  // skip header
  pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
  len = pMsg->contLen - sizeof(SMsgHead);
H
Hongze Cheng 已提交
112 113

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

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

166 167 168 169 170
  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 已提交
171
  // commit if need
H
Hongze Cheng 已提交
172
  if (vnodeShouldCommit(pVnode)) {
H
Hongze Cheng 已提交
173
    vInfo("vgId:%d commit at version %" PRId64, TD_VID(pVnode), version);
H
Hongze Cheng 已提交
174 175 176 177 178
    // commit current change
    vnodeCommit(pVnode);

    // start a new one
    vnodeBegin(pVnode);
H
Hongze Cheng 已提交
179 180 181
  }

  return 0;
H
Hongze Cheng 已提交
182 183

_err:
H
Hongze Cheng 已提交
184
  vDebug("vgId:%d process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
H
Hongze Cheng 已提交
185 186
         tstrerror(terrno), version);
  return -1;
H
Hongze Cheng 已提交
187 188 189
}

int vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
D
dapan 已提交
190
  vTrace("message in vnode query queue is processing");
191
  SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
H
Hongze Cheng 已提交
192 193
  switch (pMsg->msgType) {
    case TDMT_VND_QUERY:
D
dapan1121 已提交
194
      return qWorkerProcessQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
195
    case TDMT_VND_QUERY_CONTINUE:
D
dapan1121 已提交
196
      return qWorkerProcessCQueryMsg(&handle, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
197 198 199 200 201 202 203 204 205 206 207 208
    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:
D
dapan1121 已提交
209
      return qWorkerProcessFetchMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
210
    case TDMT_VND_FETCH_RSP:
D
dapan1121 已提交
211
      return qWorkerProcessFetchRsp(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
212
    case TDMT_VND_CANCEL_TASK:
D
dapan1121 已提交
213
      return qWorkerProcessCancelMsg(pVnode, pVnode->pQuery, pMsg, 0);
H
Hongze Cheng 已提交
214
    case TDMT_VND_DROP_TASK:
D
dapan1121 已提交
215 216 217 218
      return qWorkerProcessDropMsg(pVnode, pVnode->pQuery, pMsg, 0);
    case TDMT_VND_QUERY_HEARTBEAT:
      return qWorkerProcessHbMsg(pVnode, pVnode->pQuery, pMsg, 0);

H
Hongze Cheng 已提交
219 220 221 222
    case TDMT_VND_TABLE_META:
      return vnodeGetTableMeta(pVnode, pMsg);
    case TDMT_VND_CONSUME:
      return tqProcessPollReq(pVnode->pTq, pMsg, pInfo->workerId);
L
Liu Jicong 已提交
223

L
Liu Jicong 已提交
224 225 226 227 228
    case TDMT_VND_TASK_RUN: {
      int32_t code = tqProcessTaskRunReq(pVnode->pTq, pMsg);
      pMsg->pCont = NULL;
      return code;
    }
L
Liu Jicong 已提交
229 230 231 232 233 234 235 236
    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 已提交
237 238 239 240 241 242 243 244 245 246 247
    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);
248
  tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
H
Hongze Cheng 已提交
249 250 251
}

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

254 255 256
  if (syncEnvIsStart()) {
    SSyncNode *pSyncNode = syncNodeAcquire(pVnode->sync);
    assert(pSyncNode != NULL);
M
Minghao Li 已提交
257

258 259
    ESyncState state = syncGetMyRole(pVnode->sync);
    SyncTerm   currentTerm = syncGetMyTerm(pVnode->sync);
M
Minghao Li 已提交
260

261
    SMsgHead *pHead = pMsg->pCont;
M
Minghao Li 已提交
262

263 264 265 266 267
    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 已提交
268

269
    SRpcMsg *pRpcMsg = pMsg;
M
Minghao Li 已提交
270

271 272 273
    if (pRpcMsg->msgType == TDMT_VND_SYNC_TIMEOUT) {
      SyncTimeout *pSyncMsg = syncTimeoutFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
274

275
      ret = syncNodeOnTimeoutCb(pSyncNode, pSyncMsg);
276
      syncTimeoutDestroy(pSyncMsg);
M
Minghao Li 已提交
277

278 279 280
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING) {
      SyncPing *pSyncMsg = syncPingFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
281

282
      ret = syncNodeOnPingCb(pSyncNode, pSyncMsg);
283
      syncPingDestroy(pSyncMsg);
M
Minghao Li 已提交
284

285 286 287
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING_REPLY) {
      SyncPingReply *pSyncMsg = syncPingReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
288

289
      ret = syncNodeOnPingReplyCb(pSyncNode, pSyncMsg);
290
      syncPingReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
291

292 293 294
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_CLIENT_REQUEST) {
      SyncClientRequest *pSyncMsg = syncClientRequestFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
295

296
      ret = syncNodeOnClientRequestCb(pSyncNode, pSyncMsg);
297
      syncClientRequestDestroy(pSyncMsg);
M
Minghao Li 已提交
298

299 300 301
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE) {
      SyncRequestVote *pSyncMsg = syncRequestVoteFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
302

303
      ret = syncNodeOnRequestVoteCb(pSyncNode, pSyncMsg);
304
      syncRequestVoteDestroy(pSyncMsg);
M
Minghao Li 已提交
305

306 307 308
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE_REPLY) {
      SyncRequestVoteReply *pSyncMsg = syncRequestVoteReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
309

310
      ret = syncNodeOnRequestVoteReplyCb(pSyncNode, pSyncMsg);
311
      syncRequestVoteReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
312

313 314 315
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES) {
      SyncAppendEntries *pSyncMsg = syncAppendEntriesFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);
M
Minghao Li 已提交
316

317
      ret = syncNodeOnAppendEntriesCb(pSyncNode, pSyncMsg);
318
      syncAppendEntriesDestroy(pSyncMsg);
M
Minghao Li 已提交
319

320 321 322 323
    } else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES_REPLY) {
      SyncAppendEntriesReply *pSyncMsg = syncAppendEntriesReplyFromRpcMsg2(pRpcMsg);
      assert(pSyncMsg != NULL);

324
      ret = syncNodeOnAppendEntriesReplyCb(pSyncNode, pSyncMsg);
325
      syncAppendEntriesReplyDestroy(pSyncMsg);
M
Minghao Li 已提交
326

327 328
    } else {
      vError("==vnodeProcessSyncReq== error msg type:%d", pRpcMsg->msgType);
329
      ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
330
    }
M
Minghao Li 已提交
331

332 333 334
    syncNodeRelease(pSyncNode);
  } else {
    vError("==vnodeProcessSyncReq== error syncEnv stop");
335
    ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
336
  }
337 338

  return ret;
H
Hongze Cheng 已提交
339 340
}

H
Hongze Cheng 已提交
341
static int vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
342
  SVCreateStbReq req = {0};
H
Hongze Cheng 已提交
343
  SDecoder       coder;
H
Hongze Cheng 已提交
344

H
Hongze Cheng 已提交
345 346 347 348 349 350
  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 已提交
351
  tDecoderInit(&coder, pReq, len);
H
Hongze Cheng 已提交
352 353

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

H
Hongze Cheng 已提交
358
  if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
H
Hongze Cheng 已提交
359 360
    pRsp->code = terrno;
    goto _err;
H
Hongze Cheng 已提交
361 362
  }

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

H
Hongze Cheng 已提交
365
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
366
  return 0;
H
Hongze Cheng 已提交
367 368

_err:
H
Hongze Cheng 已提交
369
  tDecoderClear(&coder);
H
Hongze Cheng 已提交
370
  return -1;
H
Hongze Cheng 已提交
371 372
}

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

  pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
H
Hongze Cheng 已提交
385 386 387
  pRsp->code = TSDB_CODE_SUCCESS;
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
H
Hongze Cheng 已提交
388

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

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

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

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

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

H
Hongze Cheng 已提交
433
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
434

435
  tqUpdateTbUidList(pVnode->pTq, tbUids, true);
436 437
  tdUpdateTbUidList(pVnode->pSma, pStore);
  tdUidStoreFree(pStore);
C
Cary Xu 已提交
438

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

H
Hongze Cheng 已提交
453
_exit:
H
Hongze Cheng 已提交
454
  taosArrayDestroy(rsp.pArray);
455
  taosArrayDestroy(tbUids);
H
Hongze Cheng 已提交
456 457
  tDecoderClear(&decoder);
  tEncoderClear(&encoder);
H
Hongze Cheng 已提交
458
  return rcode;
H
Hongze Cheng 已提交
459 460
}

H
Hongze Cheng 已提交
461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
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 已提交
477
  }
H
Hongze Cheng 已提交
478 479 480 481 482 483 484 485 486

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

  tDecoderClear(&dc);

H
Hongze Cheng 已提交
487 488 489
  return 0;
}

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

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

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

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

  // return rsp
_exit:
  pRsp->code = rcode;
H
Hongze Cheng 已提交
515
  tDecoderClear(&decoder);
H
Hongze Cheng 已提交
516 517 518
  return 0;
}

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

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

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

_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 已提交
557 558 559
  return 0;
}

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

H
Hongze Cheng 已提交
568
  pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
H
Hongze Cheng 已提交
569 570 571
  pRsp->pCont = NULL;
  pRsp->contLen = 0;
  pRsp->code = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
572 573

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

  // process req
583
  tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
H
Hongze Cheng 已提交
584
  rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
585 586
  if (tbUids == NULL || rsp.pArray == NULL) goto _exit;

H
Hongze Cheng 已提交
587 588 589
  for (int iReq = 0; iReq < req.nReqs; iReq++) {
    SVDropTbReq *pDropTbReq = req.pReqs + iReq;
    SVDropTbRsp  dropTbRsp = {0};
H
Hongze Cheng 已提交
590

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

    taosArrayPush(rsp.pArray, &dropTbRsp);
  }

606 607
  tqUpdateTbUidList(pVnode->pTq, tbUids, false);

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

H
Hongze Cheng 已提交
619
static int vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter, const char *tags) {
D
dapan 已提交
620 621 622 623
  SSubmitBlkIter blkIter = {0};
  STSchema      *pSchema = NULL;
  tb_uid_t       suid = 0;
  STSRow        *row = NULL;
C
Cary Xu 已提交
624
  int32_t        rv = -1;
D
dapan 已提交
625 626 627

  tInitSubmitBlkIter(msgIter, pBlock, &blkIter);
  if (blkIter.row == NULL) return 0;
C
Cary Xu 已提交
628
  if (!pSchema || (suid != msgIter->suid) || rv != TD_ROW_SVER(blkIter.row)) {
D
dapan 已提交
629 630 631
    if (pSchema) {
      taosMemoryFreeClear(pSchema);
    }
C
Cary Xu 已提交
632
    pSchema = metaGetTbTSchema(pMeta, msgIter->suid, TD_ROW_SVER(blkIter.row));  // TODO: use the real schema
D
dapan 已提交
633 634
    if (pSchema) {
      suid = msgIter->suid;
C
Cary Xu 已提交
635
      rv = TD_ROW_SVER(blkIter.row);
D
dapan 已提交
636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652
    }
  }
  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 已提交
653
static int vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
C
Cary Xu 已提交
654 655 656 657 658 659 660 661 662
  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 已提交
663

D
dapan 已提交
664 665
    vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
  }
C
Cary Xu 已提交
666 667 668 669

  return 0;
}

H
Hongze Cheng 已提交
670
static int vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
H
Hongze Cheng 已提交
671
  SSubmitReq    *pSubmitReq = (SSubmitReq *)pReq;
H
Hongze Cheng 已提交
672
  SSubmitRsp     submitRsp = {0};
H
Hongze Cheng 已提交
673 674 675 676
  SSubmitMsgIter msgIter = {0};
  SSubmitBlk    *pBlock;
  SSubmitRsp     rsp = {0};
  SVCreateTbReq  createTbReq = {0};
H
Hongze Cheng 已提交
677
  SDecoder       decoder = {0};
H
Hongze Cheng 已提交
678
  int32_t        nRows;
H
Hongze Cheng 已提交
679 680
  int32_t        tsize, ret;
  SEncoder       encoder = {0};
681
  SArray        *newTbUids = NULL;
682
  terrno = TSDB_CODE_SUCCESS;
H
Hongze Cheng 已提交
683 684

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

C
Cary Xu 已提交
686 687 688 689
#ifdef TD_DEBUG_PRINT_ROW
  vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
#endif

C
Cary Xu 已提交
690 691 692 693 694
  if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
    pRsp->code = terrno;
    goto _exit;
  }

H
Hongze Cheng 已提交
695
  // handle the request
H
Hongze Cheng 已提交
696 697 698
  if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
    pRsp->code = TSDB_CODE_INVALID_MSG;
    goto _exit;
H
Hongze Cheng 已提交
699 700
  }

H
Hongze Cheng 已提交
701
  submitRsp.pArray = taosArrayInit(pSubmitReq->numOfBlocks, sizeof(SSubmitBlkRsp));
702
  newTbUids = taosArrayInit(pSubmitReq->numOfBlocks, sizeof(int64_t));
703 704 705 706 707
  if (!submitRsp.pArray) {
    pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
708
  for (;;) {
H
Hongze Cheng 已提交
709 710 711
    tGetSubmitMsgNext(&msgIter, &pBlock);
    if (pBlock == NULL) break;

H
Hongze Cheng 已提交
712 713
    SSubmitBlkRsp submitBlkRsp = {0};

H
Hongze Cheng 已提交
714 715
    // create table for auto create table mode
    if (msgIter.schemaLen > 0) {
H
Hongze Cheng 已提交
716 717
      submitBlkRsp.hashMeta = 1;

H
Hongze Cheng 已提交
718 719
      tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
      if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
H
Hongze Cheng 已提交
720
        pRsp->code = TSDB_CODE_INVALID_MSG;
H
Hongze Cheng 已提交
721
        tDecoderClear(&decoder);
H
Hongze Cheng 已提交
722 723 724 725 726
        goto _exit;
      }

      if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
        if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
H
Hongze Cheng 已提交
727
          submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
728
          tDecoderClear(&decoder);
H
Hongze Cheng 已提交
729 730 731
          goto _exit;
        }
      }
732
      taosArrayPush(newTbUids, &createTbReq.uid);
H
Hongze Cheng 已提交
733

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

H
Hongze Cheng 已提交
738 739 740 741 742 743 744
      msgIter.uid = createTbReq.uid;
      if (createTbReq.type == TSDB_CHILD_TABLE) {
        msgIter.suid = createTbReq.ctb.suid;
      } else {
        msgIter.suid = 0;
      }

D
dapan 已提交
745
      vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
H
Hongze Cheng 已提交
746
      tDecoderClear(&decoder);
D
dapan1121 已提交
747 748 749
    } else {
      submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
      sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
H
Hongze Cheng 已提交
750 751
    }

H
Hongze Cheng 已提交
752
    if (tsdbInsertTableData(pVnode->pTsdb, &msgIter, pBlock, &submitBlkRsp) < 0) {
H
Hongze Cheng 已提交
753
      submitBlkRsp.code = terrno;
H
Hongze Cheng 已提交
754 755
    }

H
Hongze Cheng 已提交
756 757 758
    submitRsp.numOfRows += submitBlkRsp.numOfRows;
    submitRsp.affectedRows += submitBlkRsp.affectedRows;
    taosArrayPush(submitRsp.pArray, &submitBlkRsp);
H
Hongze Cheng 已提交
759
  }
760
  tqUpdateTbUidList(pVnode->pTq, newTbUids, true);
761

H
Hongze Cheng 已提交
762
_exit:
763
  taosArrayDestroy(newTbUids);
H
Hongze Cheng 已提交
764 765 766 767 768 769 770 771
  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 已提交
772
    taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
H
Hongze Cheng 已提交
773 774 775
  }

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

777 778 779 780
  // 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)) {
781
    tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_DATA_TYPE_SUBMIT_BLOCK);
782
  }
C
Cary Xu 已提交
783

H
Hongze Cheng 已提交
784
  return 0;
L
Liu Jicong 已提交
785
}
786 787 788

static int vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
  SVCreateTSmaReq req = {0};
H
Hongze Cheng 已提交
789
  SDecoder        coder;
790 791 792 793 794 795 796 797 798 799 800 801

  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;
802
  }
C
Cary Xu 已提交
803

C
Cary Xu 已提交
804 805 806 807
  // record current timezone of server side
  req.timezoneInt = tsTimezone;

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

812
  tDecoderClear(&coder);
C
Cary Xu 已提交
813 814
  vDebug("vgId:%d success to create tsma %s:%" PRIi64 " for table %" PRIi64, TD_VID(pVnode), req.indexName,
         req.indexUid, req.tableUid);
H
Hongze Cheng 已提交
815
  return 0;
816 817 818

_err:
  tDecoderClear(&coder);
C
Cary Xu 已提交
819 820
  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));
821
  return -1;
L
Liu Jicong 已提交
822
}