zishuo/message-pusher
1
0
Fork 0
mirror of https://github.com/songquanpeng/message-pusher.git synced 2024-05-03 08:45:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix IsShared implementation

Browse Source
This commit is contained in:
JustSong 2023-05-06 11:49:56 +08:00
parent 3e51538fdf
commit 581d9d62dc
4 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
channel/token-store.go
View File

@ -152,6 +152,7 @@ func TokenStoreAddChannel(channel *model.Channel) {
}
item := channel2item(channel)
if item != nil {
// Do not check IsShared here, cause its useless
TokenStoreAddItem(item)
}
}
@ -161,7 +162,7 @@ func TokenStoreRemoveChannel(channel *model.Channel) {
return
}
item := channel2item(channel)
if item != nil {
if item != nil && !item.IsShared() {
TokenStoreRemoveItem(item)
}
}
@ -170,6 +171,7 @@ func TokenStoreUpdateChannel(newChannel *model.Channel, oldChannel *model.Channe
if oldChannel.Type != model.TypeWeChatTestAccount && oldChannel.Type != model.TypeWeChatCorpAccount {
return
}
// Why so complicated? Because the given channel maybe incomplete.
if oldChannel.Type == model.TypeWeChatTestAccount {
// Only keep changed parts
if newChannel.AppId == oldChannel.AppId {

5
channel/wechat-corp-account.go
View File

@ -32,8 +32,9 @@ func (i *WeChatCorpAccountTokenStoreItem) Key() string {
func (i *WeChatCorpAccountTokenStoreItem) IsShared() bool {
appId := fmt.Sprintf("%s|%s", i.CorpId, i.AgentId)
return model.DB.Where("type = ? and app_id = ? and secret = ?",
model.TypeWeChatCorpAccount, appId, i.AgentSecret).Find(&model.Channel{}).RowsAffected != 1
var count int64 = 0
model.DB.Model(&model.Channel{}).Where("type = ? and app_id = ? and secret = ?", model.TypeWeChatCorpAccount, appId, i.AgentSecret).Count(&count)
return count > 1
}
func (i *WeChatCorpAccountTokenStoreItem) IsFilled() bool {

5
channel/wechat-test-account.go
View File

@ -29,8 +29,9 @@ func (i *WeChatTestAccountTokenStoreItem) Key() string {
}
func (i *WeChatTestAccountTokenStoreItem) IsShared() bool {
return model.DB.Where("type = ? and app_id = ? and secret = ?",
model.TypeWeChatTestAccount, i.AppID, i.AppSecret).Find(&model.Channel{}).RowsAffected != 1
var count int64 = 0
model.DB.Model(&model.Channel{}).Where("type = ? and app_id = ? and secret = ?", model.TypeWeChatTestAccount, i.AppID, i.AppSecret).Count(&count)
return count > 1
}
func (i *WeChatTestAccountTokenStoreItem) IsFilled() bool {

1
controller/user.go
View File

@ -604,6 +604,7 @@ func ManageUser(c *gin.Context) {
})
return
}
channel.TokenStoreRemoveUser(&user)
case "promote":
if myRole != common.RoleRootUser {
c.JSON(http.StatusOK, gin.H{