* Changed trace.RetryLoopStartInfo.Context type from context.Context to *context.Context

Author: asmyasnikov

CHANGELOG.md

@@ -4,6 +4,7 @@
 * Fixed bug with unexpected changing of local datacenter flag in endpoint
 * Refactored errors wrapping (stackedError are not ydb error now, checking `errors.IsYdb(err)` with `errors.As` now)
 * Wrapped retry operation errors with `errors.WithStackTrace(err)`
+* Changed `trace.RetryLoopStartInfo.Context` type from `context.Context` to `*context.Context`
 
 ## v3.16.9
 * Refactored internal operation and transport errors

internal/table/client.go

@@ -135,7 +135,7 @@ func (c *client) CreateSession(ctx context.Context, opts ...table.Option) (table
 		retry.WithSlowBackoff(options.SlowBackoff),
 		retry.WithTrace(trace.Retry{
 			OnRetry: func(info trace.RetryLoopStartInfo) func(trace.RetryLoopIntermediateInfo) func(trace.RetryLoopDoneInfo) {
-				onIntermediate := trace.TableOnCreateSession(c.config.Trace(), &info.Context)
+				onIntermediate := trace.TableOnCreateSession(c.config.Trace(), info.Context)
 				return func(info trace.RetryLoopIntermediateInfo) func(trace.RetryLoopDoneInfo) {
 					onDone := onIntermediate(info.Error)
 					return func(info trace.RetryLoopDoneInfo) {

retry/retry.go

@@ -126,7 +126,7 @@ func Retry(ctx context.Context, op retryOperation, opts ...retryOption) (err err
 		attempts int
 
 		code           = int64(0)
-		onIntermediate = trace.RetryOnRetry(h.trace, ctx, h.id, h.idempotent)
+		onIntermediate = trace.RetryOnRetry(h.trace, &ctx, h.id, h.idempotent)
 	)
 	defer func() {
 		onIntermediate(err)(attempts, err)

trace/retry.go

@@ -15,7 +15,11 @@ type (
 		OnRetry func(RetryLoopStartInfo) func(RetryLoopIntermediateInfo) func(RetryLoopDoneInfo)
 	}
 	RetryLoopStartInfo struct {
-		Context    context.Context
+		// Context make available context in trace callback function.
+		// Pointer to context provide replacement of context in trace callback function.
+		// Warning: concurrent access to pointer on client side must be excluded.
+		// Safe replacement of context are provided only inside callback function
+		Context    *context.Context
 		ID         string
 		Idempotent bool
 	}