Method initializeDaggerSession

engine/server/session.go:314–365

requires that sess.stateMu is held

(
	clientMetadata *engine.ClientMetadata,
	sess *daggerSession,
	failureCleanups *cleanups.Cleanups,
)

// requires that sess.stateMu is held
func (srv *Server) initializeDaggerSession(
	clientMetadata *engine.ClientMetadata,
	sess *daggerSession,
	failureCleanups *cleanups.Cleanups,
) error {
	slog.Info("initializing new session", "session", clientMetadata.SessionID)
	defer slog.Debug("initialized new session", "session", clientMetadata.SessionID)

	sess.sessionID = clientMetadata.SessionID
	sess.mainClientCallerID = clientMetadata.ClientID
	sess.clients = map[string]*daggerClient{}
	sess.attachables = newSessionAttachableManager()
	sess.endpoints = map[string]http.Handler{}
	sess.closingCtx, sess.cancelClosing = context.WithCancelCause(context.Background())
	sess.shutdownCh = make(chan struct{})
	sess.services = core.NewServices()
	sess.authProvider = auth.NewRegistryAuthProvider()
	sess.resolver = serverresolver.New(serverresolver.Opts{
		Hosts: srv.registryHosts,
		Auth: serverresolver.NewSessionAuthSource(
			sess.authProvider,
			func(ctx context.Context) (*grpc.ClientConn, error) {
				return srv.sessionMainClientConn(ctx, sess)
			},
		),
		ContentStore: srv.contentStore,
		LeaseManager: srv.leaseManager,
	})
	failureCleanups.Add("close session resolver", sess.resolver.Close)
	sess.containers = map[bkgw.Container]struct{}{}
	sess.dagqlCond = sync.NewCond(&sess.dagqlMu)
	sess.telemetryPubSub = srv.telemetryPubSub
	sess.interactive = clientMetadata.Interactive
	sess.interactiveCommand = clientMetadata.InteractiveCommand
	sess.allowedLLMModules = clientMetadata.AllowedLLMModules

	sess.analytics = analytics.New(analytics.Config{
		DoNotTrack: clientMetadata.DoNotTrack || analytics.DoNotTrack(),
		Labels: enginetel.NewLabels(clientMetadata.Labels, nil, nil).
			WithEngineLabel(srv.engineName).
			WithServerLabels(
				engine.Version,
				runtime.GOOS,
				runtime.GOARCH,
				false,
			),
	})
	failureCleanups.Add("close session analytics", sess.analytics.Close)

	sess.state = sessionStateInitialized
	return nil
}

var errSessionClosing = errors.New("session is closing")