tarndt / sema Goto Github PK

type redirectJobSynchronizer struct {
    mapLock sync.RWMutex
    // Map of per-originID locks
    idToLock map[int64]sema.TimeoutCountingSema
}

func (this *redirectJobSynchronizer) Do(id int64, fnc func() error) error {
    this.mapLock.RLock()
    idSema, ok := this.idToLock[id]
    this.mapLock.RUnlock()
    if !ok {
        this.mapLock.Lock()
        if idSema, ok = this.idToLock[id]; !ok {
            idSema = sema.NewTimeoutSema(1, 10*time.Second)
            this.idTpLock[id] = idSema
        }
        this.mapLock.Unlock()
    }

    //Aquire lock
    if idSema.P() {
        defer idSema.V()
        return fnc()
    }
    return errors.New("timeout")