From 7dec0ed5222135201e1dcf60410d08a81356e29b Mon Sep 17 00:00:00 2001 From: zhaojianbin Date: Wed, 18 Sep 2024 09:26:53 +0800 Subject: [PATCH 1/2] add some function comments --- cmd/server/app/service/internal/dao/batchmanagerdao.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/cmd/server/app/service/internal/dao/batchmanagerdao.go b/cmd/server/app/service/internal/dao/batchmanagerdao.go index a76575f6..71b57942 100644 --- a/cmd/server/app/service/internal/dao/batchmanagerdao.go +++ b/cmd/server/app/service/internal/dao/batchmanagerdao.go @@ -5,6 +5,7 @@ import ( "gorm.io/gorm" ) +// 批次 type Batch struct { gorm.Model Name string `gorm:"type:varchar(100);not null" json:"name" binding:"required" msg:"批次名不能为空"` @@ -14,6 +15,7 @@ type Batch struct { DepartName string `gorm:"type:varchar(100)"` } +// 批次-机器 type BatchMachines struct { Batch Batch `gorm:"Foreignkey:BatchID"` BatchID uint @@ -26,6 +28,7 @@ func CreateBatchMessage(batch *Batch) error { return mysqlmanager.MySQL().Create(&batch).Error } +// 查询所有批次 func GetBatch() ([]Batch, error) { var batch []Batch err := mysqlmanager.MySQL().Find(&batch).Error @@ -40,23 +43,27 @@ func GetBatchrPaged(offset, size int) (int64, []Batch, error) { return count, batchs, err } +// 查询批次是否存在 func IsExistName(name string) (bool, error) { var batch Batch err := mysqlmanager.MySQL().Where("name=?", name).Find(&batch).Error return batch.ID != 0, err } +// 查询批次是否存在 func IsExistID(id int) (bool, error) { var batch Batch err := mysqlmanager.MySQL().Where("id=?", id).Find(&batch).Error return batch.ID != 0, err } +// 删除批次 func DeleteBatch(departid int) error { var batch Batch return mysqlmanager.MySQL().Where("id=?", departid).Delete(&batch).Error } +// 更新批次 func UpdateBatch(BatchID int, BatchName string, Descrip string) error { var batch Batch BatchNew := Batch{ @@ -79,12 +86,14 @@ func GetMachineIDPaged(offset, size, batchid int) (int64, []BatchMachines, error return count, machineids, err } +// 查询批次-机器数据 func GetMachineID(BatchID int) ([]uint, error) { var machineids []uint err := mysqlmanager.MySQL().Model(BatchMachines{}).Select("machine_node_id").Where("batch_id=?", BatchID).Find(&machineids).Error return machineids, err } +// 删除批次-机器数据 func DeleteMachineBatch(NodeID int) error { var bm BatchMachines return mysqlmanager.MySQL().Where("machine_node_id=?", NodeID).Delete(&bm).Error -- Gitee From 68d6d4b901544d820fa4d73832b13539e9d76208 Mon Sep 17 00:00:00 2001 From: zhaojianbin Date: Wed, 18 Sep 2024 09:29:44 +0800 Subject: [PATCH 2/2] add function comments --- cmd/server/app/service/internal/dao/configfiledao.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cmd/server/app/service/internal/dao/configfiledao.go b/cmd/server/app/service/internal/dao/configfiledao.go index 6785186c..c98c9cae 100644 --- a/cmd/server/app/service/internal/dao/configfiledao.go +++ b/cmd/server/app/service/internal/dao/configfiledao.go @@ -7,6 +7,7 @@ import ( "gitee.com/openeuler/PilotGo/pkg/dbmanager/mysqlmanager" ) +// ConfigFile 配置文件 type ConfigFile struct { ID uint `gorm:"primary_key;AUTO_INCREMENT" json:"id"` MachineUUID string `json:"uuid"` @@ -15,13 +16,18 @@ type ConfigFile struct { UpdatedAt time.Time } +// AddConfigFile 添加配置文件 func AddConfigFile(cf ConfigFile) error { + // 根据机器UUID获取机器信息 node, err := MachineInfoByUUID(cf.MachineUUID) if err != nil { return err } + // 如果机器信息存在 if node.ID != 0 { + // 将配置文件保存到MySQL数据库 return mysqlmanager.MySQL().Save(&cf).Error } + // 如果机器不存在,返回错误信息 return fmt.Errorf("机器不存在") } -- Gitee