status-go/services/wallet/currency/service.go

114 lines
2.5 KiB
Go

package currency
import (
"context"
"database/sql"
"time"
"github.com/ethereum/go-ethereum/event"
"github.com/status-im/status-go/services/wallet/market"
"github.com/status-im/status-go/services/wallet/token"
"github.com/status-im/status-go/services/wallet/walletevent"
)
const (
EventCurrencyTickUpdateFormat walletevent.EventType = "wallet-currency-tick-update-format"
currencyFormatUpdateInterval = 1 * time.Hour
)
type Service struct {
currency *Currency
db *DB
tokenManager *token.Manager
walletFeed *event.Feed
cancelFn context.CancelFunc
}
func NewService(db *sql.DB, walletFeed *event.Feed, tokenManager *token.Manager, marketManager *market.Manager) *Service {
return &Service{
currency: NewCurrency(marketManager),
db: NewCurrencyDB(db),
tokenManager: tokenManager,
walletFeed: walletFeed,
}
}
func (s *Service) Start() {
// Update all fiat currency formats in cache
fiatFormats, err := s.getAllFiatCurrencyFormats()
if err == nil {
_ = s.db.UpdateCachedFormats(fiatFormats)
}
ctx, cancel := context.WithCancel(context.Background())
s.cancelFn = cancel
go func() {
ticker := time.NewTicker(currencyFormatUpdateInterval)
defer ticker.Stop()
for {
select {
case <-ctx.Done():
return
case <-ticker.C:
s.walletFeed.Send(walletevent.Event{
Type: EventCurrencyTickUpdateFormat,
})
}
}
}()
}
func (s *Service) Stop() {
if s.cancelFn != nil {
s.cancelFn()
}
}
func (s *Service) GetCachedCurrencyFormats() (FormatPerSymbol, error) {
return s.db.GetCachedFormats()
}
func (s *Service) FetchAllCurrencyFormats() (FormatPerSymbol, error) {
// Only token prices can change, so we fetch those
tokenFormats, err := s.fetchAllTokenCurrencyFormats()
if err != nil {
return nil, err
}
err = s.db.UpdateCachedFormats(tokenFormats)
if err != nil {
return nil, err
}
return s.GetCachedCurrencyFormats()
}
func (s *Service) getAllFiatCurrencyFormats() (FormatPerSymbol, error) {
return GetFiatCurrencyFormats(GetAllFiatCurrencySymbols())
}
func (s *Service) fetchAllTokenCurrencyFormats() (FormatPerSymbol, error) {
tokens, err := s.tokenManager.GetAllTokensAndNativeCurrencies()
if err != nil {
return nil, err
}
tokenPerSymbolMap := make(map[string]bool)
tokenSymbols := make([]string, 0)
for _, t := range tokens {
symbol := t.Symbol
if !tokenPerSymbolMap[symbol] {
tokenPerSymbolMap[symbol] = true
tokenSymbols = append(tokenSymbols, symbol)
}
}
return s.currency.FetchTokenCurrencyFormats(tokenSymbols)
}