Golang使用map保存锁

gistfile1.txt

type Unlocker func()

type MapLocker interface {
	Lock(ctx context.Context, key string) (Unlocker, error)
}

type LMap struct {
  	lock	sync.Map
}

func NewLMap() *LMap {
  return &LMap{
    lock: sync.Map{}
  }
}

type Signal struct {
	c	chan bool
}

func (m *LMap) Lock(ctx context.Context, key string) (Unlocker, error) {
  for {
      l, ok := m.lock.LoadOrStore(&Signal{make(chan bool, 1)})
      if !ok {
        break
      }
      select {
        case: <-l.c {
        }
        case: <-ctx.Done(): {
          return nil, fmt.Errorf("timeout")
        }
      }
  }
  
  return func() {
    if tmp, ok := m.lock.Load(key); ok {
    	m.lock.Delete(key)
    	close(tmp.c)
    }    
  }, nil
}