Go语言中HTTP客户端会话管理：正确使用CookieJar处理重定向与持久化

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "net/http/cookiejar" // 导入标准库的cookiejar
    "net/url"
    "strings"
    "log"
    "time" // 用于模拟实际的用户名和密码
)

// 假设的用户名和密码，实际应用中应从安全配置中获取
const (
    username = "your_username"
    password = "your_password"
)

// NewClientWithJar 创建一个配置了标准CookieJar的HTTP客户端
func NewClientWithJar() *http.Client {
    // 创建一个新的CookieJar实例
    jar, err := cookiejar.New(nil) // nil表示使用默认的公共后缀列表
    if err != nil {
        log.Fatalf("Failed to create cookie jar: %v", err)
    }

    // 创建HTTP客户端，并将CookieJar赋值给其Jar字段
    client := &http.Client{
        Jar: jar, // 客户端将自动使用这个jar来管理cookie
        // CheckRedirect: nil, // 默认行为是自动跟随重定向，无需修改
        Timeout: 30 * time.Second, // 设置一个合理的超时时间
    }
    return client
}

// Login 模拟登录过程并维护会话
func Login(client *http.Client) {
    loginURL := "https://www.statuscake.com/App/" // 登录API的URL
    baseURL, _ := url.Parse("https://www.statuscake.com") // 用于Cookie的基URL

    values := url.Values{}
    values.Add("username", username)
    values.Add("password", password)
    values.Add("Login", "yes")
    values.Add("redirect", "")
    postBody := values.Encode()

    req, err := http.NewRequest("POST", loginURL, strings.NewReader(postBody))
    if err != nil {
        log.Fatalf("Failed to create request: %v", err)
    }

    // 设置请求头
    req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
    req.Header.Set("Accept", "text/html")
    req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.65 Safari/537.36")

    // 注意：这里不再需要手动添加Cookie，http.Client的Jar会自动处理
    // req.AddCookie(cookie) // 移除这部分代码

    fmt.Printf("Attempting to log in to: %s\n", loginURL)
    resp, err := client.Do(req)
    if err != nil {
        log.Fatalf("Login request failed: %v", err)
    }
    defer resp.Body.Close()

    bodyBytes, _ := ioutil.ReadAll(resp.Body)
    fmt.Printf("Login Response Status: %s\n", resp.Status)
    fmt.Printf("Login Response Body (partial): %s...\n", string(bodyBytes[:min(len(bodyBytes), 500)])) // 打印部分响应体

    if resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusFound { // 200 OK 或 302 Found (重定向)
        fmt.Println("\n----- Login Successful (or redirected) -----")
        fmt.Println("HTTP Code: ", resp.StatusCode)
        fmt.Println("Response Cookies (from resp.Cookies()): ", resp.Cookies()) // 这里的Cookie是当前响应设置的
        fmt.Println("Cookies stored in Jar for base URL: ", client.Jar.Cookies(baseURL)) // Jar中存储的Cookie
        fmt.Println("------------------------------------------")

        // 假设登录成功后，我们可以访问一个受保护的子页面
        accessProtectedPage(client)
    } else {
        fmt.Printf("Login failed with status: %s\n", resp.Status)
    }
}

// accessProtectedPage 访问一个需要会话的受保护页面
func accessProtectedPage(client *http.Client) {
    protectedURL := "https://www.statuscake.com/App/Dashboard.php" // 假设这是一个受保护的页面
    fmt.Printf("\nAttempting to access protected page: %s\n", protectedURL)

    req, err := http.NewRequest("GET", protectedURL, nil)
    if err != nil {
        log.Fatalf("Failed to create protected page request: %v", err)
    }

    // 客户端会自动从Jar中获取并添加必要的Cookie
    resp, err := client.Do(req)
    if err != nil {
        log.Fatalf("Protected page request failed: %v", err)
    }
    defer resp.Body.Close()

    bodyBytes, _ := ioutil.ReadAll(resp.Body)
    fmt.Printf("Protected Page Response Status: %s\n", resp.Status)
    fmt.Printf("Protected Page Response Body (partial): %s...\n", string(bodyBytes[:min(len(bodyBytes), 500)]))

    if resp.StatusCode == http.StatusOK {
        fmt.Println("\n----- Accessed Protected Page Successfully -----")
    } else {
        fmt.Printf("Failed to access protected page with status: %s\n", resp.Status)
    }
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

func main() {
    client := NewClientWithJar()
    Login(client)
}