Skip to content

Commit a4b8893

Browse files
committed
refund/deduct naming
1 parent 28bc016 commit a4b8893

File tree

4 files changed

+26
-26
lines changed

4 files changed

+26
-26
lines changed

p2p/pkg/depositmanager/deposit.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ type Store interface {
2323
GetBalance(bidder common.Address, provider common.Address) (*big.Int, error)
2424
SetBalance(bidder common.Address, provider common.Address, balance *big.Int) error
2525
DeleteBalance(bidder common.Address, provider common.Address) error
26-
IncreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
27-
DecreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
26+
RefundBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
27+
DeductBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
2828
}
2929

3030
type DepositManager struct {
@@ -275,10 +275,10 @@ func (dm *DepositManager) getDefaultBalance(
275275
return balance, nil
276276
}
277277

278-
func (dm *DepositManager) IncreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error {
279-
return dm.store.IncreaseBalanceIfExists(bidder, provider, amount)
278+
func (dm *DepositManager) RefundBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error {
279+
return dm.store.RefundBalanceIfExists(bidder, provider, amount)
280280
}
281281

282-
func (dm *DepositManager) DecreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error {
283-
return dm.store.DecreaseBalanceIfExists(bidder, provider, amount)
282+
func (dm *DepositManager) DeductBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error {
283+
return dm.store.DeductBalanceIfExists(bidder, provider, amount)
284284
}

p2p/pkg/depositmanager/store/store.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ func (s *Store) DeleteBalance(bidder common.Address, provider common.Address) er
6464
return s.st.Delete(balanceKey(bidder, provider))
6565
}
6666

67-
func (s *Store) IncreaseBalanceIfExists(
67+
func (s *Store) RefundBalanceIfExists(
6868
bidder common.Address,
6969
provider common.Address,
7070
amount *big.Int,
@@ -84,7 +84,7 @@ func (s *Store) IncreaseBalanceIfExists(
8484
return s.st.Put(balanceKey(bidder, provider), newAmount.Bytes())
8585
}
8686

87-
func (s *Store) DecreaseBalanceIfExists(
87+
func (s *Store) DeductBalanceIfExists(
8888
bidder common.Address,
8989
provider common.Address,
9090
amount *big.Int,

p2p/pkg/depositmanager/store/store_test.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -70,15 +70,15 @@ func TestStore_GetBalance_NoBalance(t *testing.T) {
7070
}
7171
}
7272

73-
func TestStore_IncreaseBalanceIfExists(t *testing.T) {
73+
func TestStore_RefundBalanceIfExists(t *testing.T) {
7474
st := inmem.New()
7575
s := store.New(st)
7676

7777
bidder := common.HexToAddress("0x123")
7878
provider := common.HexToAddress("0x456")
7979
amount := big.NewInt(20)
8080

81-
err := s.IncreaseBalanceIfExists(bidder, provider, amount)
81+
err := s.RefundBalanceIfExists(bidder, provider, amount)
8282
if err == nil {
8383
t.Fatal("expected error, got nil")
8484
}
@@ -92,7 +92,7 @@ func TestStore_IncreaseBalanceIfExists(t *testing.T) {
9292
}
9393

9494
increaseAmount := big.NewInt(5)
95-
err = s.IncreaseBalanceIfExists(bidder, provider, increaseAmount)
95+
err = s.RefundBalanceIfExists(bidder, provider, increaseAmount)
9696
if err != nil {
9797
t.Fatal(err)
9898
}
@@ -107,15 +107,15 @@ func TestStore_IncreaseBalanceIfExists(t *testing.T) {
107107
}
108108
}
109109

110-
func TestStore_DecreaseBalanceIfExists(t *testing.T) {
110+
func TestStore_DeductBalanceIfExists(t *testing.T) {
111111
st := inmem.New()
112112
s := store.New(st)
113113

114114
bidder := common.HexToAddress("0x123")
115115
provider := common.HexToAddress("0x456")
116116
initialBalance := big.NewInt(20)
117117

118-
err := s.DecreaseBalanceIfExists(bidder, provider, big.NewInt(10))
118+
err := s.DeductBalanceIfExists(bidder, provider, big.NewInt(10))
119119
if err == nil {
120120
t.Fatal("expected error, got nil")
121121
}
@@ -129,7 +129,7 @@ func TestStore_DecreaseBalanceIfExists(t *testing.T) {
129129
}
130130

131131
decreaseAmount := big.NewInt(5)
132-
err = s.DecreaseBalanceIfExists(bidder, provider, decreaseAmount)
132+
err = s.DeductBalanceIfExists(bidder, provider, decreaseAmount)
133133
if err != nil {
134134
t.Fatal(err)
135135
}

p2p/pkg/preconfirmation/tracker/tracker.go

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,8 @@ type Watcher interface {
9898
}
9999

100100
type DepositManager interface {
101-
IncreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
102-
DecreaseBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
101+
RefundBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
102+
DeductBalanceIfExists(bidder common.Address, provider common.Address, amount *big.Int) error
103103
}
104104

105105
func NewTracker(
@@ -457,9 +457,9 @@ func (t *Tracker) statusUpdater(
457457
} else {
458458
status = store.CommitmentStatusStored
459459
if t.depositMgr != nil {
460-
// Try to decrease cached balance now that commitment was successfully stored
461-
if err := t.tryDecreaseCachedBalance(task.commitment); err != nil {
462-
t.logger.Warn("failed to decrease cached balance. Bidder is likely withdrawing", "error", err)
460+
// Try to deduct cached balance now that commitment was successfully stored
461+
if err := t.tryDeductCachedBalance(task.commitment); err != nil {
462+
t.logger.Warn("failed to deduct cached balance. Bidder is likely withdrawing", "error", err)
463463
}
464464
}
465465
}
@@ -556,7 +556,7 @@ func (t *Tracker) openCommitments(
556556
)
557557
if t.depositMgr != nil {
558558
// This node isn't the winner, so try to refund relevant cached balance
559-
if err := t.tryIncreaseCachedBalance(commitment); err != nil {
559+
if err := t.tryRefundCachedBalance(commitment); err != nil {
560560
t.logger.Warn("failed to refund cached balance. Bidder is likely withdrawing", "error", err)
561561
}
562562
}
@@ -800,34 +800,34 @@ func (t *Tracker) generateBidderProof(
800800
}
801801
}
802802

803-
func (t *Tracker) tryIncreaseCachedBalance(
803+
func (t *Tracker) tryRefundCachedBalance(
804804
commitment *store.Commitment,
805805
) error {
806806
if commitment.BidderAddress == nil || commitment.ProviderAddress == nil || commitment.BidAmount == nil {
807807
return fmt.Errorf("nil commitment fields")
808808
}
809-
if err := t.depositMgr.IncreaseBalanceIfExists(
809+
if err := t.depositMgr.RefundBalanceIfExists(
810810
*commitment.BidderAddress,
811811
common.BytesToAddress(commitment.ProviderAddress),
812812
commitment.BidAmount,
813813
); err != nil {
814-
return fmt.Errorf("failed to increase balance: %w", err)
814+
return fmt.Errorf("failed to refund balance: %w", err)
815815
}
816-
t.logger.Info("increased cached balance from commitment",
816+
t.logger.Info("refunded cached balance from commitment",
817817
"bidder", commitment.BidderAddress,
818818
"provider", common.BytesToAddress(commitment.ProviderAddress),
819819
"amount", commitment.BidAmount,
820820
)
821821
return nil
822822
}
823823

824-
func (t *Tracker) tryDecreaseCachedBalance(
824+
func (t *Tracker) tryDeductCachedBalance(
825825
commitment *store.Commitment,
826826
) error {
827827
if commitment.BidderAddress == nil || commitment.ProviderAddress == nil || commitment.BidAmount == nil {
828828
return fmt.Errorf("nil commitment fields")
829829
}
830-
if err := t.depositMgr.DecreaseBalanceIfExists(
830+
if err := t.depositMgr.DeductBalanceIfExists(
831831
*commitment.BidderAddress,
832832
common.BytesToAddress(commitment.ProviderAddress),
833833
commitment.BidAmount,

0 commit comments

Comments
 (0)