Go语言当资源超出了周围函数的作用域时，如何延迟资源清理？

axzmvihb 于 2023-10-14 发布在 Go

关注(0)|答案(2)|浏览(124)

让我们以this piece of code为例，它使日志记录器写入本地文件而不是标准输出：

f, err := os.OpenFile("filename", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
if err != nil {
        log.Fatal(err)
}   
defer f.Close()
log.SetOutput(f)

作者将这段代码直接放入main()函数中，使其按预期工作。但是，如果我想把这段代码放到一个main()可以调用的专用函数中，那么它将不再工作，因为f.Close()将在日志记录器使用之前被调用。
例如（如果上面的代码现在在一个名为logToFile()的函数中）：

main() {
   logToFile()
   log.Print("I'm going to end up in stdout\n")
}

this是否可以移动到它自己的函数中，并仍然按预期工作？

我在打开/关闭数据库连接时也遇到过同样的情况。看起来唯一的方法是在main()内部完成这两件事，但我认为如果我们可以将代码划分为函数，代码看起来会更清晰，更像SoC。这是围棋中的禁忌吗

来源：https://stackoverflow.com/questions/68693616/how-to-defer-resource-cleanup-when-that-resource-outlives-the-scope-of-the-surro

2条答案

按热度按时间

dsf9zpds1#

你在找这样的东西吗？

type closerFunc func() error
func logToFile(path string) closerFunc {
    f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
    if err != nil {
        log.Fatal(err)
    }   
    log.SetOutput(f)
    return func() error {
        return f.Close()
    }
}

用途：

func main() {
    closerFn := logToFile("filename")
    defer closerFn()
    log.Print("logs to file\n")
}

展开查看全部

赞(0）回复(0）举报 2023-10-14

t8e9dugd2#

一种选择是使用continuation-passing style，将在defer块中执行的代码作为显式参数传递：

func withLogToFile(filename string, body func()) {
    f, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
    if err != nil {
        log.Fatal(err)
    }
    prevOut := log.Writer()
    log.SetOutput(f)
    defer func() {
        log.SetOutput(prevOut)
        if err := f.Close(); err != nil {
            log.Fatal(err)
        }
    }()
    body()
}

然后调用站点变为：

func main() {
    withLogToFile(filename, func() {
        log.Print("I'm going to end up in ", filename)
    })
}

（https://play.golang.org/p/ebCvtzufU5U）

展开查看全部

赞(0）回复(0）举报 2023-10-14

我来回答

Go语言当资源超出了周围函数的作用域时，如何延迟资源清理？

2条答案

相关问题

热门标签

最新问答

Go语言 当资源超出了周围函数的作用域时，如何延迟资源清理？

2条答案

相关问题

热门标签

最新问答

Go语言当资源超出了周围函数的作用域时，如何延迟资源清理？