Go App Engine中urlfetch进行POST请求的实践指南

package myapp

import (
    "bytes"
    "context"
    "encoding/json"
    "fmt"
    "io/ioutil"
    "net/http"

    "google.golang.org/appengine"
    "google.golang.org/appengine/urlfetch"
)

// RequestPayload 定义了示例POST请求的JSON结构
type RequestPayload struct {
    Key    string `json:"key"`
    Number int    `json:"number"`
}

// MakePostRequest 演示如何在GAE Go中执行POST请求
// ctx: App Engine上下文
// targetURL: 目标URL
// payload: POST请求的原始数据体 (例如JSON字节数组)
func MakePostRequest(ctx context.Context, targetURL string, payload []byte) (string, error) {
    // 1. 使用urlfetch.Client创建http客户端
    // 这是在GAE环境中进行外部HTTP请求的关键一步。
    client := urlfetch.Client(ctx)

    // 2. 准备POST请求
    // http.NewRequest 允许我们完全控制请求的各个方面，包括方法、URL和请求体。
    req, err := http.NewRequest(http.MethodPost, targetURL, bytes.NewBuffer(payload))
    if err != nil {
        return "", fmt.Errorf("创建POST请求失败: %w", err)
    }

    // 3. 设置请求头
    // 对于POST请求，Content-Type头通常是必需的，它告诉服务器请求体的格式。
    // 这里我们假设发送的是JSON数据。
    req.Header.Set("Content-Type", "application/json")
    req.Header.Set("Accept", "application/json") // 示例：可以根据需要添加其他头

    // 4. 发送请求
    // 使用 client.Do 发送请求。这个客户端已经由urlfetch包装，因此请求会通过GAE的urlfetch服务发送。
    resp, err := client.Do(req)
    if err != nil {
        // 这里的错误可能包括网络问题、DNS解析失败或urlfetch服务本身的错误。
        return "", fmt.Errorf("发送POST请求失败: %w", err)
    }
    defer resp.Body.Close() // 确保关闭响应体，释放资源

    // 5. 读取响应体
    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        return "", fmt.Errorf("读取响应体失败: %w", err)
    }

    // 6. 检查HTTP状态码
    // 2xx系列状态码通常表示成功。非2xx状态码应被视为错误。
    if resp.StatusCode < 200 || resp.StatusCode >= 300 {
        return "", fmt.Errorf("请求返回非成功状态码: %d, 响应: %s", resp.StatusCode, string(body))
    }

    return string(body), nil
}

// ExampleHandler 是一个GAE HTTP处理函数，演示如何调用MakePostRequest
func ExampleHandler(w http.ResponseWriter, r *http.Request) {
    // 获取App Engine上下文。在GAE HTTP处理函数中，这是从请求中获取的。
    ctx := appengine.NewContext(r)

    // 目标URL和请求负载
    targetURL := "https://jsonplaceholder.typicode.com/posts" // 示例公共API

    data := RequestPayload{
        Key:    "my_data",
        Number: 42,
    }
    jsonPayload, err := json.Marshal(data)
    if err != nil {
        http.Error(w, fmt.Sprintf("序列化JSON失败: %v", err), http.StatusInternalServerError)
        return
    }

    // 调用MakePostRequest函数
    responseBody, err := MakePostRequest(ctx, targetURL, jsonPayload)
    if err != nil {
        http.Error(w, fmt.Sprintf("POST请求执行失败: %v", err), http.StatusInternalServerError)
        return
    }

    // 成功处理并返回响应
    w.Header().Set("Content-Type", "application/json")
    w.WriteHeader(http.StatusOK)
    fmt.Fprintf(w, "POST请求成功，响应: %s", responseBody)
}

// 在GAE应用的init函数中注册handler
func init() {
    http.HandleFunc("/make-post-request", ExampleHandler)
}