fix: 修复QQ存在的panic风险

This commit is contained in:
mei
2025-10-06 22:15:35 +08:00
parent a253e15e95
commit e2b68fef1e
5 changed files with 44 additions and 16 deletions
+36 -13
View File
@@ -3,11 +3,11 @@ package internal
import (
"bytes"
"fmt"
"io"
"log"
"mime/multipart"
"net/http"
"strconv"
"time"
"github.com/ssdomei232/goodBaby/configs"
)
@@ -16,6 +16,7 @@ func SendQQ() {
config, err := configs.GetConfig()
if err != nil {
log.Printf("获取配置文件失败: %v", err)
return // 添加return避免继续执行
}
url := fmt.Sprintf("%s/?secret=%s", config.CatBotUrl, config.CatBotKey)
@@ -24,43 +25,65 @@ func SendQQ() {
for _, groupId := range config.QQSendGroup {
sendQQMsg(url, groupId, stopMsg)
}
}
func sendQQMsg(url string, groupId int, msg string) {
method := "POST"
payload := &bytes.Buffer{}
writer := multipart.NewWriter(payload)
_ = writer.WriteField("group_id", strconv.Itoa(groupId))
_ = writer.WriteField("message", msg)
// 处理写入字段的错误
if err := writer.WriteField("group_id", strconv.Itoa(groupId)); err != nil {
log.Printf("写入group_id字段失败: %v", err)
return
}
if err := writer.WriteField("message", msg); err != nil {
log.Printf("写入message字段失败: %v", err)
return
}
err := writer.Close()
if err != nil {
log.Println(err)
log.Printf("关闭multipart writer失败: %v", err)
return
}
client := &http.Client{}
req, err := http.NewRequest(method, url, payload)
if err != nil {
fmt.Println(err)
log.Printf("创建HTTP请求失败: %v", err)
return
}
req.Header.Set("Content-Type", writer.FormDataContentType())
var res *http.Response
var maxRetries = 3
for i := range maxRetries {
res, err = client.Do(req)
if err == nil {
break // 请求成功,跳出循环
}
log.Printf("请求QQ API失败,正在重试... (%d/%d)", i+1, maxRetries)
if i < maxRetries {
time.Sleep(retryDelay * time.Duration(i+1)) // 递增延迟
}
log.Printf("请求QQ API失败,正在重试... (%d/%d): %v", i+1, maxRetries, err)
}
defer res.Body.Close()
_, err = io.ReadAll(res.Body)
if err != nil {
fmt.Println(err)
// 检查res是否为nil,避免panic
if res == nil {
log.Println("QQ API请求失败,无法获取响应")
return
}
// 使用defer确保资源被释放,但要确保res不为nil
defer func() {
if res != nil && res.Body != nil {
res.Body.Close()
}
}()
// 可选:记录响应结果
log.Println("QQ消息发送完成")
}