fix: zbx 模式下 sessions 的管理

2024-08-27 09:33:23 +08:00 · 2024-08-27 09:33:23 +08:00 · a641d6846d
commit a641d6846d
parent 67de79e230
1 changed files with 6 additions and 1 deletions
--- a/integration/zabbixagent/gin_handler.go
+++ b/integration/zabbixagent/gin_handler.go
@ -32,6 +32,9 @@ func ONVIFDeviceRegister(c *gin.Context) {
 		return
 	}

+	// 用于替换已存在的 Sessions
+	var sessions = make(map[string]*service.Session)
+
 	for _, v := range req {
 		session, err := service.NewSession(v.Xaddr, v.Username, v.Password)
 		if err != nil {
@ -45,8 +48,10 @@ func ONVIFDeviceRegister(c *gin.Context) {
 			return
 		}

-		service.Sessions[session.Device.GetDeviceParams().Xaddr] = session
+		sessions[session.Device.GetDeviceParams().Xaddr] = session
 	}

+	service.Sessions = sessions
+
 	response.NewResponse().Success().Send(c)
 }