go - 如何实现 Python functools.wraps 等效？

Question

我知道我可以通过返回函数在 Go 中包装函数，如何在 Go 中实现等效的 Python functools.wraps？如何将属性附加到 Go 中的函数？就像下面的 Python 代码。

from functools import wraps

def d(f):
    def wrapper(*args):
        f(*args)

    return wrapper

def d_wraps(f):
    @wraps(f)
    def wrapper(*args):
        f(*args)

    return wrapper

@d
def f(a=''):
    print a

@d_wraps
def g(a=''):
    print a

if __name__ == '__main__':
    print 'function name: ', f.__name__
    f('abc')

    print 'function name: ', g.__name__
    g('abc')

d 不改变函数名，d_wraps 改变函数名。结果是

function name:  wrapper
abc
function name:  g
abc

我想在运行时使用包装函数作为键。所以我想在包裹后保持函数名称不变。如何在 Go 中实现 d_wraps 所做的工作？

Answer 1

How to attach attribute to functions in Go?

不可以，您不能在 Go 中将属性附加到函数。

How to implement the job that d_wraps does in Go?

您可以实现另一个函数调用您的函数，并使用新名称调用该函数。

package main

import "fmt"

func main() {
   g_warp("abc")
}

func g(a string){
   fmt.Println(a);
}

func g_warp(a string){
   g(a+"_mysuffix");
}

如果要改变函数内容但使用相同的函数名，可以使用全局函数变量:

package main

import (
    "fmt"
)

func main() {
    g = gV2
    g("hello")
}

var g = gV1;

func gV1(a string){
    fmt.Println(a)
}
func gV2(a string){
    gV1(a+"_suffix")
}

如果你有很多函数有相同的包装逻辑，你可以传入原始函数并返回新函数:

package main

import (
    "fmt"
)

func main() {
    g = wrap(g)
    g("hello")
    f = wrap(f)
    f("hello")
}

var g = gV1;

func gV1(a string){
    fmt.Println(a)
}
var f = fV1;

func fV1(a string){
    fmt.Println(a+" "+a)
}
func wrap(originFn func(a string)) func(a string){
    return func(a string){
         originFn(a+"_suffix")
    }
}