task.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455
  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. "sync"
  11. "github.com/astaxie/beego/orm"
  12. )
  13. //检测任务模型
  14. type T_data_task struct {
  15. Id int `orm:"pk"`
  16. Name string // '任务名称' ,
  17. Code string // '任务编号' ,
  18. StationId int // '所属变电站' ,
  19. ScdId int64 // 'SCD文件编号' ,
  20. ReportId int // '报告模板ID' ,
  21. Memo string // '任务说明' ,
  22. State int // '检测状态;0 未检查 1 检测中 2检测结束 3 异常中断' ,
  23. StartTime string // '开始检测时间' ,
  24. EndTime string // '检测结束时间' ,
  25. Cr int // '创建人' ,
  26. Ct string `orm:"-"` // '创建时间' ,
  27. Ur int // '更新人' ,
  28. Ut string `orm:"-"` // '更新时间'
  29. }
  30. //检测任务管理对象
  31. type TaskMgr struct {
  32. Model T_data_task
  33. DeviceBaseModel
  34. }
  35. var TaskSCD = sync.Map{}
  36. var modelDesc = "检测任务"
  37. //已终止的任务
  38. var stopedTaskMap = sync.Map{}
  39. func init() {
  40. orm.RegisterModel(new(T_data_task))
  41. }
  42. //重置检测异常中断的任务状态。一般在系统启动时处理
  43. func (c *TaskMgr) ChangeErr2Normal() {
  44. db := orm.NewOrm()
  45. db.Raw("update t_data_task set state=0 where state=1").Exec()
  46. }
  47. //设置指定任务的状态 flag:0 未检查 1 检测中 2检测结束 3 异常中断
  48. func (c *TaskMgr) SetActive(flag int) (err error) {
  49. dblog := new(SystemLog)
  50. dblog.SetUserInfo(c.GetUserInfo())
  51. dblog.Audittype = enum.AuditType_check_task
  52. dblog.Logtype = enum.LogType_Execute
  53. dblog.Eventtype = enum.OptEventType_Bus
  54. dblog.Eventlevel = enum.OptEventLevel_Hight
  55. db := orm.NewOrm()
  56. err = db.Read(&c.Model)
  57. if err != nil {
  58. logger.Logger.Error(err)
  59. return err
  60. }
  61. if flag == 0 {
  62. c.Model.StartTime = "1970-01-01 00:00:00"
  63. c.Model.EndTime = "1970-01-01 00:00:00"
  64. }
  65. if flag == 1 {
  66. if c.Model.State == 1 {
  67. return errors.New("当前任务正在检测中")
  68. }
  69. c.Model.StartTime = tools.NowTime()
  70. c.Model.EndTime = "1970-01-01 00:00:00"
  71. }
  72. if flag == 2 {
  73. c.Model.EndTime = tools.NowTime()
  74. }
  75. c.Model.State = flag
  76. _, err = db.Update(&c.Model)
  77. if err != nil {
  78. logger.Logger.Error(err)
  79. dblog.Description = fmt.Sprintf("更新%s%s状态失败:%s,操作数据:%+v", modelDesc, c.Model.Name, err.Error(), c.Model)
  80. dblog.Fail2()
  81. } else {
  82. dblog.Description = fmt.Sprintf("更新%s%s状态,操作数据:%+v", modelDesc, c.Model.Name, c.Model)
  83. dblog.Success2()
  84. if flag == 0 {
  85. return nil
  86. }
  87. if flag == 1 {
  88. return c.start(c.Model)
  89. }
  90. if flag == 2 {
  91. return c.finish(c.Model)
  92. }
  93. if flag == 3 {
  94. return c.stop(c.Model)
  95. }
  96. }
  97. return err
  98. }
  99. //开始检测
  100. func (c *TaskMgr) start(task T_data_task) error {
  101. scdParse := new(ScdParse)
  102. scdParse.SetUserInfo(c.GetUserInfo())
  103. scdMgr := new(ScdMgr)
  104. scdMgr.SetUserInfo(c.GetUserInfo())
  105. scdInfo, err := scdMgr.One(fmt.Sprintf("%d", task.ScdId))
  106. if err != nil {
  107. return err
  108. }
  109. c.Model.Id = task.Id
  110. _, cnt, _ := c.GetModels()
  111. if cnt == 0 {
  112. return errors.New("该检测任务还未配置检测模型")
  113. }
  114. TaskSCD.Store(task.ScdId, task)
  115. db := orm.NewOrm()
  116. //清除原步骤及状态
  117. db.Raw("delete from t_data_task_check_step where task_id=?", task.Id).Exec()
  118. db.Raw("delete from t_scd_node_rule_parse where scd_id=?", task.ScdId).Exec()
  119. //生成新的步骤
  120. db.Raw("insert into t_data_task_check_step(task_id,task_step_id,state) select ?,id,0 from global_const_code where parentcode='CHECK_STEP' order by id", task.Id).Exec()
  121. var isStop = func(taskid int) bool {
  122. v, r := stopedTaskMap.Load(taskid)
  123. if r {
  124. return v.(bool)
  125. }
  126. return false
  127. }
  128. go func(task T_data_task, scdInfo orm.Params) {
  129. _, err = scdParse.XmlIEDParse(tools.IsEmpty(scdInfo["station_id"]), tools.IsEmpty(scdInfo["path"]), tools.IsEmpty(scdInfo["scd_name"]))
  130. if isStop(task.Id) {
  131. return
  132. }
  133. if err == nil {
  134. scdParse.XmlCheckParse(task.ScdId)
  135. if isStop(task.Id) {
  136. return
  137. }
  138. //模型分析
  139. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_model_parse.Code(), 1)
  140. checkAreaMgr := new(CheckAreaMgr)
  141. checkAreaMgr.ScdId = task.ScdId
  142. err = checkAreaMgr.CheckAreaIedRelation()
  143. if isStop(task.Id) {
  144. return
  145. }
  146. if err != nil {
  147. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_model_parse.Code(), 3, err.Error())
  148. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_fcda_parse.Code(), 3, "终止")
  149. } else {
  150. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_model_parse.Code(), 2)
  151. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_fcda_parse.Code(), 1)
  152. err = checkAreaMgr.CheckIedFcda()
  153. if isStop(task.Id) {
  154. return
  155. }
  156. if err != nil {
  157. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_fcda_parse.Code(), 3, err.Error())
  158. } else {
  159. c.SetStep(tools.IsEmpty(task.ScdId), enum.TaskStep_SCD_fcda_parse.Code(), 2)
  160. }
  161. }
  162. }
  163. }(task, scdInfo)
  164. return err
  165. }
  166. //取消/终止检测
  167. func (c *TaskMgr) stop(task T_data_task) error {
  168. TaskSCD.Delete(task.ScdId)
  169. stopedTaskMap.Store(task.Id, true)
  170. db := orm.NewOrm()
  171. _, err := db.Raw("update t_data_task_check_step set state=3,ut=now() where task_id=? and state<2", task.Id).Exec()
  172. return err
  173. }
  174. //完成检测
  175. func (c *TaskMgr) finish(task T_data_task) error {
  176. db := orm.NewOrm()
  177. _, err := db.Raw("update t_data_task_check_step set state=2,ut=now() where task_id=? and state=0", task.Id).Exec()
  178. return err
  179. }
  180. //更新检测步骤状态
  181. func (c *TaskMgr) SetStep(scdid string, stepcode string, state int, msg ...string) {
  182. db := orm.NewOrm()
  183. scdidint64, _ := strconv.ParseInt(scdid, 10, 64)
  184. taskid, has := TaskSCD.Load(scdidint64)
  185. if !has {
  186. return
  187. }
  188. errormsg := ""
  189. if len(msg) > 0 {
  190. errormsg = msg[0]
  191. }
  192. t1 := "ct" //步骤开始时间字段
  193. if state > 1 {
  194. t1 = "ut" //步骤结束时间字段
  195. }
  196. db.Raw("update t_data_task_check_step set state=?,state_msg=?,"+t1+"=now() where task_id=? and task_step_id=(select id from global_const_code where parentcode='CHECK_STEP' and code=?)", state, errormsg, taskid, stepcode).Exec()
  197. // 所有步骤都不为0和1了,则分析完成
  198. hasStep := []orm.Params{}
  199. _, err := db.Raw("select count(1) cnt from t_data_task_check_step where task_id=? and state in(0,1)", taskid).Values(&hasStep)
  200. if err != nil {
  201. logger.Logger.Error(err)
  202. return
  203. }
  204. if len(hasStep) == 0 || tools.IsEmpty(hasStep[0]["cnt"]) == "0" {
  205. // 全部步骤检测完成,更改任务检测状态为已完成
  206. db.Raw("update t_data_task set state=2 where id=?", taskid).Exec()
  207. }
  208. }
  209. //保存任务信息
  210. func (c *TaskMgr) Save() (id int, err error) {
  211. dblog := new(SystemLog)
  212. dblog.SetUserInfo(c.GetUserInfo())
  213. dblog.Audittype = enum.AuditType_check_task
  214. dblog.Logtype = enum.LogType_Insert
  215. dblog.Eventtype = enum.OptEventType_Bus
  216. dblog.Eventlevel = enum.OptEventLevel_Hight
  217. db := orm.NewOrm()
  218. oldinfo := T_data_task{}
  219. oldid, _ := c.ExistName(c.Model.Name)
  220. if c.Model.Id == 0 {
  221. if oldid != "" {
  222. return 0, errors.New("任务名称(" + c.Model.Name + ")已存在")
  223. }
  224. c.Model.StartTime = "1970-01-01 00:00:00"
  225. c.Model.EndTime = "1970-01-01 00:00:00"
  226. c.Model.State = 0
  227. c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
  228. id64 := int64(0)
  229. id64, err = db.Insert(&c.Model)
  230. id = int(id64)
  231. } else {
  232. if oldid != "" && oldid != tools.IsEmpty(c.Model.Id) {
  233. return 0, errors.New("任务名称(" + c.Model.Name + ")已存在")
  234. }
  235. id = c.Model.Id
  236. oldinfo, _ = c.One()
  237. if oldinfo.State == 1 {
  238. return 0, errors.New("任务正在检测中,不允许编辑")
  239. }
  240. if oldinfo.State == 2 {
  241. return 0, errors.New("任务已检测完成,不允许编辑")
  242. }
  243. c.Model.StartTime = oldinfo.StartTime
  244. c.Model.EndTime = oldinfo.EndTime
  245. c.Model.State = oldinfo.State
  246. c.Model.Ur, _ = strconv.Atoi(c.GetUserId())
  247. _, err = db.Update(&c.Model)
  248. }
  249. if err != nil {
  250. logger.Logger.Error(err)
  251. dblog.Description = fmt.Sprintf("保存%s信息失败:%s,操作数据:%+v", modelDesc, err.Error(), c.Model)
  252. dblog.Fail2()
  253. } else {
  254. dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据:%+v", modelDesc, c.Model)
  255. dblog.Success2()
  256. if oldinfo.Id == 0 || oldinfo.ScdId != c.Model.ScdId {
  257. // 解析模型间隔及标准装置
  258. checkArea := new(CheckAreaMgr)
  259. checkArea.ScdId = c.Model.ScdId
  260. checkArea.Reset()
  261. }
  262. }
  263. return id, err
  264. }
  265. func (c *TaskMgr) One() (T_data_task, error) {
  266. if c.Model.Id == 0 {
  267. return c.Model, errors.New("未指定id")
  268. }
  269. o := orm.NewOrm()
  270. tmp := T_data_task{Id: c.Model.Id}
  271. err := o.Read(&tmp)
  272. if err != nil {
  273. logger.Logger.Error(err)
  274. }
  275. return tmp, err
  276. }
  277. func (c *TaskMgr) ExistName(name string) (string, error) {
  278. o := orm.NewOrm()
  279. rowset := []orm.Params{}
  280. _, err := o.Raw("select id from t_data_task where name=?", name).Values(&rowset)
  281. if err != nil {
  282. logger.Logger.Error(err)
  283. return "", err
  284. }
  285. if len(rowset) > 0 {
  286. return tools.IsEmpty(rowset[0]["id"]), nil
  287. }
  288. return "", nil
  289. }
  290. //根据model中指定的id删除检测任务
  291. func (c *TaskMgr) Delete() (err error) {
  292. dblog := new(SystemLog)
  293. u := c.GetUserInfo()
  294. dblog.SetUserInfo(u)
  295. dblog.Audittype = enum.AuditType_check_task
  296. dblog.Logtype = enum.LogType_Delete
  297. dblog.Eventtype = enum.OptEventType_Bus
  298. dblog.Eventlevel = enum.OptEventLevel_Hight
  299. db := orm.NewOrm()
  300. if c.Model.Id == 0 {
  301. return errors.New("无效的ID")
  302. } else {
  303. db.Read(&c.Model)
  304. if c.Model.State == 1 {
  305. return errors.New("任务正在检测中,不允许删除")
  306. }
  307. _, err = db.Delete(&c.Model)
  308. }
  309. if err != nil {
  310. logger.Logger.Error(err)
  311. dblog.Description = fmt.Sprintf("删除%s%s失败:%s", modelDesc, c.Model.Name, err.Error())
  312. dblog.Fail2()
  313. } else {
  314. TaskSCD.Delete(c.Model.ScdId)
  315. dblog.Description = fmt.Sprintf("删除%s%s成功", modelDesc, c.Model.Name)
  316. dblog.Success2()
  317. //同步删除关联数据
  318. db.Raw("delete from t_data_task_check_step where task_id=?", c.Model.Id).Exec()
  319. db.Raw("delete from t_data_check_area where scd_id=?", c.Model.ScdId).Exec()
  320. db.Raw("delete from t_data_check_area_ied where scd_id=?", c.Model.ScdId).Exec()
  321. scdmgr := new(ScdMgr)
  322. scdmgr.SetUserInfo(u)
  323. scdmgr.DeleteScd(strconv.FormatInt(c.Model.ScdId, 10), true)
  324. //删除关联模型
  325. m1 := new(TaskModelMgr)
  326. m1.SetUserInfo(u)
  327. m1.Model = T_data_task_model{TaskId: c.Model.Id}
  328. m1.Delete()
  329. //删除报告关联数据
  330. r1 := new(TaskReportMgr)
  331. r1.SetUserInfo(u)
  332. r1.Model = T_data_task_report{TaskId: c.Model.Id}
  333. r1.Delete()
  334. }
  335. return err
  336. }
  337. //根据model中的指定过滤属性条件查询任务列表
  338. func (c *TaskMgr) List(pageno, pagesize int) ([]orm.Params, int, error) {
  339. dblog := new(SystemLog)
  340. dblog.SetUserInfo(c.GetUserInfo())
  341. dblog.Audittype = enum.AuditType_check_task
  342. dblog.Logtype = enum.LogType_Query
  343. dblog.Eventtype = enum.OptEventType_Bus
  344. dblog.Eventlevel = enum.OptEventLevel_Low
  345. o := orm.NewOrm()
  346. sqlParamters := []interface{}{}
  347. sql := "select t.*,u.name username,s.scd_name,ts.area_name station_name,c1.id voltage_level,c1.name voltage_name from t_data_task t left join t_scd_scl s on t.scd_id=s.id left join t_data_user u on t.cr=u.id left join t_data_area ts on t.station_id=ts.ID left join global_const_code c1 on ts.AREA_LEVEL=c1.id and c1.parentcode='voltage_level' where 1=1 "
  348. if c.Model.Id > 0 {
  349. sql += " and t.id=?"
  350. sqlParamters = append(sqlParamters, c.Model.Id)
  351. }
  352. if c.Model.StationId > 0 {
  353. sql += " and t.station_id=?"
  354. sqlParamters = append(sqlParamters, c.Model.StationId)
  355. }
  356. if c.Model.Name != "" {
  357. sql += " and t.name like ?"
  358. sqlParamters = append(sqlParamters, "%"+c.Model.Name+"%")
  359. }
  360. if c.Model.ScdId > 0 {
  361. sql += " and t.scd_id=?"
  362. sqlParamters = append(sqlParamters, c.Model.ScdId)
  363. }
  364. if c.Model.State >= 0 {
  365. sql += " and t.state=?"
  366. sqlParamters = append(sqlParamters, c.Model.State)
  367. }
  368. if c.Model.StartTime != "" {
  369. sql += " and t.ct>=?"
  370. sqlParamters = append(sqlParamters, c.Model.StartTime+" 00:00:00")
  371. }
  372. if c.Model.EndTime != "" {
  373. sql += " and t.ct<=?"
  374. sqlParamters = append(sqlParamters, c.Model.EndTime+" 23:59:59")
  375. }
  376. limit := fmt.Sprintf(" order by t.ct desc limit %d,%d", (pageno-1)*pagesize, pagesize)
  377. r := []orm.Params{}
  378. _, err := o.Raw(sql+limit, sqlParamters).Values(&r)
  379. dblog.Description = fmt.Sprintf("SQL:%s 参数:%+v", sql+limit, sqlParamters)
  380. if err != nil {
  381. logger.Logger.Error(err, dblog.Description)
  382. dblog.Fail2()
  383. return nil, 0, err
  384. }
  385. dblog.Success2()
  386. total := []orm.Params{}
  387. _, err = o.Raw(strings.Replace(sql, "t.*,u.name username,s.scd_name,ts.area_name station_name", "count(1) cnt", 1), sqlParamters).Values(&total)
  388. if err != nil {
  389. logger.Logger.Error(err)
  390. return nil, 0, err
  391. }
  392. totalCnt := 0
  393. if len(total) > 0 {
  394. totalCnt, _ = strconv.Atoi(tools.IsEmpty(total[0]["cnt"]))
  395. }
  396. return r, totalCnt, err
  397. }
  398. //获取指定任务的已选择模型对应的系统内置模型
  399. func (c *TaskMgr) GetModels() ([]orm.Params, int, error) {
  400. if c.Model.Id == 0 {
  401. return nil, 0, errors.New("任务ID不能为空")
  402. }
  403. db := orm.NewOrm()
  404. rowset := []orm.Params{}
  405. _, err := db.Raw("select t.*,t1.from_model_id sys_model_id,t2.model_name,t2.vol_id,t2.line_link_style from t_data_task_model t,t_data_model_defualt t1,t_data_model_defualt t2 where t.model_id=t1.id and t1.from_model_id=t2.id and t.task_id=?", c.Model.Id).Values(&rowset)
  406. if err != nil {
  407. return rowset, 0, err
  408. }
  409. return rowset, len(rowset), nil
  410. }
  411. //根据SCD ID获取指定任务的已选择模型
  412. func (c *TaskMgr) GetModelsByScdID(scdid int64) ([]orm.Params, int, error) {
  413. if scdid == 0 {
  414. return nil, 0, errors.New("SCD ID不能为空")
  415. }
  416. db := orm.NewOrm()
  417. rowset := []orm.Params{}
  418. _, err := db.Raw("select t.*,t1.model_name,t1.vol_id,t1.line_link_style from t_data_task a, t_data_task_model t,t_data_model_defualt t1 where a.id=t.task_id and t.model_id=t1.id and a.scd_id=?", scdid).Values(&rowset)
  419. if err != nil {
  420. return rowset, 0, err
  421. }
  422. return rowset, len(rowset), nil
  423. }
  424. func (c *TaskMgr) GetCheckStepInfo() ([]orm.Params, int, error) {
  425. db := orm.NewOrm()
  426. rowset := []orm.Params{}
  427. _, err := db.Raw("select a.*,a1.name step_name,case a.state when 0 then '未开始' when 1 then '检测中' when 2 then '已完成' when 3 then '检测终止' else '检测异常' end state_name from t_data_task_check_step a left join global_const_code a1 on a.task_step_id=a1.id and a1.parentcode='CHECK_STEP' where a.task_id=? order by a1.id ", c.Model.Id).Values(&rowset)
  428. if err != nil {
  429. return rowset, 0, err
  430. }
  431. return rowset, len(rowset), nil
  432. }