123456789101112131415161718192021222324252627282930313233343536 |
- 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
- }
|