欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Golang中实现简单的Http Middleware

程序员文章站 2022-03-13 12:12:47
本文主要针对golang的内置库 net/http 做了简单的扩展,通过添加中间件的形式实现了管道(pipeline)模式,这样的好处是各模块之间是低耦合的,符合单一职责原则,可以很灵活的通过中间件的...

本文主要针对golang的内置库 net/http 做了简单的扩展,通过添加中间件的形式实现了管道(pipeline)模式,这样的好处是各模块之间是低耦合的,符合单一职责原则,可以很灵活的通过中间件的形式添加一些功能到管道中,一次请求和响应在管道中的执行过程如下

Golang中实现简单的Http Middleware

首先, 我定义了三个测试的中间件 middleware1,2,3 如下

这里中间件的入参和出参的类型都是 http.handler, 然后在 next.servehttp() 的前后分别输出了 in 和 out.

接下来,定义一个 pipeline 的方法,里面使用嵌套的形式, 使用了上面定义的三个测试的中间件.

然后还需要业务代码,这里我定义了 loginhandler 和 registerhandler 两个方法

最后修改程序的 main 函数, 在 login 接口上使用上面添加过中间件的 pipeline

启动程序后,访问 http://localhost:8080/login, 程序的输出如下,这和本文最上面的管道的流程图是一致的,然后访问 register 接口, 控制台没有输出信息,当然也不会执行任何中间件。

Golang中实现简单的Http Middleware

现在已经实现了中间件的机制,但是,上面添加中间件是用嵌套的方法,这种方式不能说不太优雅,只能说非常的low,接下来我们需要对管道进行优化

首先定义了一个chain 的struct,用来接收添加到管道中的中间件,在 addmiddlewares() 函数中,接收了多个handle, 然后组装到 chain 对象并返回, 接下来调用 then() 函数, 把管道中的中间件和业务的handler 关联起来。在中间件的使用方式上, 这两种方法都是一样的,只需要调用 pipeline() 方法就行了。

本文在go web中简单的实现了中间件的机制,这样带来的好处也是显而易见的,当然社区也有一些成熟的 middleware 组件,包括 gin 一些web框架中也包含了 middleware 相关的功能, 希望对您有用.

到此这篇关于golang中实现简单的http middleware的文章就介绍到这了,更多相关golang实现http middleware内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!