Use in-memory to store known delegations.

Author: Jiri Malek

internal/repository/db/account.go

@@ -48,6 +48,12 @@ type AccountRow struct {
 	ScHash   *common.Hash `bson:"-"`
 }
 
+// initAccountsCollection initializes the account collection with
+// indexes and additional parameters needed by the app.
+func (db *MongoDbBridge) initAccountsCollection() {
+	db.log.Debugf("accounts collection initialized")
+}
+
 // Account tries to load an account identified by the address given from
 // the off-chain database.
 func (db *MongoDbBridge) Account(addr *common.Address) (*types.Account, error) {
@@ -91,12 +97,6 @@ func (db *MongoDbBridge) Account(addr *common.Address) (*types.Account, error) {
 	}, nil
 }
 
-// initAccountsCollection initializes the account collection with
-// indexes and additional parameters needed by the app.
-func (db *MongoDbBridge) initAccountsCollection() {
-	db.log.Debugf("accounts collection initialized")
-}
-
 // AddAccount stores an account in the blockchain if not exists.
 func (db *MongoDbBridge) AddAccount(acc *types.Account) error {
 	// do we have account data?

internal/repository/sfc_delegation.go

@@ -46,13 +46,8 @@ func (p *proxy) UpdateDelegationBalance(addr *common.Address, valID *hexutil.Big
 		return err
 	}
 
-	// check if we need to update it
-	if !p.stakedAmounts.Update(addr, valID, val) {
-		return nil
-	}
-
 	// do the update
-	err = p.db.UpdateDelegationBalance(addr, valID, (*hexutil.Big)(val))
+	err = p.updateDelegationBalance(addr, valID, val)
 	if err == nil {
 		return nil
 	}
@@ -65,10 +60,41 @@ func (p *proxy) UpdateDelegationBalance(addr *common.Address, valID *hexutil.Big
 	return err
 }
 
+// updateDelegationBalance performs delegation balance update if needed.
+func (p *proxy) updateDelegationBalance(addr *common.Address, valID *hexutil.Big, amo *big.Int) error {
+	// get the delegation detail
+	dlg, err := p.Delegation(addr, valID)
+	if err != nil {
+		return err
+	}
+
+	// do we need to update?
+	if dlg.AmountStaked.ToInt().Cmp(amo) == 0 {
+		return nil
+	}
+
+	// update the delegation in cache
+	dlg.AmountDelegated = (*hexutil.Big)(amo)
+
+	// perform the update
+	return p.db.UpdateDelegationBalance(addr, valID, dlg.AmountDelegated)
+}
+
 // Delegation returns a detail of delegation for the given address.
 func (p *proxy) Delegation(addr *common.Address, valID *hexutil.Big) (*types.Delegation, error) {
+	// log what we do
 	p.log.Debugf("loading delegation of %s to #%d", addr.String(), valID.ToInt().Uint64())
-	return p.db.Delegation(addr, valID)
+
+	// try cache first
+
+	// pull from DB instead
+	dlg, err := p.db.Delegation(addr, valID)
+	if err != nil {
+		return nil, err
+	}
+
+	// store to cache for future reference
+	return dlg, nil
 }
 
 // DelegationAmountStaked returns the current amount of staked tokens for the given delegation.

internal/repository/stake_map.go

@@ -31,9 +31,13 @@ type Delegation struct {
 	Address         common.Address `json:"address"`
 	ToStakerId      *hexutil.Big   `json:"toStakerID"`
 	ToStakerAddress common.Address `json:"toStakerAddr"`
-	AmountStaked    *hexutil.Big   `json:"amountStaked"`
-	AmountDelegated *hexutil.Big   `json:"amountDelegated"`
 	CreatedTime     hexutil.Uint64 `json:"createdTime"`
+
+	// AmountStaked represents the current staked amount
+	AmountStaked *hexutil.Big `json:"amountStaked"`
+
+	// AmountDelegated is the original amount delegated
+	AmountDelegated *hexutil.Big `json:"amountDelegated"`
 }
 
 // BsonDelegation represents the BSON i/o struct for a delegation.