returned Do example back

This reverts commit 779d45f.

Author: LostImagin4tion

dev-1/lesson-2.1/go/main.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/ydb-platform/ydb-go-sdk/v3"
+	"github.com/ydb-platform/ydb-go-sdk/v3/query"
 )
 
 // author: Egor Danilov
@@ -32,12 +33,11 @@ func main() {
 	queryCtx, txCancel := context.WithTimeout(context.Background(), 30*time.Second)
 	defer txCancel()
 
-	// Ниже представлены примеры по обращению к клиенту для выполнения YQL запросов
-	//
+	// Ниже представлены примеры по обращению к клиенту для выполнения YQL запросов.
 	// Query(), QueryResultSet() и QueryRow() являются хелперами для неинтерактивных транзакций
-	// (т.е. когда транзакция завершается за одно обращение к серверу)
-	// Ретраи тут происходят внутри, а наружу возвращается уже материализированный в памяти в результат,
-	// Таймаутом транзакции можно управлять через контекст
+	// (т.е. когда транзакция завершается за одно обращение к серверу).
+	// Ретраи тут происходят внутри, а наружу возвращается уже материализированный в памяти в результат.
+	// Таймаутом транзакции можно управлять через контекст.
 
 	result, err := db.Query().Query(queryCtx, "SELECT 1 as id")
 	if err != nil {
@@ -90,4 +90,54 @@ func main() {
 		log.Fatal(err)
 	}
 	println(structId.Id)
+
+	// ======
+
+	// Ниже представлен код для отправки запроса. Он же используется в реализации методов-хелперов, показанных выше.
+	// В [query.Client.Do] передается анонимная функция, которая будет ретраиться.
+	// По типу ошибки, которую вернет эта анонимная функция, драйвер решит, надо ли ретраить запрос.
+	err = db.Query().Do(
+		queryCtx,
+		// Внутри анонимной функции получаем из пула сессию query.Session для выполнения запроса.
+		// После окончания работы ретраера сессия возвращается обратно в пул.
+		func(ctx context.Context, s query.Session) error {
+			res, err := s.Query(ctx, "SELECT 4 as id;")
+
+			if err != nil {
+				return err
+			}
+
+			defer func() { _ = res.Close(ctx) }()
+
+			// Итерируемся по набору результатов
+			for rs, err := range res.ResultSets(ctx) {
+				if err != nil {
+					return err
+				}
+
+				// Итерируемся по строкам в каждом наборе результате
+				for row, err := range rs.Rows(ctx) {
+					if err != nil {
+						return err
+					}
+
+					// Читаем результат
+					// Важно - код внутри ретраера должен быть готов к тому, 
+					// что он может исполниться несколько раз подряд.
+					// Например, нужно очищать слайс, куда будут записываться результаты, 
+					// иначе там могут оказаться дублирующиеся данные.
+					if err = row.ScanStruct(&structId); err != nil {
+						return err
+					}
+				}
+			}
+			return nil
+		},
+		query.WithIdempotent(),
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	println(structId.Id)
 }