iter: make close compatible with gocql

Author: Kulezi

query.go

@@ -2,6 +2,7 @@ package scylla
 
 import (
 	"context"
+	"errors"
 	"fmt"
 
 	"github.com/scylladb/scylla-go-driver/frame"
@@ -346,6 +347,8 @@ type Iter struct {
 	nextCh    chan transport.QueryResult
 	errCh     chan error
 	closed    bool
+
+	err error
 }
 
 var (
@@ -367,16 +370,18 @@ func (it *Iter) NumRows() int {
 
 func (it *Iter) Next() (frame.Row, error) {
 	if it.closed {
-		return nil, ErrClosedIter
+		return nil, nil
 	}
 
 	if it.pos >= it.rowCnt {
 		select {
 		case r := <-it.nextCh:
 			it.result = r
 		case err := <-it.errCh:
-			it.Close()
-			return nil, err
+			if !errors.Is(err, ErrNoMoreRows) {
+				it.err = err
+			}
+			return nil, it.Close()
 		}
 
 		it.pos = 0
@@ -394,12 +399,13 @@ func (it *Iter) Next() (frame.Row, error) {
 	return res, nil
 }
 
-func (it *Iter) Close() {
+func (it *Iter) Close() error {
 	if it.closed {
-		return
+		return it.err
 	}
 	it.closed = true
 	close(it.requestCh)
+	return it.err
 }
 
 type iterWorker struct {