函数的内存管理通过垃圾回收器 gc 完成，它自动释放不再使用的对象。测试用例可验证函数是否正确释放了内存，可使用 runtime.gc 强制进行内存回收。

Go 语言函数内存管理测试用例

在 Go 语言中，函数的内存管理是通过垃圾回收器 (GC) 来完成的。GC 会自动管理内存，释放不再使用的对象。在某些情况下，了解 GC 如何管理函数内存对于优化代码性能至关重要。

本篇文章将通过一个实战案例介绍如何编写测试用例来测试函数的内存管理。

立即学习“go语言免费学习笔记（深入）”；

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

实战案例

以下是一个简单的 Go 函数，用于计算斐波那契数列的第 n 项：

func fibonacci(n int) int {
    if n <= 1 {
        return n
    }
    return fibonacci(n-1) + fibonacci(n-2)
}

我们想要编写一个测试用例来验证该函数是否正确释放了内存。为此，我们可以使用 Go 内置的 runtime.GC 函数强制进行内存回收：

package main

import (
    "log"
    "runtime"
    "testing"
)

func fibonacci(n int) int {
    if n <= 1 {
        return n
    }
    return fibonacci(n-1) + fibonacci(n-2)
}

func TestFibonacciMemoryManagement(t *testing.T) {
    var allocatedBytes uint64

    // 记录调用函数前的已分配字节数
    runtime.ReadMemStats(&stats)
    allocatedBytes = stats.TotalAlloc

    // 调用函数并计算斐波那契数
    fibonacci(40)

    // 运行 GC 以释放函数创建的对象
    runtime.GC()

    // 再次记录已分配字节数
    runtime.ReadMemStats(&stats)

    // 检查已分配字节数是否恢复到调用函数前的值
    if stats.TotalAlloc != allocatedBytes {
        t.Errorf("内存泄露，已分配字节数未恢复到调用函数前的值")
    }
}

在该测试用例中，我们将记录调用 fibonacci 函数前后的已分配字节数。如果函数正确释放了内存，则 GC 运行后已分配字节数应当恢复到调用函数前的值。否则，就会出现内存泄露。