go - 我什么时候应该把对象放回 sync.Pool

Question

我在 goroutines 之间通过 channel 传递一个大对象，一个 goroutine 用于发布，许多 goroutines 订阅和等待消息，它是如此频繁，我想使用 sync.Pool 来减少数量分配，示例代码如下:

package main

import (
    "log"
    "sync"
    "time"
)

var pool *sync.Pool

type object struct {
    info string
    // other stuff
}

func initPool() {
    pool = &sync.Pool{
        New: func() interface{} {
            return new(object)
        },
    }
}

var (
    lock     sync.RWMutex
    registry = make(map[string][]chan interface{})
)

func sub(topic string) chan interface{} {
    ch := make(chan interface{}, 10)
    lock.Lock()
    registry[topic] = append(registry[topic], ch)
    lock.Unlock()
    return ch
}

// publish goroutine
func pub() {
    ticker := time.NewTicker(time.Second)
    o := pool.Get().(*object)
    o.info = "new"
    // do something

    for _ = range ticker.C {
        lock.RLock()
        for topic, chs := range registry {
            if o.info != "new" {
                log.Printf("error")
            }
            if topic == "hello" {
                for _, ch := range chs {
                    select {
                    case ch <- o:
                    default:
                    }
                }
            }
        }
        lock.RUnlock()
    }
}

func run(topic string) {
    ch := sub(topic)
    for {
        select {
        case o := <-ch:
            switch o := o.(type) {
            case *object:
                if o.info != "new" {
                    log.Printf("error")
                }

                // do something
                time.Sleep(time.Second)
                o.info = "used"
                pool.Put(o)
            }
        }
    }
}

func main() {
    initPool()

    for i := 0; i <= 100; i++ {
        go run("hello")
    }
    pub()
}

问题是从 pool 借来的对象可以被多个 goroutine 访问，所以如果一个 goroutine 使用后放回 pool，可能会被其他 goroutine 修改。我不知道什么时候所有的 goroutines 都完成了这个对象，所以我可以把它放回池中。

Answer 1

应该在处理完池对象后调用pool.Put()，即它被认为是“垃圾”。

如果您将 o 放回 pub 中的池中，那么循环的下一次迭代可以获得相同的对象并为其之前的属性分配新值它已被 run 处理。因此，在处理完对象后，将对象放回池中的正确位置是 run。