task_exit.go

//go:build scheduler.tasks || scheduler.asyncify || scheduler.cores

package task

import "sync/atomic"

var (
	mainTask           *Task
	liveTasks          uint32
	mainExitedByGoexit uint32
)

func addLiveTask(t *Task) {
	if mainTask == nil {
		mainTask = t
	}
	atomic.AddUint32(&liveTasks, 1)
}

// Goexit exits the current task because runtime.Goexit was called.
func Goexit() {
	exit(true)
}

// Exit exits the current task after its entry function returns.
func Exit() {
	exit(false)
}

func exit(goexit bool) {
	t := Current()
	if hasReleasableStack {
		t.Exited = true
	}
	exitSynctest(t)
	remaining := atomic.AddUint32(&liveTasks, ^uint32(0))
	if t == mainTask {
		if goexit {
			if remaining == 0 {
				runtimeFatal("all goroutines are asleep - deadlock!")
			}
			atomic.StoreUint32(&mainExitedByGoexit, 1)
		}
	} else if atomic.LoadUint32(&mainExitedByGoexit) != 0 && remaining == 0 {
		runtimeFatal("all goroutines are asleep - deadlock!")
	}

	// TODO: explicitly free the stack after switching back to the scheduler.
	Pause()
	runtimeFatal("unreachable")
}