From c5047186ddfd5dba0cbd2ae32592686218cb9c4a Mon Sep 17 00:00:00 2001
From: xushiwei <x@goplus.org>
Date: Mon, 8 Jul 2024 15:50:16 +0800
Subject: [PATCH] demo: ctxcancel

---
 _demo/ctxcancel/ctx.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 _demo/ctxcancel/ctx.go

diff --git a/_demo/ctxcancel/ctx.go b/_demo/ctxcancel/ctx.go
new file mode 100644
index 00000000..3b9afc26
--- /dev/null
+++ b/_demo/ctxcancel/ctx.go
@@ -0,0 +1,39 @@
+package main
+
+import (
+	"context"
+	"fmt"
+)
+
+func main() {
+	// gen generates integers in a separate goroutine and
+	// sends them to the returned channel.
+	// The callers of gen need to cancel the context once
+	// they are done consuming generated integers not to leak
+	// the internal goroutine started by gen.
+	gen := func(ctx context.Context) <-chan int {
+		dst := make(chan int)
+		n := 1
+		go func() {
+			for {
+				select {
+				case <-ctx.Done():
+					return // returning not to leak the goroutine
+				case dst <- n:
+					n++
+				}
+			}
+		}()
+		return dst
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel() // cancel when we are finished consuming integers
+
+	for n := range gen(ctx) {
+		fmt.Println(n)
+		if n == 5 {
+			break
+		}
+	}
+}