feat: Added Process struct, GetProcessList API

Author: RawanMostafa08

cmd/main.go

@@ -11,6 +11,9 @@ func main() {
 	log.Printf("%v",cpuInfo)
 	memInfo, err := psutils.GetMemInfo()
 	log.Printf("%v",memInfo)
-	log.Fatalf("err %v",err)
+	log.Printf("err %v",err)
+	procs, err := psutils.GetProcessList()
+	log.Printf("%v\n",procs)
+	log.Printf("err %v",err)
 
 }

pkg/cpuInfo.go

@@ -7,13 +7,13 @@ import (
 )
 
 type Loader interface {
-	Load() (string, error)
+	Load(filePath string) (string, error)
 }
 
 type RealCpuLoader struct{}
 
-func (l *RealCpuLoader) Load() (string, error) {
-	return loadFile("/proc/cpuinfo")
+func (l *RealCpuLoader) Load(filePath string) (string, error) {
+	return loadFile(filePath)
 }
 
 type CpuInfo struct {
@@ -72,7 +72,7 @@ func GetCpuInfo() (cpuInfo CpuInfo, err error) {
 
 func getCpuInfo(loader Loader) (cpuInfo CpuInfo, err error) {
 
-	cpuData, err := loader.Load()
+	cpuData, err := loader.Load("/proc/cpuinfo")
 	if err != nil {
 		return
 	}

pkg/cpuInfo_test.go

@@ -177,7 +177,7 @@ cpu cores	: 6
 
 type spyCpuLoader struct{}
 
-func (l *spyCpuLoader) Load() (string, error) {
+func (l *spyCpuLoader) Load(string) (string, error) {
 	return cpuinfo, nil
 }
 func TestGetCpuInfo(t *testing.T) {

pkg/memInfo.go

@@ -13,8 +13,8 @@ type MemInfo struct {
 
 type RealMemLoader struct{}
 
-func (l *RealMemLoader) Load() (string, error) {
-	return loadFile("/proc/meminfo")
+func (l *RealMemLoader) Load(filePath string) (string, error) {
+	return loadFile(filePath)
 }
 
 func extractValue(line string) (valueNoKB string) {
@@ -56,7 +56,7 @@ func GetMemInfo() (memInfo MemInfo, err error) {
 
 func getMemInfo(loader Loader) (memInfo MemInfo, err error) {
 
-	memData, err := loader.Load()
+	memData, err := loader.Load("/proc/meminfo")
 	if err != nil {
 		return
 	}

pkg/memInfo_test.go

@@ -18,7 +18,7 @@ Inactive:        6296272 kB
 
 type spyMemLoader struct{}
 
-func (l *spyMemLoader) Load() (string, error) {
+func (l *spyMemLoader) Load(string) (string, error) {
 	return memInfo, nil
 }
 func TestGetMemInfo(t *testing.T) {

pkg/processInfo.go

@@ -0,0 +1,76 @@
+package psutils
+
+import (
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+)
+
+type Process struct {
+	PID         int
+	ProcessName string
+}
+
+type RealProcLoader struct{}
+
+func (l *RealProcLoader) Load(filePath string) (string, error) {
+	return loadFile(filePath)
+}
+
+func setProcInfo(data string) (proc Process, err error) {
+	lines := strings.SplitN(data, "\n", -1)
+
+	for _, line := range lines {
+		if strings.HasPrefix(line, "Name") {
+			parts := strings.Split(line, ":")
+			value := strings.TrimSpace(parts[1])
+			proc.ProcessName = value
+
+		} else if strings.HasPrefix(line, "Pid") {
+
+			parts := strings.Split(line, ":")
+			value := strings.TrimSpace(parts[1])
+			proc.PID, err = strconv.Atoi(value)
+		}
+	}
+	return
+}
+
+func GetProcessList() (procs []Process, err error) {
+	var _ Loader = (*RealProcLoader)(nil)
+	return getProcessList(&RealProcLoader{})
+}
+
+func getProcessList(loader Loader) (procs []Process, err error) {
+	procs = make([]Process, 0)
+	dir, err := os.Open("/proc")
+	if err != nil {
+		return
+	}
+	defer dir.Close()
+
+	entries, err := dir.Readdirnames(0)
+	if err != nil {
+		return
+	}
+
+	for _, entry := range entries {
+
+		_, err = strconv.Atoi(entry)
+		if err != nil {
+			continue
+		}
+
+		statusFile := filepath.Join("/proc", entry, "status")
+		var data string
+		data, err = loader.Load(statusFile)
+		if err != nil {
+			return
+		}
+		var process Process
+		process, err = setProcInfo(data)
+		procs = append(procs, process)
+	}
+	return
+}