likes
comments
collection
share

利用函数闭包实现一个简单的中间件

作者站长头像
站长
· 阅读数 12

前言

本文是探讨的是"go语言中的中间件"

判断

为了判断你是否有阅读本文的必要,请你在脑海中快速实现一个中间件?

什么叫做中间件?

中间件设计模式是一种常见的软件设计模式,它在许多编程语言和框架中被广泛应用,包括Go、Node.js、Python等。

中间件模式的核心思想是将一系列的处理逻辑组合在一起,形成一个处理链(或称为管道),每个处理逻辑被封装成一个中间件函数。每个中间件函数可以在请求处理的不同阶段添加额外的功能或逻辑,而无需修改原始的处理函数。

中间件模式的优势在于它提供了一种可插拔、可组合的方式来扩展和定制请求处理过程。以下是中间件模式的一些关键特点:

  1. 可插拔性(Pluggability):中间件模式允许将中间件函数插入到处理链中,以添加新的功能或逻辑。这种可插拔性使得我们可以根据需要动态地添加、移除或替换中间件,而无需修改原始的处理函数。

  2. 复用性(Reusability):中间件模式可以将通用的功能或逻辑封装成可复用的中间件函数。这些中间件函数可以在不同的请求处理中重复使用,提高了代码的复用性。

  3. 可组合性(Composability):中间件模式允许将多个中间件函数按照特定的顺序组合在一起,形成一个处理链。每个中间件函数可以在调用原始处理函数之前、之后或两者之间执行自己的逻辑。这种可组合性使得我们可以构建复杂的处理逻辑,将各个中间件函数按需组合,形成一个完整的请求处理流程。

  4. 责任链(Chain of Responsibility):中间件模式的处理链形成了一个责任链结构,每个中间件函数都有机会处理请求或将处理传递给下一个中间件。这种责任链结构可以根据实际需求动态调整中间件的顺序,实现不同的处理流程。

中间件模式在Web开发中得到了广泛应用,特别是在处理HTTP请求的过程中。常见的中间件功能包括身份验证、日志记录、请求解析、缓存、错误处理、压缩等。通过将这些功能封装成中间件函数,并按照需要组合起来,我们可以轻松地构建灵活、可扩展的Web应用程序。

利用函数闭包实现一个简单的中间件

下面我以求两数之和为例,实现一个简单的中间件,可以看如下代码

package main

import "log"

func sum(a, b int) int {
   return a + b
}

func Zh(f func(a, b int) int) func(a, b int) int {
   return func(a, b int) int {
      log.Printf("日志中间件,记录操作数: a: %d b: %d\n", a, b)
      return f(a, b)
   }
}

func main(){

   addWithLogging := Zh(sum)
   
   result := addWithLogging(2, 3)

   result2 := addWithLogging(5, 6)

   fmt.Println(result,result2)

}

利用函数闭包实现一个简单的中间件

上面的代码,可以被认为是一个中间件函数的实现,我们在sum函数执行之前,先进行了日志的记录,并且又不影响原函数的执行,就像添加了一个过程一样,这个中间件函数可以被插入到处理链中,在执行原始函数之前记录操作数的日志。

当然,我们还可以更加优雅,那就是抽象为一个函数类型,并且我们可以发现,Zh中间件处理后的返回值,还可以继续被它处理。

// 优雅地处理
type F func(func(a, b int) int

对了,上面的函数是利用闭包实现的,首先闭包是指函数内部定义的函数可以访问其外部函数的变量。

而在这个例子中,匿名函数func(a, b int) int是在Zh函数内部定义的,并且可以访问外部函数的参数f

具体来说,Zh函数接受一个函数类型的参数f,并返回一个具有相同函数签名的函数。在返回的函数内部,它使用了log.Printf打印日志,并调用了外部函数的参数f

由于返回的函数是在Zh函数内部定义的,它可以访问Zh函数的参数f,即使Zh函数已经执行完毕并返回了。这是因为闭包函数捕获了其外部函数的变量,形成了一个闭包环境,使得返回的函数仍然可以访问外部函数的变量。

当然,我只是想通过这个简单的样例让你知道,什么是中间件

像go中的gin框架中的中间件的使用就很成熟,gin巧妙的把具体实现过程进行了隐藏,让你只要写好一个函数,然后插入使用即可,非常优雅,而本文的样例,其实是把隐藏的过程进行了展示,详情请看gin的源码。

转载自:https://juejin.cn/post/7284261686609018936
评论
请登录