Flush output file

Flush output file before closing it.

Signed-off-by: Leon Hwang <[email protected]>

Author: Asphaltt

internal/pwru/output.go

@@ -7,7 +7,6 @@ package pwru
 import (
 	"errors"
 	"fmt"
-	"io"
 	"log"
 	"net"
 	"os"
@@ -36,7 +35,7 @@ type output struct {
 	printSkbMap   *ebpf.Map
 	printStackMap *ebpf.Map
 	addr2name     Addr2Name
-	writer        io.Writer
+	writer        *os.File
 	kprobeMulti   bool
 	kfreeReasons  map[uint64]string
 	ifaceCache    map[uint64]map[uint32]string
@@ -81,6 +80,13 @@ func NewOutput(flags *Flags, printSkbMap *ebpf.Map, printStackMap *ebpf.Map,
 	}, nil
 }
 
+func (o *output) Close() {
+	if o.writer != os.Stdout {
+		_ = o.writer.Sync()
+		_ = o.writer.Close()
+	}
+}
+
 func (o *output) PrintHeader() {
 	if o.flags.OutputTS == "absolute" {
 		fmt.Fprintf(o.writer, "%12s ", "TIME")

main.go

@@ -299,6 +299,7 @@ func main() {
 	if err != nil {
 		log.Fatalf("Failed to create outputer: %s", err)
 	}
+	defer output.Close()
 	output.PrintHeader()
 
 	defer func() {