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

package middleware

内容来自zvvq

import (

"fmt" zvvq

"net/http" 内容来自zvvq

) zvvq好，好zvvq

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

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

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

内容来自samhan

fmt.Println("Before request")

内容来自zvvq

next.ServeHTTP(w, r) 内容来自samhan

fmt.Println("After request")

内容来自samhan666

}) zvvq.cn

} 本文来自zvvq

mux := http.NewServeMux() 本文来自zvvq

mux.Use(middleware.ExampleMiddleware)

package middleware 内容来自samhan666

import (

"log"

内容来自samhan

"net/http"

"runtime/debug"

)

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

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

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

defer func() {

copyright zvvq

if err := recover(); err != nil { 内容来自zvvq

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

debug.PrintStack()

内容来自samhan

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

}

copyright zvvq

}() zvvq

next.ServeHTTP(w, r)

内容来自zvvq

})

copyright zvvq

}

package middleware copyright zvvq

import ( zvvq好，好zvvq

"net/http"

内容来自zvvq，别采集哟

"time" 内容来自samhan

) 本文来自zvvq

// RateLimitMiddleware 限制请求速率。

func RateLimitMiddleware(next http.Handler) http.Handler { 本文来自zvvq

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

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

select {

zvvq.cn

case <-bucket: 内容来自zvvq，别采集哟

next.ServeHTTP(w, r)

default: zvvq

http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests) 内容来自samhan666

} 内容来自samhan666

bucket <- time.Now()

})

内容来自samhan666

}

内容来自zvvq