コード例 #1
0
ファイル: lock.go プロジェクト: shantanoo/atlantis-manager
func (l *DeployLock) Lock() error {
	if l.locked {
		return nil
	}
	// check if we can lock by checking if any thing in the lock file is a prefix to us
	path := helper.GetBaseLockPath("deploy")
	mutex := zookeeper.NewMutex(Zk.Conn, path)
	if err := mutex.Lock(); err != nil {
		return err
	}
	defer mutex.Unlock()
	lockedPaths, err := getLockedPaths(path)
	if err != nil {
		return err
	}
	if allID, ok := lockedPaths[allPath]; ok && allID != "" {
		return LockConflictError(allID)
	}
	for p, conflictID := range lockedPaths {
		if strings.HasPrefix(l.path, p) {
			return LockConflictError(conflictID)
		}
	}
	// if no conflicts, register our lock
	lockedPaths[l.path] = l.id
	if err := setJson(path, lockedPaths); err != nil {
		return err
	}
	l.locked = true
	return nil
}
コード例 #2
0
ファイル: lock.go プロジェクト: shantanoo/atlantis-manager
func (l *RouterPortsLock) Lock() error {
	var path string
	if l.locked {
		return nil
	}
	if l.internal {
		path = helper.GetBaseLockPath("router_ports_internal")
	} else {
		path = helper.GetBaseLockPath("router_ports_external")
	}
	l.mutex = zookeeper.NewMutex(Zk.Conn, path)
	if err := l.mutex.Lock(); err != nil {
		return err
	}
	l.locked = true
	return nil
}
コード例 #3
0
ファイル: lock.go プロジェクト: shantanoo/atlantis-manager
func (l *DeployLock) Unlock() error {
	if !l.locked {
		return nil
	}
	// remove ourselves from the lock file
	path := helper.GetBaseLockPath("deploy")
	mutex := zookeeper.NewMutex(Zk.Conn, path)
	if err := mutex.Lock(); err != nil {
		return err
	}
	defer mutex.Unlock()
	lockedPaths, err := getLockedPaths(path)
	if err != nil {
		return err
	}
	delete(lockedPaths, l.path)
	if err := setJson(path, lockedPaths); err != nil {
		return err
	}
	l.locked = false
	return nil
}