Merge pull request #41 from gateway-fm/feat/collateral-amount

feat: add get collateral amount method

perpsv3.go

@@ -144,6 +144,9 @@ type IPerpsv3 interface {
 	// GetAccountOwner is used to get accounts owner address for given account ID
 	GetAccountOwner(accountId *big.Int) (string, error)
 
+	// GetCollateralAmount is used to get accounts collateral amount for given market ID
+	GetCollateralAmount(accountId *big.Int, marketId *big.Int) (*big.Int, error)
+
 	// GetRequiredMaintenanceMargin is used to get required maintenance margin for given account ID
 	GetRequiredMaintenanceMargin(accountId *big.Int) (*big.Int, error)
 
@@ -313,6 +316,10 @@ func (p *Perpsv3) GetAccountOwner(accountId *big.Int) (string, error) {
 	return p.service.GetAccountOwner(accountId)
 }
 
+func (p *Perpsv3) GetCollateralAmount(accountId *big.Int, marketId *big.Int) (*big.Int, error) {
+	return p.service.GetCollateralAmount(accountId, marketId)
+}
+
 func (p *Perpsv3) GetRequiredMaintenanceMargin(accountId *big.Int) (*big.Int, error) {
 	return p.service.GetRequiredMaintenanceMargin(accountId)
 }

services/accounts.go

@@ -163,6 +163,16 @@ func (s *Service) GetAccountOwner(accountId *big.Int) (string, error) {
 	return owner.Hex(), nil
 }
 
+func (s *Service) GetCollateralAmount(accountId *big.Int, marketId *big.Int) (*big.Int, error) {
+	amount, err := s.perpsMarket.GetCollateralAmount(nil, accountId, marketId)
+	if err != nil {
+		logger.Log().WithField("layer", "Service-GetCollateralAmount").Errorf("get colleteral amount error: %v", err.Error())
+		return nil, errors.GetReadContractErr(err, "perps market", "GetCollateralAmount")
+	}
+
+	return amount, nil
+}
+
 // formatAccounts is used to get accounts from the contract using event filter function for 'AccountCreated' event
 // and given filter options
 func (s *Service) formatAccounts(opts *bind.FilterOpts) ([]*models.Account, error) {

services/service.go

@@ -91,6 +91,9 @@ type IService interface {
 	// GetAccountOwner is used to get accounts owner address for given account ID
 	GetAccountOwner(accountId *big.Int) (string, error)
 
+	// GetCollateralAmount is used to get accounts collateral amount for given market ID
+	GetCollateralAmount(accountId *big.Int, marketId *big.Int) (*big.Int, error)
+
 	// GetRequiredMaintenanceMargin is used to get required maintenance margin for given account ID
 	GetRequiredMaintenanceMargin(accountId *big.Int) (*big.Int, error)