feat: add small document (#628)

wundergraph · Oct 6, 2023 · fcbaaa9 · fcbaaa9
1 parent 3358a84
commit fcbaaa9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/v2/pkg/ast/ast.go b/v2/pkg/ast/ast.go
@@ -106,6 +106,17 @@ func NewDocument() *Document {
 	}
 }
 
+func NewSmallDocument() *Document {
+	return &Document{
+		BooleanValues: [2]BooleanValue{false, true},
+		Refs:          make([][8]int, 48),
+		RefIndex:      -1,
+		Index: Index{
+			nodes: make(map[uint64][]Node, 4),
+		},
+	}
+}
+
 func (d *Document) Reset() {
 	d.RootNodes = d.RootNodes[:0]
 	d.SchemaDefinitions = d.SchemaDefinitions[:0]

diff --git a/v2/pkg/engine/resolve/load.go b/v2/pkg/engine/resolve/load.go
@@ -698,9 +698,9 @@ func (l *Loader) loadWithSingleFlight(ctx *Context, source DataSource, identifie
 	}
 	data := maybeData.([]byte)
 	if shared {
-		out := res.getBuffer()
-		_, err = out.Write(data)
-		return out.Bytes(), err
+		out := make([]byte, len(data))
+		copy(out, data)
+		return out, err
 	}
 	return data, nil
 }