update

locker/source_locker.go

@@ -1,21 +1,54 @@
 package locker
 
-import "sync"
+import (
+	"fmt"
+)
 
-var locks = make(map[string]sync.Locker)
+// 资源锁
+type SourceLocker struct {
+	m     RWLocker
+	locks map[string]Locker
+}
+
+func NewSourceLocker() *SourceLocker {
+	return &SourceLocker{
+		m:     NewRWLocker(),
+		locks: make(map[string]Locker),
+	}
+}
+
+func (s *SourceLocker) Lock(key string) {
+	s.m.RLock()
+	l, ok := s.locks[key]
+
+	if ok {
+		s.m.RUnlock()
 
-func Lock(name string) {
-	if l, ok := locks[name]; ok {
 		l.Lock()
-	}
+		fmt.Println("加锁")
+	} else {
+		s.m.RUnlock()
 
-	new := &sync.Mutex{}
-	locks[name] = new
-	new.Lock()
+		s.m.Lock()
+		new := NewLocker()
+		s.locks[key] = new
+		s.m.Unlock()
+
+		new.Lock()
+		fmt.Println("初始加锁")
+	}
 }
 
-func Unlock(name string) {
-	if l, ok := locks[name]; ok {
+func (s *SourceLocker) Unlock(key string) {
+	s.m.Lock()
+	if l, ok := s.locks[key]; ok {
 		l.Unlock()
+		// delete(s.locks, key)
+		fmt.Println("解锁")
 	}
+	s.m.Unlock()
+}
+
+func (s *SourceLocker) TryLock(key string) bool {
+	return false
 }