add lock while running background task

Signed-off-by: Lakshmipathi <lakshmipathi@giis.co.in>
Lakshmipathi · Oct 9, 2023 · 9ceba80 · 9ceba80
1 parent e7fb007
commit 9ceba80
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ebpf-diskalert b/ebpf-diskalert
diff --git a/src/ebpf_loader.go b/src/ebpf_loader.go
@@ -7,6 +7,7 @@ import (
  "encoding/binary"
  "log"
  "os"
+ "sync"
 
  "github.com/cilium/ebpf/link"
  "github.com/cilium/ebpf/perf"
@@ -58,7 +59,7 @@ func ebpf_loader(ci *ConfigData, di *Devinfo) {
  if err != nil {
  logger.Fatalf("ebpf_loader: reader error")
  }
-
+ var m sync.Mutex
  for {
  ev, err := rd.Read()
  if err != nil {
@@ -79,6 +80,9 @@ func ebpf_loader(ci *ConfigData, di *Devinfo) {
  }
 
  logger.Printf("Command:%s User: %d wrote to the mountpoint:%s\n", data.Command, data.Pid, di.MountPoint)
+
+ m.Lock()
  checkDiskUSage(ci, di)
+ m.Unlock()
  }
 }