新增测点配置

model/init-db-data/init.go

@@ -21,7 +21,10 @@ func initDB(InitDBFunctions ...InitDBFunc) (err error) {
 }
 
 func InitDbData() {
-	err := initDB(initPeakValleyTimeBlockData)
+	err := initDB(
+		initPeakValleyTimeBlockData,
+		initPointNameData,
+	)
 	if err != nil {
 		fmt.Println("[-]初始化基础数据失败:", err)
 		os.Exit(0)

model/init-db-data/point_name.go

@@ -0,0 +1,56 @@
+package init_db_data
+
+import (
+	"energy-management-system/global"
+	"energy-management-system/model"
+	"github.com/gookit/color"
+	"gorm.io/gorm"
+)
+
+type InitPointNameData struct{}
+
+var initPointNameData = new(InitPointNameData)
+
+// type api struct{}
+//func buildData() (dataList []*peak_valley_model.PeakValleyTimeBlock) {
+//	// 一天有24小时，即1440分钟
+//	totalMinutes := 24 * 60
+//	// 时间块的持续时间：10分钟
+//	blockDuration := 10
+//	// 遍历一天中的所有时间块
+//	for i := 0; i < totalMinutes; i += blockDuration {
+//		startTime := i
+//		endTime := i + blockDuration
+//		dataList = append(dataList, &peak_valley_model.PeakValleyTimeBlock{
+//			StartTime: uint(startTime),
+//			EndTime:   uint(endTime),
+//		})
+//	}
+//	return dataList
+//}
+
+var pointNames = []model.PointName{
+	{Name: "测点-用电量"},
+	{Name: "测点-用水量"},
+}
+
+// Init 初始化用户数据
+func (i *InitPointNameData) Init() error {
+	return global.Db.Transaction(func(tx *gorm.DB) error {
+		m := &model.PointName{}
+		var count int64
+		err := tx.Model(&model.PointName{}).Count(&count).Error
+		if err != nil {
+			return err
+		}
+		if count > 0 {
+			color.Danger.Println("\n[PGSQL] --> " + m.TableName() + " 表的初始数据已存在!")
+			return nil
+		}
+		if err := tx.Create(&pointNames).Error; err != nil { // 遇到错误时回滚事务
+			return err
+		}
+		color.Info.Println("\n[PGSQL] --> " + m.TableName() + " 表初始数据成功!")
+		return nil
+	})
+}