balances can accept delta pushes for FAST

This commit is contained in:
Bel LaPointe
2023-10-25 06:41:04 -06:00
parent bece66b5b3
commit de4a8e9723
2 changed files with 17 additions and 4 deletions

View File

@@ -34,11 +34,10 @@ func main() {
sort.Slice(deltas, func(i, j int) bool { sort.Slice(deltas, func(i, j int) bool {
return deltas[i].Debug() < deltas[j].Debug() return deltas[i].Debug() < deltas[j].Debug()
}) })
deltasSoFar := make(ledger.Deltas, 0) balancesSoFar := make(ledger.Balances)
for i := range deltas { for i := range deltas {
deltasSoFar = append(deltasSoFar, deltas[i]) balancesSoFar.Push(deltas[i])
bal, _ := deltasSoFar.Like(ledger.LikeName("^" + deltas[i].Name + "$")).Balances() fmt.Printf("%s (%+v)\n", deltas[i].Debug(), balancesSoFar[deltas[i].Name].Debug())
fmt.Printf("%s (%+v)\n", deltas[i].Debug(), bal[deltas[i].Name].Debug())
} }
case "bal": case "bal":
balances, err := deltas.Balances() balances, err := deltas.Balances()

View File

@@ -9,6 +9,20 @@ type Balances map[string]Balance
type Balance map[Currency]float64 type Balance map[Currency]float64
func (balances Balances) Push(d Delta) {
if _, ok := balances[d.Name]; !ok {
balances[d.Name] = make(Balance)
}
balances[d.Name].Push(d)
}
func (balance Balance) Push(d Delta) {
if _, ok := balance[d.Currency]; !ok {
balance[d.Currency] = 0
}
balance[d.Currency] += d.Value
}
func (balances Balances) Debug() string { func (balances Balances) Debug() string {
result := []string{} result := []string{}
for k, v := range balances { for k, v := range balances {