settings.go

package nodedata

import (
	"sync"
	"time"

	entities "github.com/bolt-observer/agent/entities"
	api "github.com/bolt-observer/agent/lightning"
)

// PerNodeSettings struct
type PerNodeSettings struct {
	mutex sync.RWMutex
	data  map[string]Settings
}

// NewPerNodeSettings creates PerNodeSettings
func NewPerNodeSettings() *PerNodeSettings {
	return &PerNodeSettings{
		mutex: sync.RWMutex{},
		data:  make(map[string]Settings),
	}
}

// GetKeys - get all keys
func (s *PerNodeSettings) GetKeys() []string {
	s.mutex.RLock()
	defer s.mutex.RUnlock()
	keys := make([]string, 0, len(s.data))
	for k := range s.data {
		keys = append(keys, k)
	}

	return keys
}

// Get - get settings for one key
func (s *PerNodeSettings) Get(key string) Settings {
	s.mutex.RLock()
	defer s.mutex.RUnlock()
	return s.data[key]
}

// Set - set settings for one key
func (s *PerNodeSettings) Set(key string, value Settings) {
	s.mutex.Lock()
	defer s.mutex.Unlock()
	s.data[key] = value
}

// Delete - delete settings for one key
func (s *PerNodeSettings) Delete(key string) {
	s.mutex.Lock()
	defer s.mutex.Unlock()
	delete(s.data, key)
}

// Settings struct
type Settings struct {
	nodeDataCallback  entities.NodeDataReportCallback
	getAPI            api.NewAPICall
	hash              uint64
	identifier        entities.NodeIdentifier
	lastGraphCheck    time.Time
	lastReport        time.Time
	lastNodeReport    time.Time
	lastCheck         time.Time
	lastSyncedToChain time.Time
	settings          entities.ReportingSettings
}