Go语言HTTP服务器高级配置与优化-开发者社区

Go语言HTTP服务器高级配置与优化

HTTP服务器是Web应用的核心组件，Go语言的net/http包提供了强大的HTTP服务器实现。本文将深入探讨Go语言HTTP服务器的高级配置和性能优化技巧。

一、HTTP服务器基础

1.1 基础服务器配置

package main import ( "log" "net/http" "time" ) func main() { server := &http.Server{ Addr: ":8080", Handler: nil, // 使用默认的ServeMux ReadTimeout: 10 * time.Second, WriteTimeout: 10 * time.Second, IdleTimeout: 30 * time.Second, MaxHeaderBytes: 1 << 20, // 1MB } log.Println("Server starting on :8080") log.Fatal(server.ListenAndServe()) }

1.2 自定义Handler

type CustomHandler struct{} func (h *CustomHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") w.WriteHeader(http.StatusOK) w.Write([]byte(`{"message": "Hello, World!"}`)) } func main() { handler := &CustomHandler{} server := &http.Server{ Addr: ":8080", Handler: handler, } server.ListenAndServe() }

二、中间件机制

2.1 基础中间件

type Middleware func(http.Handler) http.Handler func LoggingMiddleware(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { start := time.Now() log.Printf("Started %s %s", r.Method, r.URL.Path) next.ServeHTTP(w, r) duration := time.Since(start) log.Printf("Completed %s %s in %v", r.Method, r.URL.Path, duration) }) } func AuthMiddleware(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { token := r.Header.Get("Authorization") if token == "" { http.Error(w, "Unauthorized", http.StatusUnauthorized) return } // 验证token逻辑 if !validateToken(token) { http.Error(w, "Invalid token", http.StatusForbidden) return } next.ServeHTTP(w, r) }) } func validateToken(token string) bool { // 实际的token验证逻辑 return token == "valid-token" }

2.2 中间件链

func ApplyMiddleware(handler http.Handler, middlewares ...Middleware) http.Handler { for i := len(middlewares) - 1; i >= 0; i-- { handler = middlewares[i](handler) } return handler } func main() { mux := http.NewServeMux() mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("Hello, World!")) }) middlewares := []Middleware{ LoggingMiddleware, AuthMiddleware, } handler := ApplyMiddleware(mux, middlewares...) server := &http.Server{ Addr: ":8080", Handler: handler, } server.ListenAndServe() }

三、路由配置

3.1 基本路由

func main() { mux := http.NewServeMux() mux.HandleFunc("/users", getUsers) mux.HandleFunc("/users/{id}", getUser) mux.HandleFunc("/users/{id}/posts", getUserPosts) server := &http.Server{ Addr: ":8080", Handler: mux, } server.ListenAndServe() } func getUsers(w http.ResponseWriter, r *http.Request) { w.Write([]byte("Get all users")) } func getUser(w http.ResponseWriter, r *http.Request) { id := r.PathValue("id") w.Write([]byte("Get user: " + id)) }

3.2 带参数验证的路由

type Router struct { routes []Route } type Route struct { Method string Pattern string Handler http.HandlerFunc Middlewares []Middleware } func (r *Router) Handle(method, pattern string, handler http.HandlerFunc, middlewares ...Middleware) { route := Route{ Method: method, Pattern: pattern, Handler: handler, Middlewares: middlewares, } r.routes = append(r.routes, route) } func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) { for _, route := range r.routes { if route.Method != req.Method { continue } matched, params := r.match(route.Pattern, req.URL.Path) if matched { // 设置路径参数 ctx := req.Context() for key, value := range params { ctx = context.WithValue(ctx, key, value) } req = req.WithContext(ctx) // 应用中间件 handler := http.HandlerFunc(route.Handler) for i := len(route.Middlewares) - 1; i >= 0; i-- { handler = route.Middlewares[i](handler).(http.HandlerFunc) } handler(w, req) return } } http.NotFound(w, req) }

四、性能优化

4.1 连接复用

func main() { server := &http.Server{ Addr: ":8080", ReadTimeout: 5 * time.Second, WriteTimeout: 10 * time.Second, IdleTimeout: 30 * time.Second, // 启用HTTP/2 TLSConfig: &tls.Config{ NextProtos: []string{"h2", "http/1.1"}, }, } // 使用keep-alive server.SetKeepAlivesEnabled(true) log.Fatal(server.ListenAndServeTLS("cert.pem", "key.pem")) }

4.2 并发处理优化

func main() { // 设置GOMAXPROCS为CPU核心数 runtime.GOMAXPROCS(runtime.NumCPU()) server := &http.Server{ Addr: ":8080", Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { // 使用goroutine处理耗时操作 go processAsync(r) w.WriteHeader(http.StatusAccepted) w.Write([]byte("Request accepted")) }), } server.ListenAndServe() } func processAsync(r *http.Request) { // 耗时的后台处理 time.Sleep(5 * time.Second) log.Println("Processed async request") }

4.3 响应缓存

type Cache struct { data map[string]cacheEntry mu sync.RWMutex } type cacheEntry struct { data []byte expiresAt time.Time } func NewCache() *Cache { return &Cache{ data: make(map[string]cacheEntry), } } func (c *Cache) Get(key string) ([]byte, bool) { c.mu.RLock() defer c.mu.RUnlock() entry, ok := c.data[key] if !ok { return nil, false } if time.Now().After(entry.expiresAt) { return nil, false } return entry.data, true } func (c *Cache) Set(key string, data []byte, ttl time.Duration) { c.mu.Lock() defer c.mu.Unlock() c.data[key] = cacheEntry{ data: data, expiresAt: time.Now().Add(ttl), } } func CacheMiddleware(cache *Cache, ttl time.Duration) Middleware { return func(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { key := r.URL.Path if cached, ok := cache.Get(key); ok { w.Header().Set("X-Cache", "HIT") w.Write(cached) return } w.Header().Set("X-Cache", "MISS") // 记录响应 recorder := &responseRecorder{ResponseWriter: w} next.ServeHTTP(recorder, r) // 缓存响应 cache.Set(key, recorder.body, ttl) }) } } type responseRecorder struct { http.ResponseWriter body []byte } func (r *responseRecorder) Write(data []byte) (int, error) { r.body = append(r.body, data...) return r.ResponseWriter.Write(data) }

五、错误处理

func ErrorHandler(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { defer func() { if err := recover(); err != nil { log.Printf("Panic: %v", err) http.Error(w, "Internal Server Error", http.StatusInternalServerError) } }() next.ServeHTTP(w, r) }) } type AppError struct { Code int Message string Err error } func (e *AppError) Error() string { return fmt.Sprintf("%s: %v", e.Message, e.Err) } func handleError(w http.ResponseWriter, err error) { if appErr, ok := err.(*AppError); ok { http.Error(w, appErr.Message, appErr.Code) return } http.Error(w, "Internal Server Error", http.StatusInternalServerError) }

六、HTTPS配置

func main() { certFile := "server.crt" keyFile := "server.key" server := &http.Server{ Addr: ":443", TLSConfig: &tls.Config{ MinVersion: tls.VersionTLS12, CipherSuites: []uint16{ tls.TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256, tls.TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384, tls.TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256, }, }, } log.Fatal(server.ListenAndServeTLS(certFile, keyFile)) }