feat: Added MemInfo struct, GetMemInfo API

Author: RawanMostafa08

cmd/main.go

@@ -7,7 +7,10 @@ import (
 )
 
 func main() {
-	cpuinfo, _ := psutils.GetCpuInfo()
-	log.Printf("%v",cpuinfo)
+	cpuInfo, _ := psutils.GetCpuInfo()
+	log.Printf("%v",cpuInfo)
+	memInfo, err := psutils.GetMemInfo()
+	log.Printf("%v",memInfo)
+	log.Fatalf("err %v",err)
 
 }

pkg/cpuInfo.go

@@ -10,10 +10,10 @@ type Loader interface {
 	Load() (string, error)
 }
 
-type RealLoader struct{}
+type RealCpuLoader struct{}
 
-func (l *RealLoader) Load() (string, error) {
-	return loadCpuFile()
+func (l *RealCpuLoader) Load() (string, error) {
+	return loadFile("/proc/cpuinfo")
 }
 
 type CpuInfo struct {
@@ -24,17 +24,17 @@ type CpuInfo struct {
 	CpuMHZ    float32
 }
 
-func loadCpuFile() (cpuFile string, err error) {
-	file, err := os.ReadFile("/proc/cpuinfo")
-	cpuFile = string(file)
+func loadFile(filepath string) (fileData string, err error) {
+	file, err := os.ReadFile(filepath)
+	fileData = string(file)
 
 	return
 }
 
-func setCpuInfo(cpuFile string) (cpuInfo CpuInfo, err error) {
+func setCpuInfo(cpuData string) (cpuInfo CpuInfo, err error) {
 	var processorsNum int
 	var totalFreq float32
-	lines := strings.SplitN(cpuFile, "\n", -1)
+	lines := strings.SplitN(cpuData, "\n", -1)
 	for _, line := range lines {
 		if strings.HasPrefix(line, "processor") {
 			processorsNum++
@@ -66,16 +66,16 @@ func setCpuInfo(cpuFile string) (cpuInfo CpuInfo, err error) {
 }
 
 func GetCpuInfo() (cpuInfo CpuInfo, err error) {
-	var _ Loader = (*RealLoader)(nil)
-	return getCpuInfo(&RealLoader{})
+	var _ Loader = (*RealCpuLoader)(nil)
+	return getCpuInfo(&RealCpuLoader{})
 }
 
 func getCpuInfo(loader Loader) (cpuInfo CpuInfo, err error) {
 
-	cpuFile, err := loader.Load()
+	cpuData, err := loader.Load()
 	if err != nil {
 		return
 	}
-	cpuInfo, err = setCpuInfo(cpuFile)
+	cpuInfo, err = setCpuInfo(cpuData)
 	return
 }

pkg/memInfo.go

@@ -0,0 +1,65 @@
+package psutils
+
+import (
+	"strconv"
+	"strings"
+)
+
+type MemInfo struct {
+	TotalMemoryKB     float64
+	UsedMemoryKB      float64
+	AvailableMemoryKB float64
+}
+
+type RealMemLoader struct{}
+
+func (l *RealMemLoader) Load() (string, error) {
+	return loadFile("/proc/meminfo")
+}
+
+func extractValue(line string) (valueNoKB string) {
+	parts := strings.Split(line, ":")
+	value := strings.Replace(parts[1], "kB", "", 1)
+	valueNoKB = strings.TrimSpace(value)
+	return
+}
+
+func setMemInfo(memData string) (memInfo MemInfo, err error) {
+	lines := strings.SplitN(memData, "\n", -1)
+
+	for _, line := range lines {
+		if strings.HasPrefix(line, "MemTotal") {
+
+			memTotalNoKB := extractValue(line)
+			memInfo.TotalMemoryKB, err = strconv.ParseFloat(memTotalNoKB, 64)
+
+		} else if strings.HasPrefix(line, "MemAvailable") {
+
+			memavailabeNoKB := extractValue(line)
+			memInfo.AvailableMemoryKB, err = strconv.ParseFloat(memavailabeNoKB, 64)
+
+		} else if strings.HasPrefix(line, "MemFree") {
+
+			var free float64
+			MemFreeNoKB := extractValue(line)
+			free, err = strconv.ParseFloat(MemFreeNoKB, 64)
+			memInfo.UsedMemoryKB = memInfo.TotalMemoryKB - free
+		}
+	}
+	return
+}
+
+func GetMemInfo() (memInfo MemInfo, err error) {
+	var _ Loader = (*RealMemLoader)(nil)
+	return getMemInfo(&RealMemLoader{})
+}
+
+func getMemInfo(loader Loader) (memInfo MemInfo, err error) {
+
+	memData, err := loader.Load()
+	if err != nil {
+		return
+	}
+	memInfo, err = setMemInfo(memData)
+	return
+}