From a963719ea10293ad66c72518609278d392e2e7d0 Mon Sep 17 00:00:00 2001 From: dapan1121 <89396746@qq.com> Date: Tue, 5 Jan 2021 18:17:51 +0800 Subject: [PATCH] add sql check --- src/client/src/tscServer.c | 5 +++++ src/client/src/tscSubquery.c | 2 ++ src/client/src/tscUtil.c | 5 +++++ 3 files changed, 12 insertions(+) diff --git a/src/client/src/tscServer.c b/src/client/src/tscServer.c index b785f741db..7f6fb4eb9a 100644 --- a/src/client/src/tscServer.c +++ b/src/client/src/tscServer.c @@ -42,6 +42,11 @@ static void tscCheckpSql(SSqlObj* pSql) { assert(parent->metaSubPtr == pSql && parent->self == pSql->parentRid && parent->metaSubRid == pSql->self); } +static void tscCheckpSql2(SSqlObj* pSql, SSqlObj* parent) { + assert(parent->metaSubPtr == pSql && parent->self == pSql->parentRid && parent->metaSubRid == pSql->self); +} + + static int32_t minMsgSize() { return tsRpcHeadSize + 100; } static int32_t getWaitingTimeInterval(int32_t count) { int32_t initial = 100; // 100 ms by default diff --git a/src/client/src/tscSubquery.c b/src/client/src/tscSubquery.c index 0068991825..ee74496255 100644 --- a/src/client/src/tscSubquery.c +++ b/src/client/src/tscSubquery.c @@ -1336,6 +1336,8 @@ void tscJoinQueryCallback(void* param, TAOS_RES* tres, int code) { SJoinSupporter* pSupporter = (SJoinSupporter*)param; SSqlObj* pParentSql = pSupporter->pObj; + tscCheckpSql2(pSql, pParentSql); + // There is only one subquery and table for each subquery. SQueryInfo* pQueryInfo = tscGetQueryInfoDetail(&pSql->cmd, 0); assert(pQueryInfo->numOfTables == 1 && pSql->cmd.numOfClause == 1); diff --git a/src/client/src/tscUtil.c b/src/client/src/tscUtil.c index b2ac5deb15..85401f826e 100644 --- a/src/client/src/tscUtil.c +++ b/src/client/src/tscUtil.c @@ -1997,6 +1997,7 @@ SSqlObj* createSubqueryObj(SSqlObj* pSql, int16_t tableIndex, void (*fp)(), void pNew->pTscObj = pSql->pTscObj; pNew->signature = pNew; pNew->sqlstr = strdup(pSql->sqlstr); + pNew->parentRid = pSql->self; SSqlCmd* pnCmd = &pNew->cmd; memcpy(pnCmd, pCmd, sizeof(SSqlCmd)); @@ -2145,6 +2146,10 @@ SSqlObj* createSubqueryObj(SSqlObj* pSql, int16_t tableIndex, void (*fp)(), void } registerSqlObj(pNew); + + pSql->metaSubRid = pNew->self; + pSql->metaSubPtr = pNew; + return pNew; _error: -- GitLab