context的上下文值如何進行傳遞

這篇文章運用簡單易懂的例子給大家介紹context的上下文值如何進行傳遞，代碼非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

本文開始將針對context的用法進行系統(tǒng)化討論，在這里你將能夠在工作中合理使用context解決一些比較棘手的問題。

context處理超時處理之外還可以用來保存數(shù)據(jù)，當你需要在多個上下文傳遞時傳遞數(shù)據(jù)，那么本文提到的知識可以排上用場。

示例代碼

示例代碼為一個簡單的http服務(wù)，流程是登錄之后會跳轉(zhuǎn)首頁，首頁通過guard中間件進行鑒權(quán)。當然，示例代碼未做其他諸如連接數(shù)據(jù)庫之類的處理，這不是本文的重點。
守衛(wèi)函數(shù)讀取cookie之后將cookie值寫入context并向下傳遞，在整個請求中可以說是“透明”的。當訪問到需要保護的接口時檢測到?jīng)]有提供cookie，則直接終端請求，否則通過r.WithContext將username的值存入cookie，避免的業(yè)務(wù)接口直接讀取cookie的弊端。因為如果后期更改鑒權(quán)算法的話，業(yè)務(wù)代碼可以不用更改，直接更改中間件即可。

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"
    "time"
)

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", guard(home))
    mux.HandleFunc("/login", login)
    log.Fatal(http.ListenAndServe(":8080", mux))
}

// 登錄
func login(w http.ResponseWriter, r *http.Request) {
    if r.URL.Query().Get("username") != "root" {
        http.Error(w, http.StatusText(401), 401)
        return
    }
    cookie := &http.Cookie{Name: "username", Value: "root", Expires: time.Now().Add(time.Hour)}
    http.SetCookie(w, cookie)
    http.Redirect(w, r, "/", 302)
}

func home(w http.ResponseWriter, r *http.Request) {
    username := r.Context().Value("username")
    fmt.Fprintf(w, "welcome login: %s", username.(string))
}

// 守衛(wèi)
func guard(handleFunc http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // check username
        log.Printf("%s - %s\n", r.Method, r.RequestURI)
        cookie, err := r.Cookie("username")
        if err != nil || cookie == nil { // 如果username為空直接攔截
            http.Error(w, http.StatusText(401), 401)
            return
        }
        handleFunc(w, r.WithContext(context.WithValue(r.Context(), "username", cookie.Value)))
    }
}

關(guān)于context的上下文值如何進行傳遞就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。