mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-02 11:47:36 +08:00
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
package dao
|
|
|
|
import (
|
|
gormbulkups "github.com/atcdot/gorm-bulk-upsert"
|
|
"github.com/goodrain/rainbond/api/util/bcode"
|
|
"github.com/goodrain/rainbond/db/model"
|
|
"github.com/jinzhu/gorm"
|
|
pkgerr "github.com/pkg/errors"
|
|
)
|
|
|
|
//TenantServiceMonitorDaoImpl -
|
|
type TenantServiceMonitorDaoImpl struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
//AddModel create service monitor
|
|
func (t *TenantServiceMonitorDaoImpl) AddModel(mo model.Interface) error {
|
|
m := mo.(*model.TenantServiceMonitor)
|
|
var oldTSM model.TenantServiceMonitor
|
|
if ok := t.DB.Where("name = ? and tenant_id = ?", m.Name, m.TenantID).Find(&oldTSM).RecordNotFound(); ok {
|
|
if err := t.DB.Create(m).Error; err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
return bcode.ErrServiceMonitorNameExist
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//UpdateModel update service monitor
|
|
func (t *TenantServiceMonitorDaoImpl) UpdateModel(mo model.Interface) error {
|
|
tsm := mo.(*model.TenantServiceMonitor)
|
|
if err := t.DB.Save(tsm).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//DeleteServiceMonitor delete service monitor
|
|
func (t *TenantServiceMonitorDaoImpl) DeleteServiceMonitor(mo *model.TenantServiceMonitor) error {
|
|
if err := t.DB.Delete(mo).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//DeleteServiceMonitorByServiceID delete service monitor by service id
|
|
func (t *TenantServiceMonitorDaoImpl) DeleteServiceMonitorByServiceID(serviceID string) error {
|
|
if err := t.DB.Where("service_id=?", serviceID).Delete(&model.TenantServiceMonitor{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//DeleteByComponentIDs delete service monitor by component ids
|
|
func (t *TenantServiceMonitorDaoImpl) DeleteByComponentIDs(componentIDs []string) error {
|
|
return t.DB.Where("service_id in (?)", componentIDs).Delete(&model.TenantServiceMonitor{}).Error
|
|
}
|
|
|
|
//CreateOrUpdateMonitorInBatch -
|
|
func (t *TenantServiceMonitorDaoImpl) CreateOrUpdateMonitorInBatch(monitors []*model.TenantServiceMonitor) error {
|
|
var objects []interface{}
|
|
for _, monitor := range monitors {
|
|
objects = append(objects, *monitor)
|
|
}
|
|
if err := gormbulkups.BulkUpsert(t.DB, objects, 2000); err != nil {
|
|
return pkgerr.Wrap(err, "create or update component monitors in batch")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//GetByServiceID get tsm by service id
|
|
func (t *TenantServiceMonitorDaoImpl) GetByServiceID(serviceID string) ([]*model.TenantServiceMonitor, error) {
|
|
var tsm []*model.TenantServiceMonitor
|
|
if err := t.DB.Where("service_id=?", serviceID).Find(&tsm).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return tsm, nil
|
|
}
|
|
|
|
//GetByName get by name
|
|
func (t *TenantServiceMonitorDaoImpl) GetByName(serviceID, name string) (*model.TenantServiceMonitor, error) {
|
|
var tsm model.TenantServiceMonitor
|
|
if err := t.DB.Where("service_id=? and name=?", serviceID, name).Find(&tsm).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tsm, nil
|
|
}
|