feat: 为 zbx 添加批量注册设备能力

integration/zabbixagent/gin_handler.go

@@ -3,7 +3,9 @@ package zabbixagent
 import (
 	"github.com/gin-gonic/gin"
 	"net/http"
-	"onvif-agent/service/onvif"
+	"onvif-agent/response"
+	handler "onvif-agent/router/handler/onvif"
+	service "onvif-agent/service/onvif"
 )
 
 func ONVIFDeviceDiscovery(c *gin.Context) {
@@ -12,7 +14,7 @@ func ONVIFDeviceDiscovery(c *gin.Context) {
 	}
 
 	arr := make([]ZBXDevice, 0)
-	for xaddr := range onvif.Sessions {
+	for xaddr := range service.Sessions {
 		arr = append(arr, ZBXDevice{
 			Xaddr: xaddr,
 		})
@@ -22,3 +24,29 @@ func ONVIFDeviceDiscovery(c *gin.Context) {
 		"data": arr,
 	})
 }
+
+func ONVIFDeviceRegister(c *gin.Context) {
+	var req []handler.CreateSessionRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.NewResponse().Error(err).Send(c)
+		return
+	}
+
+	for _, v := range req {
+		session, err := service.NewSession(v.Xaddr, v.Username, v.Password)
+		if err != nil {
+			response.NewResponse().Error(err).Send(c)
+			return
+		}
+
+		_, err = session.GetDeviceInfo()
+		if err != nil {
+			response.NewResponse().Error(err).Send(c)
+			return
+		}
+
+		service.Sessions[session.Device.GetDeviceParams().Xaddr] = session
+	}
+
+	response.NewResponse().Success().Send(c)
+}

router/router.go

@@ -35,5 +35,6 @@ func SetupRoutes(r *gin.Engine) {
 	zabbixGroup := r.Group("/zabbix")
 	{
 		zabbixGroup.POST("/onvifDeviceDiscovery", zabbixagent.ONVIFDeviceDiscovery)
+		zabbixGroup.POST("/onvifDeviceRegister", zabbixagent.ONVIFDeviceRegister)
 	}
 }