feat(lock): init lock

Author: jaronnie

lock/filelock/filelock.go

@@ -0,0 +1,31 @@
+package filelock
+
+import (
+	"os"
+	"syscall"
+)
+
+type FileLock struct {
+	f *os.File
+}
+
+func New(fp string) (*FileLock, error) {
+	f, err := os.Open(fp)
+	if err != nil {
+		return nil, err
+	}
+	return &FileLock{f: f}, nil
+}
+
+func (l *FileLock) Lock() error {
+	err := syscall.Flock(int(l.f.Fd()), syscall.LOCK_EX|syscall.LOCK_NB)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (l *FileLock) Unlock() error {
+	defer l.f.Close()
+	return syscall.Flock(int(l.f.Fd()), syscall.LOCK_UN)
+}

lock/filelock/filelock_test.go

@@ -0,0 +1,30 @@
+package filelock
+
+import (
+	"os"
+	"sync"
+	"testing"
+
+	"github.com/spf13/cast"
+)
+
+func TestFileLock(t *testing.T) {
+	wg := sync.WaitGroup{}
+	for i := 0; i < 1000; i++ {
+		wg.Add(1)
+		go func(num int) {
+			defer wg.Done()
+			flock, _ := New("testdata/config")
+			err := flock.Lock()
+			if err != nil {
+				t.Error(err)
+			}
+			defer flock.Unlock()
+
+			f, _ := os.OpenFile("testdata/config", os.O_APPEND|os.O_WRONLY, 0666)
+			defer f.Close()
+			f.WriteString("test" + cast.ToString(num) + "\n")
+		}(i)
+	}
+	wg.Wait()
+}