refactor(router): 改为结构化 web 项目

router/handler/onvif/connection.go

@@ -0,0 +1,41 @@
+package onvif
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"onvif-agent/service/onvif"
+)
+
+var conns = make(map[string]*onvif.Connection)
+
+func CreateConnection(c *gin.Context) {
+	conn, err := onvif.NewConnection("172.16.19.239", "admin", "admin123")
+	if err != nil {
+		c.JSON(http.StatusServiceUnavailable, gin.H{
+			"message": err.Error(),
+		})
+		return
+	}
+
+	conns[conn.Device.GetDeviceParams().Xaddr] = conn
+
+	info, err := conn.GetDeviceInformation()
+	c.JSON(http.StatusOK, gin.H{
+		"connectionParams": conn.Params,
+		"deviceInfo":       info,
+	})
+}
+
+func GetConnections(c *gin.Context) {
+	c.JSON(http.StatusOK, conns)
+}
+
+func GetConnectionByXaddr(c *gin.Context) {
+	xaddr := c.Param("xaddr")
+	c.JSON(http.StatusOK, conns[xaddr])
+}
+
+func DeleteConnection(c *gin.Context) {
+	xaddr := c.Param("xaddr")
+	delete(conns, xaddr)
+}

router/handler/onvif/subscription.go

@@ -0,0 +1,38 @@
+package onvif
+
+import (
+	"github.com/IOTechSystems/onvif/event"
+	"github.com/IOTechSystems/onvif/gosoap"
+	"github.com/gin-gonic/gin"
+	"log"
+	"net/http"
+)
+
+func CreateEventSubscription(c *gin.Context) {
+	xaddr := c.Param("xaddr")
+	// FIXME: 把参数变成传入的参数
+	result, err := conns[xaddr].SubscribeEvents("http://172.16.19.230:8080/events/callback", "PT60S")
+	if err != nil {
+		c.JSON(http.StatusServiceUnavailable, gin.H{
+			"message": err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, result)
+}
+
+func EventNotifyCallback(c *gin.Context) {
+	var notify event.Notify
+	envelope := gosoap.NewSOAPEnvelope(&notify)
+	if err := c.ShouldBindXML(&envelope); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	for _, msg := range envelope.Body.Content.(*event.Notify).NotificationMessage {
+		log.Println(msg.Topic.TopicKinds, msg.Message.Message)
+	}
+
+	c.String(http.StatusOK, "OK")
+}

router/handler/onvif_handler.go

@@ -1,16 +0,0 @@
-package handler
-
-import (
-	"github.com/gin-gonic/gin"
-	"net/http"
-)
-
-func GetConnections(c *gin.Context) {
-	// 这里可以添加获取产品的逻辑
-	c.JSON(http.StatusOK, gin.H{"message": "Get Connections"})
-}
-
-func CreateConnection(c *gin.Context) {
-	// 这里可以添加创建产品的逻辑
-	c.JSON(http.StatusCreated, gin.H{"message": "Connection Created"})
-}

router/router.go

@@ -2,7 +2,7 @@ package router
 
 import (
 	"github.com/gin-gonic/gin"
-	"onvif-agent/router/handler"
+	"onvif-agent/router/handler/onvif"
 )
 
 func SetupRoutes(r *gin.Engine) {
@@ -11,8 +11,16 @@ func SetupRoutes(r *gin.Engine) {
 	{
 		connectionGroup := userGroup.Group("/connections")
 		{
-			connectionGroup.GET("/", handler.GetConnections)
-			connectionGroup.POST("/", handler.CreateConnection)
+			connectionGroup.POST("/", onvif.CreateConnection)
+			connectionGroup.GET("/", onvif.GetConnections)
+			connectionGroup.GET("/:xaddr", onvif.GetConnectionByXaddr)
+			connectionGroup.DELETE("/:xaddr", onvif.DeleteConnection)
+		}
+
+		subscriptionGroup := userGroup.Group("/subscriptions")
+		{
+			subscriptionGroup.POST("/:xaddr", onvif.CreateEventSubscription)
+			subscriptionGroup.POST("/callback", onvif.EventNotifyCallback)
 		}
 	}
 }