ana-ledger/ledger/register.go

53 lines
961 B
Go

package ledger
import (
"slices"
"time"
)
type Register map[string]Balances
func (register Register) PushAll(other Register) {
for date := range other {
if _, ok := register[date]; !ok {
register[date] = make(Balances)
}
register[date].PushAll(other[date])
}
}
func (register Register) Names() []string {
names := map[string]int{}
for _, v := range register {
for name := range v {
names[name] = 1
}
}
result := make([]string, 0, len(register))
for k := range names {
result = append(result, k)
}
slices.Sort(result)
return result
}
func (register Register) Dates() []string {
result := make([]string, 0, len(register))
for k := range register {
result = append(result, k)
}
slices.Sort(result)
return result
}
func (register Register) Times() []time.Time {
dates := register.Dates()
result := make([]time.Time, len(dates))
for i := range dates {
v, _ := dateToTime(dates[i])
result[i] = v
}
return result
}