七叶笔记 » golang编程 » 干货|Golang拦截器的一种实现

干货|Golang拦截器的一种实现

前言

说起拦截器,大家一定会想起 Java 语言。

Java里的拦截器是动态拦截Action调用的对象,它提供了一种机制使开发者可以定义在一个action执行的前后执行的代码,也可以在一个action执行前阻止其执行,同时也提供了一种可以提取action中可重用部分的方式。在 AOP (Aspect-Oriented Programming)中拦截器用于在某个方法或字段被访问之前,进行拦截然后在之前或之后加入某些操作。

最近一段时间,笔者想在Golang代码里面使用拦截器,但在 github 上却没有找到相关的库,于是就有了自己实现一个拦截器的想法。

Golang没有 虚拟机 ,对反射的支持比Java弱很多,所以不能照搬java的实现方式。就在笔者决定彻底放弃使用动态代理实现拦截器时,突然来了灵感,并且一口气完成了实现。

本文结合代码介绍Golang拦截器的一种实现,希望能给现在或将来想用拦截器的Gopher一些思路。

产品代码

我们简单模拟一下产品代码:

  • 有一个 Account 接口,声明了方法Query和Update,分别用于查询帐户和更新帐户
  • 类AccountImpl实现了Account接口
  • 有一个简单工厂New,用于创建一个Account对象

代码如下所示:

我们写一个main函数:

运行程序:

静态代理

静态代理的 类图 很简单,如下所示:

我们在产品代码之外实现一下Proxy,如下所示:

Account对象跳转到Proxy

若要将Account对象从AccountImpl跳转到Proxy,则需要使用Monkey框架的Patch接口。

我们在Proxy所在的文件中增加init函数来完成跳转:

注:因为该代码不是测试代码,所以我们在import monkey时,没有在前面加点号”.”。

在main函数所在文件的import语句中增加一行代码(第二行):

运行程序:

和期望不符,难道是Monkey的Patch没有生效吗?

回顾上一篇文章《Monkey框架使用指南》,Monkey有 inline 函数的缺陷,我们的解决方案是通过命令行参数-gcflags=-l禁止inline。

重新运行程序:

OK!完全符合期望,而且拦截器对产品代码零入寝,仅需在main函数所在文件的import语句中增加一行代码。

小结

本文结合代码给出了Golang拦截器的一种实现,即“静态 代理模式 + Monkey框架”。从过程中可以看出,要拦截的方法必须是接口声明的,而且有一个简单工厂用于创建该接口的对象。

说明:对于不满足本方法约束的框架,如果想对其入口出口消息进行拦截,则需要具体问题具体分析,比如Beego框架可以借助 filter 功能来实现拦截器。

相关文章