aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/testdata/testgoroutineleakprofile/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/testdata/testgoroutineleakprofile/main.go')
-rw-r--r--src/runtime/testdata/testgoroutineleakprofile/main.go39
1 files changed, 39 insertions, 0 deletions
diff --git a/src/runtime/testdata/testgoroutineleakprofile/main.go b/src/runtime/testdata/testgoroutineleakprofile/main.go
new file mode 100644
index 0000000000..5787c1e2b2
--- /dev/null
+++ b/src/runtime/testdata/testgoroutineleakprofile/main.go
@@ -0,0 +1,39 @@
+// Copyright 2025 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package main
+
+import "os"
+
+// The number of times the main (profiling) goroutine should yield
+// in order to allow the leaking goroutines to get stuck.
+const yieldCount = 10
+
+var cmds = map[string]func(){}
+
+func register(name string, f func()) {
+ if cmds[name] != nil {
+ panic("duplicate registration: " + name)
+ }
+ cmds[name] = f
+}
+
+func registerInit(name string, f func()) {
+ if len(os.Args) >= 2 && os.Args[1] == name {
+ f()
+ }
+}
+
+func main() {
+ if len(os.Args) < 2 {
+ println("usage: " + os.Args[0] + " name-of-test")
+ return
+ }
+ f := cmds[os.Args[1]]
+ if f == nil {
+ println("unknown function: " + os.Args[1])
+ return
+ }
+ f()
+}