diff --git a/internal/report.go b/internal/report.go index 700346c..b043c21 100644 --- a/internal/report.go +++ b/internal/report.go @@ -93,7 +93,10 @@ func BuildReport(ctx context.Context, reader RecordReader, writer ReportWriter, store: EphemeralStore{}, }, options) - buys := make(map[string]*FillerQueue) + buys, err := optionals.store.Load() + if err != nil { + return fmt.Errorf("loading state: %w", err) + } var buysCount, sellsCount int64 var lastTimestamp time.Time @@ -115,6 +118,11 @@ func BuildReport(ctx context.Context, reader RecordReader, writer ReportWriter, rec, err := reader.ReadRecord(ctx) if err != nil { if errors.Is(err, io.EOF) { + err = optionals.store.Save(buys) + if err != nil { + return fmt.Errorf("saving state: %w", err) + } + return nil } return err @@ -138,9 +146,10 @@ func BuildReport(ctx context.Context, reader RecordReader, writer ReportWriter, if err != nil { return fmt.Errorf("processing record: %w", err) } - } } + + return nil } // processRecord either adds buys to the queue or consumes buys from the queue when processing a