From 5ce5dded752035b64dd8b82350be386054fcd698 Mon Sep 17 00:00:00 2001
From: Benguang Zhao <bgzhao@taosdata.com>
Date: Wed, 8 Feb 2023 14:41:15 +0800
Subject: [PATCH] enh: reset wal reader after walReadVer

---
 include/libs/wal/wal.h             | 1 +
 source/libs/sync/src/syncRaftLog.c | 1 +
 source/libs/wal/src/walRead.c      | 9 +++++++++
 3 files changed, 11 insertions(+)

diff --git a/include/libs/wal/wal.h b/include/libs/wal/wal.h
index a0f421212a..1eed342f8c 100644
--- a/include/libs/wal/wal.h
+++ b/include/libs/wal/wal.h
@@ -191,6 +191,7 @@ int32_t walApplyVer(SWal *, int64_t ver);
 // read
 SWalReader *walOpenReader(SWal *, SWalFilterCond *pCond);
 void        walCloseReader(SWalReader *pRead);
+void        walReadReset(SWalReader *pReader);
 int32_t     walReadVer(SWalReader *pRead, int64_t ver);
 int32_t     walReadSeekVer(SWalReader *pRead, int64_t ver);
 int32_t     walNextValidMsg(SWalReader *pRead);
diff --git a/source/libs/sync/src/syncRaftLog.c b/source/libs/sync/src/syncRaftLog.c
index e6569d9974..9299651999 100644
--- a/source/libs/sync/src/syncRaftLog.c
+++ b/source/libs/sync/src/syncRaftLog.c
@@ -249,6 +249,7 @@ int32_t raftLogGetEntry(struct SSyncLogStore* pLogStore, SyncIndex index, SSyncR
 
   int64_t ts2 = taosGetTimestampNs();
   code = walReadVer(pWalHandle, index);
+  walReadReset(pWalHandle);
   int64_t ts3 = taosGetTimestampNs();
 
   // code = walReadVerCached(pWalHandle, index);
diff --git a/source/libs/wal/src/walRead.c b/source/libs/wal/src/walRead.c
index 07228bddd8..0f8f140bd8 100644
--- a/source/libs/wal/src/walRead.c
+++ b/source/libs/wal/src/walRead.c
@@ -594,3 +594,12 @@ int32_t walReadVer(SWalReader *pReader, int64_t ver) {
 
   return 0;
 }
+
+void walReadReset(SWalReader *pReader) {
+  taosThreadMutexLock(&pReader->mutex);
+  taosCloseFile(&pReader->pIdxFile);
+  taosCloseFile(&pReader->pLogFile);
+  pReader->curInvalid = 1;
+  pReader->curFileFirstVer = -1;
+  taosThreadMutexUnlock(&pReader->mutex);
+}