Golang函数指针的陷阱和最佳实践

go 中的函数指针陷阱和最佳实践：陷阱：指针指向不可用函数最佳实践：使用局部变量或闭包捕获函数值。陷阱：修改指针所指向的函数最佳实践：保持函数指针不可修改，在另一个闭包中创建新的函数。实战案例：回调函数例如，使用函数指针创建日志函数，该函数将日志消息和严重级别作为回调函数的参数。

Go 中函数指针的陷阱和最佳实践

在 Go 中，函数指针是一种将函数作为值传递的强大机制。然而，使用函数指针时需要注意一些陷阱：

陷阱 1：函数指针指向不可用函数

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

当函数指针指向不再可用的函数时，就会出现悬空指针错误：

func newFunc() {
    // ...
}

func main() {
    newFuncPtr := newFunc // 将函数 newFunc 赋值给指针 newFuncPtr
    newFunc = nil        // 将 newFunc 设为 nil，使其不可用
    newFuncPtr()        // 这会触发悬空指针错误
}

最佳实践：使用局部变量或闭包捕获函数值

为了避免悬空指针，请将函数值捕获到局部变量或闭包中：

func main() {
    newFunc := newFunc
    newFunc() // 仍然有效，因为它捕获了 newFunc 的副本
}

陷阱 2：修改指针所指向的函数

CoCo

智谱AI推出的首个有记忆的企业自主Agent智能体

下载

对函数指针所指向的函数进行修改可能导致意外的后果：

type API interface {
    Call()
}

func makeAPI() API {
    return func() {
        fmt.Println("Hello")
    }
}

func main() {
    apiPtr := makeAPI()
    apiPtr = func() {
        fmt.Println("Goodbye")
    }
    apiPtr.Call() // 输出 "Goodbye"
}

最佳实践：保持函数指针不可修改

如果需要修改函数行为，请在另一个闭包中创建新的函数：

func main() {
    api := makeAPI()
    api = func(a API) API {
        return func() {
            fmt.Println("Goodbye")
            a.Call()
        }
    }(api)
    api.Call() // 输出 "Hello", 然后是 "Goodbye"
}

实战案例：回调函数

回调函数是一个常见的函数指针示例，它用于在事件发生时通知调用者。

例如，我们可以使用函数指针来构建一个日志函数，该函数将日志消息和严重级别作为回调函数的参数：

import "fmt"
import "log"

type Severity string

const (
    Info    Severity = "INFO"
    Warning Severity = "WARNING"
    Error   Severity = "ERROR"
)

func Log(severity Severity, message string) {
    log.Println(fmt.Sprintf("%s: %s", severity, message))
}

// 获取 severity 为 Info 的日志函数
func InfoLogger() func(string) {
    return func(message string) {
        Log(Info, message)
    }
}

然后，我们可以使用回调函数将日志消息发送到日志函数：

func main() {
    infoLogger := InfoLogger()
    infoLogger("Hello, world!") // 输出：INFO: Hello, world!
}