37 lines
929 B
Go
37 lines
929 B
Go
package dao
|
|
|
|
import (
|
|
"go-common/app/admin/ep/merlin/model"
|
|
|
|
pkgerr "github.com/pkg/errors"
|
|
)
|
|
|
|
// DelMachineNodeByMachineID delete node by machine id.
|
|
func (d *Dao) DelMachineNodeByMachineID(mID int64) error {
|
|
return pkgerr.WithStack(d.db.Where("machine_id = ?", mID).Delete(&model.MachineNode{}).Error)
|
|
}
|
|
|
|
// GenMachineNodes generate some machine nodes.
|
|
func (d *Dao) GenMachineNodes(nodes []*model.MachineNode) (err error) {
|
|
tx := d.db.Begin()
|
|
if err = tx.Error; err != nil {
|
|
return
|
|
}
|
|
for _, n := range nodes {
|
|
if err = tx.Create(n).Error; err != nil {
|
|
tx.Rollback()
|
|
return
|
|
}
|
|
}
|
|
if err = tx.Commit().Error; err != nil {
|
|
tx.Rollback()
|
|
}
|
|
return
|
|
}
|
|
|
|
// QueryMachineNodes query machine nodes by machine id.
|
|
func (d *Dao) QueryMachineNodes(mID int64) (nodes []*model.MachineNode, err error) {
|
|
err = pkgerr.WithStack(d.db.Model(&model.MachineNode{}).Where("machine_id = ?", mID).Find(&nodes).Error)
|
|
return
|
|
}
|