From 2aad82370942592a5314b880f86c23ffaf3121e3 Mon Sep 17 00:00:00 2001
From: t_max <1172915550@qq.com>
Date: Mon, 22 May 2023 10:53:10 +0800
Subject: [PATCH] fix: free assignment crash on Windows

---
 examples/c/tmq.c              | 4 ++--
 include/client/taos.h         | 1 +
 source/client/src/clientTmq.c | 8 ++++++++
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/examples/c/tmq.c b/examples/c/tmq.c
index 94545dfaad..e1133c109e 100644
--- a/examples/c/tmq.c
+++ b/examples/c/tmq.c
@@ -73,7 +73,7 @@ static int32_t init_env() {
   taos_free_result(pRes);
 
   // create database
-  pRes = taos_query(pConn, "create database tmqdb precision 'ns'");
+  pRes = taos_query(pConn, "create database tmqdb precision 'ns' WAL_RETENTION_PERIOD 3600");
   if (taos_errno(pRes) != 0) {
     printf("error in create tmqdb, reason:%s\n", taos_errstr(pRes));
     goto END;
@@ -289,7 +289,7 @@ void consume_repeatly(tmq_t* tmq) {
     }
   }
 
-  free(pAssign);
+  tmq_free_assignment(pAssign);
 
   // let's do it again
   basic_consume_loop(tmq);
diff --git a/include/client/taos.h b/include/client/taos.h
index 8811c4ab64..9f5a84fb22 100644
--- a/include/client/taos.h
+++ b/include/client/taos.h
@@ -288,6 +288,7 @@ DLL_EXPORT int32_t   tmq_commit_sync(tmq_t *tmq, const TAOS_RES *msg);
 DLL_EXPORT void      tmq_commit_async(tmq_t *tmq, const TAOS_RES *msg, tmq_commit_cb *cb, void *param);
 DLL_EXPORT int32_t   tmq_get_topic_assignment(tmq_t *tmq, const char *pTopicName, tmq_topic_assignment **assignment,
                                               int32_t *numOfAssignment);
+DLL_EXPORT void      tmq_free_assignment(tmq_topic_assignment* pAssignment);
 DLL_EXPORT int32_t   tmq_offset_seek(tmq_t *tmq, const char *pTopicName, int32_t vgId, int64_t offset);
 
 /* ----------------------TMQ CONFIGURATION INTERFACE---------------------- */
diff --git a/source/client/src/clientTmq.c b/source/client/src/clientTmq.c
index c38e206a9b..aff5846092 100644
--- a/source/client/src/clientTmq.c
+++ b/source/client/src/clientTmq.c
@@ -2584,6 +2584,14 @@ int32_t tmq_get_topic_assignment(tmq_t* tmq, const char* pTopicName, tmq_topic_a
   }
 }
 
+void tmq_free_assignment(tmq_topic_assignment* pAssignment) {
+    if (pAssignment == NULL) {
+        return;
+    }
+
+    taosMemoryFree(pAssignment);
+}
+
 int32_t tmq_offset_seek(tmq_t* tmq, const char* pTopicName, int32_t vgId, int64_t offset) {
   if (tmq == NULL) {
     tscError("invalid tmq handle, null");
-- 
GitLab