35 lines
495 B
Go
35 lines
495 B
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func gen(ctx context.Context) <-chan int {
|
|||
|
|
dst := make(chan int)
|
|||
|
|
n := 1
|
|||
|
|
go func() {
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case <-ctx.Done():
|
|||
|
|
return // return结束该goroutine，防止泄露
|
|||
|
|
case dst <- n:
|
|||
|
|
n++
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
return dst
|
|||
|
|
}
|
|||
|
|
func main() {
|
|||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|||
|
|
defer cancel() // 当我们取完需要的整数后调用cancel
|
|||
|
|
|
|||
|
|
for n := range gen(ctx) {
|
|||
|
|
fmt.Println(n)
|
|||
|
|
if n == 5 {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|