code: func WithStats(ctx context.Context, stats Stats) context.Context {
  	return context.WithValue(ctx, statsKey, stats)
  }
docs: // WithStats returns a new context.Context with the Stats implementation
  // embedded.