chore: use enableGRPCMetadata

Signed-off-by: rogerogers <[email protected]>
kitex-contrib · Oct 10, 2024 · 80c9213 · 80c9213
1 parent 6222294
commit 80c9213
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tracing/middleware.go b/tracing/middleware.go
@@ -44,7 +44,7 @@ func ClientMiddleware(cfg *config) endpoint.Middleware {
 
  Inject(ctx, cfg, md)
 
- if cfg.enableMetadata {
+ if cfg.enableGRPCMetadata {
  grpcMd, ok := metadata.FromOutgoingContext(ctx)
  if ok {
  ctx = injectMetadata(ctx, cfg, grpcMd)
@@ -82,7 +82,7 @@ func ServerMiddleware(cfg *config) endpoint.Middleware {
  md := metainfo.GetAllValues(ctx)
  peerServiceAttributes := extractPeerServiceAttributesFromMetaInfo(md)
 
- if cfg.enableMetadata {
+ if cfg.enableGRPCMetadata {
  grpcMd, ok := metadata.FromIncomingContext(ctx)
  if ok {
  ctx = extractMetadata(ctx, cfg, grpcMd)

diff --git a/tracing/options.go b/tracing/options.go
@@ -45,7 +45,7 @@ type config struct {
  textMapPropagator propagation.TextMapPropagator
 
  recordSourceOperation bool
- enableMetadata  bool
+ enableGRPCMetadata bool
 }
 
 func newConfig(opts []Option) *config {
@@ -90,9 +90,9 @@ func WithTextMapPropagator(p propagation.TextMapPropagator) Option {
  })
 }
 
-// WithEnableMetadata Enable retrieving data from metadata or adding data to metadata
-func WithEnableMetadata() Option {
+// WithEnableGRPCMetadata Enable retrieving data from metadata or adding data to metadata
+func WithEnableGRPCMetadata() Option {
  return option(func(cfg *config) {
- cfg.enableMetadata = true
+ cfg.enableGRPCMetadata = true
  })
 }