793 lines
24 KiB
C
793 lines
24 KiB
C
/*
|
|
* 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/>.
|
|
*/
|
|
|
|
#include "vnd.h"
|
|
|
|
static int vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
|
|
static int vnodeProcessAlterStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
|
|
static int vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
|
|
static int vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
|
|
static int vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
|
|
static int vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
|
|
static int vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp);
|
|
static int vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp);
|
|
|
|
int vnodePreprocessWriteReqs(SVnode *pVnode, SArray *pMsgs, int64_t *version) {
|
|
#if 0
|
|
SRpcMsg *pMsg;
|
|
SRpcMsg *pRpc;
|
|
|
|
*version = pVnode->state.processed;
|
|
for (int i = 0; i < taosArrayGetSize(pMsgs); i++) {
|
|
pMsg = *(SRpcMsg **)taosArrayGet(pMsgs, i);
|
|
pRpc = pMsg;
|
|
|
|
// set request version
|
|
if (walWrite(pVnode->pWal, pVnode->state.processed++, pRpc->msgType, pRpc->pCont, pRpc->contLen) < 0) {
|
|
vError("vnode:%d write wal error since %s", TD_VID(pVnode), terrstr());
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
walFsync(pVnode->pWal, false);
|
|
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
int vnodeProcessWriteReq(SVnode *pVnode, SRpcMsg *pMsg, int64_t version, SRpcMsg *pRsp) {
|
|
void *ptr = NULL;
|
|
void *pReq;
|
|
int len;
|
|
int ret;
|
|
|
|
vTrace("vgId:%d start to process write request %s, version %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
|
|
version);
|
|
|
|
pVnode->state.applied = version;
|
|
|
|
// skip header
|
|
pReq = POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead));
|
|
len = pMsg->contLen - sizeof(SMsgHead);
|
|
|
|
switch (pMsg->msgType) {
|
|
/* META */
|
|
case TDMT_VND_CREATE_STB:
|
|
if (vnodeProcessCreateStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_ALTER_STB:
|
|
if (vnodeProcessAlterStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_DROP_STB:
|
|
if (vnodeProcessDropStbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_CREATE_TABLE:
|
|
if (vnodeProcessCreateTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_ALTER_TABLE:
|
|
if (vnodeProcessAlterTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_DROP_TABLE:
|
|
if (vnodeProcessDropTbReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
break;
|
|
case TDMT_VND_CREATE_SMA: {
|
|
if (vnodeProcessCreateTSmaReq(pVnode, version, pReq, len, pRsp) < 0) goto _err;
|
|
} break;
|
|
/* TSDB */
|
|
case TDMT_VND_SUBMIT:
|
|
if (vnodeProcessSubmitReq(pVnode, version, pMsg->pCont, pMsg->contLen, pRsp) < 0) goto _err;
|
|
break;
|
|
/* TQ */
|
|
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;
|
|
case TDMT_VND_MQ_VG_DELETE:
|
|
if (tqProcessVgDeleteReq(pVnode->pTq, pMsg->pCont, pMsg->contLen) < 0) {
|
|
// TODO: handle error
|
|
}
|
|
break;
|
|
case TDMT_VND_TASK_DEPLOY: {
|
|
if (tqProcessTaskDeploy(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)),
|
|
pMsg->contLen - sizeof(SMsgHead)) < 0) {
|
|
}
|
|
} break;
|
|
#if 0
|
|
case TDMT_VND_TASK_WRITE_EXEC: {
|
|
if (tqProcessTaskExec(pVnode->pTq, POINTER_SHIFT(pMsg->pCont, sizeof(SMsgHead)), pMsg->contLen - sizeof(SMsgHead),
|
|
0) < 0) {
|
|
}
|
|
} break;
|
|
#endif
|
|
case TDMT_VND_ALTER_VNODE:
|
|
break;
|
|
default:
|
|
ASSERT(0);
|
|
break;
|
|
}
|
|
|
|
vDebug("vgId:%d process %s request success, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType), version);
|
|
|
|
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;
|
|
}
|
|
|
|
// commit if need
|
|
if (vnodeShouldCommit(pVnode)) {
|
|
vInfo("vgId:%d commit at version %" PRId64, TD_VID(pVnode), version);
|
|
// commit current change
|
|
vnodeCommit(pVnode);
|
|
|
|
// start a new one
|
|
vnodeBegin(pVnode);
|
|
}
|
|
|
|
return 0;
|
|
|
|
_err:
|
|
vDebug("vgId:%d process %s request failed since %s, version: %" PRId64, TD_VID(pVnode), TMSG_INFO(pMsg->msgType),
|
|
tstrerror(terrno), version);
|
|
return -1;
|
|
}
|
|
|
|
int vnodeProcessQueryMsg(SVnode *pVnode, SRpcMsg *pMsg) {
|
|
vTrace("message in vnode query queue is processing");
|
|
#if 0
|
|
SReadHandle handle = {.reader = pVnode->pTsdb, .meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
|
|
#endif
|
|
SReadHandle handle = {.meta = pVnode->pMeta, .config = &pVnode->config, .vnode = pVnode, .pMsgCb = &pVnode->msgCb};
|
|
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);
|
|
|
|
case TDMT_VND_TASK_RUN: {
|
|
int32_t code = tqProcessTaskRunReq(pVnode->pTq, pMsg);
|
|
pMsg->pCont = NULL;
|
|
return code;
|
|
}
|
|
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);
|
|
|
|
#if 0
|
|
case TDMT_VND_TASK_PIPE_EXEC:
|
|
case TDMT_VND_TASK_MERGE_EXEC:
|
|
return tqProcessTaskExec(pVnode->pTq, msgstr, msgLen, 0);
|
|
case TDMT_VND_STREAM_TRIGGER:{
|
|
// refactor, avoid double free
|
|
int code = tqProcessStreamTrigger(pVnode->pTq, pMsg->pCont, pMsg->contLen, 0);
|
|
pMsg->pCont = NULL;
|
|
return code;
|
|
}
|
|
#endif
|
|
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);
|
|
tdProcessTSmaInsert(((SVnode *)pVnode)->pSma, smaId, (const char *)data);
|
|
}
|
|
|
|
int vnodeProcessSyncReq(SVnode *pVnode, SRpcMsg *pMsg, SRpcMsg **pRsp) {
|
|
int32_t ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
|
|
|
|
if (syncEnvIsStart()) {
|
|
SSyncNode *pSyncNode = syncNodeAcquire(pVnode->sync);
|
|
assert(pSyncNode != NULL);
|
|
|
|
ESyncState state = syncGetMyRole(pVnode->sync);
|
|
SyncTerm currentTerm = syncGetMyTerm(pVnode->sync);
|
|
|
|
SMsgHead *pHead = pMsg->pCont;
|
|
|
|
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);
|
|
|
|
SRpcMsg *pRpcMsg = pMsg;
|
|
|
|
if (pRpcMsg->msgType == TDMT_VND_SYNC_TIMEOUT) {
|
|
SyncTimeout *pSyncMsg = syncTimeoutFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnTimeoutCb(pSyncNode, pSyncMsg);
|
|
syncTimeoutDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING) {
|
|
SyncPing *pSyncMsg = syncPingFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnPingCb(pSyncNode, pSyncMsg);
|
|
syncPingDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_PING_REPLY) {
|
|
SyncPingReply *pSyncMsg = syncPingReplyFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnPingReplyCb(pSyncNode, pSyncMsg);
|
|
syncPingReplyDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_CLIENT_REQUEST) {
|
|
SyncClientRequest *pSyncMsg = syncClientRequestFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnClientRequestCb(pSyncNode, pSyncMsg);
|
|
syncClientRequestDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE) {
|
|
SyncRequestVote *pSyncMsg = syncRequestVoteFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnRequestVoteCb(pSyncNode, pSyncMsg);
|
|
syncRequestVoteDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_REQUEST_VOTE_REPLY) {
|
|
SyncRequestVoteReply *pSyncMsg = syncRequestVoteReplyFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnRequestVoteReplyCb(pSyncNode, pSyncMsg);
|
|
syncRequestVoteReplyDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES) {
|
|
SyncAppendEntries *pSyncMsg = syncAppendEntriesFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnAppendEntriesCb(pSyncNode, pSyncMsg);
|
|
syncAppendEntriesDestroy(pSyncMsg);
|
|
|
|
} else if (pRpcMsg->msgType == TDMT_VND_SYNC_APPEND_ENTRIES_REPLY) {
|
|
SyncAppendEntriesReply *pSyncMsg = syncAppendEntriesReplyFromRpcMsg2(pRpcMsg);
|
|
assert(pSyncMsg != NULL);
|
|
|
|
ret = syncNodeOnAppendEntriesReplyCb(pSyncNode, pSyncMsg);
|
|
syncAppendEntriesReplyDestroy(pSyncMsg);
|
|
|
|
} else {
|
|
vError("==vnodeProcessSyncReq== error msg type:%d", pRpcMsg->msgType);
|
|
ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
|
|
}
|
|
|
|
syncNodeRelease(pSyncNode);
|
|
} else {
|
|
vError("==vnodeProcessSyncReq== error syncEnv stop");
|
|
ret = TAOS_SYNC_PROPOSE_OTHER_ERROR;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static int vnodeProcessCreateStbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
|
|
SVCreateStbReq req = {0};
|
|
SDecoder coder;
|
|
|
|
pRsp->msgType = TDMT_VND_CREATE_STB_RSP;
|
|
pRsp->code = TSDB_CODE_SUCCESS;
|
|
pRsp->pCont = NULL;
|
|
pRsp->contLen = 0;
|
|
|
|
// decode and process req
|
|
tDecoderInit(&coder, pReq, len);
|
|
|
|
if (tDecodeSVCreateStbReq(&coder, &req) < 0) {
|
|
pRsp->code = terrno;
|
|
goto _err;
|
|
}
|
|
|
|
if (metaCreateSTable(pVnode->pMeta, version, &req) < 0) {
|
|
pRsp->code = terrno;
|
|
goto _err;
|
|
}
|
|
|
|
tdProcessRSmaCreate(pVnode->pSma, pVnode->pMeta, &req, &pVnode->msgCb);
|
|
|
|
tDecoderClear(&coder);
|
|
return 0;
|
|
|
|
_err:
|
|
tDecoderClear(&coder);
|
|
return -1;
|
|
}
|
|
|
|
static int vnodeProcessCreateTbReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
|
|
SDecoder decoder = {0};
|
|
int rcode = 0;
|
|
SVCreateTbBatchReq req = {0};
|
|
SVCreateTbReq *pCreateReq;
|
|
SVCreateTbBatchRsp rsp = {0};
|
|
SVCreateTbRsp cRsp = {0};
|
|
char tbName[TSDB_TABLE_FNAME_LEN];
|
|
STbUidStore *pStore = NULL;
|
|
SArray *tbUids = NULL;
|
|
|
|
pRsp->msgType = TDMT_VND_CREATE_TABLE_RSP;
|
|
pRsp->code = TSDB_CODE_SUCCESS;
|
|
pRsp->pCont = NULL;
|
|
pRsp->contLen = 0;
|
|
|
|
// decode
|
|
tDecoderInit(&decoder, pReq, len);
|
|
if (tDecodeSVCreateTbBatchReq(&decoder, &req) < 0) {
|
|
rcode = -1;
|
|
terrno = TSDB_CODE_INVALID_MSG;
|
|
goto _exit;
|
|
}
|
|
|
|
rsp.pArray = taosArrayInit(req.nReqs, sizeof(cRsp));
|
|
tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
|
|
if (rsp.pArray == NULL || tbUids == NULL) {
|
|
rcode = -1;
|
|
terrno = TSDB_CODE_OUT_OF_MEMORY;
|
|
goto _exit;
|
|
}
|
|
|
|
// loop to create table
|
|
for (int iReq = 0; iReq < req.nReqs; iReq++) {
|
|
pCreateReq = req.pReqs + iReq;
|
|
|
|
// 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
|
|
if (metaCreateTable(pVnode->pMeta, version, pCreateReq) < 0) {
|
|
if (pCreateReq->flags & TD_CREATE_IF_NOT_EXISTS && terrno == TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
|
|
cRsp.code = TSDB_CODE_SUCCESS;
|
|
} else {
|
|
cRsp.code = terrno;
|
|
}
|
|
} else {
|
|
cRsp.code = TSDB_CODE_SUCCESS;
|
|
tdFetchTbUidList(pVnode->pSma, &pStore, pCreateReq->ctb.suid, pCreateReq->uid);
|
|
taosArrayPush(tbUids, &pCreateReq->uid);
|
|
}
|
|
|
|
taosArrayPush(rsp.pArray, &cRsp);
|
|
}
|
|
|
|
tDecoderClear(&decoder);
|
|
|
|
tqUpdateTbUidList(pVnode->pTq, tbUids, true);
|
|
tdUpdateTbUidList(pVnode->pSma, pStore);
|
|
tdUidStoreFree(pStore);
|
|
|
|
// prepare rsp
|
|
SEncoder encoder = {0};
|
|
int32_t ret = 0;
|
|
tEncodeSize(tEncodeSVCreateTbBatchRsp, &rsp, pRsp->contLen, ret);
|
|
pRsp->pCont = rpcMallocCont(pRsp->contLen);
|
|
if (pRsp->pCont == NULL) {
|
|
terrno = TSDB_CODE_OUT_OF_MEMORY;
|
|
rcode = -1;
|
|
goto _exit;
|
|
}
|
|
tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
|
|
tEncodeSVCreateTbBatchRsp(&encoder, &rsp);
|
|
tEncoderClear(&encoder);
|
|
|
|
_exit:
|
|
taosArrayDestroy(rsp.pArray);
|
|
taosArrayDestroy(tbUids);
|
|
tDecoderClear(&decoder);
|
|
tEncoderClear(&encoder);
|
|
return rcode;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
if (metaAlterSTable(pVnode->pMeta, version, &req) < 0) {
|
|
pRsp->code = terrno;
|
|
tDecoderClear(&dc);
|
|
return -1;
|
|
}
|
|
|
|
tDecoderClear(&dc);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeProcessDropStbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
|
|
SVDropStbReq req = {0};
|
|
int rcode = TSDB_CODE_SUCCESS;
|
|
SDecoder decoder = {0};
|
|
|
|
pRsp->msgType = TDMT_VND_CREATE_STB_RSP;
|
|
pRsp->pCont = NULL;
|
|
pRsp->contLen = 0;
|
|
|
|
// decode request
|
|
tDecoderInit(&decoder, pReq, len);
|
|
if (tDecodeSVDropStbReq(&decoder, &req) < 0) {
|
|
rcode = TSDB_CODE_INVALID_MSG;
|
|
goto _exit;
|
|
}
|
|
|
|
// process request
|
|
// if (metaDropSTable(pVnode->pMeta, version, &req) < 0) {
|
|
// rcode = terrno;
|
|
// goto _exit;
|
|
// }
|
|
|
|
// return rsp
|
|
_exit:
|
|
pRsp->code = rcode;
|
|
tDecoderClear(&decoder);
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeProcessAlterTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
|
|
SVAlterTbReq vAlterTbReq = {0};
|
|
SVAlterTbRsp vAlterTbRsp = {0};
|
|
SDecoder dc = {0};
|
|
int rcode = 0;
|
|
int ret;
|
|
SEncoder ec = {0};
|
|
|
|
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) {
|
|
vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
|
|
tDecoderClear(&dc);
|
|
rcode = -1;
|
|
goto _exit;
|
|
}
|
|
|
|
// process
|
|
if (metaAlterTable(pVnode->pMeta, version, &vAlterTbReq) < 0) {
|
|
vAlterTbRsp.code = TSDB_CODE_INVALID_MSG;
|
|
tDecoderClear(&dc);
|
|
rcode = -1;
|
|
goto _exit;
|
|
}
|
|
tDecoderClear(&dc);
|
|
|
|
_exit:
|
|
tEncodeSize(tEncodeSVAlterTbRsp, &vAlterTbRsp, pRsp->contLen, ret);
|
|
pRsp->pCont = rpcMallocCont(pRsp->contLen);
|
|
tEncoderInit(&ec, pRsp->pCont, pRsp->contLen);
|
|
tEncodeSVAlterTbRsp(&ec, &vAlterTbRsp);
|
|
tEncoderClear(&ec);
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeProcessDropTbReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
|
|
SVDropTbBatchReq req = {0};
|
|
SVDropTbBatchRsp rsp = {0};
|
|
SDecoder decoder = {0};
|
|
SEncoder encoder = {0};
|
|
int ret;
|
|
SArray *tbUids = NULL;
|
|
|
|
pRsp->msgType = TDMT_VND_DROP_TABLE_RSP;
|
|
pRsp->pCont = NULL;
|
|
pRsp->contLen = 0;
|
|
pRsp->code = TSDB_CODE_SUCCESS;
|
|
|
|
// decode req
|
|
tDecoderInit(&decoder, pReq, len);
|
|
ret = tDecodeSVDropTbBatchReq(&decoder, &req);
|
|
if (ret < 0) {
|
|
terrno = TSDB_CODE_INVALID_MSG;
|
|
pRsp->code = terrno;
|
|
goto _exit;
|
|
}
|
|
|
|
// process req
|
|
tbUids = taosArrayInit(req.nReqs, sizeof(int64_t));
|
|
rsp.pArray = taosArrayInit(req.nReqs, sizeof(SVDropTbRsp));
|
|
if (tbUids == NULL || rsp.pArray == NULL) goto _exit;
|
|
|
|
for (int iReq = 0; iReq < req.nReqs; iReq++) {
|
|
SVDropTbReq *pDropTbReq = req.pReqs + iReq;
|
|
SVDropTbRsp dropTbRsp = {0};
|
|
|
|
/* code */
|
|
ret = metaDropTable(pVnode->pMeta, version, pDropTbReq, tbUids);
|
|
if (ret < 0) {
|
|
if (pDropTbReq->igNotExists && terrno == TSDB_CODE_VND_TABLE_NOT_EXIST) {
|
|
dropTbRsp.code = TSDB_CODE_SUCCESS;
|
|
} else {
|
|
dropTbRsp.code = terrno;
|
|
}
|
|
} else {
|
|
dropTbRsp.code = TSDB_CODE_SUCCESS;
|
|
}
|
|
|
|
taosArrayPush(rsp.pArray, &dropTbRsp);
|
|
}
|
|
|
|
tqUpdateTbUidList(pVnode->pTq, tbUids, false);
|
|
|
|
_exit:
|
|
taosArrayDestroy(tbUids);
|
|
tDecoderClear(&decoder);
|
|
tEncodeSize(tEncodeSVDropTbBatchRsp, &rsp, pRsp->contLen, ret);
|
|
pRsp->pCont = rpcMallocCont(pRsp->contLen);
|
|
tEncoderInit(&encoder, pRsp->pCont, pRsp->contLen);
|
|
tEncodeSVDropTbBatchRsp(&encoder, &rsp);
|
|
tEncoderClear(&encoder);
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeDebugPrintSingleSubmitMsg(SMeta *pMeta, SSubmitBlk *pBlock, SSubmitMsgIter *msgIter, const char *tags) {
|
|
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);
|
|
}
|
|
pSchema = metaGetTbTSchema(pMeta, msgIter->suid, 1); // TODO: use the real schema
|
|
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;
|
|
}
|
|
|
|
static int vnodeDebugPrintSubmitMsg(SVnode *pVnode, SSubmitReq *pMsg, const char *tags) {
|
|
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;
|
|
|
|
vnodeDebugPrintSingleSubmitMsg(pMeta, pBlock, &msgIter, tags);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeProcessSubmitReq(SVnode *pVnode, int64_t version, void *pReq, int32_t len, SRpcMsg *pRsp) {
|
|
SSubmitReq *pSubmitReq = (SSubmitReq *)pReq;
|
|
SSubmitRsp submitRsp = {0};
|
|
SSubmitMsgIter msgIter = {0};
|
|
SSubmitBlk *pBlock;
|
|
SSubmitRsp rsp = {0};
|
|
SVCreateTbReq createTbReq = {0};
|
|
SDecoder decoder = {0};
|
|
int32_t nRows;
|
|
int32_t tsize, ret;
|
|
SEncoder encoder = {0};
|
|
terrno = TSDB_CODE_SUCCESS;
|
|
|
|
pRsp->code = 0;
|
|
|
|
#ifdef TD_DEBUG_PRINT_ROW
|
|
vnodeDebugPrintSubmitMsg(pVnode, pReq, __func__);
|
|
#endif
|
|
|
|
if (tsdbScanAndConvertSubmitMsg(pVnode->pTsdb, pSubmitReq) < 0) {
|
|
pRsp->code = terrno;
|
|
goto _exit;
|
|
}
|
|
|
|
// handle the request
|
|
if (tInitSubmitMsgIter(pSubmitReq, &msgIter) < 0) {
|
|
pRsp->code = TSDB_CODE_INVALID_MSG;
|
|
goto _exit;
|
|
}
|
|
|
|
submitRsp.pArray = taosArrayInit(pSubmitReq->numOfBlocks, sizeof(SSubmitBlkRsp));
|
|
if (!submitRsp.pArray) {
|
|
pRsp->code = TSDB_CODE_OUT_OF_MEMORY;
|
|
goto _exit;
|
|
}
|
|
|
|
for (int i = 0;;) {
|
|
tGetSubmitMsgNext(&msgIter, &pBlock);
|
|
if (pBlock == NULL) break;
|
|
|
|
SSubmitBlkRsp submitBlkRsp = {0};
|
|
|
|
// create table for auto create table mode
|
|
if (msgIter.schemaLen > 0) {
|
|
submitBlkRsp.hashMeta = 1;
|
|
|
|
tDecoderInit(&decoder, pBlock->data, msgIter.schemaLen);
|
|
if (tDecodeSVCreateTbReq(&decoder, &createTbReq) < 0) {
|
|
pRsp->code = TSDB_CODE_INVALID_MSG;
|
|
tDecoderClear(&decoder);
|
|
goto _exit;
|
|
}
|
|
|
|
if (metaCreateTable(pVnode->pMeta, version, &createTbReq) < 0) {
|
|
if (terrno != TSDB_CODE_TDB_TABLE_ALREADY_EXIST) {
|
|
submitBlkRsp.code = terrno;
|
|
tDecoderClear(&decoder);
|
|
goto _exit;
|
|
}
|
|
}
|
|
|
|
submitBlkRsp.uid = createTbReq.uid;
|
|
submitBlkRsp.tblFName = taosMemoryMalloc(strlen(pVnode->config.dbname) + strlen(createTbReq.name) + 2);
|
|
sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
|
|
|
|
msgIter.uid = createTbReq.uid;
|
|
if (createTbReq.type == TSDB_CHILD_TABLE) {
|
|
msgIter.suid = createTbReq.ctb.suid;
|
|
} else {
|
|
msgIter.suid = 0;
|
|
}
|
|
|
|
vnodeDebugPrintSingleSubmitMsg(pVnode->pMeta, pBlock, &msgIter, "real uid");
|
|
tDecoderClear(&decoder);
|
|
} else {
|
|
submitBlkRsp.tblFName = taosMemoryMalloc(TSDB_TABLE_FNAME_LEN);
|
|
sprintf(submitBlkRsp.tblFName, "%s.", pVnode->config.dbname);
|
|
}
|
|
|
|
if (tsdbInsertTableData(pVnode->pTsdb, &msgIter, pBlock, &submitBlkRsp) < 0) {
|
|
submitBlkRsp.code = terrno;
|
|
}
|
|
|
|
submitRsp.numOfRows += submitBlkRsp.numOfRows;
|
|
submitRsp.affectedRows += submitBlkRsp.affectedRows;
|
|
taosArrayPush(submitRsp.pArray, &submitBlkRsp);
|
|
}
|
|
|
|
_exit:
|
|
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++) {
|
|
taosMemoryFree(((SSubmitBlkRsp *)taosArrayGet(submitRsp.pArray, i))[0].tblFName);
|
|
}
|
|
|
|
taosArrayDestroy(submitRsp.pArray);
|
|
|
|
// 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)) {
|
|
tdProcessRSmaSubmit(pVnode->pSma, pReq, STREAM_DATA_TYPE_SUBMIT_BLOCK);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int vnodeProcessCreateTSmaReq(SVnode *pVnode, int64_t version, void *pReq, int len, SRpcMsg *pRsp) {
|
|
SVCreateTSmaReq req = {0};
|
|
SDecoder coder;
|
|
|
|
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;
|
|
}
|
|
|
|
// record current timezone of server side
|
|
req.timezoneInt = tsTimezone;
|
|
|
|
if (tdProcessTSmaCreate(pVnode->pSma, version, (const char *)&req) < 0) {
|
|
pRsp->code = terrno;
|
|
goto _err;
|
|
}
|
|
|
|
tDecoderClear(&coder);
|
|
vDebug("vgId:%d success to create tsma %s:%" PRIi64 " for table %" PRIi64, TD_VID(pVnode), req.indexName,
|
|
req.indexUid, req.tableUid);
|
|
return 0;
|
|
|
|
_err:
|
|
tDecoderClear(&coder);
|
|
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));
|
|
return -1;
|
|
}
|