github.com/searKing/golang/go@v1.2.117/go/generator/example_test.go

// Copyright 2020 The searKing Author. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package generator_test

import (
	"fmt"

	"github.com/searKing/golang/go/go/generator"
)

func ExampleGeneratorFunc() {
	g := func(i int) *generator.Generator {
		return generator.GeneratorFunc(func(yield generator.Yield) {
			if !yield(i) {
				return
			}
			if !yield(i + 10) {
				return
			}
		})
	}

	gen := g(10)

	for msg := range gen.C {
		fmt.Println(msg)
	}

	// Output:
	// 10
	// 20
}
func ExampleGenerator_Next() {
	g := func(i int) *generator.Generator {
		return generator.GeneratorFunc(func(yield generator.Yield) {
			if !yield(i) {
				return
			}
			if !yield(i + 10) {
				return
			}
		})
	}

	gen := g(10)

	for {
		msg, ok := gen.Next()
		if !ok {
			return
		}
		fmt.Println(msg)
	}

	// Output:
	// 10
	// 20
}

func ExampleGeneratorWithSupplier() {
	var g *generator.Generator

	supplierC := make(chan any)
	supplierF := func(i int) {
		consumer := g.Yield(supplierC)
		if !consumer(i) {
			return
		}
		if !consumer(i + 10) {
			return
		}
		close(supplierC)
	}
	g = generator.GeneratorWithSupplier(supplierC)
	go supplierF(10)

	for msg := range g.C {
		fmt.Println(msg)
	}

	// Output:
	// 10
	// 20
}

func ExampleGeneratorVariadicFunc() {
	g := func(i int) *generator.Generator {
		g := generator.GeneratorVariadicFunc(func(yield generator.Yield, args ...any) {
			i := (args[0]).(int)
			if !yield(i) {
				return
			}
			if !yield(i + 10) {
				return
			}
		})

		gen := g(i)
		return gen
	}

	gen := g(10)

	for msg := range gen.C {
		fmt.Println(msg)
	}

	// Output:
	// 10
	// 20
}