add internal retryer's to lazy discovery, scheme, scripting and table clients

Author: asmyasnikov

CHANGELOG.md

@@ -1,7 +1,13 @@
 ## 3.8.12
-* Fixed `test.TestTable`
+* Unwrap sub-tests called as `t.Run(...)` in integration tests 
 * Updated `grpc` dependency (from `v1.38.0` to `v1.43.0`)
 * Updated `protobuf` dependency (from `v1.26.0` to `v1.27.1`)
+* Added internal retryers into `lazy.Ratelimiter`
+* Added internal retryers into `lazy.Coordination`
+* Added internal retryers into `lazy.Discovery`
+* Added internal retryers into `lazy.Scheme`
+* Added internal retryers into `lazy.Scripting`
+* Added internal retryer into `lazy.Table.CreateSession`
 
 ## 3.8.11
 * Fixed version

internal/lazy/discovery.go

@@ -8,6 +8,7 @@ import (
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/db"
 	builder "github.com/ydb-platform/ydb-go-sdk/v3/internal/discovery"
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/endpoint"
+	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
 	"github.com/ydb-platform/ydb-go-sdk/v3/trace"
 )
 
@@ -27,14 +28,22 @@ func Discovery(db db.Connection, trace trace.Driver) discovery.Client {
 	}
 }
 
-func (d *lazyDiscovery) Discover(ctx context.Context) ([]endpoint.Endpoint, error) {
+func (d *lazyDiscovery) Discover(ctx context.Context) (endpoints []endpoint.Endpoint, err error) {
 	d.init()
-	return d.client.Discover(ctx)
+	err = retry.Retry(ctx, true, func(ctx context.Context) (err error) {
+		endpoints, err = d.client.Discover(ctx)
+		return err
+	})
+	return endpoints, err
 }
 
-func (d *lazyDiscovery) WhoAmI(ctx context.Context) (*discovery.WhoAmI, error) {
+func (d *lazyDiscovery) WhoAmI(ctx context.Context) (whoAmI *discovery.WhoAmI, err error) {
 	d.init()
-	return d.client.WhoAmI(ctx)
+	err = retry.Retry(ctx, true, func(ctx context.Context) (err error) {
+		whoAmI, err = d.client.WhoAmI(ctx)
+		return err
+	})
+	return whoAmI, err
 }
 
 func (d *lazyDiscovery) Close(ctx context.Context) error {

internal/lazy/scheme.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/db"
 	builder "github.com/ydb-platform/ydb-go-sdk/v3/internal/scheme"
+	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
 	"github.com/ydb-platform/ydb-go-sdk/v3/scheme"
 )
 
@@ -27,7 +28,9 @@ func (s *lazyScheme) ModifyPermissions(
 	opts ...scheme.PermissionsOption,
 ) (err error) {
 	s.init()
-	return s.client.ModifyPermissions(ctx, path, opts...)
+	return retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		return s.client.ModifyPermissions(ctx, path, opts...)
+	})
 }
 
 func (s *lazyScheme) Close(ctx context.Context) error {
@@ -52,20 +55,32 @@ func (s *lazyScheme) init() {
 
 func (s *lazyScheme) DescribePath(ctx context.Context, path string) (e scheme.Entry, err error) {
 	s.init()
-	return s.client.DescribePath(ctx, path)
+	err = retry.Retry(ctx, true, func(ctx context.Context) (err error) {
+		e, err = s.client.DescribePath(ctx, path)
+		return err
+	})
+	return e, err
 }
 
 func (s *lazyScheme) MakeDirectory(ctx context.Context, path string) (err error) {
 	s.init()
-	return s.client.MakeDirectory(ctx, path)
+	return retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		return s.client.MakeDirectory(ctx, path)
+	})
 }
 
 func (s *lazyScheme) ListDirectory(ctx context.Context, path string) (d scheme.Directory, err error) {
 	s.init()
-	return s.client.ListDirectory(ctx, path)
+	err = retry.Retry(ctx, true, func(ctx context.Context) (err error) {
+		d, err = s.client.ListDirectory(ctx, path)
+		return err
+	})
+	return d, err
 }
 
 func (s *lazyScheme) RemoveDirectory(ctx context.Context, path string) (err error) {
 	s.init()
-	return s.client.RemoveDirectory(ctx, path)
+	return retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		return s.client.RemoveDirectory(ctx, path)
+	})
 }

internal/lazy/scripting.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/db"
 	builder "github.com/ydb-platform/ydb-go-sdk/v3/internal/scripting"
+	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
 	"github.com/ydb-platform/ydb-go-sdk/v3/scripting"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table/result"
@@ -21,27 +22,39 @@ func (s *lazyScripting) Execute(
 	ctx context.Context,
 	query string,
 	params *table.QueryParameters,
-) (result.Result, error) {
+) (res result.Result, err error) {
 	s.init()
-	return s.client.Execute(ctx, query, params)
+	err = retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		res, err = s.client.Execute(ctx, query, params)
+		return err
+	})
+	return res, err
 }
 
 func (s *lazyScripting) Explain(
 	ctx context.Context,
 	query string,
 	mode scripting.ExplainMode,
-) (table.ScriptingYQLExplanation, error) {
+) (e table.ScriptingYQLExplanation, err error) {
 	s.init()
-	return s.client.Explain(ctx, query, mode)
+	err = retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		e, err = s.client.Explain(ctx, query, mode)
+		return err
+	})
+	return e, err
 }
 
 func (s *lazyScripting) StreamExecute(
 	ctx context.Context,
 	query string,
 	params *table.QueryParameters,
-) (result.StreamResult, error) {
+) (res result.StreamResult, err error) {
 	s.init()
-	return s.client.StreamExecute(ctx, query, params)
+	err = retry.Retry(ctx, false, func(ctx context.Context) (err error) {
+		res, err = s.client.StreamExecute(ctx, query, params)
+		return err
+	})
+	return res, err
 }
 
 func (s *lazyScripting) Close(ctx context.Context) error {

internal/lazy/table.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/db"
 	builder "github.com/ydb-platform/ydb-go-sdk/v3/internal/table"
+	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table/config"
 )
@@ -26,7 +27,11 @@ func Table(db db.Connection, options []config.Option) table.Client {
 
 func (t *lazyTable) CreateSession(ctx context.Context) (s table.ClosableSession, err error) {
 	t.init(ctx)
-	return t.client.CreateSession(ctx)
+	err = retry.Retry(ctx, true, func(ctx context.Context) (err error) {
+		s, err = t.client.CreateSession(ctx)
+		return err
+	})
+	return s, err
 }
 
 func (t *lazyTable) Do(ctx context.Context, op table.Operation, opts ...table.Option) (err error) {