From b1cca65a05285854e7ec4f327ca90b04988124ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Wed, 20 Sep 2023 13:49:54 +0800
Subject: [PATCH] Add memory package

---
 common/memory/memory.go        | 16 ++++++++++++++++
 common/memory/memory_darwin.go | 18 ++++++++++++++++++
 common/memory/memory_stub.go   |  9 +++++++++
 3 files changed, 43 insertions(+)
 create mode 100644 common/memory/memory.go
 create mode 100644 common/memory/memory_darwin.go
 create mode 100644 common/memory/memory_stub.go

diff --git a/common/memory/memory.go b/common/memory/memory.go
new file mode 100644
index 0000000..63b8d35
--- /dev/null
+++ b/common/memory/memory.go
@@ -0,0 +1,16 @@
+package memory
+
+import "runtime"
+
+func Total() uint64 {
+	if nativeAvailable {
+		return usageNative()
+	}
+	return Inuse()
+}
+
+func Inuse() uint64 {
+	var memStats runtime.MemStats
+	runtime.ReadMemStats(&memStats)
+	return memStats.StackInuse + memStats.HeapInuse + memStats.HeapIdle - memStats.HeapReleased
+}
diff --git a/common/memory/memory_darwin.go b/common/memory/memory_darwin.go
new file mode 100644
index 0000000..4fe2e50
--- /dev/null
+++ b/common/memory/memory_darwin.go
@@ -0,0 +1,18 @@
+package memory
+
+// #include <mach/mach.h>
+import "C"
+import "unsafe"
+
+const nativeAvailable = true
+
+func usageNative() uint64 {
+	var memoryUsageInByte uint64
+	var vmInfo C.task_vm_info_data_t
+	var count C.mach_msg_type_number_t = C.TASK_VM_INFO_COUNT
+	var kernelReturn C.kern_return_t = C.task_info(C.vm_map_t(C.mach_task_self_), C.TASK_VM_INFO, (*C.integer_t)(unsafe.Pointer(&vmInfo)), &count)
+	if kernelReturn == C.KERN_SUCCESS {
+		memoryUsageInByte = uint64(vmInfo.phys_footprint)
+	}
+	return memoryUsageInByte
+}
diff --git a/common/memory/memory_stub.go b/common/memory/memory_stub.go
new file mode 100644
index 0000000..3781f58
--- /dev/null
+++ b/common/memory/memory_stub.go
@@ -0,0 +1,9 @@
+//go:build (darwin && !cgo) || !darwin
+
+package memory
+
+const nativeAvailable = false
+
+func usageNative() uint64 {
+	return 0
+}