Go语言中的装饰器模式：优雅增强功能，提升代码可维护性

一、GoLang装饰器

1、简单timer装饰器

package main

import (
    "fmt"
    "time"
)

func timer(fn func()) func() {
    return func() {
        startTime := time.Now().Unix()
        fn()
        endTime := time.Now().Unix()
        fmt.PrintIn("运行时间：",endTime - startTime)
    }
}

func testFunc() {
    fmt.PrintIn("运行 testFunc")
    time.Sleep(time.Second * 2)
}

func main() {
    testFunc := timer(testFunc)
    testFunc()
}
/*
运行 testFunc
运行时间： 2
 */

2、项目中认证实现

package main

import (
	"fmt"
	"log"
	"net/http"
)

// DecoratorHandler 是一个用于包装 http.HandlerFunc 的装饰器类型
type DecoratorHandler func(http.HandlerFunc) http.HandlerFunc

// MiddlewareHandlerFunc 将多个 DecoratorHandler 应用于给定的 http.HandlerFunc，并返回新的 http.HandlerFunc。
// 这个函数的目的是创建一个中间件链，每个中间件都能够修改或验证请求。
func MiddlewareHandlerFunc(hp http.HandlerFunc, decors ...DecoratorHandler) http.HandlerFunc {
	// 遍历所有的 DecoratorHandler，并将其应用于 hp
	for _, fn := range decors {
		dp := fn
		hp = dp(hp) // dp(hp) => VerifyHeader(Pong)
	}
	return hp
}

// VerifyHeader 是一个装饰器函数，用于验证请求中的 token 是否为空。
// 如果 token 为空，将返回错误响应；否则，将调用下一个处理函数。
func VerifyHeader(h http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		token := r.URL.Query().Get("token")
		if token == "" {
			fmt.Fprintf(w, r.URL.Path+" VerifyHeader:token不能为空")
			return
		}
		h(w, r)
	}
}

// VerifyHeader2 是另一个装饰器函数，用于验证请求中的 token 是否为 "mimi"。
// 如果 token 不为 "mimi"，将返回错误响应；否则，将调用下一个处理函数。
func VerifyHeader2(h http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		token := r.URL.Query().Get("token")
		if token != "mimi" {
			fmt.Fprintf(w, r.URL.Path+" VerifyHeader:token必须为：mimi")
			return
		}
		h(w, r)
	}
}

// Pong 是一个简单的 http.HandlerFunc，用于返回成功响应。
func Pong(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, r.URL.Path+"--> 验证通过，返回数据")
	return
}

func main() {
	// 在 "/test" 路径上注册 MiddlewareHandlerFunc，应用 VerifyHeader 和 VerifyHeader2 装饰器
	// http://127.0.0.1:8080/test?token=mimi
	http.HandleFunc("/test", MiddlewareHandlerFunc(Pong, VerifyHeader, VerifyHeader2))

	// 启动 HTTP 服务器，监听在 8080 端口
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}