×Ô¶¨Òåhttp·þÎñ+Öмä¼þ°¸Àý

package main

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

// MiddlewareFunc Öмä¼þµÄº¯Êý
type MiddlewareFunc func(ctx context.Context, req interface{}) (resp interface{}, err error)

// Middleware ¹¹½¨Öмä¼þº¯ÊýʹÓÃ
type Middleware func(MiddlewareFunc) MiddlewareFunc

// LogFile LogFile
type LogFile struct {
	logName string
}

// buildMiddleWare ¹¹½¨Ò»¸öÖмä¼þº¯Êý
func buildMiddleWare(handle MiddlewareFunc) MiddlewareFunc {
	var chain []Middleware
	var LogFiler LogFile
	chain = append(chain, frontHook)
	chain = append(chain, newPrintLog(LogFiler))

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

// buildChain ¹¹½¨Á´
/**
°ÑÖмä¼þµÄÊý×é¹¹½¨³É¸öÁ´
×îÏÈÖ´ÐеÄÔÚ×îÍâ²ã
nextÖ´ÐÐÏÂÒ»¸ö  Ö´ÐÐ˳ÐòºÍÊý×é˳ÐòÏà·´
*/
func buildChain(chain []Middleware) Middleware {
	return func(next MiddlewareFunc) MiddlewareFunc {
		for i := len(chain) - 1; i >= 0; i-- {
			next = chain[i](next)
		}
		return next
	}
}

// frontHook Ö´ÐÐÇ°ÖÃÖмä¼þ
func frontHook(next MiddlewareFunc) MiddlewareFunc {
	return func(ctx context.Context, req interface{}) (resp interface{}, err error) {
		fmt.Println("Ö´ÐÐÇ°ÖÃÖмä¼þ")
		startTime := time.Now().UnixNano() / 1e6
		resp, err = next(ctx, req) // Ö´ÐÐÏÂÒ»¸ö
		if err != nil {
			log.Fatal(err)
			return
		}
		endTime := time.Now().UnixNano() / 1e6
		fmt.Println("º¯ÊýÖ´ÐÐʱ¼äΪ:", endTime-startTime)
		return
	}
}

// newPrintLog ´òÓ¡ÈÕÖ¾Öмä¼þ
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, req)
			resp, err = next(ctx, req)
			return
		}
	}
}

// customHandler ×Ô¶¨Ò庯Êý
func customHandler(ctx context.Context, req interface{}) (resp interface{}, err error) {
	resp = req
	fmt.Println("customHandle")
	time.Sleep(time.Second)
	return
}

// router router
func router(writer http.ResponseWriter, request *http.Request) {

	MiddlewareFunction := buildMiddleWare(customHandler)
	_, err := MiddlewareFunction(context.Background(), request)
	if err != nil {
		log.Fatal(err)
		return
	}

	_, _ = writer.Write([]byte("<h1 style='color:red;'>2023ˮһÌù</h1>"))

}

func main() {

	//×¢²á·ÓÉ
	http.HandleFunc("/", router)
	//½¨Á¢¼àÌý
	err := http.ListenAndServe(fmt.Sprintf("127.0.0.1:%d", 8089), nil)

	if err != nil {
		log.Fatal(err)
		return
	}

}