types: fix net asset field adding

c9s · Dec 16, 2024 · e06c510 · e06c510
1 parent adf5a90
commit e06c510
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/types/balance.go b/pkg/types/balance.go
@@ -47,6 +47,14 @@ func (b Balance) Add(b2 Balance) Balance {
 	newB.Locked = b.Locked.Add(b2.Locked)
 	newB.Borrowed = b.Borrowed.Add(b2.Borrowed)
 	newB.Interest = b.Interest.Add(b2.Interest)
+
+	if !b.NetAsset.IsZero() && !b2.NetAsset.IsZero() {
+		newB.NetAsset = b.NetAsset.Add(b2.NetAsset)
+	} else {
+		// do not use this field, reset it
+		newB.NetAsset = fixedpoint.Zero
+	}
+
 	return newB
 }