From 2aa6d7e38482a76211f265c6edbc31d52ba7e387 Mon Sep 17 00:00:00 2001
From: Hongze Cheng <hzcheng@taosdata.com>
Date: Wed, 23 Mar 2022 10:30:52 +0000
Subject: [PATCH] start the query

---
 source/libs/tdb/src/db/tdbDb.c   |  5 +++++
 source/libs/tdb/src/inc/tdbDb.h  |  1 +
 source/libs/tdb/test/tdbTest.cpp | 35 ++++++++++++++++++++++++--------
 3 files changed, 33 insertions(+), 8 deletions(-)

diff --git a/source/libs/tdb/src/db/tdbDb.c b/source/libs/tdb/src/db/tdbDb.c
index 00f38a19bb..27cbaeaa01 100644
--- a/source/libs/tdb/src/db/tdbDb.c
+++ b/source/libs/tdb/src/db/tdbDb.c
@@ -85,5 +85,10 @@ int tdbDbInsert(STDb *pDb, const void *pKey, int keyLen, const void *pVal, int v
     return -1;
   }
 
+  return 0;
+}
+
+int tdbDbGet(STDb *pDb, const void *pKey, int kLen, void *pVal, int *vLen) {
+  // TODO
   return 0;
 }
\ No newline at end of file
diff --git a/source/libs/tdb/src/inc/tdbDb.h b/source/libs/tdb/src/inc/tdbDb.h
index 06ea74a83e..fdf4c4aca5 100644
--- a/source/libs/tdb/src/inc/tdbDb.h
+++ b/source/libs/tdb/src/inc/tdbDb.h
@@ -26,6 +26,7 @@ int tdbDbOpen(const char *fname, int keyLen, int valLen, FKeyComparator keyCmprF
 int tdbDbClose(STDb *pDb);
 int tdbDbDrop(STDb *pDb);
 int tdbDbInsert(STDb *pDb, const void *pKey, int keyLen, const void *pVal, int valLen);
+int tdbDbGet(STDb *pDb, const void *pKey, int kLen, void *pVal, int *vLen);
 
 #ifdef __cplusplus
 }
diff --git a/source/libs/tdb/test/tdbTest.cpp b/source/libs/tdb/test/tdbTest.cpp
index 63b53b1732..c5a2787aaf 100644
--- a/source/libs/tdb/test/tdbTest.cpp
+++ b/source/libs/tdb/test/tdbTest.cpp
@@ -6,6 +6,7 @@ TEST(tdb_test, simple_test) {
   int    ret;
   STEnv *pEnv;
   STDb  *pDb;
+  int    nData = 1000;
 
   // Open Env
   ret = tdbEnvOpen("tdb", 1024, 256, &pEnv);
@@ -15,15 +16,33 @@ TEST(tdb_test, simple_test) {
   ret = tdbDbOpen("db.db", TDB_VARIANT_LEN, TDB_VARIANT_LEN, NULL, pEnv, &pDb);
   GTEST_ASSERT_EQ(ret, 0);
 
-  {  // Insert some data
-    char key[64];
-    char val[64];
+  {
+    char  key[64];
+    char  val[64];
+    void *pVal;
+    int   vLen;
 
-    for (int i = 1; i <= 1000; i++) {
-      sprintf(key, "key%d", i);
-      sprintf(val, "value%d", i);
-      ret = tdbDbInsert(pDb, key, strlen(key), val, strlen(val));
-      GTEST_ASSERT_EQ(ret, 0);
+    {  // Insert some data
+
+      for (int i = 1; i <= nData; i++) {
+        sprintf(key, "key%d", i);
+        sprintf(val, "value%d", i);
+        ret = tdbDbInsert(pDb, key, strlen(key), val, strlen(val));
+        GTEST_ASSERT_EQ(ret, 0);
+      }
+    }
+
+    {  // Query the data
+      for (int i = 1; i <= nData; i++) {
+        sprintf(key, "key%d", i);
+        sprintf(val, "value%d", i);
+
+        ret = tdbDbGet(pDb, key, strlen(key), pVal, &vLen);
+        GTEST_ASSERT_EQ(ret, 0);
+
+        GTEST_ASSERT_EQ(vLen, strlen(val));
+        GTEST_ASSERT_EQ(memcmp(val, pVal, vLen), 0);
+      }
     }
   }