111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
// Copyright 2019 clair authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package pgsql
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
searchLock = `SELECT until FROM Lock WHERE name = $1`
|
|
updateLock = `UPDATE Lock SET until = $3 WHERE name = $1 AND owner = $2`
|
|
removeLock = `DELETE FROM Lock WHERE name = $1 AND owner = $2`
|
|
removeLockExpired = `DELETE FROM LOCK WHERE until < CURRENT_TIMESTAMP`
|
|
|
|
soiLock = `WITH new_lock AS (
|
|
INSERT INTO lock (name, owner, until)
|
|
VALUES ( $1, $2, $3)
|
|
WHERE NOT EXISTS (SELECT id FROM lock WHERE name = $1)
|
|
RETURNING owner, until
|
|
)
|
|
SELECT * FROM new_lock
|
|
UNION
|
|
SELECT owner, until FROM lock WHERE name = $1`
|
|
)
|
|
|
|
var (
|
|
errLockNotFound = errors.New("lock is not in database")
|
|
)
|
|
|
|
func (tx *pgSession) AcquireLock(lockName, whoami string, desiredDuration time.Duration) (bool, time.Time, error) {
|
|
if lockName == "" || whoami == "" || desiredDuration == 0 {
|
|
panic("invalid lock parameters")
|
|
}
|
|
|
|
if err := tx.pruneLocks(); err != nil {
|
|
return false, time.Time{}, err
|
|
}
|
|
|
|
var (
|
|
desiredLockedUntil = time.Now().Add(desiredDuration)
|
|
|
|
lockedUntil time.Time
|
|
lockOwner string
|
|
)
|
|
|
|
defer observeQueryTime("Lock", "soiLock", time.Now())
|
|
err := tx.QueryRow(soiLock, lockName, whoami, desiredLockedUntil).Scan(&lockOwner, &lockedUntil)
|
|
return lockOwner == whoami, lockedUntil, err
|
|
}
|
|
|
|
func (tx *pgSession) ExtendLock(lockName, whoami string, desiredDuration time.Duration) (bool, time.Time, error) {
|
|
if lockName == "" || whoami == "" || desiredDuration == 0 {
|
|
panic("invalid lock parameters")
|
|
}
|
|
|
|
desiredLockedUntil := time.Now().Add(desiredDuration)
|
|
|
|
defer observeQueryTime("Lock", "update", time.Now())
|
|
result, err := tx.Exec(updateLock, lockName, whoami, desiredLockedUntil)
|
|
if err != nil {
|
|
return false, time.Time{}, handleError("updateLock", err)
|
|
}
|
|
|
|
if numRows, err := result.RowsAffected(); err == nil {
|
|
// This is the only happy path.
|
|
return numRows > 0, desiredLockedUntil, nil
|
|
}
|
|
|
|
return false, time.Time{}, handleError("updateLock", err)
|
|
}
|
|
|
|
func (tx *pgSession) ReleaseLock(name, owner string) error {
|
|
if name == "" || owner == "" {
|
|
panic("invalid lock parameters")
|
|
}
|
|
|
|
defer observeQueryTime("Unlock", "all", time.Now())
|
|
_, err := tx.Exec(removeLock, name, owner)
|
|
return err
|
|
}
|
|
|
|
// pruneLocks removes every expired locks from the database
|
|
func (tx *pgSession) pruneLocks() error {
|
|
defer observeQueryTime("pruneLocks", "all", time.Now())
|
|
|
|
if r, err := tx.Exec(removeLockExpired); err != nil {
|
|
return handleError("removeLockExpired", err)
|
|
} else if affected, err := r.RowsAffected(); err != nil {
|
|
return handleError("removeLockExpired", err)
|
|
} else {
|
|
log.Debugf("Pruned %d Locks", affected)
|
|
}
|
|
|
|
return nil
|
|
}
|