异步任务调用流程

package main

import (
  "bytes"
  "encoding/json"
  "fmt"
  "io"
  "net/http"
  "os"
  "strconv"
  "time"
)

type mediaSource struct {
  URL string `json:"url"`
}

type imageRequest struct {
  Model string `json:"model"`
  Prompt string `json:"prompt,omitempty"`
  AspectRatio string `json:"aspect_ratio"`
  Resolution string `json:"resolution"`
}

type taskResponse struct {
  ID     string          `json:"id"`
  Status string          `json:"status"`
  Output json.RawMessage `json:"output"`
  Error  json.RawMessage `json:"error"`
}

func requestJSON(client *http.Client, method, url string, payload any) taskResponse {
  var body io.Reader
  if payload != nil {
    encoded, err := json.Marshal(payload)
    if err != nil { panic(err) }
    body = bytes.NewReader(encoded)
  }
  request, err := http.NewRequest(method, url, body)
  if err != nil { panic(err) }
  request.Header.Set("Authorization", "Bearer "+os.Getenv("FIREFLY_API_KEY"))
  request.Header.Set("Content-Type", "application/json")
  response, err := client.Do(request)
  if err != nil { panic(err) }
  defer response.Body.Close()
  raw, err := io.ReadAll(response.Body)
  if err != nil { panic(err) }
  if response.StatusCode < 200 || response.StatusCode >= 300 {
    fmt.Fprintln(os.Stderr, string(raw))
    os.Exit(1)
  }
  var result taskResponse
  if err := json.Unmarshal(raw, &result); err != nil { panic(err) }
  return result
}

func main() {
  client := &http.Client{Timeout: 60 * time.Second}
  task := requestJSON(client, http.MethodPost, "https://api.fireflyfusion.ai/v1/aigc/images", imageRequest{
    Model: "openai/gpt-image-2/text-to-image",
    Prompt: "夜晚森林中的萤火虫，电影感光影",
    AspectRatio: "1:1",
    Resolution: "1k",
  })
  intervalMS, _ := strconv.Atoi(os.Getenv("FIREFLY_POLL_INTERVAL_MS"))
  if intervalMS <= 0 { intervalMS = 2000 }

  for task.Status == "queued" || task.Status == "running" {
    time.Sleep(time.Duration(intervalMS) * time.Millisecond)
    task = requestJSON(client, http.MethodGet, "https://api.fireflyfusion.ai/v1/aigc/tasks/"+task.ID, nil)
  }
  if task.Status == "succeeded" {
    fmt.Println(string(task.Output))
    return
  }
  if task.Status == "failed" {
    fmt.Fprintln(os.Stderr, string(task.Error))
  } else {
    fmt.Fprintln(os.Stderr, "未知任务状态："+task.Status)
  }
  os.Exit(1)
}