From 5b1a138ca1c24e5fbfc86aa67e6b54cf6f59275f Mon Sep 17 00:00:00 2001 From: fuzhongyun <15339891972@163.com> Date: Thu, 22 Jan 2026 13:33:53 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E8=B0=83=E6=95=B4=E9=92=89?= =?UTF-8?q?=E9=92=89=E6=9C=BA=E5=99=A8=E4=BA=BA=E5=9B=9E=E8=B0=83=E4=B8=B4?= =?UTF-8?q?=E6=97=B6=E8=BD=AC=E5=8F=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/server/router/router.go | 2 +- internal/services/callback.go | 18 +++--------------- 2 files changed, 4 insertions(+), 16 deletions(-) diff --git a/internal/server/router/router.go b/internal/server/router/router.go index 349a1ec..d8c8c73 100644 --- a/internal/server/router/router.go +++ b/internal/server/router/router.go @@ -69,7 +69,7 @@ func SetupRoutes(app *fiber.App, ChatService *services.ChatService, sessionServi // 回调 r.Post("/callback", callbackService.Callback) // 钉钉机器人回调 - r.Get("/callback/dingtalk-robot", callbackService.CallbackDingtalkRobot) + r.Post("/callback/dingtalk-robot", callbackService.CallbackDingtalkRobot) // 企业微信回调 r.Get("/qywx/callback", callbackService.QywxCallback) //广播 diff --git a/internal/services/callback.go b/internal/services/callback.go index dde5a97..864d1f2 100644 --- a/internal/services/callback.go +++ b/internal/services/callback.go @@ -367,21 +367,9 @@ func getString(str, endstr string, start int, msg *string) int { // CallbackDingtalkRobot 钉钉机器人回调 func (s *CallbackService) CallbackDingtalkRobot(c *fiber.Ctx) (err error) { - // query 中获取 dev 参数 - query := c.Request().URI().QueryString() - str, _ := url.QueryUnescape(string(query)) - // 转map - params := make(map[string]string) - for _, param := range strings.Split(str, "&") { - kv := strings.Split(param, "=") - if len(kv) == 2 { - params[kv[0]] = kv[1] - } - } - // token 校验 - isDev := params["dev"] - - if isDev != "1" { + // header 中获取 X-IsDev 参数 + isDev := c.Request().Header.Peek("X-IsDev") + if string(isDev) != "1" { // 非开发环境,直接返回 return nil }