milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

enhance: print total memory when milvus starts (#29351) (#29354) 

issue: https://github.com/milvus-io/milvus/issues/29349
pr: https://github.com/milvus-io/milvus/pull/29351

---------

Signed-off-by: longjiquan <jiquan.long@zilliz.com>
pull/29565/head
Jiquan Long 2023-12-21 14:20:43 +08:00 committed by GitHub
parent e19efd42a5
commit 739828a9ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/milvus/run.go
View File

@ -11,6 +11,7 @@ import (
"go.uber.org/zap"
"github.com/milvus-io/milvus/internal/log"
"github.com/milvus-io/milvus/internal/util/hardware"
"github.com/milvus-io/milvus/internal/util/metricsinfo"
"github.com/milvus-io/milvus/internal/util/paramtable"
"github.com/milvus-io/milvus/internal/util/typeutil"
@ -44,6 +45,7 @@ func (c *run) execute(args []string, flags *flag.FlagSet) {
c.printBanner(flags.Output())
c.injectVariablesToEnv()
c.printHardwareInfo(flags.Output())
lock, err := createPidFile(flags.Output(), filename, runtimeDir)
if err != nil {
panic(err)
@ -67,6 +69,14 @@ func (c *run) printBanner(w io.Writer) {
fmt.Fprintln(w)
}
func (c *run) printHardwareInfo(w io.Writer) {
totalMem := hardware.GetMemoryCount()
usedMem := hardware.GetUsedMemoryCount()
fmt.Fprintf(w, "TotalMem: %d\n", totalMem)
fmt.Fprintf(w, "UsedMem: %d\n", usedMem)
fmt.Fprintln(w)
}
func (c *run) injectVariablesToEnv() {
// inject in need