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对象

代码如下所示:

package account

import (
    "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 main

import (
    "interceptor/account"
)

func main() {
    id := "100111"
    a := account.New("ZhangSan", id)
    a.Query(id)
    a.Update(id, 500)
}

运行程序:

$ go run main.go
AccountImpl.Query
AccountImpl.Update

静态代理

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

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

package proxy

import (
    "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 proxy

import (
    "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.go
AccountImpl.Query
AccountImpl.Update

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

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

重新运行程序:

$ go run -gcflags=-l main.go
Proxy.Query begin
AccountImpl.Query
Proxy.Query end
Proxy.Update begin
AccountImpl.Update
Proxy.Update end

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

小结

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

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

原文发布于微信公众号 - Golang语言社区(Golangweb)

原文发表时间:2017-11-09

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Linux驱动

C-fopen,fwrite,fread,fseek笔记

FILE * fopen(const char * path,const char * mode);

1892
来自专栏IT可乐

mybatis 详解(八)------ 懒加载

  本章我们讲如何通过懒加载来提高mybatis的查询效率。   本章所有代码:http://pan.baidu.com/s/1o8p2Drs 密码:trd6 ...

3509
来自专栏微信公众号:Java团长

彻底解决MySQL中文乱码

mysql是我们项目中非常常用的数据型数据库。但是因为我们需要在数据库保存中文字符,所以经常遇到数据库乱码情况。下面就来介绍一下如何彻底解决数据库中文乱码情况。

2052
来自专栏程序员的SOD蜜

消息服务框架使用案例之--大文件上传(断点续传)功能

消息服务框架使用案例之--大文件上传(断点续传)功能 一、分块上传和断点续传原理 在我们的一个产品应用中,客户需要上传大量的文件到服务器,其中不乏很大的视频文件...

8455
来自专栏JMCui

Hibernate问题浅析

1、什么是SessionFactory?什么是Session?httpsession和hibernate的session的有什么区别?     SessionF...

2713
来自专栏python3

pymysql--插入300万数据

        (2)使用python协程(遇到I/O操作就切换任务,无需等待--提高效率)

1442
来自专栏Java技术分享

高并发分布式系统中生成全局唯一Id汇总

数据在分片时,典型的是分库分表,就有一个全局ID生成的问题。 单纯的生成全局ID并不是什么难题,但是生成的ID通常要满足分片的一些要求:    1 不能有单...

2855
来自专栏猿天地

mongodb java操作语法

之前都是通过spring-data框架来对mongodb操作,当然在很多老的项目中,由于jar包版本的问题,不太好升级到最新的版本,往往会采用mongodb-d...

28310
来自专栏hbbliyong

ShellExecute 启动外部程序 参数详细介绍

ShellExecute的功能是运行一个外部程序(或者是打开一个已注册的文件、打开一个目录、打印一个文件等等),并对外部程序有一定的控制。 目录 1基本简介 2...

50410
来自专栏张善友的专栏

DataDirectory是什么?

asp.net 2.0有一个特殊目录app_data,通常Sql server 2005 express数据文件就放在这个目录,相应的数据库连接串就是: con...

2079

扫码关注云+社区

领取腾讯云代金券