enhance: reduce mmap_rss after chunkcache warmup (milvus-io#35974)

related pr: milvus-io#35965 Signed-off-by: cqy123456 <[email protected]>
chyezh · Sep 11, 2024 · 44742c8 · 44742c8
1 parent 6e28b27
commit 44742c8
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/internal/core/src/segcore/SegmentSealedImpl.cpp b/internal/core/src/segcore/SegmentSealedImpl.cpp
@@ -152,9 +152,13 @@ SegmentSealedImpl::WarmupChunkCache(const FieldId field_id, bool mmap_enabled) {
     auto field_info = it->second;
 
     auto cc = storage::MmapManager::GetInstance().GetChunkCache();
+    bool mmap_rss_not_need = true;
     for (const auto& data_path : field_info.insert_files) {
-        auto column =
-            cc->Read(data_path, mmap_descriptor_, field_meta, mmap_enabled);
+        auto column = cc->Read(data_path,
+                               mmap_descriptor_,
+                               field_meta,
+                               mmap_enabled,
+                               mmap_rss_not_need);
     }
 }
 

diff --git a/internal/core/src/storage/ChunkCache.cpp b/internal/core/src/storage/ChunkCache.cpp
@@ -24,7 +24,8 @@ std::shared_ptr<ColumnBase>
 ChunkCache::Read(const std::string& filepath,
                  const MmapChunkDescriptorPtr& descriptor,
                  const FieldMeta& field_meta,
-                 bool mmap_enabled) {
+                 bool mmap_enabled,
+                 bool mmap_rss_not_need) {
     // use rlock to get future
     {
         std::shared_lock lck(mutex_);
@@ -64,6 +65,22 @@ ChunkCache::Read(const std::string& filepath,
         field_data = DownloadAndDecodeRemoteFile(cm_.get(), filepath);
         column = Mmap(
             field_data->GetFieldData(), descriptor, field_meta, mmap_enabled);
+        if (mmap_enabled && mmap_rss_not_need) {
+            auto ok = madvise(reinterpret_cast<void*>(
+                                  const_cast<char*>(column->MmappedData())),
+                              column->ByteSize(),
+                              ReadAheadPolicy_Map["dontneed"]);
+            if (ok != 0) {
+                LOG_WARN(
+                    "failed to madvise to the data file {}, addr {}, size {}, "
+                    "err: "
+                    "{}",
+                    filepath,
+                    static_cast<const void*>(column->MmappedData()),
+                    column->ByteSize(),
+                    strerror(errno));
+            }
+        }
         allocate_success = true;
     } catch (const SegcoreError& e) {
         err_code = e.get_error_code();
@@ -113,7 +130,7 @@ ChunkCache::Prefetch(const std::string& filepath) {
         LOG_WARN(
             "failed to madvise to the data file {}, addr {}, size {}, err: {}",
             filepath,
-            column->MmappedData(),
+            static_cast<const void*>(column->MmappedData()),
             column->ByteSize(),
             strerror(errno));
     }

diff --git a/internal/core/src/storage/ChunkCache.h b/internal/core/src/storage/ChunkCache.h
@@ -48,7 +48,8 @@ class ChunkCache {
     Read(const std::string& filepath,
          const MmapChunkDescriptorPtr& descriptor,
          const FieldMeta& field_meta,
-         bool mmap_enabled);
+         bool mmap_enabled,
+         bool mmap_rss_not_need = false);
 
     void
     Remove(const std::string& filepath);