5

这篇文章指出：“defer 语句将函数调用推送到列表中。” 我想知道是否可以从程序中的另一个位置访问该列表中的元素然后调用它们？我可以多次调用它们吗？我假设我引用了具有延迟行为的函数（如果有帮助的话）。

所以，这是我想做的一个简短的例子：

func main {
    doStuff = func() {
        // open database connections
        // write temporary files
        // etc...

        defer func() {
            // close database connections
            // delete temporary files
            // etc...
        }()
    }

    AwesomeApplication(doStuff)
}

func AwesomeApplication(doStuff func()) {
    // Now, can I get a reference to the defer function within `doStuff`?
    // No, I can't just define the defer function somewhere an pass it
    // with `doStuff`.  Think of this as a curiosity I want to satisfy,
    // not a real use case.
}
4

1 回答 1

11

存储调用的“列表”defer完全是特定于实现的，因此您没有可靠的方法来访问此列表。1 , 2 *g 编译器系列的实现细节（虽然有点旧）可以在 Russ Cox 的研究博客中找到。

延迟函数与当前的 goroutine ( g->Defer) 相关联，并且（在 *g 系列的情况下）由当前堆栈指针标识。如果当前堆栈帧与存储在最顶层Defer条目中的堆栈帧匹配，则调用此函数。

有了这些知识，就可以使用 cgo 访问延迟函数列表。你得知道

  • 当前堆栈指针
  • 函数的地址
  • 当前的 goroutine

但是，我不建议使用这个。您描述的用例的一般解决方案是具有这样的功能：

func setupRoutines() (setUp, tearDown func()) {
    // store db connection object and such

    return func() { /* connect db and such */ }, func() { /* close db and such */ }
}

然后，您可以在您的代码中共享该tearDown函数，该函数将使用defer. 这样，您仍然可以获得所有数据库连接和此类本地连接的好处，但您可以共享初始化/断开连接功能。

小提琴玩

如果你真的对玩unsafe和 C 感兴趣，可以使用以下代码作为模板。

检查/runtime.c：

// +build gc
#include <runtime.h>

void ·FirstDeferred(void* foo) {
    foo = g->defer->fn;

    FLUSH(&foo);
}

检查/检查.go

package inspect

import "unsafe"

func FirstDeferred() unsafe.Pointer

defer.go

package main

import "defer/inspect"

func f(a, b int) {
    println("deferred f(", a, b, ")")
}

func main() {
    defer f(1, 2)
    println( inspect.FirstDeferred() )
}

这段代码（基于this）让你可以访问当前的 goroutine ( g) 以及defer它的属性。因此，您应该能够访问指向函数的指针并将其包装FuncVal并返回。

于 2013-08-26T23:40:56.407 回答