当前位置：首页 > 文章列表 > Golang > Go教程 > 在 GO 中动态获取代码输入的出现

在 GO 中动态获取代码输入的出现

来源：dev.to 2024-12-20 10:34:10 0浏览收藏

golang学习网今天将给大家带来《在 GO 中动态获取代码输入的出现》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

advent of code 是程序员测试和提高解决问题能力的一种有趣方式。在解决谜题时，您可能希望直接使用其 url 自动获取个性化谜题输入，而不是将输入复制到本地可用的文本文件中。但是，尝试使用简单的 http 请求访问输入 url 会导致以下消息：

谜题输入因用户而异。请登录以获取您的谜题输入。

本文解释了为什么会发生这种情况以及如何使用 go 编程语言动态正确获取输入。

问题：为什么我们不能直接获取输入？

《代码的降临》要求您登录才能访问您的个性化谜题输入。当您通过浏览器登录时，advent of code 会在您的浏览器中设置会话 cookie。此 cookie 用于识别您的帐户并提供您独特的输入。

如果您的 http 请求不包含此会话 cookie，advent of code 服务器无法将您识别为登录用户，因此会出现错误消息。

解决方案：在 http 请求中使用 session cookie

我们必须在 http 请求中包含会话 cookie 才能获取拼图输入。以下是分步指南：

登录 advent of code。
打开浏览器的开发者工具（按 f12 键）并导航到“网络”选项卡。
刷新 advent of code 页面并在请求标头中查找 cookie 标头。

在 GO 中动态获取代码输入的出现

提取会话 cookie 的值。

在 GO 中动态获取代码输入的出现

注意：对您的会话 cookie 保密非常重要，因为其他人可以访问您的 advent of code 帐户（如果他们可以访问它）。

获取输入的代码

下面是一个简单的程序，我们将用它来动态获取拼图输入：

设置基本 url

我们首先定义用于获取输入的基本 url，并创建一个函数来读取特定日期的输入。

const baseurl = "https://adventofcode.com/2024/day/%s/input"

func readinput(day string) {
    url := fmt.sprintf(baseurl, day)
    fmt.println(url)
}

创建 http 请求

接下来，我们创建一个 http 请求并包含会话 cookie。

client := &http.client{}
    req, err := http.newrequest("get", url, nil)
    if err != nil {
        fmt.printf("error creating http request: %v\n", err)
        return
    }

    // add the session cookie
    req.header.add("cookie", "session=[your_session_token]")

http.newrequest：为输入 url 创建 http get 请求。

req.header.add：使用会话令牌向请求添加标头以进行身份验证。（将 [your_session_token] 替换为您的实际令牌）。

发送请求并处理响应

现在我们发送 http 请求并读取服务器的响应。

resp, err := client.do(req)
    if err != nil {
        fmt.printf("error making http request: %v\n", err)
        return
    }
    defer resp.body.close()

    if resp.statuscode != http.statusok {
        fmt.printf("unexpected http status: %d\n", resp.statuscode)
        return
    }

client.do(req)：发送 http 请求并存储响应。

defer resp.body.close()：确保读取后关闭响应正文。

resp.statuscode：检查 http 状态代码。 200 以外的代码表示错误。

读取并打印输入

最后，我们读取响应正文并打印拼图输入。

body, err := io.readall(resp.body)
    if err != nil {
        fmt.printf("error reading response body: %v\n", err)
        return
    }

    fmt.println(string(body))

io.readall(resp.body)：读取响应正文。

string(body)：将正文从字节切片转换为字符串，以便于显示。

定义主要函数

我们从主函数调用 readinput 函数来获取第一天的输入。

func main() {
    readinput("1") // fetches input puzzle for day 1
}

增强安全性

在我们的代码中对会话令牌进行硬编码并不安全。相反，我们应该使用以下步骤将其存储为环境变量：

使用终端导出会话令牌：

export aoc_session="[your_session_token]"

修改代码以从环境变量中读取会话令牌。（确保您的导入中包含“os”）：

req.header.add("cookie", "session="+os.getenv("aoc_session"))

这有助于将会话令牌保留在源代码之外，从而降低意外暴露的风险。

完整程序代码

以下是完整的程序供参考：

package main

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

const baseURL = "https://adventofcode.com/2024/day/%s/input"

func readInput(day string) {
    url := fmt.Sprintf(baseURL, day)

    client := &http.Client{}
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        fmt.Printf("Error creating HTTP request: %v\n", err)
        return
    }

    // Use session token from environment variable
    req.Header.Add("Cookie", "session="+os.Getenv("AOC_SESSION"))

    resp, err := client.Do(req)
    if err != nil {
        fmt.Printf("Error making HTTP request: %v\n", err)
        return
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        fmt.Printf("Unexpected HTTP status: %d\n", resp.StatusCode)
        return
    }

    body, err := io.ReadAll(resp.Body)
    if err != nil {
        fmt.Printf("Error reading response body: %v\n", err)
        return
    }

    fmt.Println(string(body))
}

func main() {
    readInput("1")
}

要记住的事情