status-im/status-desktop
2
0
Fork 0
mirror of https://github.com/status-im/status-desktop.git synced 2025-02-17 00:56:39 +00:00
Code Issues Projects Releases Wiki Activity

cache requests

Browse Source
This commit is contained in:
Iuri Matias 2020-06-25 13:31:53 -04:00
parent 59b8e525b0
commit 0fc9105f0a
1 changed files with 28 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/status/wallet/balance_manager.nim
View File

@ -1,30 +1,37 @@
import tables, strformat, strutils, stint, httpclient, json import tables, strformat, strutils, stint, httpclient, json, times
import ../libstatus/wallet as status_wallet import ../libstatus/wallet as status_wallet
import ../libstatus/tokens as status_tokens import ../libstatus/tokens as status_tokens
import account import account
type PriceTime* = ref object
price: string
timestamp: DateTime
type BalanceManager* = ref object type BalanceManager* = ref object
pricePairs: Table[string, string] pricePairs: Table[string, PriceTime]
tokenBalances: Table[string, string] tokenBalances: Table[string, PriceTime]
proc isCached(self: Table[string, PriceTime], cacheKey: string): bool =
self.hasKey(cacheKey) and ((self[cacheKey].timestamp + initDuration(minutes = 1)) >= now())
proc cacheValue(self: var Table[string, PriceTime], cacheKey: string, value: string) =
self[cacheKey] = PriceTime(price: value, timestamp: now())
proc newBalanceManager*(): BalanceManager = proc newBalanceManager*(): BalanceManager =
result = BalanceManager() result = BalanceManager()
result.pricePairs = initTable[string, string]() result.pricePairs = initTable[string, PriceTime]()
result.tokenBalances = initTable[string, string]() result.tokenBalances = initTable[string, PriceTime]()
var balanceManager = newBalanceManager() var balanceManager = newBalanceManager()
proc getPrice(crypto: string, fiat: string): string = proc getPrice(crypto: string, fiat: string): string =
try: try:
if balanceManager.pricePairs.hasKey(fiat):
return balanceManager.pricePairs[fiat]
let url: string = fmt"https://min-api.cryptocompare.com/data/price?fsym={crypto}&tsyms={fiat}" let url: string = fmt"https://min-api.cryptocompare.com/data/price?fsym={crypto}&tsyms={fiat}"
let client = newHttpClient() let client = newHttpClient()
client.headers = newHttpHeaders({ "Content-Type": "application/json" }) client.headers = newHttpHeaders({ "Content-Type": "application/json" })
let response = client.request(url) let response = client.request(url)
result = $parseJson(response.body)[fiat.toUpper] result = $parseJson(response.body)[fiat.toUpper]
# balanceManager.pricePairs[fiat] = result
except Exception as e: except Exception as e:
echo "error getting price" echo "error getting price"
echo e.msg echo e.msg
@ -33,20 +40,28 @@ proc getPrice(crypto: string, fiat: string): string =
proc getEthBalance(address: string): string = proc getEthBalance(address: string): string =
var balance = status_wallet.getBalance(address) var balance = status_wallet.getBalance(address)
result = status_wallet.hex2Eth(balance) result = status_wallet.hex2Eth(balance)
# balanceManager.tokenBalances["ETH"] = result
proc getBalance*(symbol: string, accountAddress: string, tokenAddress: string): string = proc getBalance*(symbol: string, accountAddress: string, tokenAddress: string): string =
if balanceManager.tokenBalances.hasKey(symbol): let cacheKey = fmt"{symbol}-{accountAddress}-{tokenAddress}"
return balanceManager.tokenBalances[symbol] if balanceManager.tokenBalances.isCached(cacheKey):
return balanceManager.tokenBalances[cacheKey].price
if symbol == "ETH": if symbol == "ETH":
return getEthBalance(accountAddress) let ethBalance = getEthBalance(accountAddress)
balanceManager.tokenBalances.cacheValue(cacheKey, ethBalance)
return ethBalance
result = $status_tokens.getTokenBalance(tokenAddress, accountAddress) result = $status_tokens.getTokenBalance(tokenAddress, accountAddress)
# balanceManager.tokenBalances[symbol] = result balanceManager.tokenBalances.cacheValue(cacheKey, result)
proc getFiatValue*(crypto_balance: string, crypto_symbol: string, fiat_symbol: string): float = proc getFiatValue*(crypto_balance: string, crypto_symbol: string, fiat_symbol: string): float =
if crypto_balance == "0.0": return 0.0 if crypto_balance == "0.0": return 0.0
let cacheKey = fmt"{crypto_symbol}-{fiat_symbol}"
if balanceManager.pricePairs.isCached(cacheKey):
return parseFloat(crypto_balance) * parseFloat(balanceManager.pricePairs[cacheKey].price)
var fiat_crypto_price = getPrice(crypto_symbol, fiat_symbol) var fiat_crypto_price = getPrice(crypto_symbol, fiat_symbol)
balanceManager.pricePairs.cacheValue(cacheKey, fiat_crypto_price)
parseFloat(crypto_balance) * parseFloat(fiat_crypto_price) parseFloat(crypto_balance) * parseFloat(fiat_crypto_price)
proc updateBalance*(asset: Asset, currency: string) = proc updateBalance*(asset: Asset, currency: string) =