系列開篇
本文開始將針對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
", 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與協程超時控制將在下一篇文章中講到。
我的部落格
每天進步一點點,歡迎大家作客!