task_model.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. package bo
  2. import (
  3. "errors"
  4. "fmt"
  5. "scd_check_tools/logger"
  6. "scd_check_tools/models/enum"
  7. "scd_check_tools/tools"
  8. "strconv"
  9. "strings"
  10. "github.com/astaxie/beego/orm"
  11. )
  12. //检测任务模型
  13. type T_data_task_model struct {
  14. Id int `orm:"pk"`
  15. TaskId int // '任务ID' ,
  16. AreaId int // 间隔ID
  17. ModelId int // '模型ID' ,
  18. Cr int // '创建人' ,
  19. Ct string `orm:"-"` // '创建时间' ,
  20. Ur int // '更新人' ,
  21. Ut string `orm:"-"` // '更新时间'
  22. }
  23. //检测任务管理对象
  24. type TaskModelMgr struct {
  25. Model T_data_task_model
  26. DeviceBaseModel
  27. }
  28. var modelDesc2 = "检测任务关联模型"
  29. func init() {
  30. orm.RegisterModel(new(T_data_task_model))
  31. }
  32. //保存任务模型信息
  33. func (c *TaskModelMgr) Save(modelids []int) (err error) {
  34. dblog := new(SystemLog)
  35. dblog.SetUserInfo(c.GetUserInfo())
  36. dblog.Audittype = enum.AuditType_check_task
  37. dblog.Logtype = enum.LogType_Insert
  38. dblog.Eventtype = enum.OptEventType_Bus
  39. dblog.Eventlevel = enum.OptEventLevel_Hight
  40. err = c.Delete()
  41. if err != nil {
  42. return err
  43. }
  44. db := orm.NewOrm()
  45. c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
  46. for _, ids := range modelids {
  47. //c.Model.AreaId = (ids[0])
  48. //c.Model.ModelId = (ids[1])
  49. c.Model.ModelId = ids
  50. _, err = db.Insert(&c.Model)
  51. if err != nil {
  52. break
  53. }
  54. }
  55. if err != nil {
  56. logger.Logger.Error(err)
  57. dblog.Description = fmt.Sprintf("保存%s信息失败:%s,操作数据:%+v", modelDesc2, err.Error(), c.Model)
  58. dblog.Fail2()
  59. } else {
  60. dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据:%+v", modelDesc2, c.Model)
  61. dblog.Success2()
  62. }
  63. return err
  64. }
  65. func (c *TaskModelMgr) One() (T_data_task_model, error) {
  66. if c.Model.Id == 0 {
  67. return c.Model, errors.New("未指定id")
  68. }
  69. o := orm.NewOrm()
  70. err := o.Read(&c.Model)
  71. if err != nil {
  72. logger.Logger.Error(err)
  73. }
  74. return c.Model, err
  75. }
  76. //根据model中指定的id删除检测任务
  77. func (c *TaskModelMgr) Delete() (err error) {
  78. dblog := new(SystemLog)
  79. dblog.SetUserInfo(c.GetUserInfo())
  80. dblog.Audittype = enum.AuditType_check_task
  81. dblog.Logtype = enum.LogType_Delete
  82. dblog.Eventtype = enum.OptEventType_Bus
  83. dblog.Eventlevel = enum.OptEventLevel_Hight
  84. db := orm.NewOrm()
  85. if c.Model.Id > 0 {
  86. db.Read(&c.Model)
  87. _, err = db.Delete(&c.Model)
  88. }
  89. if c.Model.TaskId > 0 {
  90. _, err = db.Raw("delete from t_data_task_model where task_id=?", c.Model.TaskId).Exec()
  91. }
  92. if c.Model.ModelId > 0 {
  93. _, err = db.Raw("delete from t_data_task_model where model_id=?", c.Model.ModelId).Exec()
  94. }
  95. if err != nil {
  96. logger.Logger.Error(err)
  97. dblog.Description = fmt.Sprintf("删除%s%d失败:%s", modelDesc2, c.Model.TaskId, err.Error())
  98. dblog.Fail2()
  99. } else {
  100. dblog.Description = fmt.Sprintf("删除%s%d成功", modelDesc2, c.Model.TaskId)
  101. dblog.Success2()
  102. }
  103. return err
  104. }
  105. //根据model中的指定过滤属性条件查询任务列表
  106. func (c *TaskModelMgr) List(pageno, pagesize int) ([]orm.Params, int, error) {
  107. dblog := new(SystemLog)
  108. dblog.SetUserInfo(c.GetUserInfo())
  109. dblog.Audittype = enum.AuditType_check_task
  110. dblog.Logtype = enum.LogType_Query
  111. dblog.Eventtype = enum.OptEventType_Bus
  112. dblog.Eventlevel = enum.OptEventLevel_Low
  113. o := orm.NewOrm()
  114. sqlParamters := []interface{}{}
  115. sql := "select t.*,t1.model_name from t_data_task_model t,t_data_model_defualt t1 where t.model_id=t1.id "
  116. if c.Model.Id > 0 {
  117. sql += " and t.id=?"
  118. sqlParamters = append(sqlParamters, c.Model.Id)
  119. }
  120. if c.Model.TaskId > 0 {
  121. sql += " and t.task_id=?"
  122. sqlParamters = append(sqlParamters, c.Model.TaskId)
  123. }
  124. limit := fmt.Sprintf(" order by t.ct desc limit %d,%d", (pageno-1)*pagesize, pagesize)
  125. r := []orm.Params{}
  126. _, err := o.Raw(sql+limit, sqlParamters).Values(&r)
  127. dblog.Description = fmt.Sprintf("SQL:%s 参数:%+v", sql+limit, sqlParamters)
  128. if err != nil {
  129. logger.Logger.Error(err, dblog.Description)
  130. dblog.Fail2()
  131. return nil, 0, err
  132. }
  133. dblog.Success2()
  134. total := []orm.Params{}
  135. _, err = o.Raw(strings.Replace(sql, "t.*,t1.model_name", "count(1) cnt", 1), sqlParamters).Values(&total)
  136. if err != nil {
  137. logger.Logger.Error(err)
  138. return nil, 0, err
  139. }
  140. totalCnt := 0
  141. if len(total) > 0 {
  142. totalCnt, _ = strconv.Atoi(tools.IsEmpty(total[0]["cnt"]))
  143. }
  144. return r, totalCnt, err
  145. }