code: func ContextSegment(ctx context.Context) *Segment {
  	if s := ctx.Value(segKey); s != nil {
  		return s.(*Segment)
  	}
  	return nil
  }
docs: // ContextSegment extracts the segment set in the context with WithSegment.