From 03bd0222a8ef360c332f78f9e6a3762bde0aea58 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Mon, 29 Aug 2022 21:44:40 -0600
Subject: wip

---
 cmd/btrfs-rec/inspect_rebuildnodes.go | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

(limited to 'cmd/btrfs-rec/inspect_rebuildnodes.go')

diff --git a/cmd/btrfs-rec/inspect_rebuildnodes.go b/cmd/btrfs-rec/inspect_rebuildnodes.go
index df9a977..36a9207 100644
--- a/cmd/btrfs-rec/inspect_rebuildnodes.go
+++ b/cmd/btrfs-rec/inspect_rebuildnodes.go
@@ -5,14 +5,17 @@
 package main
 
 import (
-	"encoding/json"
+	"bufio"
+	"io"
 	"os"
 
+	"git.lukeshu.com/go/lowmemjson"
 	"github.com/datawire/dlib/dlog"
 	"github.com/datawire/ocibuild/pkg/cliutil"
 	"github.com/spf13/cobra"
 
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfs"
+	"git.lukeshu.com/btrfs-progs-ng/lib/btrfs/btrfsvol"
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect/rebuildnodes"
 )
 
@@ -38,9 +41,7 @@ func init() {
 			}
 
 			dlog.Info(ctx, "Writing re-built nodes to stdout...")
-			encoder := json.NewEncoder(os.Stdout)
-			encoder.SetIndent("", "    ")
-			if err := encoder.Encode(rebuiltNodes); err != nil {
+			if err := writeNodesJSON(os.Stdout, rebuiltNodes); err != nil {
 				return err
 			}
 			dlog.Info(ctx, "... done writing")
@@ -49,3 +50,18 @@ func init() {
 		},
 	})
 }
+
+func writeNodesJSON(w io.Writer, rebuiltNodes map[btrfsvol.LogicalAddr]*rebuildnodes.RebuiltNode) (err error) {
+	buffer := bufio.NewWriter(w)
+	defer func() {
+		if _err := buffer.Flush(); err == nil && _err != nil {
+			err = _err
+		}
+	}()
+	return lowmemjson.Encode(&lowmemjson.ReEncoder{
+		Out: buffer,
+
+		Indent:                "\t",
+		ForceTrailingNewlines: true,
+	}, rebuiltNodes)
+}
-- 
cgit v1.2.3-2-g168b