set device name

Author: huangzhiran

api/http.go

@@ -47,8 +47,9 @@ type pubkeyResp struct {
 }
 
 type queryReq struct {
-	DeviceID  string `json:"deviceID"                   binding:"required"`
-	Signature string `json:"signature,omitempty"        binding:"required"`
+	DeviceID   string `json:"deviceID"                   binding:"required"`
+	DeviceIMEI string `json:"deviceIMEI,omitempty"`
+	Signature  string `json:"signature,omitempty"        binding:"required"`
 }
 
 type queryResp struct {
@@ -141,6 +142,14 @@ func (s *httpServer) query(c *gin.Context) {
 		return
 	}
 
+	if req.DeviceIMEI != "" {
+		if err := s.handleName(strings.ToLower(req.DeviceID), req.DeviceIMEI); err != nil {
+			slog.Error("failed to update device name", "device_id", req.DeviceID)
+			c.JSON(http.StatusBadRequest, newErrResp(errors.Wrap(err, "failed to update device name")))
+			return
+		}
+	}
+
 	metrics.TrackDeviceCount(req.DeviceID)
 	metrics.TrackRequestCount("get")
 	now := time.Now()
@@ -495,6 +504,14 @@ func (s *httpServer) handle(id string, pkg *proto.BinPackage, data goproto.Messa
 	return errors.Wrapf(err, "failed to handle %T", data)
 }
 
+func (s *httpServer) handleName(id, name string) error {
+	err := s.db.UpdateByID(id, map[string]any{
+		"name":       name,
+		"updated_at": time.Now(),
+	})
+	return errors.Wrapf(err, "failed to update device name: %s", id)
+}
+
 func (s *httpServer) handleConfig(id string, data *proto.SensorConfig) error {
 	err := s.db.UpdateByID(id, map[string]any{
 		"bulk_upload":               int32(data.GetBulkUpload()),