use stream.ReadAll where it makes sense

Signed-off-by: Achille Roussel <[email protected]>
dispatchrun · May 30, 2023 · 8905c14 · 8905c14
1 parent adac0ed
commit 8905c14
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/internal/timemachine/registry.go b/internal/timemachine/registry.go
@@ -427,11 +427,8 @@ func (reg *Registry) LookupLogManifest(ctx context.Context, processID format.UUI
  segments := reg.ListLogSegments(ctx, processID)
  defer segments.Close()
 
- it := stream.Iter[format.LogSegment](segments)
- for it.Next() {
- m.Segments = append(m.Segments, it.Value())
- }
- if err := it.Err(); err != nil {
+ m.Segments, err = stream.ReadAll[format.LogSegment](segments)
+ if err != nil {
  return nil, err
  }