task_model.go 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. package bo
  2. import (
  3. "errors"
  4. "fmt"
  5. "scd_check_tools/arrayex"
  6. "scd_check_tools/logger"
  7. "scd_check_tools/models/enum"
  8. "scd_check_tools/tools"
  9. "strconv"
  10. "strings"
  11. "github.com/astaxie/beego/orm"
  12. )
  13. //检测任务模型
  14. type T_data_task_model struct {
  15. Id int `orm:"pk"`
  16. TaskId int // '任务ID' ,
  17. AreaId int // 间隔ID
  18. ModelId int // '模型ID' ,
  19. Cr int // '创建人' ,
  20. Ct string `orm:"-"` // '创建时间' ,
  21. Ur int // '更新人' ,
  22. Ut string `orm:"-"` // '更新时间'
  23. }
  24. //检测任务管理对象
  25. type TaskModelMgr struct {
  26. Model T_data_task_model
  27. DeviceBaseModel
  28. }
  29. var modelDesc2 = "检测任务关联模型"
  30. func init() {
  31. orm.RegisterModel(new(T_data_task_model))
  32. }
  33. //保存任务模型信息
  34. func (c *TaskModelMgr) Save(modelids []string) (err error) {
  35. dblog := new(SystemLog)
  36. dblog.SetUserInfo(c.GetUserInfo())
  37. dblog.Audittype = enum.AuditType_check_task
  38. dblog.Logtype = enum.LogType_Insert
  39. dblog.Eventtype = enum.OptEventType_Bus
  40. dblog.Eventlevel = enum.OptEventLevel_Hight
  41. //获取检测任务原来的模型
  42. oldModels := []string{}
  43. rowset := []orm.Params{}
  44. db := orm.NewOrm()
  45. db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&rowset)
  46. for _, row := range rowset {
  47. oldModels = append(oldModels, tools.IsEmpty(row["model_id"]))
  48. }
  49. c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
  50. m1 := new(SysCheckModelMgr)
  51. m1.SetUserInfo(c.GetUserInfo())
  52. for _, item := range oldModels {
  53. //判断模型有没有删减,如果有删除模型则同步删除对应的模型配置数据
  54. if arrayex.IndexOf(modelids, item) == -1 {
  55. c.Model.ModelId, _ = strconv.Atoi(item)
  56. c.Delete()
  57. }
  58. }
  59. for _, ids := range modelids {
  60. //判断是否新增了模型
  61. if arrayex.IndexOf(oldModels, ids) == -1 {
  62. //将选择的系统模型复制为检测模型
  63. m1.Model.ModelName = fmt.Sprintf("[%d]", c.Model.TaskId) //生成不唯一的模型名称,规则:检测任务id作为复制源模型的名称前缀
  64. tmpid, _ := strconv.Atoi(ids)
  65. err, newid := m1.Copy(tmpid)
  66. if err != nil {
  67. break
  68. }
  69. if newid == 0 {
  70. continue
  71. }
  72. c.Model.ModelId = newid
  73. _, err = db.Insert(&c.Model)
  74. if err != nil {
  75. break
  76. }
  77. }
  78. }
  79. if err != nil {
  80. logger.Logger.Error(err)
  81. dblog.Description = fmt.Sprintf("保存%s信息失败:%s,操作数据:%+v", modelDesc2, err.Error(), c.Model)
  82. dblog.Fail2()
  83. } else {
  84. dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据:%+v", modelDesc2, c.Model)
  85. dblog.Success2()
  86. }
  87. return err
  88. }
  89. func (c *TaskModelMgr) One() (T_data_task_model, error) {
  90. if c.Model.Id == 0 {
  91. return c.Model, errors.New("未指定id")
  92. }
  93. o := orm.NewOrm()
  94. err := o.Read(&c.Model)
  95. if err != nil {
  96. logger.Logger.Error(err)
  97. }
  98. return c.Model, err
  99. }
  100. //根据model中指定的id删除检测任务模型对应关系
  101. func (c *TaskModelMgr) Delete() (err error) {
  102. dblog := new(SystemLog)
  103. dblog.SetUserInfo(c.GetUserInfo())
  104. dblog.Audittype = enum.AuditType_check_task
  105. dblog.Logtype = enum.LogType_Delete
  106. dblog.Eventtype = enum.OptEventType_Bus
  107. dblog.Eventlevel = enum.OptEventLevel_Hight
  108. db := orm.NewOrm()
  109. if c.Model.Id > 0 {
  110. db.Read(&c.Model)
  111. _, err = db.Delete(&c.Model)
  112. }
  113. models := []orm.Params{}
  114. if c.Model.TaskId > 0 {
  115. if c.Model.ModelId > 0 {
  116. _, err = db.Raw("delete from t_data_task_model where task_id=? and model_id=?", c.Model.TaskId, c.Model.ModelId).Exec()
  117. } else {
  118. _, err = db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&models)
  119. _, err = db.Raw("delete from t_data_task_model where task_id=?", c.Model.TaskId).Exec()
  120. }
  121. } else {
  122. if c.Model.ModelId > 0 {
  123. _, err = db.Raw("delete from t_data_task_model where model_id=?", c.Model.ModelId).Exec()
  124. }
  125. }
  126. if err != nil {
  127. logger.Logger.Error(err)
  128. dblog.Description = fmt.Sprintf("删除%s%d失败:%s", modelDesc2, c.Model.TaskId, err.Error())
  129. dblog.Fail2()
  130. } else {
  131. if c.Model.TaskId > 0 && c.Model.ModelId == 0 {
  132. for _, r := range models {
  133. id, _ := strconv.Atoi(tools.IsEmpty(r["model_id"]))
  134. m1 := new(SysCheckModelMgr)
  135. m1.SetUserInfo(c.GetUserInfo())
  136. m1.Model = T_data_model_defualt{Id: id, IsSys: 2}
  137. m1.Delete()
  138. }
  139. } else {
  140. m1 := new(SysCheckModelMgr)
  141. m1.SetUserInfo(c.GetUserInfo())
  142. m1.Model = T_data_model_defualt{Id: c.Model.ModelId, IsSys: 2}
  143. m1.Delete()
  144. }
  145. dblog.Description = fmt.Sprintf("删除%s%d成功", modelDesc2, c.Model.TaskId)
  146. dblog.Success2()
  147. }
  148. return err
  149. }
  150. //根据model中的指定过滤属性条件查询任务列表
  151. func (c *TaskModelMgr) List(pageno, pagesize int) ([]orm.Params, int, error) {
  152. dblog := new(SystemLog)
  153. dblog.SetUserInfo(c.GetUserInfo())
  154. dblog.Audittype = enum.AuditType_check_task
  155. dblog.Logtype = enum.LogType_Query
  156. dblog.Eventtype = enum.OptEventType_Bus
  157. dblog.Eventlevel = enum.OptEventLevel_Low
  158. o := orm.NewOrm()
  159. sqlParamters := []interface{}{}
  160. sql := "select t.*,t1.model_name from t_data_task_model t,t_data_model_defualt t1 where t.model_id=t1.id "
  161. if c.Model.Id > 0 {
  162. sql += " and t.id=?"
  163. sqlParamters = append(sqlParamters, c.Model.Id)
  164. }
  165. if c.Model.TaskId > 0 {
  166. sql += " and t.task_id=?"
  167. sqlParamters = append(sqlParamters, c.Model.TaskId)
  168. }
  169. limit := fmt.Sprintf(" order by t.ct desc limit %d,%d", (pageno-1)*pagesize, pagesize)
  170. r := []orm.Params{}
  171. _, err := o.Raw(sql+limit, sqlParamters).Values(&r)
  172. dblog.Description = fmt.Sprintf("SQL:%s 参数:%+v", sql+limit, sqlParamters)
  173. if err != nil {
  174. logger.Logger.Error(err, dblog.Description)
  175. dblog.Fail2()
  176. return nil, 0, err
  177. }
  178. dblog.Success2()
  179. total := []orm.Params{}
  180. _, err = o.Raw(strings.Replace(sql, "t.*,t1.model_name", "count(1) cnt", 1), sqlParamters).Values(&total)
  181. if err != nil {
  182. logger.Logger.Error(err)
  183. return nil, 0, err
  184. }
  185. totalCnt := 0
  186. if len(total) > 0 {
  187. totalCnt, _ = strconv.Atoi(tools.IsEmpty(total[0]["cnt"]))
  188. }
  189. return r, totalCnt, err
  190. }