Передача параметра запроса обработчику HTTP-запроса Go с помощью пакета MUX

Я пытаюсь передать дополнительный параметр в запросе, который я пытаюсь отправить на сервер Go -

websocket.create_connection("ws://<ip>:port/x/y?token="qwerty")

реализация Go server выглядит следующим образом -

func main() {
    err := config.Parse()
    if err != nil {
        glog.Error(err)
        os.Exit(1)
        return
    }

    flag.Parse()
    defer glog.Flush()

    router := mux.NewRouter()
    http.Handle("/", httpInterceptor(router))

    router.Handle("/v1/x", common.ErrorHandler(stats.GetS)).Methods("GET")
    router.Handle("/v1/x/y", common.ErrorHandler(stats.GetS)).Methods("GET")

    var listen = fmt.Sprintf("%s:%d", config.Config.Ip, config.Config.Port)
    err = http.ListenAndServe(listen, nil)

    if err != nil {
        glog.Error(err)
        os.Exit(1)
    }
}

func httpInterceptor(router http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
        startTime := time.Now()

        if !auth.Auth(w, req) {
            http.Error(w, "Failed authentication", 401)
            return
        }

        router.ServeHTTP(w, req)

        finishTime := time.Now()
        elapsedTime := finishTime.Sub(startTime)

        switch req.Method {
        case "GET":
        case "POST":
        }

    })
}

как я должен искать и анализировать токен на сервере Go, чтобы аутентификация была успешной?

функции библиотеки

func ParseFromRequest(req *http.Request, keyFunc Keyfunc) (token *Token, err error) {

    // Look for an Authorization header
    if ah := req.Header.Get("Authorization"); ah != "" {
        // Should be a bearer token
        if len(ah) > 6 && strings.ToUpper(ah[0:6]) == "BEARER" {
            return Parse(ah[7:], keyFunc)
        }
    }

    // Look for "access_token" parameter
    req.ParseMultipartForm(10e6)
    if tokStr := req.Form.Get("access_token"); tokStr != "" {
        return Parse(tokStr, keyFunc)
    }

    return nil, ErrNoTokenInRequest

}

3 ответов


вызов FormValue чтобы получить параметр запроса:

token := req.FormValue("token")

req - это *http.Request

альтернативой является вызов ParseForm и доступ req.Форма напрямую:

if err := req.ParseForm(); err != nil {
   // handle error
}
token := req.Form.Get("token")

OP спрашивает в комментарии, как сопоставить "токен" с "access_token" для внешнего пакета, который ищет "access_token". Выполните этот код перед вызовом внешнего пакета:

if err := req.ParseForm(); err != nil {
   // handle error
}
req.Form["access_token"] = req.Form["token"]

когда внешний пакет звонки req.Form.Get("access_token"), это будет получите то же значение, что и параметр "token".


в зависимости от того , как вы хотите проанализировать токен, если он исходит из формы или URL-адреса.

первый ответ можно использовать, если токен отправляется из формы, а в случае URL-адреса я бы предложил использовать это. Это работает для меня

token := req.URL.Query().Get("token")

для параметров запроса url:

mux.Vars(r)["token"]