在 Golang 中如何验证 HTTP 响应状态代码？

在 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)
  }

  // 解析响应并返回数据
}

通过验证响应状态代码，你可以确保只有在服务器成功处理请求时才返回数据。