repositories

type DowntimeRepo struct {
	// contains filtered or unexported fields
}

type DowntimeRepository interface {
	Save(downtime *models.Downtime) error
	// FindDowntimesInsideInterval returns all models.Downtime that started or ended inside interval
	// defined with arguments from and to
	FindDowntimesInsideInterval(nodeID string, from time.Time, to time.Time) ([]models.Downtime, error)
}

type FeeRepository interface {
	RecordNewFee(nodeID string, newFee int64) error
	GetAllFees() (*[]models.Fee, error)
}

type MetricsRepository interface {
	FindByID(ID string) (*models.Metrics, error)
	Save(metrics *models.Metrics) error
	SaveAndCheckIfFirstEntry(metrics *models.Metrics) (bool, error)
	GetAll() (*[]models.Metrics, error)
	GetLatestBlockMetrics() (*models.LatestBlockMetrics, error)
}

type NodeRepository interface {
	FindByID(ID string) (*models.Node, error)
	Save(node *models.Node) error
	GetAll() (*[]models.Node, error)
	GetActiveNodes(selection string) *[]models.Node
	GetPenalizedNodes() (*[]models.Node, error)
	GetAllActiveNodes() *[]models.Node
	IsNodeActive(ID string) bool
	RemoveNodeFromActive(ID string) error
	AddNodeToActive(ID string) error
	UpdateNodeUsed(node models.Node)
	IncreaseNodeCooldown(ID string) (*models.Node, error)
	ResetNodeCooldown(ID string) (*models.Node, error)
	IsNodeOnCooldown(ID string) (bool, error)
}

type PayoutRepository interface {
	Save(payment *models.Payout) error
	GetAll() (*[]models.Payout, error)
	FindLatestPayout() (*models.Payout, error)
}

type PingRepository interface {
	FindByNodeID(nodeId string) (*models.Ping, error)
	Save(ping *models.Ping) error
	GetAll() (*[]models.Ping, error)
	ResetAllPings() error
	CalculateDowntime(nodeId string, pingTime time.Time) (time.Time, time.Duration, error)
}

type RecordRepository interface {
	Save(record *models.Record) error
	// FindSuccessfulRecordsInsideInterval returns all models.Record that happened inside interval
	// defined with arguments from and to
	FindSuccessfulRecordsInsideInterval(nodeID string, from time.Time, to time.Time) ([]models.Record, error)
	CountSuccessfulRequests() (int, error)
	CountFailedRequests() (int, error)
}

type Repos struct {
	NodeRepo     NodeRepository
	PingRepo     PingRepository
	MetricsRepo  MetricsRepository
	RecordRepo   RecordRepository
	DowntimeRepo DowntimeRepository
	PayoutRepo   PayoutRepository
	FeeRepo      FeeRepository
}