Compute balances when serializing core.Volume.

formancehq · Jun 9, 2022 · 50987cf · 50987cf
1 parent 2746192
commit 50987cf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/core/volumes.go b/pkg/core/volumes.go
@@ -10,6 +10,17 @@ type Volume struct {
 	Output int64 `json:"output"`
 }
 
+func (v Volume) MarshalJSON() ([]byte, error) {
+	type volume Volume
+	return json.Marshal(struct {
+		volume
+		Balance int64 `json:"balance"`
+	}{
+		volume:  volume(v),
+		Balance: v.Input - v.Output,
+	})
+}
+
 func (v Volume) Balance() int64 {
 	return v.Input - v.Output
 }