pool.go

/*
2020 © Postgres.ai
*/

package resources

import (
	"path"
	"sync"
	"time"

	"gitlab.com/postgres-ai/database-lab/v3/pkg/util"
)

// PoolStatus represents a pool status.
type PoolStatus string

const (
	// ActivePool defines an active pool status.
	ActivePool PoolStatus = "active"
	// RefreshingPool defines the status of a pool when data retrieval in progress.
	RefreshingPool PoolStatus = "refreshing"
	// EmptyPool defines the status of an inactive pool.
	EmptyPool PoolStatus = "empty"
)

// Pool describes a storage pool.
type Pool struct {
	Name           string
	Mode           string
	DSA            time.Time
	PoolDirName    string
	MountDir       string
	CloneSubDir    string
	DataSubDir     string
	SocketSubDir   string
	ObserverSubDir string
	mu             sync.RWMutex
	status         PoolStatus
}

// NewPool creates a new Pool.
func NewPool(name string) *Pool {
	return &Pool{Name: name, status: EmptyPool}
}

// IsEmpty checks if Pool is empty.
func (p *Pool) IsEmpty() bool {
	return p.Name == "" && p.Mode == ""
}

// SetDSA sets a dataStateAt value.
func (p *Pool) SetDSA(dsa time.Time) {
	p.DSA = dsa
}

// DataDir returns a path to the data directory of the storage pool.
func (p *Pool) DataDir() string {
	return path.Join(p.MountDir, p.PoolDirName, p.DataSubDir)
}

// SocketDir returns a path to the sockets directory of the storage pool.
func (p *Pool) SocketDir() string {
	return path.Join(p.MountDir, p.PoolDirName, p.SocketSubDir)
}

// ObserverDir returns a path to the observer directory of the storage pool.
func (p *Pool) ObserverDir(port uint) string {
	return path.Join(p.ClonePath(port), p.ObserverSubDir)
}

// ClonesDir returns a path to the clones directory of the storage pool.
func (p *Pool) ClonesDir() string {
	return path.Join(p.MountDir, p.PoolDirName, p.CloneSubDir)
}

// ClonePath returns a path to the initialized clone directory.
func (p *Pool) ClonePath(port uint) string {
	return path.Join(p.MountDir, p.PoolDirName, p.CloneSubDir, util.GetCloneName(port), p.DataSubDir)
}

// SocketCloneDir returns a path to the socket clone directory.
func (p *Pool) SocketCloneDir(name string) string {
	return path.Join(p.SocketDir(), name)
}

// Status gets the pool status.
func (p *Pool) Status() PoolStatus {
	p.mu.RLock()
	defer p.mu.RUnlock()

	return p.status
}

// SetStatus sets a status to the pool.
func (p *Pool) SetStatus(status PoolStatus) {
	p.mu.Lock()
	p.status = status
	p.mu.Unlock()
}