golang框架中间件在通信系统中的角色

import (
    "log"
    "net/http"

    "github.com/gorilla/mux"
    "github.com/gorilla/securecookie"
)

// 创建一个密钥，用于加密 cookie
var cookieKey = []byte(" rahasia ")

// 定义认证中间件
func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 从 cookie 中获取用户名
        username, err := r.Cookie("username")
        if err != nil {
            // 如果 cookie 不存在或无效，则返回 401 Unauthorized
            w.WriteHeader(http.StatusUnauthorized)
            return
        }

        // 解密用户名
        usernameBytes, err := securecookie.Decode("username", username.Value, cookieKey)
        if err != nil {
            log.Println("Failed to decode username:", err)
            w.WriteHeader(http.StatusInternalServerError)
            return
        }

        // 将用户名添加到请求上下文中
        ctx := r.Context()
        ctx = context.WithValue(ctx, "username", string(usernameBytes))
        r = r.WithContext(ctx)

        // 调用下一个处理程序
        next.ServeHTTP(w, r)
    })
}

func main() {
    r := mux.NewRouter()

    // 为需要认证的路由添加认证中间件
    r.Use(AuthMiddleware)

    r.HandleFunc("/protected", func(w http.ResponseWriter, r *http.Request) {
        // 获取请求上下文中的用户名
        username := r.Context().Value("username").(string)

        w.Write([]byte("Selamat datang, " + username + "!"))
    })

    log.Fatal(http.ListenAndServe(":8080", r))
}