|
|
/*
|
|
|
*
|
|
|
@author: sre
|
|
|
@date: 2023/11/19 0019
|
|
|
@desc: todo
|
|
|
*
|
|
|
*/
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
// getMemoryInfo 返回总内存 内存使用率
|
|
|
func getMemoryInfo(ip, port string) (float64, float64, error) {
|
|
|
var memTotal, memAvailable float64
|
|
|
var memFree, memCached, memBuffrs float64
|
|
|
data, err := getMetrics(ip, port)
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
// 解析metrics数据
|
|
|
lines := strings.Split(data, "\n")
|
|
|
for _, line := range lines {
|
|
|
//memTotal 两个版本 node_memory_MemTotal node_memory_MemTotal_bytes
|
|
|
if strings.HasPrefix(line, "node_memory_MemTotal_bytes ") || strings.HasPrefix(line, "node_memory_MemTotal ") {
|
|
|
fields := strings.Fields(line)
|
|
|
if len(fields) == 2 {
|
|
|
memTotal, err = extractMetricValue(fields[1])
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
|
|
//memAvailable node_memory_MemAvailable node_memory_MemAvailable_bytes
|
|
|
if strings.HasPrefix(line, "node_memory_MemAvailable_bytes ") || strings.HasPrefix(line, "node_memory_MemAvailable ") {
|
|
|
fields := strings.Fields(line)
|
|
|
if len(fields) == 2 {
|
|
|
memAvailable, err = extractMetricValue(fields[1])
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
}
|
|
|
} else {
|
|
|
//下面是另外一个版本的情况 没有可用内存数据 需要用原始定义来算
|
|
|
fields := strings.Fields(line)
|
|
|
switch {
|
|
|
case strings.HasPrefix(line, "node_memory_MemFree_bytes ") || strings.HasPrefix(line, "node_memory_MemFree "):
|
|
|
if len(fields) == 2 {
|
|
|
memFree, err = extractMetricValue(fields[1])
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
}
|
|
|
case strings.HasPrefix(line, "node_memory_Buffers_bytes ") || strings.HasPrefix(line, "node_memory_Buffers "):
|
|
|
if len(fields) == 2 {
|
|
|
memCached, err = extractMetricValue(fields[1])
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
}
|
|
|
case strings.HasPrefix(line, "node_memory_Cached_bytes ") || strings.HasPrefix(line, "node_memory_Cached "):
|
|
|
if len(fields) == 2 {
|
|
|
memBuffrs, err = extractMetricValue(fields[1])
|
|
|
if err != nil {
|
|
|
return 0, 0, err
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
//第二种情况下的输出
|
|
|
if (memFree + memCached + memBuffrs) > 0 {
|
|
|
used := memTotal - memFree - memCached - memBuffrs
|
|
|
return memTotal, used / memTotal, nil
|
|
|
}
|
|
|
|
|
|
//第一种情况
|
|
|
return memTotal, (memTotal - memAvailable) / memTotal, nil
|
|
|
}
|
|
|
|
|
|
//// 解析Node Exporter的响应内容，提取内存使用率指标
|
|
|
//func parseMemoryUsage(response string) float64 {
|
|
|
// // 在响应中找到相关的指标行，例如 "node_memory_MemTotal: 167772160000 149510252888 149510252888"
|
|
|
// lines := strings.Split(response, "\n")
|
|
|
// for _, line := range lines {
|
|
|
// if strings.Contains(line, "node_memory_MemTotal") {
|
|
|
// fields := strings.Fields(line)
|
|
|
// if len(fields) >= 4 {
|
|
|
// // 从指标行中提取内存使用率指标，并将其转换为百分比形式
|
|
|
// memoryUsage := strings.TrimSuffix(fields[2], " ") + "%"
|
|
|
// return getPercentage(memoryUsage)
|
|
|
// }
|
|
|
// }
|
|
|
// }
|
|
|
//
|
|
|
// // 如果找不到相关的指标行，则返回0%
|
|
|
// return 0.0
|
|
|
//}
|
|
|
//
|
|
|
//// 从字符串中提取百分比值，并将其转换为浮点数形式
|
|
|
//func getPercentage(percentage string) float64 {
|
|
|
// // 去掉百分号，并将其转换为浮点数形式
|
|
|
// percentage = strings.ReplaceAll(percentage, "%", "")
|
|
|
// return parseFloat(percentage) * (100.0 / 100) // 将字符串转换为浮点数，并乘以100转换为百分比形式
|
|
|
//}
|
