custom iterator

Author: James Cor

sql/rowexec/dml_iters.go

@@ -565,6 +565,10 @@ func getRowHandler(clientFoundRowsToggled bool, iter sql.RowIter) accumulatorRow
 
 func AddAccumulatorIter(ctx *sql.Context, iter sql.RowIter) (sql.RowIter, sql.Schema) {
 	switch i := iter.(type) {
+	case sql.CustomRowIter:
+		childIter := i.GetChildIter()
+		childIter, sch := AddAccumulatorIter(ctx, childIter)
+		return i.SetChildIter(childIter), sch
 	case *callIter:
 		childIter, sch := AddAccumulatorIter(ctx, i.innerIter)
 		i.innerIter = childIter

sql/rows.go

@@ -201,3 +201,11 @@ func (i *sliceRowIter) Close(*Context) error {
 	i.rows = nil
 	return nil
 }
+
+// CustomRowIter is an extension of RowIter for integrators that wrap RowIters.
+// It allows for analysis rules to inspect the underlying RowIters.
+type CustomRowIter interface {
+	RowIter
+	GetChildIter() RowIter
+	SetChildIter(childIter RowIter) RowIter
+}