TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-12-05 08:30:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Websocket信息变更时通知更新

Browse Source
This commit is contained in:
GoEdgeLab
2021-01-19 13:14:25 +08:00
parent fa9dcc2f04
commit 5d021ec0b5
4 changed files with 44 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
internal/db/models/http_websocket_dao.go
View File

@@ -47,12 +47,15 @@ func (this *HTTPWebsocketDAO) EnableHTTPWebsocket(tx *dbs.Tx, id int64) error {
}
// 禁用条目
func (this *HTTPWebsocketDAO) DisableHTTPWebsocket(tx *dbs.Tx, id int64) error {
func (this *HTTPWebsocketDAO) DisableHTTPWebsocket(tx *dbs.Tx, websocketId int64) error {
_, err := this.Query(tx).
Pk(id).
Pk(websocketId).
Set("state", HTTPWebsocketStateDisabled).
Update()
return err
if err != nil {
return err
}
return this.NotifyUpdate(tx, websocketId)
}
// 查找启用中的条目
@@ -150,5 +153,20 @@ func (this *HTTPWebsocketDAO) UpdateWebsocket(tx *dbs.Tx, websocketId int64, han
op.RequestSameOrigin = requestSameOrigin
op.RequestOrigin = requestOrigin
err := this.Save(tx, op)
return err
if err != nil {
return err
}
return this.NotifyUpdate(tx, websocketId)
}
// 通知更新
func (this *HTTPWebsocketDAO) NotifyUpdate(tx *dbs.Tx, websocketId int64) error {
webId, err := SharedHTTPWebDAO.FindEnabledWebIdWithWebsocketId(tx, websocketId)
if err != nil {
return err
}
if webId > 0 {
return SharedHTTPWebDAO.NotifyUpdate(tx, webId)
}
return nil
}