定时器陷阱：用select + timer，你可能正在制造内存泄漏！

后端

2023-11-07 11:32:13

在 Go 中，定时器是一个非常有用的工具，可以让我们在指定的时间后执行一些任务。定时器通常与 select 一起使用，可以实现同时等待多个事件。但是，如果我们不注意，很容易造成内存泄漏。

内存泄漏的原理

内存泄漏是指程序中存在着一些对象，这些对象不再被任何变量引用，但是却无法被垃圾回收器回收。这会导致程序的内存使用量不断增加，最终可能导致程序崩溃。

在 Go 中，定时器对象也是一种引用类型，因此也可能造成内存泄漏。如果我们创建了一个定时器，但是没有将其添加到 select 中，那么这个定时器就不会被垃圾回收器回收，从而导致内存泄漏。

如何避免内存泄漏

避免定时器内存泄漏的方法很简单，就是确保定时器始终被添加到 select 中。如果我们不再需要这个定时器了，那么我们应该将其从 select 中移除，这样它就会被垃圾回收器回收。

另外，我们还应该注意不要在定时器中创建新的对象，因为这些对象也会造成内存泄漏。

举个例子

下面是一个使用 select 和定时器来实现超时机制的例子：

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    timer := time.NewTimer(5 * time.Second)
    defer timer.Stop()

    select {
    case <-ctx.Done():
        fmt.Println("Timeout")
    case <-timer.C:
        fmt.Println("Timer fired")
    }
}