Go 语言 Web 编程系列——设置、读取和删除 Cookie

Golang
330
0
0
2023-08-15
吐血整理的GO语言部分框架、库必学清单！带你开启机器学习！
上一篇
Go项目标准布局
下一篇

1、 Cookie 简介

介绍完了 Go 语言的 HTTP 请求和响应处理，接下来，我们来看看 Go 语言中 Cookie 技术的实现，由于 HTTP 协议本身是无状态的，所以引入了 Cookie 来实现客户端用户识别和状态管理，关于 Cookie 本身这里不多做介绍，你可以在维基百科或者阅读 HTTP 报文首部字段（五）：扩展字段篇（Cookie）这篇教程了解更多细节。

我们可以在 HTTP 响应头中通过 Set-Cookie 字段设置 Cookie，然后在下次请求时就会在请求头 Cookie 中自动包含新增的 Cookie。你可以在 Chrome 浏览器的控制台中通过 Application->Storage-> Cookies 查看指定域名下的所有 Cookie：

可以在 Cookie 列表顶部抬头看到 Cookie 的所有属性。

2、Cookie 结构体

Go 专门提供了一个 http.Cookie 结构体来表示 Cookie，其中的字段对应着 Cookie 的所有属性：

 // A Cookie represents an HTTP cookie as sent in the Set-Cookie  Header  of an
// HTTP response or the Cookie header of an HTTP  Request .
//
// See 
type Cookie struct {
        Name  string
        Value string
        Path       string    // optional
        Domain     string    // optional
        Expires    time.Time // optional
        RawExpires string    // for reading cookies only
        // MaxAge= means no 'Max-Age' attribute specified.
        // MaxAge< means delete cookie now, equivalently 'Max-Age: 0'
        // MaxAge> means Max-Age attribute present and given in seconds
        MaxAge   int
        Secure   bool
        HttpOnly bool
        SameSite SameSite
        Raw      string
        Unparsed []string // Raw text of unparsed attribute-value pairs
}

我们可以将其与上面截图中的 Cookie 属性一一对应，如果你了解 Cookie 的基本结构的话，很容易知道每个字段的含义。如果你不了解的话，可以通过点击上面 Cookie 简介中给出的链接去了解。

这里，我们重点介绍下 Expires 字段：

  • 如果 Expires 字段为空，则 Cookie 就是 Session Cookie 或者叫临时 Cookie，这个 Cookie 会随着浏览器的关闭而销毁。
  • 如果设置了 Expires，那么这个 Cookie 就是持久 Cookie，直到过期时间后才会销毁。

有两种方法来设置过期时间：一种是直接设置 Expires 字段，一种是设置 MaxAge 字段。前者表示到期的具体时间点，后者表示 Cookie 的有效时长（单位是秒）。这并不是 Go 语言的设计，而是不同浏览器的混乱标准使然，比如虽然 HTTP/1.1 有意废弃 Expires，不过 IE 6、7、8 却不支持 MaxAge 字段。通常，考虑到默认时区问题，本地时间不可靠，推荐通过 MaxAge 字段设置 Cookie 过期时间，不过对于 Web 应用而言，通常不设置过期时间，让 Cookie 随着浏览器关闭而失效即可。

3、发送 Cookie 到客户端

了解了 Cookie 的基本结构，以及如何在 Go 语言中表示后，我们尝试在 HTTP 响应中通过设置 Set-Cookie 头新增 Cookie 并将其发送给客户端浏览器，在此之前的在线论坛项目中，已经在用户认证时用到了 Cookie 设置：

 cookie := http.Cookie{
        Name:     "_cookie",
        Value:    session.Uuid,
        HttpOnly: true,
        Path: "/",
}
http.SetCookie(writer, &cookie)
http.Redirect(writer, request, "/", )

上述代码是在用户认证通过后将 Session ID 通过 Cookie 存储到客户端，以便后续记住用户登录状态，直到用户关闭浏览器（没有设置过期时间）。通过这段代码，可以看出要在响应中发送 Cookie，需要先通过 http.Cookie 初始化一个 Cookie 对象，再通过 http.SetCookie 方法将这个 Cookie 写入到 HTTP 响应中，这样发送响应给客户端的时候就会带上这个 Cookie 了。

