mirror of
https://github.com/fatedier/frp.git
synced 2024-12-16 03:19:00 +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))
|
||
|
|
}
|