TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-20 04:50:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

waf/ip名单有变更立即发送通知

Browse Source
This commit is contained in:
刘祥超
2020-11-10 09:22:17 +08:00
parent 1a6b05f81b
commit c96e32e257
2 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/nodes/api_stream.go
View File

@@ -9,6 +9,7 @@ import (
"github.com/TeaOSLab/EdgeNode/internal/caches"
"github.com/TeaOSLab/EdgeNode/internal/errors"
"github.com/TeaOSLab/EdgeNode/internal/events"
"github.com/TeaOSLab/EdgeNode/internal/iplibrary"
"github.com/TeaOSLab/EdgeNode/internal/logs"
"github.com/TeaOSLab/EdgeNode/internal/rpc"
"io"
@@ -95,6 +96,8 @@ func (this *APIStream) loop() error {
err = this.handlePreheatCache(message)
case messageconfigs.MessageCodeConfigChanged: // 配置变化
err = this.handleConfigChanged(message)
case messageconfigs.MessageCodeIPListChanged: // IPList变化
err = this.handleIPListChanged(message)
default:
err = this.handleUnknownMessage(message)
}
@@ -438,6 +441,17 @@ func (this *APIStream) handleConfigChanged(message *pb.NodeStreamMessage) error
return nil
}
// 处理IPList变化
func (this *APIStream) handleIPListChanged(message *pb.NodeStreamMessage) error {
select {
case iplibrary.IPListUpdateNotify <- true:
default:
}
this.replyOk(message.RequestId, "ok")
return nil
}
// 处理未知消息
func (this *APIStream) handleUnknownMessage(message *pb.NodeStreamMessage) error {
this.replyFail(message.RequestId, "unknown message code '"+message.Code+"'")