iter: expose columns

Kulezi · Kulezi · commit 4c55d4e34160 · 2022-09-26T19:01:16.000+02:00
diff --git a/query.go b/query.go
@@ -309,6 +309,9 @@ func (q *Query) Iter(ctx context.Context) Iter {
 		errCh:     make(chan error, 1),
 	}
 
+	if q.stmt.Metadata != nil {
+		it.result.ColSpec = q.stmt.Metadata.Columns
+	}
 	info, err := q.info()
 	if err != nil {
 		it.errCh <- err
@@ -350,6 +353,10 @@ var (
 	ErrNoMoreRows = fmt.Errorf("no more rows left")
 )
 
+func (it *Iter) Columns() []frame.ColumnSpec {
+	return it.result.ColSpec
+}
+
 func (it *Iter) PageState() []byte {
 	return it.result.PagingState
 }

Original file line number	Diff line number	Diff line change
`@@ -309,6 +309,9 @@ func (q *Query) Iter(ctx context.Context) Iter {`
`309`	`309`	`errCh: make(chan error, 1),`
`310`	`310`	`}`
`311`	`311`
	`312`	`+ if q.stmt.Metadata != nil {`
	`313`	`+ it.result.ColSpec = q.stmt.Metadata.Columns`
	`314`	`+ }`
`312`	`315`	`info, err := q.info()`
`313`	`316`	`if err != nil {`
`314`	`317`	`it.errCh <- err`
`@@ -350,6 +353,10 @@ var (`
`350`	`353`	`ErrNoMoreRows = fmt.Errorf("no more rows left")`
`351`	`354`	`)`
`352`	`355`
	`356`	`+func (it *Iter) Columns() []frame.ColumnSpec {`
	`357`	`+ return it.result.ColSpec`
	`358`	`+}`
	`359`	`+`
`353`	`360`	`func (it *Iter) PageState() []byte {`
`354`	`361`	`return it.result.PagingState`
`355`	`362`	`}`