mirror of
https://github.com/arnaucube/hermez-node.git
synced 2026-02-07 03:16:45 +01:00
Connect price updater to historydb
This commit is contained in:
Arnau B
@@ -1,13 +1,13 @@
|
||||
package priceupdater
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"sync"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/dghubble/sling"
|
||||
"github.com/hermeznetwork/hermez-node/db/historydb"
|
||||
"github.com/hermeznetwork/hermez-node/log"
|
||||
)
|
||||
|
||||
const (
|
||||
@@ -15,116 +15,59 @@ const (
|
||||
defaultIdleConnTimeout = 10
|
||||
)
|
||||
|
||||
var (
|
||||
// ErrSymbolDoesNotExistInDatabase is used when trying to get a token that is not in the DB
|
||||
ErrSymbolDoesNotExistInDatabase = errors.New("symbol does not exist in database")
|
||||
)
|
||||
|
||||
// ConfigPriceUpdater contains the configuration set by the coordinator
|
||||
type ConfigPriceUpdater struct {
|
||||
RecommendedFee uint64 // in dollars
|
||||
RecommendedCreateAccountFee uint64 // in dollars
|
||||
TokensList []string
|
||||
APIURL string
|
||||
}
|
||||
|
||||
// TokenInfo contains the updated value for the token
|
||||
type TokenInfo struct {
|
||||
Symbol string
|
||||
Value float64
|
||||
LastUpdated time.Time
|
||||
}
|
||||
|
||||
// PriceUpdater definition
|
||||
type PriceUpdater struct {
|
||||
db map[string]TokenInfo
|
||||
config ConfigPriceUpdater
|
||||
mu sync.RWMutex
|
||||
db *historydb.HistoryDB
|
||||
apiURL string
|
||||
tokenSymbols []string
|
||||
}
|
||||
|
||||
// NewPriceUpdater is the constructor for the updater
|
||||
func NewPriceUpdater(config ConfigPriceUpdater) PriceUpdater {
|
||||
func NewPriceUpdater(apiURL string, db *historydb.HistoryDB) PriceUpdater {
|
||||
tokenSymbols := []string{}
|
||||
return PriceUpdater{
|
||||
db: make(map[string]TokenInfo),
|
||||
config: config,
|
||||
db: db,
|
||||
apiURL: apiURL,
|
||||
tokenSymbols: tokenSymbols,
|
||||
}
|
||||
}
|
||||
|
||||
// UpdatePrices is triggered by the Coordinator, and internally will update the token prices in the db
|
||||
func (p *PriceUpdater) UpdatePrices() error {
|
||||
func (p *PriceUpdater) UpdatePrices() {
|
||||
tr := &http.Transport{
|
||||
MaxIdleConns: defaultMaxIdleConns,
|
||||
IdleConnTimeout: defaultIdleConnTimeout * time.Second,
|
||||
DisableCompression: true,
|
||||
}
|
||||
httpClient := &http.Client{Transport: tr}
|
||||
client := sling.New().Base(p.config.APIURL).Client(httpClient)
|
||||
client := sling.New().Base(p.apiURL).Client(httpClient)
|
||||
|
||||
state := [10]float64{}
|
||||
|
||||
for _, tokenSymbol := range p.config.TokensList {
|
||||
for _, tokenSymbol := range p.tokenSymbols {
|
||||
resp, err := client.New().Get("ticker/t" + tokenSymbol + "USD").ReceiveSuccess(&state)
|
||||
errString := tokenSymbol + " not updated, error: "
|
||||
if err != nil {
|
||||
return err
|
||||
log.Error(errString + err.Error())
|
||||
continue
|
||||
}
|
||||
// if resp.StatusCode != 200 {
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return fmt.Errorf("Unexpected response status code: %v", resp.StatusCode)
|
||||
log.Error(errString + "response is not 200, is " + strconv.Itoa(resp.StatusCode))
|
||||
continue
|
||||
}
|
||||
|
||||
tinfo := TokenInfo{
|
||||
Symbol: tokenSymbol,
|
||||
Value: state[6],
|
||||
LastUpdated: time.Now(),
|
||||
err = p.db.UpdateTokenValue(tokenSymbol, state[6])
|
||||
if err != nil {
|
||||
log.Error(errString + err.Error())
|
||||
}
|
||||
|
||||
p.UpdateTokenInfo(tinfo)
|
||||
}
|
||||
}
|
||||
|
||||
// UpdateTokenList get the registered token symbols from HistoryDB
|
||||
func (p *PriceUpdater) UpdateTokenList() error {
|
||||
tokenSymbols, err := p.db.GetTokenSymbols()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
p.tokenSymbols = tokenSymbols
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateConfig allows to update the price-updater configuration
|
||||
func (p *PriceUpdater) UpdateConfig(config ConfigPriceUpdater) {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
|
||||
p.config = config
|
||||
}
|
||||
|
||||
// Get one token information
|
||||
func (p *PriceUpdater) Get(tokenSymbol string) (TokenInfo, error) {
|
||||
var info TokenInfo
|
||||
|
||||
// Check if symbol exists in database
|
||||
p.mu.RLock()
|
||||
defer p.mu.RUnlock()
|
||||
|
||||
if info, ok := p.db[tokenSymbol]; ok {
|
||||
return info, nil
|
||||
}
|
||||
|
||||
return info, ErrSymbolDoesNotExistInDatabase
|
||||
}
|
||||
|
||||
// GetPrices gets all the prices contained in the db
|
||||
func (p *PriceUpdater) GetPrices() map[string]TokenInfo {
|
||||
var info = make(map[string]TokenInfo)
|
||||
|
||||
p.mu.RLock()
|
||||
defer p.mu.RUnlock()
|
||||
|
||||
for key, value := range p.db {
|
||||
info[key] = value
|
||||
}
|
||||
|
||||
return info
|
||||
}
|
||||
|
||||
// UpdateTokenInfo updates one token info
|
||||
func (p *PriceUpdater) UpdateTokenInfo(tokenInfo TokenInfo) {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
|
||||
p.db[tokenInfo.Symbol] = tokenInfo
|
||||
}
|
||||
|
||||
@@ -1,38 +1,60 @@
|
||||
package priceupdater
|
||||
|
||||
import (
|
||||
"math/big"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
ethCommon "github.com/ethereum/go-ethereum/common"
|
||||
"github.com/hermeznetwork/hermez-node/common"
|
||||
dbUtils "github.com/hermeznetwork/hermez-node/db"
|
||||
"github.com/hermeznetwork/hermez-node/db/historydb"
|
||||
"github.com/hermeznetwork/hermez-node/test"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestCon(t *testing.T) {
|
||||
config := ConfigPriceUpdater{
|
||||
|
||||
RecommendedFee: 1,
|
||||
RecommendedCreateAccountFee: 1,
|
||||
TokensList: []string{"ETH", "NEC"},
|
||||
APIURL: "https://api-pub.bitfinex.com/v2/",
|
||||
func TestPriceUpdater(t *testing.T) {
|
||||
// Init DB
|
||||
pass := os.Getenv("POSTGRES_PASS")
|
||||
db, err := dbUtils.InitSQLDB(5432, "localhost", "hermez", pass, "hermez")
|
||||
assert.NoError(t, err)
|
||||
historyDB := historydb.NewHistoryDB(db)
|
||||
// Clean DB
|
||||
assert.NoError(t, historyDB.Reorg(-1))
|
||||
// Populate DB
|
||||
// Gen blocks and add them to DB
|
||||
blocks := test.GenBlocks(1, 2)
|
||||
assert.NoError(t, historyDB.AddBlocks(blocks))
|
||||
// Gen tokens and add them to DB
|
||||
tokens := []common.Token{}
|
||||
tokens = append(tokens, common.Token{
|
||||
TokenID: 0,
|
||||
EthBlockNum: blocks[0].EthBlockNum,
|
||||
EthAddr: ethCommon.BigToAddress(big.NewInt(1)),
|
||||
Name: "Ether",
|
||||
Symbol: "ETH",
|
||||
Decimals: 18,
|
||||
})
|
||||
tokens = append(tokens, common.Token{
|
||||
TokenID: 1,
|
||||
EthBlockNum: blocks[0].EthBlockNum,
|
||||
EthAddr: ethCommon.BigToAddress(big.NewInt(2)),
|
||||
Name: "DAI",
|
||||
Symbol: "DAI",
|
||||
Decimals: 18,
|
||||
})
|
||||
assert.NoError(t, historyDB.AddTokens(tokens))
|
||||
// Init price updater
|
||||
pu := NewPriceUpdater("https://api-pub.bitfinex.com/v2/", historyDB)
|
||||
// Update token list
|
||||
assert.NoError(t, pu.UpdateTokenList())
|
||||
// Update prices
|
||||
pu.UpdatePrices()
|
||||
// Check that prices have been updated
|
||||
fetchedTokens, err := historyDB.GetTokens()
|
||||
assert.NoError(t, err)
|
||||
for _, token := range fetchedTokens {
|
||||
assert.NotNil(t, token.USD)
|
||||
assert.NotNil(t, token.USDUpdate)
|
||||
}
|
||||
|
||||
pud := NewPriceUpdater(config)
|
||||
|
||||
err := pud.UpdatePrices()
|
||||
assert.Equal(t, err, nil)
|
||||
|
||||
info, _ := pud.Get("ETH")
|
||||
assert.NotZero(t, info.Value)
|
||||
|
||||
info2, _ := pud.Get("NEC")
|
||||
assert.NotZero(t, info2.Value)
|
||||
|
||||
info3, err := pud.Get("INVENTED")
|
||||
if assert.Error(t, err) {
|
||||
assert.Equal(t, ErrSymbolDoesNotExistInDatabase, err)
|
||||
}
|
||||
assert.Equal(t, info3.Value, float64(0))
|
||||
|
||||
prices := pud.GetPrices()
|
||||
assert.Equal(t, prices["ETH"], info)
|
||||
assert.Equal(t, prices["NEC"], info2)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user