update test

Author: litongxin

src/storage/disk_table.cc

@@ -533,7 +533,6 @@ void DiskTable::GcTTL() {
             PDLOG(WARNING, "Manual Compaction failed");
         }
     }
-    PDLOG(ERROR, "Manual Compaction Finished");
 }
 
 void DiskTable::GcTTLOrHead() {}
@@ -1196,28 +1195,11 @@ bool DiskTable::DeleteIndex(const std::string& idx_name) {
 
 uint64_t DiskTable::GetRecordIdxCnt() {
     auto inner_indexs = table_index_.GetAllInnerIndex();
-    rocksdb::ReadOptions ro = rocksdb::ReadOptions();
-    const rocksdb::Snapshot* snapshot = db_->GetSnapshot();
-
-    // for (size_t i = 0; i < inner_indexs->size(); i++) {
-    //     bool is_valid = false;
-    //     for (const auto& index_def : inner_indexs->at(i)->GetIndex()) {
-    //         if (index_def && index_def->IsReady()) {
-    //             rocksdb::Iterator* it = db_->NewIterator(ro, cf_hs_[i + 1]);
-
-    //             for (; it->Valid(); it->Next()) {
-    //                 uint32_t cur_ts_idx = UINT32_MAX;
-    //                 std::string cur_pk;
-    //                 uint64_t cur_ts;
-
-    //                 ParseKeyAndTs(has_ts_idx, it->key(), cur_pk, cur_ts, cur_ts_idx);
-    //             }
-    //         }
-    //     }
-        
-    // }
-    // TODO(litongxin)
-    return 0;
+    uint64_t count = 0;
+    for (const auto& inner_index : *inner_indexs) {
+        count += idx_cnt_vec_[inner_index->GetIndex().front()->GetId()]->load(std::memory_order_relaxed);
+    }
+    return count;
 }
 
 bool DiskTable::GetRecordIdxCnt(uint32_t idx, uint64_t** stat, uint32_t* size) {

src/storage/disk_table.h

@@ -21,7 +21,6 @@
 #include <memory>
 #include <string>
 #include <vector>
-#include <bitset>
 #include "base/endianconv.h"
 #include "base/slice.h"
 #include "boost/lexical_cast.hpp"
@@ -42,7 +41,6 @@
 #include "storage/table.h"
 #include "base/glog_wapper.h"  // NOLINT
 
-
 namespace openmldb {
 namespace storage {
 
@@ -180,7 +178,6 @@ class AbsoluteTTLCompactionFilter : public rocksdb::CompactionFilter {
 
     bool Filter(int /*level*/, const rocksdb::Slice& key, const rocksdb::Slice& /*existing_value*/,
                 std::string* /*new_value*/, bool* /*value_changed*/) const override {
-        PDLOG(ERROR, "using compaction filter");
         if (key.size() < TS_LEN) {
             return false;
         }
@@ -555,7 +552,6 @@ class DiskTable : public Table {
     KeyTSComparator cmp_;
     std::atomic<uint64_t> offset_;
     std::string table_path_;
-    std::atomic<uint64_t> pk_cnt_;
     std::vector<std::shared_ptr<std::atomic<uint64_t>>> idx_cnt_vec_;
     std::vector<std::shared_ptr<std::atomic<uint64_t>>> pk_cnt_vec_;
     std::vector<BloomFilter> bloom_filter_vec_;