From 179dc390752161448acaea297acb98e70f56ac43 Mon Sep 17 00:00:00 2001
From: Hongze Cheng <hzcheng@taosdata.com>
Date: Mon, 6 Jun 2022 01:51:10 +0000
Subject: [PATCH] more refact

---
 source/dnode/vnode/src/inc/tsdb.h          | 4 ++--
 source/dnode/vnode/src/tsdb/tsdbMemTable.c | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/source/dnode/vnode/src/inc/tsdb.h b/source/dnode/vnode/src/inc/tsdb.h
index 329b93e73c..35bdd0e312 100644
--- a/source/dnode/vnode/src/inc/tsdb.h
+++ b/source/dnode/vnode/src/inc/tsdb.h
@@ -223,8 +223,8 @@ struct STbData {
 };
 
 struct SMemTable {
-  SVBufPool *pPool;
-  T_REF_DECLARE()
+  STsdb     *pTsdb;
+  int32_t    nRef;
   SRWLatch   latch;
   TSDBKEY    minKey;
   TSDBKEY    maxKey;
diff --git a/source/dnode/vnode/src/tsdb/tsdbMemTable.c b/source/dnode/vnode/src/tsdb/tsdbMemTable.c
index 1427751eae..a235a6523b 100644
--- a/source/dnode/vnode/src/tsdb/tsdbMemTable.c
+++ b/source/dnode/vnode/src/tsdb/tsdbMemTable.c
@@ -36,8 +36,8 @@ int tsdbMemTableCreate(STsdb *pTsdb, SMemTable **ppMemTable) {
     return -1;
   }
 
-  pMemTable->pPool = pTsdb->pVnode->inUse;
-  T_REF_INIT_VAL(pMemTable, 1);
+  pMemTable->pTsdb = pTsdb;
+  pMemTable->nRef = 1;
   taosInitRWLatch(&pMemTable->latch);
   pMemTable->minKey = (TSDBKEY){.ts = TSKEY_MAX, .version = INT64_MAX};
   pMemTable->maxKey = (TSDBKEY){.ts = TSKEY_MIN, .version = -1};
@@ -287,7 +287,7 @@ int tsdbInsertTableData(STsdb *pTsdb, SSubmitMsgIter *pMsgIter, SSubmitBlk *pBlo
 
   // copy data to buffer pool
   int32_t tlen = pMsgIter->dataLen + pMsgIter->schemaLen + sizeof(*pBlock);
-  pBlkCopy = (SSubmitBlk *)vnodeBufPoolMalloc(pTsdb->mem->pPool, tlen);
+  pBlkCopy = (SSubmitBlk *)vnodeBufPoolMalloc(pTsdb->pVnode->inUse, tlen);
   memcpy(pBlkCopy, pBlock, tlen);
 
   tInitSubmitBlkIter(pMsgIter, pBlkCopy, &blkIter);
-- 
GitLab