refactor

notify.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"time"
+)
+
+// BarkPushPayload 定义 Bark 推送的请求 JSON 结构（单 device_key）
+type BarkPushPayload struct {
+	Title     string `json:"title"`
+	Body      string `json:"body"`
+	Group     string `json:"group,omitempty"`
+	DeviceKey string `json:"device_key"`
+	IsArchive string `json:"isArchive,omitempty"`
+	TTL       int    `json:"ttl,omitempty"`
+}
+
+// SendNotification 依次对每个 device_key 发送 Bark 通知，遇错即返回
+func SendNotification(pushURL string, deviceKeys []string, title, body string) error {
+	for _, key := range deviceKeys {
+		if err := sendOne(pushURL, key, title, body); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// sendOne 向单个 device_key 发送推送
+func sendOne(pushURL, deviceKey, title, body string) error {
+	payload := BarkPushPayload{
+		Title:     title,
+		Body:      body,
+		Group:     "Model Development Platform",
+		DeviceKey: deviceKey,
+		IsArchive: "1",
+		TTL:       3600,
+	}
+
+	jsonBytes, err := json.Marshal(payload)
+	if err != nil {
+		return fmt.Errorf("JSON 序列化失败: %w", err)
+	}
+
+	req, err := http.NewRequest("POST", pushURL, bytes.NewBuffer(jsonBytes))
+	if err != nil {
+		return fmt.Errorf("创建请求失败: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json; charset=utf-8")
+
+	client := &http.Client{Timeout: 5 * time.Second}
+	resp, err := client.Do(req)
+	if err != nil {
+		return fmt.Errorf("发送推送请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode >= 400 {
+		bodyBytes, _ := io.ReadAll(io.LimitReader(resp.Body, 512))
+		bodyText := string(bytes.TrimSpace(bodyBytes))
+		if bodyText != "" {
+			return fmt.Errorf("推送服务器返回异常状态码: %d, 响应: %s", resp.StatusCode, bodyText)
+		}
+		return fmt.Errorf("推送服务器返回异常状态码: %d", resp.StatusCode)
+	}
+
+	return nil
+}