feat: implement buffered row iterator

To be used for reading interleaved rows where the rows need to be
buffered before iterating.

internal/codegen/databasecodegen/rowiterator.go

@@ -18,8 +18,13 @@ func (g RowIteratorCodeGenerator) StreamingType() string {
 	return "streaming" + strcase.UpperCamelCase(string(g.Table.Name)) + "RowIterator"
 }
 
+func (g RowIteratorCodeGenerator) BufferedType() string {
+	return "buffered" + strcase.UpperCamelCase(string(g.Table.Name)) + "RowIterator"
+}
+
 func (g RowIteratorCodeGenerator) GenerateCode(f *codegen.File) {
 	spannerPkg := f.Import("cloud.google.com/go/spanner")
+	iteratorPkg := f.Import("google.golang.org/api/iterator")
 	row := RowCodeGenerator(g)
 	f.P()
 	f.P("type ", g.InterfaceType(), " interface {")
@@ -53,4 +58,39 @@ func (g RowIteratorCodeGenerator) GenerateCode(f *codegen.File) {
 	f.P("return f(&row)")
 	f.P("})")
 	f.P("}")
+	f.P()
+	f.P("type ", g.BufferedType(), " struct {")
+	f.P("rows []*", row.Type())
+	f.P("err error")
+	f.P("}")
+	f.P()
+	f.P("func (i *", g.BufferedType(), ") Next() (*", row.Type(), ", error) {")
+	f.P("if i.err != nil {")
+	f.P("return nil, i.err")
+	f.P("}")
+	f.P("if len(i.rows) == 0 {")
+	f.P("return nil, ", iteratorPkg, ".Done")
+	f.P("}")
+	f.P("next := i.rows[0]")
+	f.P("i.rows = i.rows[1:]")
+	f.P("return next, nil")
+	f.P("}")
+	f.P()
+	f.P("func (i *", g.BufferedType(), ") Do(f func(row *", row.Type(), ") error) error {")
+	f.P("for {")
+	f.P("row, err := i.Next()")
+	f.P("switch err {")
+	f.P("case ", iteratorPkg, ".Done:")
+	f.P("return nil")
+	f.P("case nil:")
+	f.P("if err = f(row); err != nil {")
+	f.P("return err")
+	f.P("}")
+	f.P("default:")
+	f.P("return err")
+	f.P("}")
+	f.P("}")
+	f.P("}")
+	f.P()
+	f.P("func (i *", g.BufferedType(), ") Stop() {}")
 }