快捷導(dǎo)航

一文詳解Golang的中間件設(shè)計(jì)模式

更新時(shí)間：2023年03月06日 14:49:49 作者：白象孫國(guó)帥

最近在看一些rpc框架的使用原理和源碼的時(shí)候，對(duì)中間件的實(shí)現(xiàn)非常感興趣，所以這篇文章就來(lái)和大家聊聊Golang的中間件設(shè)計(jì)模式，希望對(duì)大家有所幫助

背景

記錄一下自己在go開(kāi)發(fā)和學(xué)習(xí)上的一些筆記

最近在看一些rpc框架的使用原理和源碼的時(shí)候，對(duì)中間件的實(shí)現(xiàn)非常感興趣，然后也看了一下grpc的中間件的用法，也看了別的框架的中間件的設(shè)計(jì)，感覺(jué)grpc的還算是比較容易弄懂，于是記錄一下這個(gè)常用中間件的實(shí)現(xiàn)的一個(gè)原理的demo(吐槽一下其他的rpc框架分為inbound和outbound的middleware感覺(jué)好像有點(diǎn)復(fù)雜化了，所以我也不知道哪種設(shè)計(jì)會(huì)比較好，樓主是java出身，所以對(duì)反射走aop的那種模式比較熟悉，對(duì)鏈?zhǔn)秸{(diào)用的middleware有點(diǎn)不太熟悉，當(dāng)然現(xiàn)在已經(jīng)熟悉了)

Demo

所以接下來(lái)我們就來(lái)看看demo吧首先先定義好中間件的類(lèi)型，這里我就簡(jiǎn)單定義為以下的格式

type middleware func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error)

ctx: 協(xié)程間通信帶著
req: 請(qǐng)求的格式，這里圖簡(jiǎn)便，直接interface{}類(lèi)型
resp: 同req
err: error

handler: endpoint類(lèi)型，真正用來(lái)發(fā)起請(qǐng)求的一個(gè)處理方法或者是經(jīng)過(guò)N層中間件包裝的后的發(fā)起請(qǐng)求的處理方法

type endpoint func(ctx context.Context, req interface{}) (resp interface{}, err error)
//ctx: 協(xié)程間通信帶著 

//req: 請(qǐng)求的格式，這里圖簡(jiǎn)便，直接interface{}類(lèi)型
//resp: 同req  
//err: error

然后既然我們要將上方的endpoint進(jìn)行包裝然后產(chǎn)生一個(gè)新的endpoint那么也就是需要一個(gè)函數(shù)去做一步的事情，input是endpoint,ouput也是endpoint

type warp func(endpoint) endpoint //就是這個(gè)warp函數(shù)

然后我們通過(guò)每次調(diào)用這個(gè)warp的定義去生成一個(gè)新的endpoint就可以產(chǎn)生一個(gè)類(lèi)似于dfs鏈?zhǔn)秸{(diào)用的一個(gè)中間件的過(guò)程，因?yàn)閷?huì)一層套一層的endpoint下去，然后當(dāng)最后一層有返回了以后就可以接著返回了，然后不斷的彈?；厝プ铋_(kāi)始的地方，因?yàn)槲覀冎虚g件的實(shí)現(xiàn)必然是要調(diào)用handler的

func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error) {
		fmt.Printf("before1\n")
		resp, err = handler(ctx, req)
		fmt.Printf("end1\n")
		return
}

// handler0
var handler endpoint = func(ctx context.Context, req interface{}) (resp interface{}, err error) {
	fmt.Printf("make msg\n")
	return nil, nil
} 
// middleware
var md Middleware = func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error) {
		fmt.Printf("before1\n")
		resp, err = handler(ctx, req)
		fmt.Printf("end1\n")
		return
}
// warp ->> handler1
handler = warp(func(e endpoint) endpoint {
		return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
				return md(ctx, req, e)
			}
		})(handler)

驗(yàn)證結(jié)論

package main

import (
	"context"
	"fmt"
)

type endpoint func(ctx context.Context, req interface{}) (resp interface{}, err error)

type middleware func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error)

type warp func(endpoint) endpoint

func main() {
	mds := []middleware{}
	mds = append(mds, func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error) {
		fmt.Printf("before1\n")
		resp, err = handler(ctx, req)
		fmt.Printf("end1\n")
		return
	})
	mds = append(mds, func(ctx context.Context, req interface{}, handler endpoint) (resp interface{}, err error) {
		fmt.Printf("before2\n")
		resp, err = handler(ctx, req)
		fmt.Printf("end2\n")
		return
	})
	var handler endpoint = func(ctx context.Context, req interface{}) (resp interface{}, err error) {
		fmt.Printf("make msg\n")
		return nil, nil
	}
	for i := len(mds) - 1; i >= 0; i-- {
		handler = warp(func(e endpoint) endpoint {
			// 由于go的機(jī)制問(wèn)題如果不用tmp去存下當(dāng)前的i，那么mds[i]就會(huì)取最終的那一個(gè)，就會(huì)溢出，所以在return前先保存一下i的量，然后每一個(gè)stack去存的變量就是對(duì)的
			cur := i
			return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
				return mds[cur](ctx, req, e)
			}
		})(handler)
	}
	resp, err := handler(context.Background(), "ster")
	if resp != nil && err != nil {
		return
	}
}

結(jié)果是