keep committed data in TDB

source/libs/tdb/src/db/tdbBtree.c

@@ -297,7 +297,8 @@ static int tdbBtreeOpenImpl(SBTree *pBt) {
 
   {
     // 1. TODO: Search the main DB to check if the DB exists
-    pgno = 0;
+    ret = tdbPagerOpenDB(pBt->pPager, &pgno, true);
+    ASSERT(ret == 0);
   }
 
   if (pgno != 0) {

source/libs/tdb/src/db/tdbPager.c

@@ -76,6 +76,8 @@ int tdbPagerOpen(SPCache *pCache, const char *fileName, SPager **ppPager) {
 
   // pPager->jfd = -1;
   pPager->pageSize = tdbPCacheGetPageSize(pCache);
+  // pPager->dbOrigSize
+  ret = tdbGetFileSize(pPager->fd, pPager->pageSize, &(pPager->dbOrigSize));
 
   *ppPager = pPager;
   return 0;
@@ -91,9 +93,15 @@ int tdbPagerOpenDB(SPager *pPager, SPgno *ppgno, bool toCreate) {
   SPage *pPage;
   int    ret;
 
+  if (pPager->dbOrigSize > 0) {
+    pgno = 1;
+  } else {
+    pgno = 0;
+  }
+
   {
       // TODO: try to search the main DB to get the page number
-    pgno = 0;
+      // pgno = 0;
   }
 
       // if (pgno == 0 && toCreate) {

source/libs/tdb/src/db/tdbUtil.c

@@ -32,3 +32,16 @@ int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique) {
 
   return 0;
 }
+
+int tdbGetFileSize(tdb_fd_t fd, int szPage, SPgno *size) {
+  int     ret;
+  int64_t szBytes;
+
+  ret = tdbOsFileSize(fd, &szBytes);
+  if (ret < 0) {
+    return -1;
+  }
+
+  *size = szBytes / szPage;
+  return 0;
+}
\ No newline at end of file

source/libs/tdb/src/inc/tdbOs.h

@@ -53,6 +53,7 @@ typedef TdFilePtr tdb_fd_t;
 #define tdbOsFSync               taosFsyncFile
 #define tdbOsLSeek               taosLSeekFile
 #define tdbOsRemove              remove
+#define tdbOsFileSize(FD, PSIZE) taosFStatFile(FD, PSIZE, NULL)
 
 /* directory */
 #define tdbOsMkdir taosMkDir
@@ -110,6 +111,7 @@ i64 tdbOsWrite(tdb_fd_t fd, const void *pData, i64 nBytes);
 #define tdbOsFSync  fsync
 #define tdbOsLSeek  lseek
 #define tdbOsRemove remove
+#define tdbOsFileSize(FD, PSIZE)
 
 /* directory */
 #define tdbOsMkdir mkdir

source/libs/tdb/src/inc/tdbUtil.h

@@ -29,6 +29,7 @@ extern "C" {
 #define TDB_ROUND8(x) (((x) + 7) & ~7)
 
 int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique);
+int tdbGetFileSize(tdb_fd_t fd, int szPage, SPgno *size);
 
 #define TDB_REALLOC(PTR, SIZE)                                                               \
   ({                                                                                         \