autorestic/internal/lock.go
Boris Bera 8de8d0070e
chore: move lockfile code to internal module
In a future commit, I'll need the lockfile code to access the config
file. This solves an import cycle.
2024-11-10 13:37:59 -05:00

89 lines
1.7 KiB
Go

package internal
import (
"os"
"path"
"sync"
"github.com/cupcakearmy/autorestic/internal/colors"
"github.com/cupcakearmy/autorestic/internal/flags"
"github.com/spf13/viper"
)
var lock *viper.Viper
var file string
var lockOnce sync.Once
const (
RUNNING = "running"
)
// getLockfilePath returns the path to the lockfile. If flags.LOCKFILE_PATH is
// set, its value is used, otherwise the path is generated relative to the
// config file.
func getLockfilePath() string {
if flags.LOCKFILE_PATH != "" {
return flags.LOCKFILE_PATH
} else {
p := viper.ConfigFileUsed()
if p == "" {
colors.Error.Println("cannot lock before reading config location")
os.Exit(1)
}
return path.Join(path.Dir(p), ".autorestic.lock.yml")
}
}
func getLock() *viper.Viper {
if lock == nil {
lockOnce.Do(func() {
lock = viper.New()
lock.SetDefault("running", false)
file = getLockfilePath()
if !flags.CRON_LEAN {
colors.Faint.Println("Using lock:\t", file)
}
lock.SetConfigFile(file)
lock.SetConfigType("yml")
lock.ReadInConfig()
})
}
return lock
}
func setLockValue(key string, value interface{}) (*viper.Viper, error) {
lock := getLock()
if key == RUNNING {
value := value.(bool)
if value && lock.GetBool(key) {
colors.Error.Println("an instance is already running. exiting")
os.Exit(1)
}
}
lock.Set(key, value)
if err := lock.WriteConfigAs(file); err != nil {
return nil, err
}
return lock, nil
}
func GetCron(location string) int64 {
return getLock().GetInt64("cron." + location)
}
func SetCron(location string, value int64) {
setLockValue("cron."+location, value)
}
func Lock() error {
_, err := setLockValue(RUNNING, true)
return err
}
func Unlock() error {
_, err := setLockValue(RUNNING, false)
return err
}