Mining/pkg/database/interface.go

package database

import (
	"context"
	"time"
)

// HashrateStore defines the interface for hashrate data persistence.
// This interface allows for dependency injection and easier testing.
type HashrateStore interface {
	// store.InsertHashratePoint(ctx, "xmrig", "xmrig", HashratePoint{Timestamp: time.Now(), Hashrate: 1234}, ResolutionHigh)
	InsertHashratePoint(ctx context.Context, minerName, minerType string, point HashratePoint, resolution Resolution) error

	// points, err := store.GetHashrateHistory("xmrig", ResolutionHigh, time.Now().Add(-time.Hour), time.Now())
	GetHashrateHistory(minerName string, resolution Resolution, since, until time.Time) ([]HashratePoint, error)

	// stats, err := store.GetHashrateStats("xmrig")
	// if stats != nil { log.Printf("avg: %d H/s", stats.AverageRate) }
	GetHashrateStats(minerName string) (*HashrateStats, error)

	// allStats, err := store.GetAllMinerStats()
	// for _, stats := range allStats { log.Printf("%s avg: %d H/s", stats.MinerName, stats.AverageRate) }
	GetAllMinerStats() ([]HashrateStats, error)

	// store.Cleanup(30) // remove data older than 30 days
	Cleanup(retentionDays int) error

	// store.Close()
	Close() error
}

// defaultStore implements HashrateStore using the global database connection.
// This provides backward compatibility while allowing interface-based usage.
type defaultStore struct{}

// DefaultStore returns a HashrateStore that uses the global database connection.
// This is useful for gradual migration from package-level functions to interface-based usage.
func DefaultStore() HashrateStore {
	return &defaultStore{}
}

func (s *defaultStore) InsertHashratePoint(ctx context.Context, minerName, minerType string, point HashratePoint, resolution Resolution) error {
	return InsertHashratePoint(ctx, minerName, minerType, point, resolution)
}

func (s *defaultStore) GetHashrateHistory(minerName string, resolution Resolution, since, until time.Time) ([]HashratePoint, error) {
	return GetHashrateHistory(minerName, resolution, since, until)
}

func (s *defaultStore) GetHashrateStats(minerName string) (*HashrateStats, error) {
	return GetHashrateStats(minerName)
}

func (s *defaultStore) GetAllMinerStats() ([]HashrateStats, error) {
	return GetAllMinerStats()
}

func (s *defaultStore) Cleanup(retentionDays int) error {
	return Cleanup(retentionDays)
}

func (s *defaultStore) Close() error {
	return Close()
}

// NopStore returns a HashrateStore that does nothing.
// Useful for testing or when database is disabled.
func NopStore() HashrateStore {
	return &nopStore{}
}

type nopStore struct{}

func (s *nopStore) InsertHashratePoint(ctx context.Context, minerName, minerType string, point HashratePoint, resolution Resolution) error {
	return nil
}

func (s *nopStore) GetHashrateHistory(minerName string, resolution Resolution, since, until time.Time) ([]HashratePoint, error) {
	return nil, nil
}

func (s *nopStore) GetHashrateStats(minerName string) (*HashrateStats, error) {
	return nil, nil
}

func (s *nopStore) GetAllMinerStats() ([]HashrateStats, error) {
	return nil, nil
}

func (s *nopStore) Cleanup(retentionDays int) error {
	return nil
}

func (s *nopStore) Close() error {
	return nil
}