From 079f6358aa38cf14ce9d0bbd7c4aa670c9278aca Mon Sep 17 00:00:00 2001 From: Haojun Liao Date: Fri, 2 Aug 2024 11:49:21 +0800 Subject: [PATCH] refactor: do some internal refactor. --- source/dnode/mnode/impl/src/mndStream.c | 7 ++----- source/dnode/mnode/impl/src/mndStreamTrans.c | 13 ++++++------- 2 files changed, 8 insertions(+), 12 deletions(-) diff --git a/source/dnode/mnode/impl/src/mndStream.c b/source/dnode/mnode/impl/src/mndStream.c index 25c94fda56..ccf22cfe85 100644 --- a/source/dnode/mnode/impl/src/mndStream.c +++ b/source/dnode/mnode/impl/src/mndStream.c @@ -749,6 +749,7 @@ static int32_t mndProcessCreateStreamReq(SRpcMsg *pReq) { int32_t sqlLen = 0; const char *pMsg = "create stream tasks on dnodes"; int32_t code = TSDB_CODE_SUCCESS; + int32_t lino = 0; terrno = TSDB_CODE_SUCCESS; SCMCreateStreamReq createReq = {0}; @@ -788,11 +789,7 @@ static int32_t mndProcessCreateStreamReq(SRpcMsg *pReq) { if (createReq.sql != NULL) { sqlLen = strlen(createReq.sql); sql = taosMemoryMalloc(sqlLen + 1); - if (sql == NULL) { - code = terrno; - goto _OVER; - } - + TSDB_CHECK_NULL(sql, code, lino, _OVER, terrno); memset(sql, 0, sqlLen + 1); memcpy(sql, createReq.sql, sqlLen); } diff --git a/source/dnode/mnode/impl/src/mndStreamTrans.c b/source/dnode/mnode/impl/src/mndStreamTrans.c index 54f4189194..40f3d34055 100644 --- a/source/dnode/mnode/impl/src/mndStreamTrans.c +++ b/source/dnode/mnode/impl/src/mndStreamTrans.c @@ -167,29 +167,28 @@ int32_t mndStreamGetRelTrans(SMnode *pMnode, int64_t streamId) { } int32_t doCreateTrans(SMnode *pMnode, SStreamObj *pStream, SRpcMsg *pReq, ETrnConflct conflict, const char *name, - const char *pMsg, STrans ** pTrans1) { + const char *pMsg, STrans **pTrans1) { *pTrans1 = NULL; terrno = 0; + int32_t code = 0; STrans *p = mndTransCreate(pMnode, TRN_POLICY_RETRY, conflict, pReq, name); if (p == NULL) { - mError("failed to build trans:%s, reason: %s", name, tstrerror(TSDB_CODE_OUT_OF_MEMORY)); - terrno = TSDB_CODE_OUT_OF_MEMORY; + mError("failed to build trans:%s, reason: %s", name, tstrerror(terrno)); return terrno; } mInfo("stream:0x%" PRIx64 " start to build trans %s, transId:%d", pStream->uid, pMsg, p->id); mndTransSetDbName(p, pStream->sourceDb, pStream->targetSTbName); - if (mndTransCheckConflict(pMnode, p) != 0) { - terrno = TSDB_CODE_MND_TRANS_CONFLICT; + if ((code = mndTransCheckConflict(pMnode, p)) != 0) { mError("failed to build trans:%s for stream:0x%" PRIx64 " code:%s", name, pStream->uid, tstrerror(terrno)); mndTransDrop(p); - return terrno; + return code; } *pTrans1 = p; - return 0; + return code; } SSdbRaw *mndStreamActionEncode(SStreamObj *pStream) {