72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
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
|
||
}
|