接下来，我们在 goblog 项目中演示发送 Cookie 到客户端，这一次，我们不通过写入 Cookie 到 HTTP 响应，而是直接通过 HTTP 响应头 Set-Cookie 来设置 Cookie，在 goblog/handlers/common.go 中新增 SetCookie 处理器：

 func SetCookie(w http.ResponseWriter, r *http.Request)  {
        c := http.Cookie{
                Name: "username",
                Value: url.QueryEscape("学院君"),
                HttpOnly: true,
        }
        c := http.Cookie{
                Name: "website",
                Value: "#;,
                HttpOnly: true,
        }
        w.Header().Add("Set-Cookie", c.String())
        w.Header().Add("Set-Cookie", c.String())
        fmt.Fprintln(w, "通过 HTTP 响应头发送 Cookie 信息")
}

这里我们新增了两个 Cookie，所以使用了 w.Header().Add 方法，如果用 Set 方法，后面的 Set-Cookie 头会覆盖前面的，另外，由于 Cookie 值包含了中文字符，需要通过 url.QueryEscape 方法进行 URL 编码，否则无法正常显示。

在 routes/web.go 中注册对应的路由：

 WebRoute{
        "SetCookie",
        "GET",
        "/setcookies",
        handlers.SetCookie,
},

重启 HTTP 服务器，在浏览器中访问 ，就可以在响应结果中看到对应的 Cookie 信息了：

当然也可以通过 http.SetCookie 方法写入 Cookie 到 HTTP 响应来实现，对应的代码如下，这样做更便捷：

 http.SetCookie(w, &c)
http.SetCookie(w, &c)

设置 Cookie 有效期

还可以通过 Expires/MaxAge 设置 Cookie 的有效期：

 c := http.Cookie{
        Name: "username",
        ...
        Expires: time.Now().AddDate(, 0, 1),  // Cookie 有效期设置为1天
}
c := http.Cookie{
        Name: "website",
        ...
        MaxAge: ,    // Cookie 有效期设置为 1000s
}

这样，就可以在响应中看到对应的 Cookie 有效期了：

4、从请求中读取 Cookie

一旦通过 Set-Cookie 响应头将 Cookie 信息发送到客户端浏览器，那么在 Cookie 有效期内，下次同域名下的用户请求将自动在请求头中包含对应的 Cookie 信息，比如我们访问 ，就可以在请求头 Cookie 中看到上次响应返回的 Cookie：

要在服务端获取这些 Cookie 信息，可以通过读取请求头的方式：

 cookie := r.Header.Get("Cookie")

但是这种方式读取的 Cookie 字符串值还需要进行解析，才能得到每个 Cookie 的值，为此可以通过更加便捷的专门用于读取每个 Cookie 的 r.Cookie 方法（r 表示 HTTP 请求对象实例），我们在 handlers/common.go 中新增一个 GetCookie 方法：

 func GetCookie(w http.ResponseWriter, r *http.Request)  {
        c, err := r.Cookie("username")
        if err != nil {
                fmt.Fprintln(w, "名为 username 的 Cookie 不存在")
                return
        }
        username, _ := url.QueryUnescape(c.Value)
        c, err := r.Cookie("website")
        if err != nil {
                fmt.Fprintln(w, "名为 website 的 Cookie 不存在")
                return
        }
        website := c.Value
    fmt.Fprintf(w, "从用户请求中读取的 Cookie: {username: %s, website: %s}\n", username, website)
}

需要注意的是 r.Cookie 方法返回的是指针类型的 Cookie 对象和一个错误信息，需要通过调用 Cookie 对象上的 Value 属性返回对应的 Cookie 值，对于 username 而言，还需要通过 url.QueryUnescape 对编码值进行解码。

在 routes/web.go 中注册一个对应的路由：

 WebRoute{
        "GetCookie",
        "GET",
        "/getcookies",
        handlers.GetCookie,
},

然后重启 HTTP 服务器，在浏览器中访问 ，website Cookie 如果已经过期，则会打印错误消息：

重新访问 ，再访问 ，就可以成功获取并打印出所有的 Cookie 信息：

