用户登录
用户注册

分享至

Go基础系列:惰性数值生成器

  • 作者: 武大地
  • 来源: 51数据库
  • 2021-10-03

惰性数值生成器是指在需要的时候才生成下一个数值,不需要的时候就卡在那。这和python的列表推导表达式类似。惰性生成器的好处是不会一次性将全部结果返回或放进内存,而是每次只返回一个,这样不会在某一时刻大量占用内存和其它资源。

比如,要生成10w个数值,如果要迭代这10w个数值,有两种方法。第一种方法是将10w个数值全部生成好放进一个数组(或其它数据结构),然后再去数组中取数据。第二种方法是迭代到哪个数值的时候临时去生成这个数值。它们的区别是显然的:第一种方法会占用大量内存,且速度有可能会很慢,第二种方法每次只占用一个数值的内存空间,用完就丢了。

下面是一个不算完美的惰性数值生成器示例:

package main

import (
    "fmt"
)

func generatenums(nums chan int) {
    num := 0
    go func() {
        for {
            num++
            nums <- num
        }
    }()
}

func getnums(nums chan int) int {
    return <-nums
}

func main() {
    nums := make(chan int)
    generatenums(nums)
    for i := 0; i < 10; i++ {
        fmt.println(getnums(nums))
    }
}

其中generatenums()函数是惰性数值生成器,它使用一个nums channel作为参数,每次生成的数值都会放进这个channel中。

getnums()函数是取出存入nums channel中的数并返回。

理论上一切都很简单,只要在需要的地方调用generatenums()函数即可。但问题在于如果多个地方调用generatenums(),各个地方的nums通道将互相影响。所以,应该改进一下,让generatenums()自带属于自己的nums通道,而不是多个generatenums()共享一个nums通道。

func generatenums(nums chan int) {}
        |
       \|/
func generatenums(){
    nums := make(chan int)
}

因为通道私有了,要想从这个通道中取数据,需要将这个通道作为返回值:

func generatenums() chan int{
    nums := make(chan int)
    ...
    return nums
}

这样每次调用generatenums()就取得了它的数值生成器通道:

nums := generatenums()

下面是改良后的惰性数值生成器:

package main

import (
    "fmt"
)

func generatenums() chan int {
    nums := make(chan int)
    num := 0
    go func() {
        for {
            num++
            nums <- num
        }
    }()
    return nums
}

func getnums(nums chan int) int {
    return <-nums
}

func main() {
    nums := generatenums()
    for i := 0; i < 10; i++ {
        fmt.println(getnums(nums))
    }
}
软件
前端设计
程序设计
Java相关