mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
19 lines
315 B
Go
19 lines
315 B
Go
|
package mux
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func contextGet(r *http.Request, key interface{}) interface{} {
|
||
|
|
return r.Context().Value(key)
|
||
|
|
}
|
||
|
|
|
||
|
|
func contextSet(r *http.Request, key, val interface{}) *http.Request {
|
||
|
|
if val == nil {
|
||
|
|
return r
|
||
|
|
}
|
||
|
|
|
||
|
|
return r.WithContext(context.WithValue(r.Context(), key, val))
|
||
|
|
}
|