在上一节中,我们讲述了如何使用net/http接收和处理用户请求,在本节,将继续讲述如何使用net/http实现网站响应和Cookie设置。
网站响应
首先讲述如何生成网站的响应内容,从路由的处理函数看到,参数w http.ResponseWriter负责生成网址的响应内容,对比参数r *http.Request发现,参数w是以传值方式表示;参数r以指针方式表示,使用指针方式表示是为了让服务器能够察觉到程序对Request结构的实时修改,而参数w虽然以传值方式表示,但在net/http源码中,我们发现,参数w(ResponseWriter类型)被调用时,使用过程中指向response结构的指针。换句话说,路由处理函数的两个参数在使用过程都是以指针的形式表示,只不过两者在源码中的实现方式不同而导致参数的类型有所不同。
参数w为ResponseWriter类型,它一共设有3个方法,分别为Write、WriteHeader和Header,每个方法实现的功能说明如下。
- Write:接受一个字节数组作为参数,并将数组中的字节写入 HTTP 的响应内容,简单来说,就是讲HTML的代码生成响应的网页内容。
- WriteHeader:这是设置HTTP 响应状态码,如200、302、404和500等。
- Header:设置响应内容的响应头,如设置响应方式(Content-Type)、Cookie(Set-Cookie)等等,响应头的内容如下图所示。
大致了解参数w的语法结构,我们将其在实际中加以应用,详细的代码如下所示。
package main
import (
"encoding/json"
"net/http"
)
// Write接受一个字节数组作为参数,并将数组中的字节写入HTTP响应的主体
func indexExample(w http.ResponseWriter, r *http.Request) {
str := `<html>
<head><title>My Go</title></head>
<body><h1>Hello World</h1></body>
</html>`
w.Write([]byte(str))
}
// WriteHeader设置响应状态码
func errorExample(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(501)
str := `<html>
<head><title>My Go</title></head>
<body><h1>Hello World</h1></body>
</html>`
w.Write([]byte(str))
}
// 在Header中设置参数Location,并使用WriteHeader设置302状态码即可实现URL重定向
// 重定向的URL为参数Location的参数值
func redirectExample(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Location", "http://google.com")
w.WriteHeader(302)
}
// 定义结构体Post,用于生成JSON数据
type Post struct {
User string
Threads []string
}
// 在Header中设置参数Content-Type,参数值为application/json,将响应内容以JSON数据输出
// 使用结构体Post生成JSON数据,并由Write方法将JSON数据作为响应内容输出
func jsonExample(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
post := &Post{
User: "Go",
Threads: []string{"first", "second", "third"},
}
json, _ := json.Marshal(post)
w.Write(json)
}
func main() {
server := http.Server{
Addr: "127.0.0.1:8080",
}
http.HandleFunc("/", indexExample)
http.HandleFunc("/error", errorExample)
http.HandleFunc("/redirect", redirectExample)
http.HandleFunc("/json", jsonExample)
server.ListenAndServe()
}
Cookie
net/http定义了结构体Cookie,主要实现身份验证,Cookie是一种存储在客户端的、体积较小的信息,这些信息最初都是由服务器通过 HTTP 响应报文发送的。每当客户端向服务器发送一个 HTTP 请求时,Cookie都会随着请求被一同发送至服务器。net/http处理Cookie分为:将 Cookie发送至浏览器、从浏览器获取 Cookie。我们将这两种处理方式分别在函数setCookie和getCookie实现,代码如下所示。
package main
import (
"fmt"
"net/http"
)
func setCookie(w http.ResponseWriter, r *http.Request) {
// 设置Cookie
c1 := http.Cookie{
Name:"first_cookie",
Value:"Go Web Programming",
HttpOnly: true,
}
c2 := http.Cookie{
Name: "second_cookie",
Value: "Manning Publications Co",
HttpOnly: true,
}
// 方法1:使用Header方法设置参数Set-Cookie即可设置Cookie
//w.Header().Set("Set-Cookie", c1.String())
//w.Header().Add("Set-Cookie", c2.String())
// 方法2:使用SetCookie方法设置Cookie
http.SetCookie(w, &c1)
http.SetCookie(w, &c2)
w.Write([]byte("<p>Done</p>"))
}
func getCookie(w http.ResponseWriter, r *http.Request) {
// 方法1:获取全部Cookie信息,以切片类型表示
//h := r.Header["Cookie"]
// 方法2:获取全部Cookie信息,以切片类型表示
//h := r.Cookies()
// 获取某个Cookie信息
h, _ := r.Cookie("first_cookie")
fmt.Fprintln(w, h)
}
func main() {
server := http.Server{
Addr: "127.0.0.1:8080",
}
http.HandleFunc("/set_cookie", setCookie)
http.HandleFunc("/get_cookie", getCookie)
server.ListenAndServe()
}
上述代码中,我们简单演示了Cookie的应用,在实际开发中,当我们设置一条Cookie的时候,可以将某些数据进行加密并设置在Cookie的参数value里面,这样可以实现用户数据的临时存储和标记。
我们知道Session比Cookie的安全系数要高,而且存储的数据也比Cookie要多,但是net/http没有提供Session的功能函数,因此想要在net/http使用Session,需要开发者自己敲代码实现了。
综合上述,本文简单讲述如何使用net/http实现网站响应和Cookies设置,下一节将讲述如何使用模版引擎(html/template)。