使用 HTTP Post 并使用 Cookies

Question

我正在尝试使用 Go 登录网站并存储 cookie 供以后使用。

您能否提供用于发布表单、存储 cookie 和使用 cookie 访问另一个页面的示例代码？

我想我可能需要通过研究 http://gotour.golang.org/src/pkg/net/http/client.go 来创建一个客户端来存储 cookie

 package main

import ("net/http"
        "log"
        "net/url"
        )

func Login(user, password string) string {
        postUrl := "http://www.pge.com/eum/login"

        // Set up Login
        values := make(url.Values)
        values.Set("user", user)
        values.Set("password", password)

        // Submit form
        resp, err := http.PostForm(postUrl, values)
        if err != nil {
                log.Fatal(err)
        }
        defer resp.Body.Close()

        // How do I store cookies?
        return "Hello"
}

func ViewBill(url string, cookies) string {

//What do I put here?

}

原文由 Lionel 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

Go 1.1 引入了一个 cookie jar 实现 net/http/cookiejar 。

 import (
    "net/http"
    "net/http/cookiejar"
)

jar, err := cookiejar.New(nil)
if err != nil {
  // error handling
}

client := &http.Client{
    Jar: jar,
}

原文由 greenimpala 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

首先，您需要实现 http.CookieJar 接口。然后您可以将它传递给您创建的客户端，它将用于向客户端发出的请求。作为一个基本的例子：

 package main

import (
    "fmt"
    "net/http"
    "net/url"
    "io/ioutil"
    "sync"
)

type Jar struct {
    lk      sync.Mutex
    cookies map[string][]*http.Cookie
}

func NewJar() *Jar {
    jar := new(Jar)
    jar.cookies = make(map[string][]*http.Cookie)
    return jar
}

// SetCookies handles the receipt of the cookies in a reply for the
// given URL.  It may or may not choose to save the cookies, depending
// on the jar's policy and implementation.
func (jar *Jar) SetCookies(u *url.URL, cookies []*http.Cookie) {
    jar.lk.Lock()
    jar.cookies[u.Host] = cookies
    jar.lk.Unlock()
}

// Cookies returns the cookies to send in a request for the given URL.
// It is up to the implementation to honor the standard cookie use
// restrictions such as in RFC 6265.
func (jar *Jar) Cookies(u *url.URL) []*http.Cookie {
    return jar.cookies[u.Host]
}

func main() {
    jar := NewJar()
    client := http.Client{nil, nil, jar}

    resp, _ := client.PostForm("http://www.somesite.com/login", url.Values{
        "email": {"myemail"},
        "password": {"mypass"},
    })
    resp.Body.Close()

    resp, _ = client.Get("http://www.somesite.com/protected")

    b, _ := ioutil.ReadAll(resp.Body)
    resp.Body.Close()

    fmt.Println(string(b))
}

原文由 dskinner 发布，翻译遵循 CC BY-SA 3.0 许可协议