github.com/ydb-platform/ydb-go-sdk/v3@v3.89.2/meta/context.go (about)

     1  package meta
     2  
     3  import (
     4  	"context"
     5  
     6  	"google.golang.org/grpc/metadata"
     7  
     8  	"github.com/ydb-platform/ydb-go-sdk/v3/internal/meta"
     9  )
    10  
    11  // WithTraceID returns a copy of parent context with traceID
    12  func WithTraceID(ctx context.Context, traceID string) context.Context {
    13  	return meta.WithTraceID(ctx, traceID)
    14  }
    15  
    16  // WithUserAgent returns a copy of parent context with custom user-agent info
    17  //
    18  // Deprecated: use WithApplicationName instead.
    19  // Will be removed after Oct 2024.
    20  // Read about versioning policy: https://github.com/ydb-platform/ydb-go-sdk/blob/master/VERSIONING.md#deprecated
    21  func WithUserAgent(ctx context.Context, _ string) context.Context {
    22  	return ctx
    23  }
    24  
    25  // WithApplicationName returns a copy of parent context with application name
    26  func WithApplicationName(ctx context.Context, applicationName string) context.Context {
    27  	return meta.WithApplicationName(ctx, applicationName)
    28  }
    29  
    30  // WithRequestType returns a copy of parent context with custom request type
    31  func WithRequestType(ctx context.Context, requestType string) context.Context {
    32  	return meta.WithRequestType(ctx, requestType)
    33  }
    34  
    35  // WithAllowFeatures returns a copy of parent context with allowed client feature
    36  func WithAllowFeatures(ctx context.Context, features ...string) context.Context {
    37  	return meta.WithAllowFeatures(ctx, features...)
    38  }
    39  
    40  // WithTrailerCallback attaches callback to context for listening incoming metadata
    41  func WithTrailerCallback(
    42  	ctx context.Context,
    43  	callback func(md metadata.MD),
    44  ) context.Context {
    45  	return meta.WithTrailerCallback(ctx, callback)
    46  }
    47  
    48  // WithTraceParent returns a copy of parent context with traceparent header
    49  func WithTraceParent(ctx context.Context, traceparent string) context.Context {
    50  	return meta.WithTraceParent(ctx, traceparent)
    51  }