huoxue1
/
study_xxqg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改小bug

This commit is contained in:
johlanse 2022-09-10 18:04:38 +08:00
parent 3850c224c9
commit 07239a556b
2 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model/wechat_user.go
View File

@ -1,5 +1,7 @@
package model package model
import "database/sql"
type WechatUser struct { type WechatUser struct {
OpenID string `json:"open_id"` OpenID string `json:"open_id"`
Remark string `json:"remark"` Remark string `json:"remark"`
@ -33,6 +35,9 @@ func FindWechatUser(openID string) (*WechatUser, error) {
ping() ping()
w := new(WechatUser) w := new(WechatUser)
err := db.QueryRow(`select * from wechat_user where open_id=?;`, openID).Scan(&w.OpenID, &w.Remark, &w.Status, &w.LastRequestTime) err := db.QueryRow(`select * from wechat_user where open_id=?;`, openID).Scan(&w.OpenID, &w.Remark, &w.Status, &w.LastRequestTime)
if err == sql.ErrNoRows {
return nil, err
}
return w, err return w, err
} }

30
push/wx.go
View File

@ -1,6 +1,7 @@
package push package push
import ( import (
"database/sql"
"encoding/json" "encoding/json"
"fmt" "fmt"
"net/http" "net/http"
@ -273,21 +274,26 @@ func handleTextPass(id, msg string) {
* @param msg * @param msg
*/ */
func handleEventUseRequest(id, msg string) { func handleEventUseRequest(id, msg string) {
count := model.WechatUserCount(id) user, err := model.FindWechatUser(id)
if count < 0 { if err != nil {
err := model.AddWechatUser(&model.WechatUser{ if err == sql.ErrNoRows {
OpenID: id, err := model.AddWechatUser(&model.WechatUser{
Remark: "", OpenID: id,
Status: 0, Remark: "",
LastRequestTime: time.Now().Unix(), Status: 0,
}) LastRequestTime: time.Now().Unix(),
if err != nil { })
log.Errorln("添加用户出现错误" + err.Error()) if err != nil {
log.Errorln("添加用户出现错误" + err.Error())
return
}
sendMsg(conf.GetConfig().Wechat.SuperOpenID, fmt.Sprintf("用户%v申请使用测试号，通过则回复信息：\n通过 %v\n\n拒绝则回复:\n拒绝 %v", id, id, id))
} else {
log.Errorln("查询用户出现未知错误" + err.Error())
return return
} }
sendMsg(conf.GetConfig().Wechat.SuperOpenID, fmt.Sprintf("用户%v申请使用测试号，通过则回复信息：\n通过 %v\n\n拒绝则回复:\n拒绝 %v", id, id, id))
} else { } else {
user, err := model.FindWechatUser(id)
if err != nil { if err != nil {
log.Errorln("查询wechat用户错误" + err.Error()) log.Errorln("查询wechat用户错误" + err.Error())
return return