golang编程技巧：利用GC机制优雅地关闭协程，避免内存泄漏

最新推荐文章于 2026-04-22 07:14:43 发布

原创最新推荐文章于 2026-04-22 07:14:43 发布 · 3.4k 阅读

8 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#golang #技术 #编程 #Gc

本文探讨了Go语言中如何利用GC机制优雅地关闭协程，防止内存泄漏。通过示例解释了在创建协程时可能会遇到的内存泄漏问题，特别是封装定时清理的TtlMap库。提出了通过chan关闭协程和使用SetFinalizer在对象被GC回收前执行清理的方法。最后解答了关于直接给TtlMap设置SetFinalizer以及Wrapper对象是否会被回收的问题。

go语言有完备的垃圾回收（gc）机制，但仍然可能出现内存泄漏的情况，如下：

开启一个协程：

func test() {
    ......
    go func(){
        for {
            // do sth..
        }
    }()
    .....
}

如果for循环内没有return或break，即使test()函数结束，该协程也会继续运行，直到程序结束。
如果这不是我们的本意，就发生了协程泄漏。协程泄漏的同时也会导致内存泄漏，因为协程一直在运行，里边用到的内存无法被gc回收。
当然，只要小心一点，我们基本不会犯这种错误。

但还有更隐蔽的情况，比如：封装了一个具有定期清理到期key功能的map库，我们就叫它TtlMap。
TtlMap在初始化时会开启一个协程，负责定期删除到期的key，代码框架如下:

strcut TtlMap {
    ...
}

func New() *TtlMap {
    map := &TtlMap{
        ...
    }

    //启动定期清理协程
    go map.clear()
    return map
}

func (m *TtlMap) clear() {
    for {
        //定期清理...
    }
}

有一定go语言编程经验的人能看得出，这个map库是有问题的：没有提供结束map.clear()协程的方法。
TtlMap作为一个局部变量时，定义它的函数结束后，由于clear()协程还在运行，变量无法被Gc回收，导致内存泄漏。
解决此问题的通用方法是，TtlMap中加一个chan，通过chan发信号来关闭clear协程：

strcut TtlMap {
    ...
    stop chan bool
}

func New() *TtlMap {
    map := &TtlMap{
        ...
    }

    //启动定期清理协程
    go map.clear()
    return map
}

func (m *TtlMap) clear() {
    for {
        select {
        // 关闭
		case <-m.stop:
			return

        //定期清理...
		}
    }
}

func (m *TtlMap) Close() {
    m.stop <- true
}

这么做不太安全，当使用者忘记调用Close()方法关闭map，仍然会导致泄漏。
如何不调用Close，又能让Gc回收呢？可以在初始化时，把ttlmap包一层再返回：

strcut TtlMap {
    ...
    stop chan bool
}

// 包裹定义
struct MapWarpper {
    *TtlMap
}

func New() *MapWarpper {
    map := &TtlMap{
        ...
    }
    go map.clear()

    // 包一层
    mw := &MapWarpper{map}
    
    // 重点在此:设置被回收时操作
    runtime.SetFinalizer(mw, onGarbageCollect)
    return mw
}

func (m *TtlMap) clear() {
    for {
        select {
        // 关闭
		case <-m.stop:
			return

        //定期清理...
		}
    }
}

func onGarbageCollect(m *TtlMap) {
    m.stop <- true
}

SetFinalizer
如果需要在一个对象object被从内存中移除前执行一些特殊操作，比如写日志等，可以通过调用以下方式调用函数来实现
runtime.SetFinalizer(obj, func(obj *typeObj))
在对象被GC进程选中并从内存中移除前，SetFinalizer都不会执行，即使程序正常结束或者发生错误

MapWarpper作为局部变量时，定义它的函数结束后，MapWarpper的生命周期已结束，Gc会将其回收。
Gc回收MapWarpper时执行了onGarbageCollect()函数，将Ttlmap的clear协程关闭，进而将Ttlmap回收。
这样，借助go的Gc机制，我们实现了一个安全的map库，无需担心内存泄漏发生。

问题1：不能直接给ttlmap设置SetFinalizer吗？
答：不能。只要ttlmap.clear()协程一直在运行，Gc就无法选中ttlmap对象执行垃圾回收。
实际上，给ttlmap设置了SetFinalizer后，如果ttlmap.clear()协程没有return，而是阻塞了，Gc就可以将ttlmap回收掉。
问题2：使用了Warpper，却没有给Warpper设置SetFinalizer，Warpper还会被回收吗？
答：经测试，不会被回收。