From d1a6b8bd22a3b69a7e48159e46abb88cf98789bc Mon Sep 17 00:00:00 2001
From: Haojun Liao <hjliao@taosdata.com>
Date: Fri, 22 Sep 2023 22:50:21 +0800
Subject: [PATCH] refactor: do some internal refactor.

---
 source/dnode/vnode/src/tq/tq.c  | 2 ++
 source/libs/stream/src/stream.c | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/source/dnode/vnode/src/tq/tq.c b/source/dnode/vnode/src/tq/tq.c
index 7315b8d91a..677e71efd3 100644
--- a/source/dnode/vnode/src/tq/tq.c
+++ b/source/dnode/vnode/src/tq/tq.c
@@ -1593,6 +1593,8 @@ FAIL:
   }
 
   pRspHead->vgId = htonl(req.upstreamNodeId);
+  ASSERT(pRspHead->vgId > 0);
+
   SStreamDispatchRsp* pRsp = POINTER_SHIFT(pRspHead, sizeof(SMsgHead));
   pRsp->streamId = htobe64(req.streamId);
   pRsp->upstreamTaskId = htonl(req.upstreamTaskId);
diff --git a/source/libs/stream/src/stream.c b/source/libs/stream/src/stream.c
index e1b9a2fb2f..2891c93e9b 100644
--- a/source/libs/stream/src/stream.c
+++ b/source/libs/stream/src/stream.c
@@ -140,6 +140,8 @@ static int32_t buildDispatchRsp(const SStreamTask* pTask, const SStreamDispatchR
   }
 
   ((SMsgHead*)(*pBuf))->vgId = htonl(pReq->upstreamNodeId);
+  ASSERT(((SMsgHead*)(*pBuf))->vgId > 0);
+
   SStreamDispatchRsp* pDispatchRsp = POINTER_SHIFT((*pBuf), sizeof(SMsgHead));
 
   pDispatchRsp->stage = htobe64(pReq->stage);