优化代码

internal/utils/mem/system.go

@@ -0,0 +1,48 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package memutils
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
+	"github.com/shirou/gopsutil/v3/mem"
+)
+
+var systemTotalMemory = -1
+var systemMemoryBytes uint64
+
+func init() {
+	if !teaconst.IsMain {
+		return
+	}
+
+	_ = SystemMemoryGB()
+}
+
+// SystemMemoryGB 系统内存GB数量
+// 必须保证不小于1
+func SystemMemoryGB() int {
+	if systemTotalMemory > 0 {
+		return systemTotalMemory
+	}
+
+	stat, err := mem.VirtualMemory()
+	if err != nil {
+		return 1
+	}
+
+	systemMemoryBytes = stat.Total
+
+	systemTotalMemory = int(stat.Total / (1 << 30))
+	if systemTotalMemory <= 0 {
+		systemTotalMemory = 1
+	}
+
+	setMaxMemory(systemTotalMemory)
+
+	return systemTotalMemory
+}
+
+// SystemMemoryBytes 系统内存总字节数
+func SystemMemoryBytes() uint64 {
+	return systemMemoryBytes
+}

internal/utils/mem/system_1.19.go

@@ -0,0 +1,18 @@
+// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build go1.19
+
+package memutils
+
+import (
+	"runtime/debug"
+)
+
+// 设置软内存最大值
+func setMaxMemory(memoryGB int) {
+	if memoryGB <= 0 {
+		memoryGB = 1
+	}
+
+	var maxMemoryBytes = (int64(memoryGB) << 30) * 80 / 100 // 默认 80%
+	debug.SetMemoryLimit(maxMemoryBytes)
+}

internal/utils/mem/system_before_1.19.go

@@ -0,0 +1,9 @@
+// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build !go1.19
+
+package memutils
+
+// 设置软内存最大值
+func setMaxMemory(memoryGB int) {
+
+}