Function withEngine

cmd/dagger/engine.go:80–170

(
	ctx context.Context,
	params client.Params,
	fn runClientCallback,
)

type runClientCallback func(context.Context, *client.Client) error

func withEngine(
	ctx context.Context,
	params client.Params,
	fn runClientCallback,
) (rerr error) {
	if !moduleNoURL {
		if modRef, _ := getExplicitModuleSourceRef(); modRef != "" {
			params.Module = modRef
		}
	}
	return Frontend.Run(ctx, opts, func(ctx context.Context) (_ cleanups.CleanupF, rerr error) {
		var cleanup cleanups.Cleanups

		// Init tracing as early as possible and shutdown after the command
		// completes, ensuring progress is fully flushed to the frontend.
		ctx, cleanupTelemetry := initEngineTelemetry(ctx)

		otel.SetErrorHandler(otel.ErrorHandlerFunc(func(err error) {
			if opts.Debug {
				slog.Error("failed to emit telemetry", "error", err)
			}
			Frontend.SetTelemetryError(err)
		}))
		cleanup.Add("close telemetry", func() error {
			cleanupTelemetry(rerr)
			return nil
		})

		if debugFlag {
			params.LogLevel = slog.LevelDebug
		}

		if useCloudEngine {
			params.RunnerHost = engine.DefaultCloudRunnerHost
		} else if params.RunnerHost == "" {
			params.RunnerHost = RunnerHost
		}

		if RunnerImageLoader != "" {
			backend, err := imageload.GetBackend(RunnerImageLoader)
			if err != nil {
				return cleanup.Run, err
			}
			params.ImageLoaderBackend = backend
		}

		params.AllowedLLMModules = allowedLLMModules

		params.CloudURLCallback = Frontend.SetCloudURL

		params.EngineTrace = telemetry.SpanForwarder{
			Processors: telemetry.SpanProcessors,
		}
		params.EngineLogs = telemetry.LogForwarder{
			Processors: telemetry.LogProcessors,
		}
		params.EngineMetrics = telemetry.MetricExporters