58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package ledger
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"slices"
|
|
"time"
|
|
)
|
|
|
|
func RegisterWithContributionPrediction(reg map[string]Balances, windowAsPercentOfTotalDuration float64) (map[string]Balances, error) {
|
|
times, err := registerTimesInUnix(reg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
windowDuration := time.Second * time.Duration(float64(slices.Max(times)-slices.Min(times))*windowAsPercentOfTotalDuration)
|
|
log.Println(windowDuration)
|
|
|
|
lastReal := time.Unix(slices.Max(times), 0)
|
|
lastPredicted := lastReal
|
|
for lastPredicted.Before(lastReal.Add(windowDuration)) {
|
|
lastPredicted = lastPredicted.Add(time.Hour * 24 * time.Duration((45 - lastPredicted.Day())))
|
|
log.Println(lastPredicted)
|
|
}
|
|
|
|
result := make(map[string]Balances)
|
|
return result, io.EOF
|
|
}
|
|
|
|
func RegisterWithCompoundingInterestPrediction(reg map[string]Balances, windowAsPercentOfTotalDuration float64, name string, rate float64) (map[string]Balances, error) {
|
|
result := make(map[string]Balances)
|
|
return result, io.EOF
|
|
}
|
|
|
|
func registerTimesInUnix(reg map[string]Balances) ([]int64, error) {
|
|
result := make([]int64, 0, len(reg))
|
|
for k := range reg {
|
|
v, err := registerTime(k)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result = append(result, v.Unix())
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func registerTime(s string) (time.Time, error) {
|
|
for _, layout := range []string{
|
|
"2006-01-02",
|
|
"2006-01",
|
|
} {
|
|
if t, err := time.ParseInLocation(layout, s, time.Local); err == nil {
|
|
return t, err
|
|
}
|
|
}
|
|
return time.Time{}, fmt.Errorf("no layout matching %q", s)
|
|
}
|