GoLang—使用net/http构建Web服务（网站响应和Cookies）

﹏ヽ暗。殇╰゛Y 2024-04-20 09:32 45阅读 0赞

在上一节中，我们讲述了如何[使用net/http接收和处理用户请求][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）等等，响应头的内容如下图所示。  
    ![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly94eWh1YW5nLmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70]

大致了解参数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）。

[net_http]: https://xyhuang.blog.csdn.net/article/details/101025831
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly94eWh1YW5nLmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/2ed371f2312e49409248161a5263639e.png