package bo import ( "errors" "fmt" "scd_check_tools/arrayex" "scd_check_tools/logger" "scd_check_tools/models/enum" "scd_check_tools/tools" "strconv" "strings" "github.com/astaxie/beego/orm" ) //检测任务模型 type T_data_task_model struct { Id int `orm:"pk"` TaskId int // '任务ID' , AreaId int // 间隔ID ModelId int // '模型ID' , Cr int // '创建人' , Ct string `orm:"-"` // '创建时间' , Ur int // '更新人' , Ut string `orm:"-"` // '更新时间' } //检测任务管理对象 type TaskModelMgr struct { Model T_data_task_model DeviceBaseModel } var modelDesc2 = "检测任务关联模型" func init() { orm.RegisterModel(new(T_data_task_model)) } //保存任务模型信息 func (c *TaskModelMgr) Save(modelids []string) (err error) { dblog := new(SystemLog) dblog.SetUserInfo(c.GetUserInfo()) dblog.Audittype = enum.AuditType_check_task dblog.Logtype = enum.LogType_Insert dblog.Eventtype = enum.OptEventType_Bus dblog.Eventlevel = enum.OptEventLevel_Hight //获取检测任务原来的模型 oldModels := []string{} rowset := []orm.Params{} db := orm.NewOrm() db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&rowset) for _, row := range rowset { oldModels = append(oldModels, tools.IsEmpty(row["model_id"])) } c.Model.Cr, _ = strconv.Atoi(c.GetUserId()) m1 := new(SysCheckModelMgr) m1.SetUserInfo(c.GetUserInfo()) for _, item := range oldModels { //判断模型有没有删减,如果有删除模型则同步删除对应的模型配置数据 if arrayex.IndexOf(modelids, item) == -1 { c.Model.ModelId, _ = strconv.Atoi(item) c.Delete() } } for _, ids := range modelids { if ids == "" { continue } //判断是否新增了模型 if arrayex.IndexOf(oldModels, ids) == -1 { logger.Logger.Debug(fmt.Sprintf("正在从模型%s复制新模型", ids)) //将选择的系统模型复制为检测模型 m1.Model.ModelName = fmt.Sprintf("[%d]", c.Model.TaskId) //生成不唯一的模型名称,规则:检测任务id作为复制源模型的名称前缀 tmpid, _ := strconv.Atoi(ids) m1.Model.IsSys = 2 err, newid := m1.Copy(tmpid) if err != nil { logger.Logger.Error(err) break } if newid == 0 { continue } c.Model.ModelId = newid _, err = db.Insert(&c.Model) if err != nil { break } } } if err != nil { logger.Logger.Error(err) dblog.Description = fmt.Sprintf("保存%s信息失败:%s,操作数据:%+v", modelDesc2, err.Error(), c.Model) dblog.Fail2() } else { dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据:%+v", modelDesc2, c.Model) dblog.Success2() } return err } func (c *TaskModelMgr) One() (T_data_task_model, error) { if c.Model.Id == 0 { return c.Model, errors.New("未指定id") } o := orm.NewOrm() err := o.Read(&c.Model) if err != nil { logger.Logger.Error(err) } return c.Model, err } //根据model中指定的id删除检测任务模型对应关系 func (c *TaskModelMgr) Delete() (err error) { dblog := new(SystemLog) dblog.SetUserInfo(c.GetUserInfo()) dblog.Audittype = enum.AuditType_check_task dblog.Logtype = enum.LogType_Delete dblog.Eventtype = enum.OptEventType_Bus dblog.Eventlevel = enum.OptEventLevel_Hight db := orm.NewOrm() if c.Model.Id > 0 { db.Read(&c.Model) _, err = db.Delete(&c.Model) } models := []orm.Params{} if c.Model.TaskId > 0 { if c.Model.ModelId > 0 { _, err = db.Raw("delete from t_data_task_model where task_id=? and model_id=?", c.Model.TaskId, c.Model.ModelId).Exec() } else { _, err = db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&models) _, err = db.Raw("delete from t_data_task_model where task_id=?", c.Model.TaskId).Exec() } } else { if c.Model.ModelId > 0 { _, err = db.Raw("delete from t_data_task_model where model_id=?", c.Model.ModelId).Exec() } } if err != nil { logger.Logger.Error(err) dblog.Description = fmt.Sprintf("删除%s%d失败:%s", modelDesc2, c.Model.TaskId, err.Error()) dblog.Fail2() } else { if c.Model.TaskId > 0 && c.Model.ModelId == 0 { for _, r := range models { id, _ := strconv.Atoi(tools.IsEmpty(r["model_id"])) m1 := new(SysCheckModelMgr) m1.SetUserInfo(c.GetUserInfo()) m1.Model = T_data_model_defualt{Id: id, IsSys: 2} m1.Delete() } } else { m1 := new(SysCheckModelMgr) m1.SetUserInfo(c.GetUserInfo()) m1.Model = T_data_model_defualt{Id: c.Model.ModelId, IsSys: 2} m1.Delete() } dblog.Description = fmt.Sprintf("删除%s%d成功", modelDesc2, c.Model.TaskId) dblog.Success2() } return err } //根据model中的指定过滤属性条件查询任务列表 func (c *TaskModelMgr) List(pageno, pagesize int) ([]orm.Params, int, error) { dblog := new(SystemLog) dblog.SetUserInfo(c.GetUserInfo()) dblog.Audittype = enum.AuditType_check_task dblog.Logtype = enum.LogType_Query dblog.Eventtype = enum.OptEventType_Bus dblog.Eventlevel = enum.OptEventLevel_Low o := orm.NewOrm() sqlParamters := []interface{}{} sql := "select t.*,t1.model_name from t_data_task_model t,t_data_model_defualt t1 where t.model_id=t1.id " if c.Model.Id > 0 { sql += " and t.id=?" sqlParamters = append(sqlParamters, c.Model.Id) } if c.Model.TaskId > 0 { sql += " and t.task_id=?" sqlParamters = append(sqlParamters, c.Model.TaskId) } limit := fmt.Sprintf(" order by t.ct desc limit %d,%d", (pageno-1)*pagesize, pagesize) r := []orm.Params{} _, err := o.Raw(sql+limit, sqlParamters).Values(&r) dblog.Description = fmt.Sprintf("SQL:%s 参数:%+v", sql+limit, sqlParamters) if err != nil { logger.Logger.Error(err, dblog.Description) dblog.Fail2() return nil, 0, err } dblog.Success2() total := []orm.Params{} _, err = o.Raw(strings.Replace(sql, "t.*,t1.model_name", "count(1) cnt", 1), sqlParamters).Values(&total) if err != nil { logger.Logger.Error(err) return nil, 0, err } totalCnt := 0 if len(total) > 0 { totalCnt, _ = strconv.Atoi(tools.IsEmpty(total[0]["cnt"])) } return r, totalCnt, err }