mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-23 17:20:24 +08:00
30 lines
622 B
Go
30 lines
622 B
Go
package pathvar
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
var pathVars = contextKey("pathVars")
|
|
|
|
// Vars parses path variables and returns a map.
|
|
func Vars(r *http.Request) map[string]string {
|
|
vars, ok := r.Context().Value(pathVars).(map[string]string)
|
|
if ok {
|
|
return vars
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// WithVars writes params into given r and returns a new http.Request.
|
|
func WithVars(r *http.Request, params map[string]string) *http.Request {
|
|
return r.WithContext(context.WithValue(r.Context(), pathVars, params))
|
|
}
|
|
|
|
type contextKey string
|
|
|
|
func (c contextKey) String() string {
|
|
return "rest/pathvar/context key: " + string(c)
|
|
}
|