Golang 框架性能优化的代码重构技巧

// handler1 处理来自 "/route1" 的所有请求
func handler1(w http.ResponseWriter, r *http.Request) {}

// handler2 处理来自 "/route2" 的所有请求
func handler2(w http.ResponseWriter, r *http.Request) {}

// 使用中间件重构代码
func middleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 共同的预处理操作
        next.ServeHTTP(w, r)
        // 共同的后处理操作
    })
}

// 使用中间件优化处理程序
func main() {
    router := chi.NewRouter()
    router.Use(middleware)
    router.Handle("/route1", handler1)
    router.Handle("/route2", handler2)
}

import (
    "context"
    "net/http"

    "github.com/go-redis/redis/v8"
)

// 使用 Redis 缓存响应
func cachedHandler(w http.ResponseWriter, r *http.Request) {
    ctx := context.Background()
    key := r.URL.Path

    rdb := redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", // 如果已设置密码则填写
    })

    val, err := rdb.Get(ctx, key).Result()
    if err == redis.Nil {
        // 如果缓存中没有数据，则从数据库中获取
        data := getFromDB(key)
        if err := rdb.Set(ctx, key, data, 0).Err(); err != nil {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
        fmt.Fprint(w, data)
    } else if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
    } else {
        fmt.Fprint(w, val)
    }
}

import (
    "context"
    "net/http"
    "sync"
)

// 使用 goroutine 并发处理请求
func concurrentHandler(w http.ResponseWriter, r *http.Request) {
    ctx := context.Background()
    var wg sync.WaitGroup

    // 创建 10 个并发 goroutine
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            // 并发执行的任务
        }()
    }

    wg.Wait()
    fmt.Fprint(w, "任务完成")
}