flamegraph: stream gob encode to zstd writer

1 files changed, 3 insertions, 4 deletions

diff --git a/internal/flamegraph/iordata.go b/internal/flamegraph/iordata.go
index af4ad9d..db5bad6 100644
--- a/internal/flamegraph/iordata.go
+++ b/internal/flamegraph/iordata.go
@@ -118,12 +118,11 @@ func (iod iorData) serializeToFile() error {
 	encoder := zstd.NewWriter(file)
 	defer encoder.Close()
 
-	bytes, err := iod.serialize()
-	if err != nil {
+	gobEncoder := gob.NewEncoder(encoder)
+	if err := gobEncoder.Encode(iod.records); err != nil {
 		return err
 	}
-
-	if _, err := encoder.Write(bytes); err != nil {
+	if err := encoder.Flush(); err != nil {
 		return err
 	}