From 4224616e8af94fb5498fdcb40c81e7704b4880b8 Mon Sep 17 00:00:00 2001
From: dapan1121 <wpan@taosdata.com>
Date: Fri, 20 Dec 2024 15:12:34 +0800
Subject: [PATCH] fix: fetch last ack invalid write issue

---
 source/libs/qworker/src/qworker.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/source/libs/qworker/src/qworker.c b/source/libs/qworker/src/qworker.c
index aae07c7512..4a9eea66e2 100644
--- a/source/libs/qworker/src/qworker.c
+++ b/source/libs/qworker/src/qworker.c
@@ -1049,12 +1049,16 @@ _return:
 
   if (code || rsp) {
     bool rsped = false;
+
+    ctx = NULL;
+    (void)qwAcquireTaskCtx(QW_FPARAMS(), &ctx);
+
     if (ctx) {
       qwDbgSimulateRedirect(qwMsg, ctx, &rsped);
       qwDbgSimulateDead(QW_FPARAMS(), ctx, &rsped);
     }
 
-    if (!rsped) {
+    if (!rsped && ctx) {
       code = qwBuildAndSendFetchRsp(ctx, qwMsg->msgType + 1, &qwMsg->connInfo, rsp, dataLen, code);
       if (TSDB_CODE_SUCCESS != code) {
         QW_TASK_ELOG("fetch rsp send fail, msgType:%s, handle:%p, code:%x - %s, dataLen:%d",
@@ -1067,6 +1071,8 @@ _return:
       qwFreeFetchRsp(rsp);
       rsp = NULL;
     }
+
+    qwReleaseTaskCtx(mgmt, ctx);    
   } else {
     // qwQuickRspFetchReq(QW_FPARAMS(), ctx, qwMsg, code);
   }