From daa29bca42e3d91787d1d45db563a1fa8cdae60d Mon Sep 17 00:00:00 2001
From: Hongze Cheng <hzcheng@taosdata.com>
Date: Fri, 21 Jan 2022 09:35:33 +0000
Subject: [PATCH] more tdb

---
 source/libs/tdb/src/db/tdb_mpool.c  | 36 ++++++++++++++++++++++++++---
 source/libs/tdb/src/inc/tdb_mpool.h | 11 +++++----
 2 files changed, 39 insertions(+), 8 deletions(-)

diff --git a/source/libs/tdb/src/db/tdb_mpool.c b/source/libs/tdb/src/db/tdb_mpool.c
index 69b9a8a12f..a6df79c9b3 100644
--- a/source/libs/tdb/src/db/tdb_mpool.c
+++ b/source/libs/tdb/src/db/tdb_mpool.c
@@ -21,6 +21,7 @@ static void        tdbMPoolUnregFile(TDB_MPOOL *mp, TDB_MPFILE *mpf);
 static TDB_MPFILE *tdbMPoolGetFile(TDB_MPOOL *mp, uint8_t *fileid);
 static int         tdbMPoolFileReadPage(TDB_MPFILE *mpf, pgno_t pgno, void *p);
 static int         tdbMPoolFileWritePage(TDB_MPFILE *mpf, pgno_t pgno, const void *p);
+static void        tdbMPoolClockEvictPage(TDB_MPOOL *mp, pg_t **pagepp);
 
 int tdbMPoolOpen(TDB_MPOOL **mpp, uint64_t cachesize, pgsize_t pgsize) {
   TDB_MPOOL *mp = NULL;
@@ -44,6 +45,7 @@ int tdbMPoolOpen(TDB_MPOOL **mpp, uint64_t cachesize, pgsize_t pgsize) {
   mp->cachesize = cachesize;
   mp->pgsize = pgsize;
   mp->npages = cachesize / pgsize;
+  mp->clockHand = 0;
 
   TD_DLIST_INIT(&mp->freeList);
 
@@ -193,10 +195,11 @@ int tdbMPoolFileGetPage(TDB_MPFILE *mpf, pgno_t pgno, void *addr) {
     TD_DLIST_POP_WITH_FIELD(&(mp->freeList), pagep, free);
   } else {
     // no free page available
-    // pagep = tdbMpoolEvict(mp);
+    tdbMPoolClockEvictPage(mp, &pagep);
     if (pagep) {
-    } else {
-      // TODO: Cannot find a page to evict
+      if (pagep->dirty) {
+        // TODO: Handle dirty page eviction
+      }
     }
   }
 
@@ -346,4 +349,31 @@ static int tdbMPoolFileWritePage(TDB_MPFILE *mpf, pgno_t pgno, const void *p) {
   // TODO: handle error
 
   return 0;
+}
+
+static void tdbMPoolClockEvictPage(TDB_MPOOL *mp, pg_t **pagepp) {
+  pg_t *     pagep;
+  frame_id_t och;
+
+  *pagepp = NULL;
+  och = mp->clockHand;
+
+  do {
+    pagep = mp->pages + mp->clockHand;
+    mp->clockHand = (mp->clockHand + 1) % mp->npages;
+
+    if (pagep->pinRef == 0) {
+      if (pagep->rbit == 1) {
+        pagep->rbit = 0;
+      } else {
+        break;
+      }
+    }
+
+    if (mp->clockHand == och) {
+      return;
+    }
+  } while (1);
+
+  *pagepp = pagep;
 }
\ No newline at end of file
diff --git a/source/libs/tdb/src/inc/tdb_mpool.h b/source/libs/tdb/src/inc/tdb_mpool.h
index 0c7613da63..d951040d8d 100644
--- a/source/libs/tdb/src/inc/tdb_mpool.h
+++ b/source/libs/tdb/src/inc/tdb_mpool.h
@@ -45,11 +45,12 @@ typedef struct {
   TD_DLIST(TDB_MPFILE);
 } mpf_bucket_t;
 struct TDB_MPOOL {
-  int64_t   cachesize;
-  pgsize_t  pgsize;
-  int32_t   npages;
-  pg_t *    pages;
-  pg_list_t freeList;
+  int64_t    cachesize;
+  pgsize_t   pgsize;
+  int32_t    npages;
+  pg_t *     pages;
+  pg_list_t  freeList;
+  frame_id_t clockHand;
   struct {
     int32_t    nbucket;
     pg_list_t *hashtab;
-- 
GitLab