首次提交

internal/handler/device.go

@@ -0,0 +1,84 @@
+package handler
+
+import (
+	"encoding/json"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/redis/go-redis/v9"
+	"go.uber.org/zap"
+)
+
+type DeviceHandler struct {
+	redis *redis.Client
+	log   *zap.Logger
+}
+
+func NewDeviceHandler(redis *redis.Client, log *zap.Logger) *DeviceHandler {
+	return &DeviceHandler{
+		redis: redis,
+		log:   log,
+	}
+}
+
+func (h *DeviceHandler) ReportDevInfo(c *gin.Context) {
+	mac := strings.TrimSpace(c.Query("mac"))
+	model := strings.TrimSpace(c.Query("model"))
+	ver := strings.TrimSpace(c.Query("ver"))
+	xForwardedFor := c.GetHeader("X-Forwarded-For")
+
+	if mac == "" || model == "" {
+		c.JSON(http.StatusOK, gin.H{
+			"code": 400,
+			"msg":  "参数校验失败",
+		})
+		return
+	}
+
+	if ver == "" {
+		ver = ""
+	}
+	if xForwardedFor == "" {
+		xForwardedFor = ""
+	}
+
+	h.log.Info("report dev info",
+		zap.String("remote_ip", xForwardedFor),
+		zap.String("time", time.Now().Format("yyyy-MM-dd HH:mm:ss")),
+	)
+
+	deviceInfo := map[string]string{
+		"mac_addr":    mac,
+		"model":       model,
+		"active_date": time.Now().Format("yyyy-MM-dd"),
+		"ip_addr":     xForwardedFor,
+		"ver":         ver,
+	}
+
+	jsonData, err := json.Marshal(deviceInfo)
+	if err != nil {
+		h.log.Error("marshal device info failed", zap.Error(err))
+		c.JSON(http.StatusOK, gin.H{
+			"code": 500,
+			"msg":  "系统错误",
+		})
+		return
+	}
+
+	ctx := c.Request.Context()
+	if err := h.redis.HSet(ctx, "devices", mac, jsonData).Err(); err != nil {
+		h.log.Error("redis hset failed", zap.Error(err))
+		c.JSON(http.StatusOK, gin.H{
+			"code": 500,
+			"msg":  "系统错误",
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"code": 200,
+		"msg":  "操作成功",
+	})
+}