中介軟體是什麼

開發者在處理請求的過程中，加入使用者自己的鉤子（Hook）函式。這個鉤子函式就叫中介軟體，中介軟體適合處理一些公共的業務邏輯，比如登入認證、許可權校驗、資料分頁、記錄日誌、耗時統計等。

程式碼實現

package main

import (
	"context"
	"fmt"
	"time"
)

// 中介軟體的函式
type MiddlewareFunc func(ctx context.Context, req interface{}) (resp interface{}, err error)

// 構建中介軟體函式使用
type Middleware func(MiddlewareFunc) MiddlewareFunc

type LogFile struct {
	logName string
}

// 構建一箇中間件函式 
func buildMiddleWare(handle MiddlewareFunc) MiddlewareFunc {
	var chain []Middleware
	var LogFiler LogFile
	chain = append(chain,ExeaTime)
	chain = append(chain,NewPrintLOg(LogFiler))
	chain = append(chain,judgeReq)

	middle := buildChain(chain)
	return middle(handle)
}

// 把中介軟體的陣列構建成個鏈 
// 最先執行的在最外層
// next執行下一個 下面的為執行為請求函式handle之後執行 執行順序和陣列順序相反
func buildChain(chain []Middleware) Middleware {
	return func(next MiddlewareFunc) MiddlewareFunc {
		for i := len(chain) - 1; i >= 0; i-- {
			next = chain[i](next)
		}
		return next
	}
}

// 執行時間中介軟體
func ExeaTime(next MiddlewareFunc) MiddlewareFunc {
	return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
		fmt.Println("執行時間中介軟體")
		startTime := time.Now().UnixNano()
		resp, err = next(ctx, req) // 執行下一個
		if err != nil {
			fmt.Println("執行函式失敗")
			return
		}
		endTime := time.Now().UnixNano()
		fmt.Println("函式執行時間為:", endTime-startTime)
		return
	}
}

// 列印日誌中介軟體
func NewPrintLOg(LogFiler LogFile) Middleware {
	return func(next MiddlewareFunc) MiddlewareFunc {
		return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
			fmt.Println("列印日誌中介軟體")
			fmt.Println(LogFiler.logName)
			resp, err = next(ctx, req)
			return
		}
	}
}

// 判斷req是否為空中介軟體
func judgeReq(next MiddlewareFunc) MiddlewareFunc {
	return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
		fmt.Println("判斷req是否為空中介軟體")
		if req == "" {
			fmt.Println("req為空")
			return
		}
		resp, err = next(ctx, req)
		return
	}
}

// 請求函式
func Handle(ctx context.Context, req interface{}) (resp interface{}, err error) {
	resp = req
	fmt.Println("Handle")
	time.Sleep(time.Second)
	return
}

func main() {
	MiddlewareFunction := buildMiddleWare(Handle) // 吧請求函式穿進去 並形成個帶中介軟體的函式
	resp,err := MiddlewareFunction(context.TODO(),"zhy")	// 執行函式
	if err != nil {
		fmt.Println("main err:",err)
		return
	}
	fmt.Println("main resp:",resp)
}
 

執行結果:

執行時間中介軟體
列印日誌中介軟體

判斷req是否為空中介軟體
Handle
函式執行時間為: 1001447800
main resp: zhy

執行順序