fix: close when query finish (#31)

hantmac · web-flow · commit f5e3d74c4a69 · 2022-12-28T19:02:44.000+08:00
diff --git a/rows.go b/rows.go
@@ -62,7 +62,12 @@ func (r *nextRows) Columns() []string {
 }
 
 func (r *nextRows) Close() error {
-	// FIXME: should check & call final here
+	if len(r.respData.NextURI) != 0 {
+		_, err := r.dc.rest.QueryPage(r.respData.Id, r.respData.NextURI)
+		if err != nil {
+			return err
+		}
+	}
 	r.dc.cancel = nil
 	return nil
 }

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,12 @@ func (r *nextRows) Columns() []string {`
`62`	`62`	`}`
`63`	`63`
`64`	`64`	`func (r *nextRows) Close() error {`
`65`		`- // FIXME: should check & call final here`
	`65`	`+ if len(r.respData.NextURI) != 0 {`
	`66`	`+ _, err := r.dc.rest.QueryPage(r.respData.Id, r.respData.NextURI)`
	`67`	`+ if err != nil {`
	`68`	`+ return err`
	`69`	`+ }`
	`70`	`+ }`
`66`	`71`	`r.dc.cancel = nil`
`67`	`72`	`return nil`
`68`	`73`	`}`