generators: add go alternative: generator func

hiennv9 · Apr 17, 2019 · 3498c6d · 3498c6d
1 parent e53c9f3
commit 3498c6d
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -1769,10 +1769,24 @@ func Generator() chan string {
 	return c
 }
 
+func GeneratorFunc() func() (string, bool) {
+
+	s := []string{"hello", "world"}
+	i := -1
+
+	return func() (string, bool) {
+		i++
+		if i >= len(s) {
+			return "", false
+		}
+		return s[i], true
+	}
+}
+
 func main() {
-	gen := Generator()
 
-	for true {
+	gen := Generator()
+	for {
 		value, more := <-gen
 		fmt.Println(value, more)
 
@@ -1785,6 +1799,17 @@ func main() {
 	for value := range Generator() {
 		fmt.Println(value)
 	}
+
+	// alternatively
+	genfn := GeneratorFunc()
+	for {
+		value, more := genfn()
+		fmt.Println(value, more)
+
+		if !more {
+			break
+		}
+	}
 }
 ```
 
@@ -1796,6 +1821,9 @@ world true
  false
 hello
 world
+hello true
+world true
+ false
 ```
 
 ### datetime

diff --git a/examples/generators.go b/examples/generators.go
@@ -15,10 +15,24 @@ func Generator() chan string {
 	return c
 }
 
+func GeneratorFunc() func() (string, bool) {
+
+	s := []string{"hello", "world"}
+	i := -1
+
+	return func() (string, bool) {
+		i++
+		if i >= len(s) {
+			return "", false
+		}
+		return s[i], true
+	}
+}
+
 func main() {
-	gen := Generator()
 
-	for true {
+	gen := Generator()
+	for {
 		value, more := <-gen
 		fmt.Println(value, more)
 
@@ -31,4 +45,15 @@ func main() {
 	for value := range Generator() {
 		fmt.Println(value)
 	}
+
+	// alternatively
+	genfn := GeneratorFunc()
+	for {
+		value, more := genfn()
+		fmt.Println(value, more)
+
+		if !more {
+			break
+		}
+	}
 }