mirror of
https://github.com/fankes/komari-agent.git
synced 2025-12-13 13:11:14 +08:00
feat: #28 支持使用环境变量 / 配置文件来传入 agent 参数
This commit is contained in:
@@ -5,10 +5,12 @@ import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"github.com/komari-monitor/komari-agent/cmd/flags"
|
||||
pkg_flags "github.com/komari-monitor/komari-agent/cmd/flags"
|
||||
monitoring "github.com/komari-monitor/komari-agent/monitoring/unit"
|
||||
)
|
||||
|
||||
var flags = pkg_flags.GlobalConfig
|
||||
|
||||
func GenerateReport() []byte {
|
||||
message := ""
|
||||
data := map[string]interface{}{}
|
||||
@@ -92,7 +94,7 @@ func GenerateReport() []byte {
|
||||
// 成功获取详细信息
|
||||
gpuData := make([]map[string]interface{}, len(gpuInfo))
|
||||
totalGPUUsage := 0.0
|
||||
|
||||
|
||||
for i, info := range gpuInfo {
|
||||
gpuData[i] = map[string]interface{}{
|
||||
"name": info.Name,
|
||||
@@ -103,13 +105,13 @@ func GenerateReport() []byte {
|
||||
}
|
||||
totalGPUUsage += info.Utilization
|
||||
}
|
||||
|
||||
|
||||
avgGPUUsage := totalGPUUsage / float64(len(gpuInfo))
|
||||
|
||||
|
||||
data["gpu"] = map[string]interface{}{
|
||||
"count": len(gpuInfo),
|
||||
"average_usage": avgGPUUsage,
|
||||
"detailed_info": gpuData,
|
||||
"count": len(gpuInfo),
|
||||
"average_usage": avgGPUUsage,
|
||||
"detailed_info": gpuData,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user