如果想要一次性获取所有 Cookie，还可以通过 r.Cookies() 方法，该方法会返回的是 Cookie 对象指针类型切片：

 cookies := r.Cookies()
c := cookies[0]  // username=%E5%AD%A6%E9%99%A2%E5%90%9B
c := cookies[1]  // website=

5、删除 Cookie

如果想要在 Cookie 过期之前提前删除 Cookie，可以将 MaxAge 设置为小于 0 的值即可：

 c := http.Cookie{
        Name: "website",
        Value: "#;,
        HttpOnly: true,
        MaxAge: -,    // Cookie 有效期设置为 -1，就会在当前响应发送给客户端后销毁该 Cookie
}

如果用 Expires 字段来设置的话，可以设置 Unix 时间戳的值为 1（对应的绝对时间是 1970-01-01 08:00:01 +0800 CST，也就是一个过去的时间）：

 c := http.Cookie{
        Name: "website",
        Value: "#;,
        HttpOnly: true,
        Expires: time.Unix(, 0),    // Cookie 有效期设置为过去的时间
}

使用 Cookie 设置一次性消息

所谓一次性消息，指的是页面重新加载后消息就不存在了，也就是该消息只能被读取一次，不管你用不用它都不复存在了。我们可以结合上面的删除 Cookie 功能来实现这个一次性消息功能，首先在 common.go 中新增一个 SetWelcomeMessage 处理器：

 func SetWelcomeMessage(w http.ResponseWriter, r *http.Request)  {
         msg  := "欢迎访问学院君网站"
        cookie := http.Cookie{
                Name: "welcome_message",
                Value: base.URLEncoding.EncodeToString([]byte(msg)),
        }
        http.SetCookie(w, &cookie)
        http.Redirect(w, r, "/get_welcome_message", )
}

这里发送 Cookie 到客户端和前面的新增 Cookie 逻辑完全一样，最后通过 http.Redirect 将请求重定向到 /get_welcome_message 路由，另外这一次我们通过 Base64 对 Cookie 值进行编码，因为其中包含中文和特殊字符。接下来我们继续在 common.go 中编写 /get_welcome_message 路由对应的处理器 GetWelcomeMessage：

 func GetWelcomeMessage(w http.ResponseWriter, r *http.Request)  {
        cookie, err := r.Cookie("welcome_message")
        if err != nil {
                fmt.Fprintln(w, "没有在 Cookie 中找到欢迎消息")
        } else {
                delCookie := http.Cookie{
                        Name: "welcome_message",
                        MaxAge: -,
                }
                http.SetCookie(w, &delCookie)
                msg, _ := base.URLEncoding.DecodeString(cookie.Value)
                fmt.Fprintln(w, string(msg))
        }
}

在这段代码中，首先通过 r.Cookie 方法从 Cookie 中读取欢迎消息，读取之后，通过 Base64 对其进行解码，然后作为响应实体返回给客户端，这一块和之前读取 Cookie 逻辑一样，只是新增了读取成功之后，删除这个 Cookie 的功能。

最后，在 routes/web.go 中注册上述处理器方法对应的路由：

 WebRoute{
        "SetMessage",
        "GET",
        "/set_welcome_message",
        handlers.SetWelcomeMessage,
},
WebRoute{
        "GetMessage",
        "GET",
        "/get_welcome_message",
        handlers.GetWelcomeMessage,
},

重启 HTTP 服务器，在浏览器中访问 ，页面会重定向到 ，然后打印出欢迎消息，说明 Cookie 读取成功：

如果查看 Chrome 控制台的话，可以在 Network 标签页看到对应的 Cookie 响应头和请求头：

在 get_welcome_message 请求头中可以看到对应的 Cookie，此外这个请求的响应头还包含了删除 Cookie 的逻辑：

如果在「Application」标签页查看 Cookie 的话，会发现已经没有 welcome_message 了（website 到了过期时间自动销毁，所以也看不到了）：

好了，关于 HTTP Cookie 功能我们简单介绍到这里，下篇教程，我们来探讨在 Go 语言中如何基于 Cookie 实现 Session 技术。

（全文完）

吐血整理的GO语言部分框架、库必学清单！带你开启机器学习！
上一篇
Go项目标准布局
下一篇