在频道上发送指针

时间:2018-03-06 03:57:37

标签: pointers go goroutine channels

我正在尝试使用渠道来实现一种工作池。请看下面的代码

https://play.golang.org/p/g7aKxDoP9lf(围棋游乐场)

package main

import (
    "fmt"
    "time"
)

func main() {
    q1 := make(chan int)

    fmt.Printf("worker 1\n")
    go worker1(q1)
    for i := 0; i < 10; i++ {
        fmt.Printf("sending: %v\n", i)
        q1 <- i
    }

    time.Sleep(time.Second)

    fmt.Printf("\n\nworker 2\n")
    q2 := make(chan *int)
    go worker2(q2)
    for i := 0; i < 10; i++ {
        fmt.Printf("sending: %v\n", i)
        q2 <- &i
    }
    time.Sleep(time.Second)
}

func worker1(qTodo <-chan int) {
    var curr int
    for {
        select {
        case curr = <-qTodo:
            fmt.Printf("got: %v\n", curr)
        }
    }
}

func worker2(qTodo <-chan *int) {
    var curr *int
    for {
        select {
        case curr = <-qTodo:
            fmt.Printf("got: %v\n", *curr)
        }
    }
}

这是一个示例输出

worker 1
sending: 0
got: 0
sending: 1
sending: 2
got: 1
got: 2
sending: 3
sending: 4
got: 3
got: 4
sending: 5
sending: 6
got: 5
got: 6
sending: 7
sending: 8
got: 7
got: 8
sending: 9
got: 9


worker 2
sending: 0
got: 0
sending: 1
sending: 2
got: 2
got: 2
sending: 3
sending: 4
got: 4
got: 4
sending: 5
sending: 6
got: 6
got: 6
sending: 7
sending: 8
got: 8
got: 8
sending: 9
got: 10

似乎在worker2收到指针的时候,原始变量中的值已经改变,这反映在打印的值中。

问题是如何避免这种情况?如何解决这个问题?

2 个答案:

答案 0 :(得分:4)

接收指针指向的值不是您所期望的值,因为您每次都会向同一个变量发送指针,因此工作人员会看到该变量在取消引用指针时所具有的值。解决此类问题的一种典型方法是在for循环内复制变量并发送指向该变量的指针。这样,您每次都会向不同的对象发送指针。试试这个:

for i := 0; i < 10; i++ {
    fmt.Printf("sending: %v\n", i)
    iCopy := i
    q2 <- &iCopy
}

答案 1 :(得分:4)

the Channels section of Effective Go中介绍了此问题。这是一个简短的摘录,变量名称已更改为与您的代码匹配:

  

错误是在Go for循环中,循环变量会重复用于每次迭代,因此i变量在所有goroutine中共享。那不是我们想要的。我们需要确保i对于每个goroutine都是唯一的。

继续描述两种解决方案:

  1. i的值作为参数传递给goroutine
  2. 中的函数
  3. 在循环中创建一个新变量并使用该变量
  4. 由于您的goroutine是在循环之外启动的,因此只有#2适用于您的代码。