fix: 修复QQ存在的panic风险
This commit is contained in:
+36
-13
@@ -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消息发送完成")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user