在 golang 中验证 http 响应状态代码的方法有:使用 http.response.statuscode,直接比较响应状态码与预期值。创建自定义错误类型,包含状态码和错误消息,并通过捕获错误来验证状态码。
如何在 Golang 中验证 HTTP 响应状态代码
在 Golang 中使用 net/http 包发送 HTTP 请求时,验证响应状态代码非常重要。这有助于处理服务器端错误、配置问题和其他网络问题。
使用 http.Response.StatusCode
http.Response 类型包含一个 StatusCode 字段,表示 HTTP 响应的状态代码。要验证状态代码,可以简单地比较 StatusCode 与预期的值:
package main
import (
"fmt"
"net/http"
"net/http/httptest"
)
func main() {
// 创建一个模拟响应
resp := httptest.NewRecorder()
// 设置状态代码
resp.WriteHeader(http.StatusOK)
if resp.Code == http.StatusOK {
fmt.Println("成功")
}
}
使用自定义错误类型
另一种验证响应状态代码的方法是创建一个自定义错误类型,其中包含状态代码和错误消息:
package main
import (
"errors"
"fmt"
"net/http"
)
type HTTPError struct {
Code int
Msg string
}
func (e HTTPError) Error() string {
return fmt.Sprintf("HTTP Error: %d - %s", e.Code, e.Msg)
}
func main() {
// 发送请求并捕获响应
resp, err := http.Get("https://example.com")
if err != nil {
fmt.Println(err)
return
}
if resp.StatusCode != http.StatusOK {
// 创建自定义错误
err = HTTPError{Code: resp.StatusCode, Msg: "请求失败"}
fmt.Println(err)
}
}
实战案例
以下是使用 http.Response.StatusCode 验证状态代码的实战案例:
假设你正在开发一个 API,该 API 使用 HTTP GET 从数据库中检索数据。你需要验证响应的状态代码以确保请求成功完成:
func GetUserData(userID int) (UserData, error) {
resp, err := http.Get(fmt.Sprintf("https://myapi.com/users/%d", userID))
if err != nil {
return UserData{}, err
}
if resp.StatusCode != http.StatusOK {
return UserData{}, fmt.Errorf("请求失败: %d", resp.StatusCode)
}
// 解析响应并返回数据
}
通过验证响应状态代码,你可以确保只有在服务器成功处理请求时才返回数据。