34f860309f
Reliability fixes: - Fix HTTP response body drainage in xmrig, ttminer, miner - Fix database init race condition (nil before close) - Fix empty minerType bug in P2P StartMinerPayload - Add context timeout to InsertHashratePoint (5s default) Architecture improvements: - Extract HashrateStore interface with DefaultStore/NopStore - Create ServiceContainer for centralized initialization - Extract protocol response handler (ValidateResponse, ParseResponse) - Create generic FileRepository[T] with atomic writes 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
95 lines
3 KiB
Go
95 lines
3 KiB
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 {
|
|
// InsertHashratePoint stores a hashrate measurement.
|
|
// If ctx is nil, a default timeout will be used.
|
|
InsertHashratePoint(ctx context.Context, minerName, minerType string, point HashratePoint, resolution Resolution) error
|
|
|
|
// GetHashrateHistory retrieves hashrate history for a miner within a time range.
|
|
GetHashrateHistory(minerName string, resolution Resolution, since, until time.Time) ([]HashratePoint, error)
|
|
|
|
// GetHashrateStats retrieves aggregated statistics for a specific miner.
|
|
GetHashrateStats(minerName string) (*HashrateStats, error)
|
|
|
|
// GetAllMinerStats retrieves statistics for all miners.
|
|
GetAllMinerStats() ([]HashrateStats, error)
|
|
|
|
// Cleanup removes old data based on retention settings.
|
|
Cleanup(retentionDays int) error
|
|
|
|
// Close closes the store and releases resources.
|
|
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
|
|
}
|