From b8fde1baf4d35bc47b13af3ee589ee1448320802 Mon Sep 17 00:00:00 2001
From: Haojun Liao <hjliao@taosdata.com>
Date: Thu, 12 Dec 2024 13:53:53 +0800
Subject: [PATCH] fix(stream): check return value.

---
 source/libs/stream/src/streamHb.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/source/libs/stream/src/streamHb.c b/source/libs/stream/src/streamHb.c
index 35d520cae8..53b6a38b35 100644
--- a/source/libs/stream/src/streamHb.c
+++ b/source/libs/stream/src/streamHb.c
@@ -434,6 +434,7 @@ void streamMetaGetHbSendInfo(SMetaHbInfo* pInfo, int64_t* pStartTs, int32_t* pSe
 int32_t streamProcessHeartbeatRsp(SStreamMeta* pMeta, SMStreamHbRspMsg* pRsp) {
   SMetaHbInfo* pInfo = pMeta->pHbInfo;
   SEpSet       epset = {0};
+  int32_t      code = 0;
 
   stDebug("vgId:%d process hbMsg rsp, msgId:%d rsp confirmed", pMeta->vgId, pRsp->msgId);
   streamMetaWLock(pMeta);
@@ -446,8 +447,8 @@ int32_t streamProcessHeartbeatRsp(SStreamMeta* pMeta, SMStreamHbRspMsg* pRsp) {
     pInfo->hbCount += 1;
     pInfo->msgSendTs = -1;
 
-    streamTaskGetMndEpset(pMeta, &epset);
-    if (!isEpsetEqual(&pRsp->mndEpset, &epset)) {
+    code = streamTaskGetMndEpset(pMeta, &epset);
+    if (!isEpsetEqual(&pRsp->mndEpset, &epset) && (code == 0)) {
       // we need to update the mnode epset for each tasks
       stInfo("vgId:%d mnode epset updated, update mnode epset for all tasks", pMeta->vgId);
       streamTaskUpdateMndEpset(pMeta, &pRsp->mndEpset);