milvus/internal/util/segcore/jemalloc_stats.go
congqixia 6452d146af
enhance: move jemalloc_stats from pkg to internal/util/segcore (#46560)
Related to #46133

Move jemalloc_stats.go and its test file from pkg/util/hardware to
internal/util/segcore. This is a more appropriate location because:
- jemalloc_stats depends on milvus_core C++ library via cgo
- The pkg directory should remain independent of internal C++
dependencies
- segcore is the natural home for core memory allocator utilities

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

* **Refactor**
* Improved internal code organization by reorganizing memory statistics
collection infrastructure for better maintainability and modularity. No
impact on end-user functionality or behavior.

<sub>✏️ Tip: You can customize this high-level summary in your review
settings.</sub>

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2025-12-24 19:03:18 +08:00

61 lines
2.2 KiB
Go

// Copyright (C) 2019-2025 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License.
//go:build linux || darwin
package segcore
/*
#cgo pkg-config: milvus_core
#include <stdlib.h>
#include "monitor/jemalloc_stats_c.h"
*/
import "C"
// JemallocStats represents comprehensive jemalloc memory statistics
// All sizes are in bytes
type JemallocStats struct {
// Core memory metrics (from jemalloc)
Allocated uint64 // Total bytes allocated by the application
Active uint64 // Total bytes in active pages (includes fragmentation)
Metadata uint64 // Total bytes dedicated to jemalloc metadata
Resident uint64 // Total bytes in physically resident data pages (RSS)
Mapped uint64 // Total bytes in virtual memory mappings
Retained uint64 // Total bytes in retained virtual memory (could be returned to OS)
// Derived metrics (calculated by C code)
Fragmentation uint64 // Internal fragmentation (active - allocated)
Overhead uint64 // Memory overhead (resident - active)
// Status
Success bool // Whether stats were successfully retrieved
}
// GetJemallocStats retrieves comprehensive jemalloc memory statistics
// Returns JemallocStats with detailed memory information
// On platforms without jemalloc support, all metrics will be 0 and Success will be false
func GetJemallocStats() JemallocStats {
cStats := C.GetJemallocStats()
return JemallocStats{
Allocated: uint64(cStats.allocated),
Active: uint64(cStats.active),
Metadata: uint64(cStats.metadata),
Resident: uint64(cStats.resident),
Mapped: uint64(cStats.mapped),
Retained: uint64(cStats.retained),
Fragmentation: uint64(cStats.fragmentation),
Overhead: uint64(cStats.overhead),
Success: bool(cStats.success),
}
}