code: func WithSegment(ctx context.Context, s *Segment) context.Context {
  	return context.WithValue(ctx, segKey, s)
  }
docs: // WithSegment creates a context containing the given segment. Use ContextSegment
  // to retrieve it.