建立仓库

actions/edges.go

@@ -0,0 +1,73 @@
+package actions
+
+import (
+	"fmt"
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+	"zzman/model"
+)
+
+const batchSize = 1000
+
+func FindEdgesByCity(tx *gorm.DB, cityId int) ([]model.Edge, error) {
+	var edges []model.Edge
+	err := tx.Find(&edges, "city_id = ?", cityId).Error
+	if err != nil {
+		return nil, fmt.Errorf("failed to find edges: %w", err)
+	}
+	return edges, nil
+}
+
+func SliceActiveEdgesByCity(tx *gorm.DB, cityId int, offset int, limit int) ([]model.Edge, error) {
+	var edges []model.Edge
+	err := tx.Limit(limit).Offset(offset).Find(&edges, "city_id = ? and active = 1", cityId).Error
+	if err != nil {
+		return nil, fmt.Errorf("failed to find edges with offset: %w", err)
+	}
+	return edges, nil
+}
+
+func SaveEdges(tx *gorm.DB, edges []model.Edge) error {
+	if len(edges) == 0 {
+		return nil
+	}
+
+	// 分批处理边缘设备数据
+	for i := 0; i < len(edges); i += batchSize {
+		end := i + batchSize
+		if end > len(edges) {
+			end = len(edges)
+		}
+
+		batch := edges[i:end]
+		err := tx.Clauses(clause.OnConflict{
+			Columns:   []clause.Column{{Name: "macaddr"}},
+			UpdateAll: true,
+		}).Create(&batch).Error
+		if err != nil {
+			return fmt.Errorf("failed to save edges batch %d-%d: %w", i, end-1, err)
+		}
+	}
+	return nil
+}
+
+func DisableEdgesByMacs(tx *gorm.DB, macs []string) error {
+	if len(macs) == 0 {
+		return nil
+	}
+
+	// 分批处理MAC地址列表
+	for i := 0; i < len(macs); i += batchSize {
+		end := i + batchSize
+		if end > len(macs) {
+			end = len(macs)
+		}
+
+		batch := macs[i:end]
+		err := tx.Model(&model.Edge{}).Where("macaddr IN ?", batch).Update("active", false).Error
+		if err != nil {
+			return fmt.Errorf("failed to disable edges batch %d-%d: %w", i, end-1, err)
+		}
+	}
+	return nil
+}