90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"go-common/app/admin/main/config/model"
|
||
|
|
"go-common/library/log"
|
||
|
|
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
//UpdateForce update force.
|
||
|
|
func (s *Service) UpdateForce(ctx context.Context, treeID, version int64, env, zone, build, username string, hosts map[string]string) (err error) {
|
||
|
|
var (
|
||
|
|
app *model.App
|
||
|
|
force *model.Force
|
||
|
|
ups map[string]interface{}
|
||
|
|
)
|
||
|
|
if app, err = s.AppByTree(treeID, env, zone); err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
tx := s.dao.DB.Begin()
|
||
|
|
for key, val := range hosts {
|
||
|
|
force = &model.Force{}
|
||
|
|
force.Hostname = key
|
||
|
|
force.AppID = app.ID
|
||
|
|
force.IP = val
|
||
|
|
force.Operator = username
|
||
|
|
force.Version = version
|
||
|
|
if err = s.dao.DB.Where("app_id = ? and hostname = ? and ip = ?", app.ID, key, val).First(&model.Force{}).Error; err != nil {
|
||
|
|
if err != gorm.ErrRecordNotFound {
|
||
|
|
tx.Rollback()
|
||
|
|
log.Error("UpdateForce first error(%v)", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
//create
|
||
|
|
if err = s.dao.DB.Create(force).Error; err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
log.Error("UpdateForce(%s) error(%v)", force, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
//update
|
||
|
|
ups = map[string]interface{}{
|
||
|
|
"hostname": key,
|
||
|
|
"app_id": app.ID,
|
||
|
|
"ip": val,
|
||
|
|
"operator": username,
|
||
|
|
"version": version,
|
||
|
|
}
|
||
|
|
if err = s.dao.DB.Model(&model.Force{}).Where("app_id = ? and hostname = ? and ip = ?", app.ID, key, val).Updates(ups).Error; err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
log.Error("UpdateForce(%s) error(%v)", force, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
if err = s.PushForce(ctx, treeID, env, zone, build, version, hosts, 1); err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
tx.Commit()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
//ClearForce delete force.
|
||
|
|
func (s *Service) ClearForce(ctx context.Context, treeID int64, env, zone, build string, hosts map[string]string) (err error) {
|
||
|
|
var (
|
||
|
|
app *model.App
|
||
|
|
)
|
||
|
|
if app, err = s.AppByTree(treeID, env, zone); err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
tx := s.dao.DB.Begin()
|
||
|
|
for key, val := range hosts {
|
||
|
|
if err = s.dao.DB.Where("app_id = ? and hostname = ?", app.ID, key).Delete(model.Force{}).Error; err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
log.Error("ClearForce hostname(%s) ip(%v) error(%v)", key, val, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if err = s.PushForce(ctx, treeID, env, zone, build, 0, hosts, 0); err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
tx.Commit()
|
||
|
|
return
|
||
|
|
}
|