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

优化代码/商业版支持L2节点

Browse Source
This commit is contained in:
GoEdgeLab
2022-04-04 12:06:53 +08:00
parent 144bef4fe3
commit 4aeb3cc7b0
10 changed files with 242 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/nodes/origin_state_manager.go
View File

@@ -124,9 +124,10 @@ func (this *OriginStateManager) Loop() error {
// Fail 添加失败的源站
func (this *OriginStateManager) Fail(origin *serverconfigs.OriginConfig, reverseProxy *serverconfigs.ReverseProxyConfig, callback func()) {
if origin == nil {
if origin == nil || origin.Id <= 0 {
return
}
this.locker.Lock()
state, ok := this.stateMap[origin.Id]
var timestamp = time.Now().Unix()
@@ -164,7 +165,7 @@ func (this *OriginStateManager) Fail(origin *serverconfigs.OriginConfig, reverse
// Success 添加成功的源站
func (this *OriginStateManager) Success(origin *serverconfigs.OriginConfig, callback func()) {
if origin == nil {
if origin == nil || origin.Id <= 0 {
return
}
@@ -182,6 +183,10 @@ func (this *OriginStateManager) Success(origin *serverconfigs.OriginConfig, call
// IsAvailable 检查是否正常
func (this *OriginStateManager) IsAvailable(originId int64) bool {
if originId <= 0 {
return true
}
this.locker.RLock()
_, ok := this.stateMap[originId]
this.locker.RUnlock()