首頁 > Java > java教程 > context的上下文值傳遞的介紹(程式碼範例)

context的上下文值傳遞的介紹(程式碼範例)

不言
發布: 2018-10-18 15:25:41
轉載
3241 人瀏覽過

這篇文章帶給大家的內容是關於context的上下文值傳遞的介紹(程式碼範例),有一定的參考價值,有需要的朋友可以參考一下,希望對你有幫助。

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

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

範例程式碼

範例程式碼為簡單的http服務,流程是登入之後會跳到首頁,首頁透過guard中間件進行鑑權。當然,範例程式碼未做其他諸如連接資料庫之類的處理,這不是本文的重點。
守衛函數讀取cookie之後將cookie值寫入context並向下傳遞,在整個請求中可以說是「透明」的。當存取到需要保護的介面時偵測到沒有提供cookie,則直接終端請求,否則透過r.WithContext將username的值存入cookie,避免的業務介面直接讀取cookie的弊端。因為如果後期更改鑑權演算法的話,業務程式碼可以不用更改,直接更改中間件即可。

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))
}

// 守卫
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)))
    }
}
登入後複製

本文的程式碼就這麼多,內容也很少,希望大家能好好用上這個利器。

#

以上是context的上下文值傳遞的介紹(程式碼範例)的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:segmentfault.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板