最近一段时间,笔者想在Golang代码里面使用拦截器,但在github上却没有找到相关的库,于是就有了自己实现一个拦截器的想法。
Golang没有虚拟机,对反射的支持比Java弱很多,所以不能照搬java的实现方式。就在笔者决定彻底放弃使用动态代理实现拦截器时,突然来了灵感,并且一口气完成了实现。
本文结合代码介绍Golang拦截器的一种实现,希望能给现在或将来想用拦截器的Gopher一些思路。
产品代码
我们简单模拟一下产品代码:
有一个Account接口,声明了方法Query和Update,分别用于查询帐户和更新帐户
类AccountImpl实现了Account接口
有一个简单工厂New,用于创建一个Account对象
有一个Account接口,声明了方法Query和Update,分别用于查询帐户和更新帐户
类AccountImpl实现了Account接口
有一个简单工厂New,用于创建一个Account对象
代码如下所示:
package accountimport ( "fmt")type Account interface { Query(id string) int Update(id string, value int)}type AccountImpl struct { Name string Id string Value int}func (a *AccountImpl) Query(_ string) int { fmt.Println("AccountImpl.Query") return 100}func (a *AccountImpl) Update(_ string, _ int) { fmt.Println("AccountImpl.Update")}func New(name, id string) Account { a := &AccountImpl{name, id, 100} return a}
我们写一个main函数:
package mainimport ( "interceptor/account")func main() { id := "100111" a := account.New("ZhangSan", id) a.Query(id) a.Update(id, 500)}
运行程序:
$ go run main.goAccountImpl.QueryAccountImpl.Update
静态代理
静态代理的类图很简单,如下所示:
我们在产品代码之外实现一下Proxy,如下所示:
package proxyimport ( "interceptor/account" "fmt")type Proxy struct { Account account.Account}func (p *Proxy) Query(id string) int { fmt.Println("Proxy.Query begin") value := p.Account.Query(id) fmt.Println("Proxy.Query end") return value}func (p *Proxy) Update(id string, value int) { fmt.Println("Proxy.Update begin") p.Account.Update(id, value) fmt.Println("Proxy.Update end")}
Account对象跳转到Proxy
若要将Account对象从AccountImpl跳转到Proxy,则需要使用Monkey框架的Patch接口。
我们在Proxy所在的文件中增加init函数来完成跳转:
package proxyimport ( "interceptor/account" "fmt" "github.com/bouk/monkey")...func init() { monkey.Patch(account.New, func(name, id string) account.Account { a := &account.AccountImpl{name, id, 100} p := &Proxy{a} return p })}
注:因为该代码不是测试代码,所以我们在import monkey时,没有在前面加点号"."。
在main函数所在文件的import语句中增加一行代码(第二行):
import ( "interceptor/account" _ "interceptor/proxy")
运行程序:
$ go run main.goAccountImpl.QueryAccountImpl.Update
和期望不符,难道是Monkey的Patch没有生效吗?
回顾上一篇文章《Monkey框架使用指南》,Monkey有inline函数的缺陷,我们的解决方案是通过命令行参数-gcflags=-l禁止inline。
重新运行程序:
$ go run -gcflags=-l main.goProxy.Query beginAccountImpl.QueryProxy.Query endProxy.Update beginAccountImpl.UpdateProxy.Update end
OK!完全符合期望,而且拦截器对产品代码零入寝,仅需在main函数所在文件的import语句中增加一行代码。
小结
本文结合代码给出了Golang拦截器的一种实现,即“静态代理模式 + Monkey框架”。从过程中可以看出,要拦截的方法必须是接口声明的,而且有一个简单工厂用于创建该接口的对象。
说明:对于不满足本方法约束的框架,如果想对其入口出口消息进行拦截,则需要具体问题具体分析,比如Beego框架可以借助filter功能来实现拦截器。