功能设计-如何设计一个回调功能

package main

import (
    "fmt"
    "log"
    "net/http"
    "strings"
)

func main() {
    fmt.Println("启动服务A (监听 :8080)")

    // 设置HTTP路由
    http.HandleFunc("/start", startHandler)
    http.HandleFunc("/callback", callbackHandler)

    log.Fatal(http.ListenAndServe(":8080", nil))
}

// 开始请求处理
func startHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Println("服务A: 收到新请求")

    // 调用服务B
    go callServiceB()

    fmt.Fprint(w, "任务已提交，处理中...")
}

// 调用服务B
func callServiceB() {
    fmt.Println("服务A: 调用服务B")

    // 设置回调地址
    callbackURL := "http://localhost:8080/callback"

    // 构造服务B请求
    serviceBURL := "http://localhost:8081/task"
    payload := fmt.Sprintf(`{"callback_url": "%s"}`, callbackURL)

    // 发送HTTP请求
    _, err := http.Post(serviceBURL, "application/json",
       strings.NewReader(payload))

    if err != nil {
       log.Printf("调用服务B失败: %v", err)
    }
}

// 回调处理
func callbackHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Println("服务A: 收到服务B的回调")
    fmt.Println("服务A: 执行后续操作...")

    fmt.Fprint(w, "回调已接收")
}

package main

import (
    "encoding/json"
    "fmt"
    "log"
    "net/http"
    "strings"
    "time"
)

func main() {
    fmt.Println("启动服务B (监听 :8081)")

    // 设置HTTP路由
    http.HandleFunc("/task", taskHandler)

    log.Fatal(http.ListenAndServe(":8081", nil))
}

// 请求结构体
type TaskRequest struct {
    CallbackURL string `json:"callback_url"`
}

// 任务处理
func taskHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Println("服务B: 收到新任务")

    // 解析JSON请求
    var req TaskRequest
    if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
       http.Error(w, "无效请求数据: "+err.Error(), http.StatusBadRequest)
       log.Printf("请求解析错误: %v", err)
       return
    }

    // 获取回调URL
    callbackURL := req.CallbackURL
    if callbackURL == "" {
       http.Error(w, "缺少回调URL", http.StatusBadRequest)
       return
    }

    fmt.Printf("服务B: 开始处理任务，回调URL: %s\n", callbackURL)

    // 异步处理任务
    go func() {
       // 模拟处理时间（3秒）
       time.Sleep(3 * time.Second)

       fmt.Println("服务B: 任务处理完成")

       // 发送回调通知
       sendCallback(callbackURL)
    }()

    fmt.Fprint(w, "任务已开始处理")
}

// 发送回调
func sendCallback(callbackURL string) {
    fmt.Printf("服务B: 发送回调至 %s\n", callbackURL)

    // 发送回调通知
    payload := `{"status": "完成", "message": "任务处理成功"}`
    _, err := http.Post(callbackURL, "application/json", strings.NewReader(payload))

    if err != nil {
       log.Printf("回调发送失败: %v", err)
       return
    }

    fmt.Println("服务B: 回调发送成功")
}