Fix db_query field value to string

This commit is contained in:
若海
2024-01-12 09:02:19 +08:00
parent cd19e7d3f8
commit c9d8115c65
10 changed files with 102 additions and 61 deletions
+13 -13
View File
@@ -110,7 +110,7 @@ func (wc *Controller) dbSqlQuery(c *gin.Context) {
return
}
c.Set("Payload", wc.CmdClient.DbSqlQueryMap(req.Db, req.Sql))
c.Set("Payload", wc.CmdClient.DbSqlQuery(req.Db, req.Sql))
}
@@ -531,14 +531,14 @@ func (wc *Controller) receiveTransfer(c *gin.Context) {
}
// @Summary 开启转发消息到URL
// @Summary 开启推送消息到URL
// @Produce json
// @Param body body ForwardMsgRequest true "消息转发请求参数"
// @Param body body ReceiverRequest true "消息推送请求参数"
// @Success 200 {object} RespPayload
// @Router /enable_forward_msg [post]
func (wc *Controller) enableForwardMsg(c *gin.Context) {
// @Router /enable_receiver [post]
func (wc *Controller) enabledReceiver(c *gin.Context) {
var req ForwardMsgRequest
var req ReceiverRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.Set("Error", err)
return
@@ -549,7 +549,7 @@ func (wc *Controller) enableForwardMsg(c *gin.Context) {
return
}
err := wc.enableForwardToUrl(req.Url)
err := wc.enableUrlReceiver(req.Url)
c.Set("Payload", RespPayload{
Success: err == nil,
Error: err,
@@ -557,20 +557,20 @@ func (wc *Controller) enableForwardMsg(c *gin.Context) {
}
// @Summary 关闭转发消息到URL
// @Summary 关闭推送消息到URL
// @Produce json
// @Param body body ForwardMsgRequest true "消息转发请求参数"
// @Param body body ReceiverRequest true "消息推送请求参数"
// @Success 200 {object} RespPayload
// @Router /disable_forward_msg [post]
func (wc *Controller) disableForwardMsg(c *gin.Context) {
// @Router /disable_receiver [post]
func (wc *Controller) disableReceiver(c *gin.Context) {
var req ForwardMsgRequest
var req ReceiverRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.Set("Error", err)
return
}
err := wc.disableForwardToUrl(req.Url)
err := wc.disableUrlReceiver(req.Url)
c.Set("Payload", RespPayload{
Success: err == nil,
Error: err,
@@ -9,15 +9,15 @@ import (
"github.com/opentdp/wechat-rest/wcferry"
)
var forwardToUrlStat = false
var forwardToUrlList = map[string]bool{}
var urlReceiverStat = false
var urlReceiverList = map[string]bool{}
func (wc *Controller) enableForwardToUrl(url string) error {
func (wc *Controller) enableUrlReceiver(url string) error {
if !forwardToUrlStat {
if !urlReceiverStat {
err := wc.EnrollReceiver(true, func(msg *wcferry.WxMsg) {
ret := wcferry.WxMsgParser(msg)
for url := range forwardToUrlList {
ret := wcferry.ParseWxMsg(msg)
for url := range urlReceiverList {
logman.Info("forward msg", "url", url, "Id", ret.Id)
go request.JsonPost(url, ret, request.H{})
}
@@ -27,30 +27,30 @@ func (wc *Controller) enableForwardToUrl(url string) error {
}
}
if _, ok := forwardToUrlList[url]; ok {
if _, ok := urlReceiverList[url]; ok {
return errors.New("url already exists")
}
forwardToUrlStat = true
forwardToUrlList[url] = true
urlReceiverStat = true
urlReceiverList[url] = true
return nil
}
func (wc *Controller) disableForwardToUrl(url string) error {
func (wc *Controller) disableUrlReceiver(url string) error {
if _, ok := forwardToUrlList[url]; !ok {
if _, ok := urlReceiverList[url]; !ok {
return errors.New("url not exists")
}
delete(forwardToUrlList, url)
delete(urlReceiverList, url)
if len(forwardToUrlList) == 0 {
if len(urlReceiverList) == 0 {
if err := wc.DisableReceiver(false); err != nil {
return err
}
forwardToUrlStat = false
urlReceiverStat = false
}
return nil
+2 -2
View File
@@ -45,7 +45,7 @@ func Route(rg *gin.RouterGroup) {
rg.POST("accept_new_friend", ctrl.acceptNewFriend)
rg.POST("receive_transfer", ctrl.receiveTransfer)
rg.POST("enable_forward_msg", ctrl.enableForwardMsg)
rg.POST("disable_forward_msg", ctrl.disableForwardMsg)
rg.POST("enable_receiver", ctrl.enabledReceiver)
rg.POST("disable_receiver", ctrl.disableReceiver)
}
+1 -1
View File
@@ -14,7 +14,7 @@ type DbSqlQueryRequest struct {
}
// 消息转发参数
type ForwardMsgRequest struct {
type ReceiverRequest struct {
Url string `json:"url"`
}