web-dev-qa-db-ja.com

Go HTTPクライアントがリダイレクトを自動的に追跡しないようにする方法はありますか？

私は現在、GoでREST APIと対話するソフトウェアを作成しています。クエリしようとしているREST APIエンドポイントはHTTP 302リダイレクトを返しますリソースURIを指すHTTP Locationヘッダーを使用します。

Goスクリプトを使用して、後で処理するためにHTTP Locationヘッダーを取得しようとしています。

現在、この機能を実現するために現在行っていることは次のとおりです。

package main

import (
        "errors"
        "fmt"
        "io/ioutil"
        "net/http"
)

var BASE_URL = "https://api.stormpath.com/v1"
var STORMPATH_API_KEY_ID = "xxx"
var STORMPATH_API_KEY_SECRET = "xxx"

func noRedirect(req *http.Request, via []*http.Request) error {
        return errors.New("Don't redirect!")
}

func main() {

        client := &http.Client{
            CheckRedirect: noRedirect
        }
        req, err := http.NewRequest("GET", BASE_URL+"/tenants/current", nil)
        req.SetBasicAuth(STORMPATH_API_KEY_ID, STORMPATH_API_KEY_SECRET)

        resp, err := client.Do(req)

        // If we get here, it means one of two things: either this http request
        // actually failed, or we got an http redirect response, and should process it.
        if err != nil {
            if resp.StatusCode == 302 {
                fmt.Println("got redirect")
            } else {
                panic("HTTP request failed.")
            }
        }
        defer resp.Body.Close()

}

これは私にとってちょっとしたハックのように感じます。 http.ClientのCheckRedirect関数をオーバーライドすることで、本質的にHTTPリダイレクトをエラーのように扱うことを強制されます（そうではありません）。

私はHTTPクライアントの代わりにHTTPトランスポートを使用することを提案している他のいくつかの場所を見ましたが、これと通信するためにHTTP Basic Authを使用する必要があるため、HTTPクライアントが必要なので、これを動作させる方法がわかりませんREST API。

リダイレクトに従わずに、基本認証でHTTP要求を行う方法を教えてもらえますか？

ありがとうございました。

httprestredirectgo
44
rdegges

現在、はるかに簡単なソリューションがあります。

client: &http.Client{
    CheckRedirect: func(req *http.Request, via []*http.Request) error {
        return http.ErrUseLastResponse
    },
}

このように、httpパッケージは自動的に認識します：「ああ、リダイレクトに従うべきではありません」が、エラーはスローしません。ソースコードのコメントから：

特殊なケースとして、CheckRedirectがErrUseLastResponseを返す場合、nilエラーとともに、本文が閉じられていない最新の応答が返されます。

93
Etienne Bruines

RoundTripを使用せずにクライアント自体を使用する別のオプション：

// create a custom error to know if a redirect happened
var RedirectAttemptedError = errors.New("redirect")

client := &http.Client{}
// return the error, so client won't attempt redirects
client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
        return RedirectAttemptedError
}
// Work with the client...
resp, err := client.Head(urlToAccess)

// test if we got the custom error
if urlError, ok := err.(*url.Error); ok && urlError.Err == RedirectAttemptedError{
        err = nil   
}

PDATE：このソリューションはgo <1.7

11
uvgroovy

可能ですが、解決策は問題を少し逆にします。これはgolangテストとして作成されたサンプルです。

package redirects

import (
    "github.com/codegangsta/martini-contrib/auth"
    "github.com/go-martini/martini"
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestBasicAuthRedirect(t *testing.T) {
    // Start a test server
    server := setupBasicAuthServer()
    defer server.Close()

    // Set up the HTTP request
    req, err := http.NewRequest("GET", server.URL+"/redirect", nil)
    req.SetBasicAuth("username", "password")
    if err != nil {
        t.Fatal(err)
    }

    transport := http.Transport{}
    resp, err := transport.RoundTrip(req)
    if err != nil {
        t.Fatal(err)
    }
    // Check if you received the status codes you expect. There may
    // status codes other than 200 which are acceptable.
    if resp.StatusCode != 200 && resp.StatusCode != 302 {
        t.Fatal("Failed with status", resp.Status)
    }

    t.Log(resp.Header.Get("Location"))
}


// Create an HTTP server that protects a URL using Basic Auth
func setupBasicAuthServer() *httptest.Server {
    m := martini.Classic()
    m.Use(auth.Basic("username", "password"))
    m.Get("/ping", func() string { return "pong" })
    m.Get("/redirect", func(w http.ResponseWriter, r *http.Request) {
        http.Redirect(w, r, "/ping", 302)
    })
    server := httptest.NewServer(m)
    return server
}

上記のコードを「リダイレクト」と呼ばれる独自のパッケージに入れ、必要な依存関係を取得した後に実行することができるはずです

mkdir redirects
cd redirects
# Add the above code to a file with an _test.go suffix
go get github.com/codegangsta/martini-contrib/auth
go get github.com/go-martini/martini
go test -v

お役に立てれば！

6
Sridhar Venkatakrishnan

リダイレクトに従わない基本認証で要求を行うには、 RoundTrip を受け入れる関数を使用します* Request

このコード

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "os"
)

func main() {
    var DefaultTransport http.RoundTripper = &http.Transport{}

    req, _ := http.NewRequest("GET", "http://httpbin.org/headers", nil)
    req.SetBasicAuth("user", "password")

    resp, _ := DefaultTransport.RoundTrip(req)
    defer resp.Body.Close()
    contents, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Printf("%s", err)
        os.Exit(1)
    }
    fmt.Printf("%s\n", string(contents))
}

出力

{
  "headers": {
    "Accept-Encoding": "gzip", 
    "Authorization": "Basic dXNlcjpwYXNzd29yZA==", 
    "Connection": "close", 
    "Host": "httpbin.org", 
    "User-Agent": "Go 1.1 package http", 
    "X-Request-Id": "45b512f1-22e9-4e49-8acb-2f017e0a4e35"
  }
}
4
Alex Dvoretsky