Golang 实现 monkey patch

七秒钟回忆待续 · · 5734 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

背景:项目进行 mock 操作,选择了 monkey patch 的方式。

在 github 上找了一个开源项目 https://github.com/bouk/monkey

使用如下:

下载go get github.com/bouk/monkey

package github.com/bouk/monkey: code in directory /xxxx/src/github.com/bouk/monkey expects import "bou.ke/monkey"

查看它的源码:github.com/bouk/monkey/monkey.go 第一行显示 package monkey // import "bou.ke/monkey"
正常的 import 方式为 import "github.com/bouk/monkey", 源码指定了 import 方式,
因此需要:mv github.com/bouk bou.ke

下载完后使用结果:

func foo1(s string) string {
    return fmt.Sprintf("i am %s, calling function foo1", s)
}
func foo2(s string) string {
    return fmt.Sprintf("i am %s, calling function foo2", s)
}

// result
fmt.Println(foo1("apple"))  // 输出:i am apple, calling function foo1
fmt.Println(foo2("pear")) // 输出:i am pear, calling function foo2

结果是符合预期的,场景是有时候我需要对函数 mock 操作,进行测试(例如 mock 掉数据库操作),代码如下:

func foo1(s string) string {
    return fmt.Sprintf("i am %s, calling function foo1", s)
}

func foo2(s string) string {
    return fmt.Sprintf("i am %s, calling function foo2", s)
}
monkey.Patch(foo1, foo2)
fmt.Println(foo1("apple"))  //输出:i am apple, calling function foo2

这也是符合预期的,因为 foo1 函数被 foo2 给替换了。

下面看另外的例子:

func foo3() bool {
    return true
}

func foo4() bool {
    return false
}
monkey.Patch(foo3, foo4)
fmt.Println(foo3())  //实际输出:true

问题是我们期待的结果是应该是输出 false,因为 foo3 进行了 monkey patch 操作。看一下说明:

Monkey sometimes fails to patch a function if inlining is enabled. Try running your tests with inlining disabled, for example: go test -gcflags=-l. The same command line argument can also be used for build.

大概原因就是此 monkey patch 方式对 inline function 无效,因此需要 disable inlining ,添加参数 -gcflags=-l, 因为上述的例子执行 go run -gcflags=-l main.go 得到符合预期的和正确的输出结果 false

上面是使用 https://github.com/bouk/monkey 的一些心得。

: 开源项目的原作者实现的原理: https://bou.ke/blog/monkey-patching-in-go/


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:七秒钟回忆待续

查看原文:Golang 实现 monkey patch

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

5734 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传