优化代码

server/service/backup.go

@@ -220,7 +220,7 @@ func (service backupService) Import(backup *dto.Backup) error {
 					Rule:     item.Rule,
 					Priority: item.Priority,
 				}
-				security.GlobalSecurityManager.Add <- rule
+				security.GlobalSecurityManager.Add(rule)
 			}
 		}
 

server/service/gateway.go

@@ -60,11 +60,11 @@ func (r gatewayService) ReConnect(m *model.AccessGateway) *gateway.Gateway {
 	} else {
 		g = gateway.NewGateway(m.ID, true, "", sshClient)
 	}
-	gateway.GlobalGatewayManager.Add <- g
+	gateway.GlobalGatewayManager.Add(g)
 	log.Debugf("重建接入网关「%v」完成", m.Name)
 	return g
 }
 
 func (r gatewayService) DisconnectById(accessGatewayId string) {
-	gateway.GlobalGatewayManager.Del <- accessGatewayId
+	gateway.GlobalGatewayManager.Del(accessGatewayId)
 }

server/service/security.go

@@ -25,7 +25,7 @@ func (service securityService) ReloadAccessSecurity() error {
 				Rule:     rules[i].Rule,
 				Priority: rules[i].Priority,
 			}
-			security.GlobalSecurityManager.Add <- rule
+			security.GlobalSecurityManager.Add(rule)
 		}
 	}
 	return nil

server/service/session.go

@@ -96,14 +96,13 @@ func (service sessionService) CloseSessionById(sessionId string, code int, reaso
 		service.WriteCloseMessage(nextSession, nextSession.Mode, code, reason)
 
 		if nextSession.Observer != nil {
-			obs := nextSession.Observer.All()
-			for _, ob := range obs {
+			nextSession.Observer.Range(func(key string, ob *session.Session) {
 				service.WriteCloseMessage(ob, ob.Mode, code, reason)
 				log.Debugf("[%v] 强制踢出会话的观察者: %v", sessionId, ob.ID)
-			}
+			})
 		}
 	}
-	session.GlobalSessionManager.Del <- sessionId
+	session.GlobalSessionManager.Del(sessionId)
 
 	service.DisDBSess(sessionId, code, reason)
 }