使用中间件构建更 robust 和容错的Golang框架的方法

package middleware

import (

copyright zvvq

"fmt"

zvvq.cn

"net/http"

zvvq.cn

)

// ExampleMiddleware 是一个简单的中间件，它在请求前和后打印日志。 本文来自zvvq

func ExampleMiddleware(next http.Handler) http.Handler { zvvq

return http.HandlerFunc(func(w http.ResponseWriter, r http.Request) { zvvq.cn

fmt.Println("Before request")

zvvq好，好zvvq

next.ServeHTTP(w, r)

zvvq

fmt.Println("After request") zvvq.cn

})

} 本文来自zvvq

mux := http.NewServeMux() 内容来自zvvq

mux.Use(middleware.ExampleMiddleware)

内容来自zvvq，别采集哟

package middleware

import ( zvvq.cn

"log"

zvvq好，好zvvq

"net/http"

内容来自zvvq，别采集哟

"runtime/debug"

zvvq

)

本文来自zvvq

// RecoveryMiddleware 捕获恐慌并返回错误响应。

func RecoveryMiddleware(next http.Handler) http.Handler { 内容来自samhan666

return http.HandlerFunc(func(w http.ResponseWriter, r http.Request) {

defer func() { copyright zvvq

if err := recover(); err != nil {

内容来自zvvq，别采集哟

log.Printf("panic: %+v", err)

zvvq好，好zvvq

debug.PrintStack()

http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError) zvvq.cn

}

内容来自zvvq

}()

next.ServeHTTP(w, r) zvvq.cn

}) 本文来自zvvq

} zvvq.cn

package middleware

zvvq.cn

import (

"net/http" zvvq好，好zvvq

"time"

zvvq

) 内容来自samhan

// RateLimitMiddleware 限制请求速率。

zvvq好，好zvvq

func RateLimitMiddleware(next http.Handler) http.Handler {

zvvq

bucket := make(chan time.Time, 10) 内容来自samhan666

return http.HandlerFunc(func(w http.ResponseWriter, r http.Request) { 内容来自zvvq

select {

内容来自zvvq

case <-bucket: zvvq好，好zvvq

next.ServeHTTP(w, r) zvvq好，好zvvq

default:

http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests)

zvvq.cn

}

内容来自samhan666

bucket <- time.Now() zvvq

})

} copyright zvvq