runtime: implement newcoro, coroswitch to support package iter

1 files changed, 31 insertions, 0 deletions

diff --git a/src/runtime/coro.go b/src/runtime/coro.go
new file mode 100644
index 000000000..204a5c2be
--- /dev/null
+++ b/src/runtime/coro.go
@@ -0,0 +1,31 @@
+package runtime
+
+// A naive implementation of coroutines that supports
+// package iter.
+
+type coro struct {
+	f  func(*coro)
+	ch chan struct{}
+}
+
+//go:linkname newcoro
+
+func newcoro(f func(*coro)) *coro {
+	c := &coro{
+		ch: make(chan struct{}),
+		f:  f,
+	}
+	go func() {
+		defer close(c.ch)
+		<-c.ch
+		f(c)
+	}()
+	return c
+}
+
+//go:linkname coroswitch
+
+func coroswitch(c *coro) {
+	c.ch <- struct{}{}
+	<-c.ch
+}