yangy
85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
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": "操作成功",
|
|
})
|
|
}
|