时间戳

首页 > 解决方案 > 我什么时候应该把对象放回sync.Pool

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

问题描述

我通过 goroutine 之间的通道传递一个大对象，一个用于发布的 goroutine，许多 goroutine 订阅并等待消息，它是如此频繁，以至于我想用它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()
}

问题是从池中借来的对象可以被多个goroutine访问，所以如果一个goroutine在使用后放回池中，可能会被其他goroutine修改。我不知道对象何时被所有 goroutine 处理，因此我可以将其放回池中。

标签： gomemory-managementchannelpoolgoroutine

解决方案

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

如果您将o在. 因此，在您处理完对象后，将对象放回池中的正确位置是 in 。pubrunrun

推荐阅读