Apply suggestions from code review

Author: asmyasnikov

CHANGELOG.md

@@ -1,4 +1,4 @@
-* Added `ResultSet.Index()` method
+* Added `query.ResultSet.Index()` method
 
 ## v3.74.5
 * Fixed bug with reading empty result set parts.

internal/query/result.go

@@ -25,7 +25,7 @@ var (
 type (
 	materializedResult struct {
 		resultSets []query.ResultSet
-		idx        int64
+		idx        int
 	}
 	result struct {
 		stream         Ydb_Query_V1.QueryService_ExecuteQueryClient
@@ -51,7 +51,7 @@ func (r *materializedResult) Close(ctx context.Context) error {
 }
 
 func (r *materializedResult) NextResultSet(ctx context.Context) (query.ResultSet, error) {
-	if r.idx == int64(len(r.resultSets)) {
+	if r.idx == len(r.resultSets) {
 		return nil, xerrors.WithStackTrace(io.EOF)
 	}
 

internal/query/result_set.go

@@ -175,5 +175,5 @@ func (rs *resultSet) NextRow(ctx context.Context) (_ query.Row, err error) {
 }
 
 func (rs *resultSet) Index() int {
-	return int(rs.index)
+	return rs.idx
 }