Function AsProfiled

profiling/profiled.go:57–81

AsProfiled wraps e2e.Runnable with ProfiledRunnable. If runnable is running during invocation AsProfiled panics. NOTE(bwplotka): Caller is expected to discard passed `r` runnable and use returned ProfiledRunnable.Runnable instead.

(r e2e.Runnable, pprofPortName string, opts ...ProfiledOption)

// If runnable is running during invocation AsProfiled panics.
// NOTE(bwplotka): Caller is expected to discard passed `r` runnable and use returned ProfiledRunnable.Runnable instead.
func AsProfiled(r e2e.Runnable, pprofPortName string, opts ...ProfiledOption) *ProfiledRunnable {
	if r.IsRunning() {
		panic("can't use AsProfiled with running runnable")
	}

	opt := rOpt{
		scheme: "http",
	}
	for _, o := range opts {
		o(&opt)
	}

	if r.InternalEndpoint(pprofPortName) == "" {
		return &ProfiledRunnable{Runnable: e2e.NewFailedRunnable(r.Name(), errors.Newf("pprof port name %v does not exist in given runnable ports", pprofPortName))}
	}

	instr := &ProfiledRunnable{
		Runnable:  r,
		pprofPort: pprofPortName,
		scheme:    opt.scheme,
		config:    opt.config,
	}
	r.SetMetadata(metaKey, Profiled(instr))
	return instr
}

func (r *ProfiledRunnable) ProfileTargets() []Target {
	return []Target{{Name: r.Name(), Scheme: r.scheme, Config: r.config, InternalEndpoint: r.InternalEndpoint(r.pprofPort)}}