Warning.go 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399
  1. package controllers
  2. import (
  3. "Cold_Api/Nats/NatsServer"
  4. "Cold_Api/conf"
  5. "Cold_Api/controllers/lib"
  6. db "Cold_Api/initialize"
  7. "Cold_Api/models/Account"
  8. "Cold_Api/models/Device"
  9. "Cold_Api/models/System"
  10. "Cold_Api/models/Warning"
  11. "encoding/json"
  12. "fmt"
  13. "github.com/beego/beego/v2/core/logs"
  14. "github.com/robfig/cron/v3"
  15. "github.com/xuri/excelize/v2"
  16. "math"
  17. "net/http"
  18. "os"
  19. "strconv"
  20. "time"
  21. )
  22. // 设备告警 ------------------------------------------
  23. // 告警列表
  24. func (c *DeviceController) DeviceWarning_List() {
  25. type R_JSONS struct {
  26. //必须的大写开头
  27. Data []Warning.Warning_R
  28. Num int64
  29. Page int
  30. Page_size int
  31. }
  32. var r_jsons R_JSONS
  33. page, _ := c.GetInt("page")
  34. if page < 1 {
  35. page = 1
  36. }
  37. page_z, _ := c.GetInt("page_z")
  38. if page_z < 1 {
  39. page_z = conf.Page_size
  40. }
  41. T_name := c.GetString("T_name")
  42. T_tp := c.GetString("T_tp")
  43. Time_start := c.GetString("Time_start")
  44. Time_end := c.GetString("Time_end")
  45. T_handle, _ := c.GetInt("T_handle")
  46. T_admin, _ := c.GetInt("T_admin")
  47. T_history, _ := c.GetInt("T_history")
  48. tpCount := Warning.Read_WarningType_Count()
  49. var tpList []string
  50. if len(T_tp) > 0 {
  51. tpList = lib.SplitStringIds(T_tp, "T")
  52. if len(tpList) == int(tpCount) {
  53. // 类型全选,替换为空,不进行查询筛选(查询全部)
  54. tpList = []string{}
  55. }
  56. } else {
  57. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  58. if power.T_warning != "*" {
  59. tpList = lib.SplitStringIds(power.T_warning, "W")
  60. }
  61. }
  62. countRedisKey := fmt.Sprintf("Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, T_admin, T_history)
  63. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  64. if err != nil {
  65. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  66. c.ServeJSON()
  67. return
  68. }
  69. var T_year, T_month string
  70. if T_history == 1 {
  71. date, err := time.Parse("2006-01-02 15:04:05", Time_start)
  72. if err != nil {
  73. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  74. c.ServeJSON()
  75. return
  76. }
  77. T_year, T_month = date.Format("2006"), date.Format("01")
  78. }
  79. // 管理员界面
  80. if T_admin == 1 {
  81. if T_history == 1 {
  82. // 获取备份
  83. r_jsons.Data, r_jsons.Num = Warning.Read_Admin_Warning_Backups(*c.Admin_r, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  84. } else {
  85. // 获取最新
  86. r_jsons.Data, r_jsons.Num = Warning.Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  87. }
  88. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  89. c.ServeJSON()
  90. return
  91. }
  92. // 公司报警界面
  93. if T_history == 1 {
  94. r_jsons.Data, r_jsons.Num = Warning.Read_Warning_Backups(c.T_pid, bindSN, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  95. } else {
  96. // 获取最新
  97. r_jsons.Data, r_jsons.Num = Warning.Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  98. }
  99. r_jsons.Page = page
  100. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  101. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  102. c.ServeJSON()
  103. return
  104. }
  105. func (c *DeviceController) DeviceWarning_List2() {
  106. type R_JSONS struct {
  107. //必须的大写开头
  108. Data []Warning.Warning_R
  109. Num int64
  110. Page int
  111. Page_size int
  112. }
  113. var r_jsons R_JSONS
  114. page, _ := c.GetInt("page")
  115. if page < 1 {
  116. page = 1
  117. }
  118. page_z, _ := c.GetInt("page_z")
  119. if page_z < 1 {
  120. page_z = conf.Page_size
  121. }
  122. T_name := c.GetString("T_name")
  123. T_tp := c.GetString("T_tp")
  124. Time_start := c.GetString("Time_start")
  125. Time_end := c.GetString("Time_end")
  126. T_handle, _ := c.GetInt("T_handle")
  127. T_admin, _ := c.GetInt("T_admin")
  128. T_history, _ := c.GetInt("T_history")
  129. tpCount := Warning.Read_WarningType_Count()
  130. var tpList []string
  131. if len(T_tp) > 0 {
  132. tpList = lib.SplitStringIds(T_tp, "T")
  133. if len(tpList) == int(tpCount) {
  134. // 类型全选,替换为空,不进行查询筛选(查询全部)
  135. tpList = []string{}
  136. }
  137. } else {
  138. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  139. if power.T_warning != "*" {
  140. tpList = lib.SplitStringIds(power.T_warning, "W")
  141. }
  142. }
  143. countRedisKey := fmt.Sprintf("Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, T_admin, T_history)
  144. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  145. if err != nil {
  146. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  147. c.ServeJSON()
  148. return
  149. }
  150. // 管理员界面
  151. if T_admin == 1 {
  152. // 获取最新
  153. r_jsons.Data, r_jsons.Num = Warning.Gorm_Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  154. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  155. c.ServeJSON()
  156. return
  157. }
  158. r_jsons.Data, r_jsons.Num = Warning.Gorm_Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, countRedisKey)
  159. r_jsons.Page = page
  160. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  161. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  162. c.ServeJSON()
  163. return
  164. }
  165. func (c *DeviceController) DeviceWarning_List_Sse() {
  166. page, _ := c.GetInt("page")
  167. if page < 1 {
  168. page = 1
  169. }
  170. page_z, _ := c.GetInt("page_z")
  171. if page_z < 1 {
  172. page_z = conf.Page_size
  173. }
  174. T_name := c.GetString("T_name")
  175. T_tp := c.GetString("T_tp")
  176. Time_start := c.GetString("Time_start")
  177. Time_end := c.GetString("Time_end")
  178. T_handle, _ := c.GetInt("T_handle")
  179. T_admin, _ := c.GetInt("T_admin")
  180. tpCount := Warning.Read_WarningType_Count()
  181. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  182. if err != nil {
  183. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  184. c.ServeJSON()
  185. return
  186. }
  187. var tpList []string
  188. if len(T_tp) > 0 {
  189. tpList = lib.SplitStringIds(T_tp, "T")
  190. if len(tpList) == int(tpCount) {
  191. // 类型全选,替换为空,不进行查询筛选(查询全部)
  192. tpList = []string{}
  193. }
  194. } else {
  195. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  196. if power.T_warning != "*" {
  197. tpList = lib.SplitStringIds(power.T_warning, "W")
  198. }
  199. }
  200. countRedisKey := fmt.Sprintf("Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, T_admin, 0)
  201. tableNames := Warning.GenerateWarningTableNames(Time_start, Time_end)
  202. // 设置响应头
  203. c.Ctx.ResponseWriter.Header().Set("Content-Type", "text/event-stream")
  204. c.Ctx.ResponseWriter.Header().Set("Cache-Control", "no-cache")
  205. c.Ctx.ResponseWriter.Header().Set("Connection", "keep-alive")
  206. c.Ctx.ResponseWriter.WriteHeader(http.StatusOK)
  207. var cnt int64
  208. tableCount := make(map[string]int64)
  209. // 管理员界面
  210. if T_admin == 1 {
  211. gormReadWarningListScopes := Warning.GormReadAdminWarningListScopes(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end)
  212. cnt, tableCount, err = Warning.Redis_WarningCount_GetOne(c.Admin_r.Id, countRedisKey)
  213. offset := (page - 1) * page_z
  214. remaining := offset
  215. var warningsCount int
  216. for _, tableName := range tableNames {
  217. tableDB := db.DB.Table(tableName)
  218. var count int64
  219. if tableName == fmt.Sprintf("warning_%d_%02d", time.Now().Year(), time.Now().Month()) {
  220. if saveTime, ok := tableCount["SaveTime"]; ok {
  221. t1 := time.Unix(saveTime, 0)
  222. if time.Now().Sub(t1) < 5*time.Minute {
  223. if count, ok = tableCount[tableName]; !ok {
  224. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  225. tableCount[tableName] = count
  226. }
  227. } else {
  228. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  229. tableCount[tableName] = count
  230. }
  231. }
  232. } else {
  233. var ok bool
  234. if count, ok = tableCount[tableName]; !ok {
  235. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  236. tableCount[tableName] = count
  237. }
  238. }
  239. cnt += count
  240. if remaining >= int(count) {
  241. remaining -= int(count)
  242. continue
  243. }
  244. var tableWarning []Warning.Warning
  245. if warningsCount < page_z {
  246. tableDB.Scopes(gormReadWarningListScopes).
  247. Order("t__ut desc").Offset(remaining).Limit(page_z - warningsCount).Find(&tableWarning)
  248. }
  249. warningsCount += len(tableWarning)
  250. remaining = 0
  251. var r_jsons lib.R_JSONS
  252. r_jsons.Num = cnt
  253. r_jsons.Data = tableWarning
  254. json_b, _ := json.Marshal(r_jsons)
  255. c.Ctx.ResponseWriter.Write(json_b)
  256. c.Ctx.ResponseWriter.Flush()
  257. // 可以根据需要调整间隔时间
  258. time.Sleep(1 * time.Second)
  259. }
  260. Warning.Redis_WarningCount_Set(c.Admin_r.Id, countRedisKey, cnt, tableCount)
  261. return
  262. }
  263. //从redis获取报警统计数量
  264. gormReadWarningListScopes := Warning.GormReadWarningListScopes(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end)
  265. cnt, tableCount, err = Warning.Redis_WarningCount_GetOne(c.T_pid, countRedisKey)
  266. offset := (page - 1) * page_z
  267. remaining := offset
  268. var warningsCount int
  269. for _, tableName := range tableNames {
  270. tableDB := db.DB.Table(tableName)
  271. var count int64
  272. if tableName == fmt.Sprintf("warning_%d_%02d", time.Now().Year(), time.Now().Month()) {
  273. if saveTime, ok := tableCount["SaveTime"]; ok {
  274. t1 := time.Unix(saveTime, 0)
  275. if time.Now().Sub(t1) < 5*time.Minute {
  276. if count, ok = tableCount[tableName]; !ok {
  277. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  278. tableCount[tableName] = count
  279. }
  280. } else {
  281. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  282. tableCount[tableName] = count
  283. }
  284. }
  285. } else {
  286. var ok bool
  287. if count, ok = tableCount[tableName]; !ok {
  288. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  289. tableCount[tableName] = count
  290. }
  291. }
  292. cnt += count
  293. if remaining >= int(count) {
  294. remaining -= int(count)
  295. continue
  296. }
  297. var tableWarning []Warning.Warning
  298. if warningsCount < page_z {
  299. tableDB.Scopes(gormReadWarningListScopes).
  300. Order("t__ut desc").Offset(remaining).Limit(page_z - warningsCount).Find(&tableWarning)
  301. }
  302. warningsCount += len(tableWarning)
  303. remaining = 0
  304. var r_jsons lib.R_JSONS
  305. r_jsons.Num = cnt
  306. r_jsons.Data = tableWarning
  307. json_b, _ := json.Marshal(r_jsons)
  308. c.Ctx.ResponseWriter.Write(json_b)
  309. c.Ctx.ResponseWriter.Flush()
  310. // 可以根据需要调整间隔时间
  311. time.Sleep(1 * time.Second)
  312. }
  313. Warning.Redis_WarningCount_Set(c.T_pid, countRedisKey, cnt, tableCount)
  314. }
  315. func (c *DeviceController) CompanyWarning_List() {
  316. var r_jsons lib.R_JSONS
  317. page, _ := c.GetInt("page")
  318. if page < 1 {
  319. page = 1
  320. }
  321. page_z, _ := c.GetInt("page_z")
  322. if page_z < 1 {
  323. page_z = conf.Page_size
  324. }
  325. T_name := c.GetString("T_name")
  326. T_tp := c.GetString("T_tp")
  327. Time_start := c.GetString("Time_start")
  328. Time_end := c.GetString("Time_end")
  329. T_handle, _ := c.GetInt("T_handle")
  330. T_history, _ := c.GetInt("T_history")
  331. tpCount := Warning.Read_WarningType_Count()
  332. var tpList []string
  333. if len(T_tp) > 0 {
  334. tpList = lib.SplitStringIds(T_tp, "T")
  335. if len(tpList) == int(tpCount) {
  336. // 类型全选,替换为空,不进行查询筛选(查询全部)
  337. tpList = []string{}
  338. }
  339. } else {
  340. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  341. if power.T_warning != "*" {
  342. tpList = lib.SplitStringIds(power.T_warning, "W")
  343. }
  344. }
  345. countRedisKey := fmt.Sprintf("Company_Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0, T_history)
  346. Company_r, err := Account.Read_Company_ById(c.T_pid)
  347. if err != nil {
  348. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
  349. c.ServeJSON()
  350. return
  351. }
  352. T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
  353. var T_year, T_month string
  354. if T_history == 1 {
  355. date, err := time.Parse("2006-01-02 15:04:05", Time_start)
  356. if err != nil {
  357. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  358. c.ServeJSON()
  359. return
  360. }
  361. T_year, T_month = date.Format("2006"), date.Format("01")
  362. }
  363. Account.Read_Company_All_Maps()
  364. if T_history == 1 {
  365. r_jsons.Data, r_jsons.Num = Warning.Read_Company_Warning_Backups(T_pids, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, c.T_pid, countRedisKey)
  366. } else {
  367. // 获取最新
  368. r_jsons.Data, r_jsons.Num = Warning.Read_Company_Warning_List(T_pids, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, c.T_pid, countRedisKey)
  369. }
  370. r_jsons.Page = page
  371. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  372. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  373. c.ServeJSON()
  374. return
  375. }
  376. func (c *DeviceController) CompanyWarning_List2() {
  377. var r_jsons lib.R_JSONS
  378. page, _ := c.GetInt("page")
  379. if page < 1 {
  380. page = 1
  381. }
  382. page_z, _ := c.GetInt("page_z")
  383. if page_z < 1 {
  384. page_z = conf.Page_size
  385. }
  386. T_name := c.GetString("T_name")
  387. T_tp := c.GetString("T_tp")
  388. Time_start := c.GetString("Time_start")
  389. Time_end := c.GetString("Time_end")
  390. T_handle, _ := c.GetInt("T_handle")
  391. tpCount := Warning.Read_WarningType_Count()
  392. var tpList []string
  393. if len(T_tp) > 0 {
  394. tpList = lib.SplitStringIds(T_tp, "T")
  395. if len(tpList) == int(tpCount) {
  396. // 类型全选,替换为空,不进行查询筛选(查询全部)
  397. tpList = []string{}
  398. }
  399. } else {
  400. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  401. if power.T_warning != "*" {
  402. tpList = lib.SplitStringIds(power.T_warning, "W")
  403. }
  404. }
  405. countRedisKey := fmt.Sprintf("Company_Warning_Count_%d_%s_%s_%s_%s_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0)
  406. Company_r, err := Account.Read_Company_ById(c.T_pid)
  407. if err != nil {
  408. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
  409. c.ServeJSON()
  410. return
  411. }
  412. T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
  413. Account.Read_Company_All_Maps()
  414. // 获取最新
  415. r_jsons.Data, r_jsons.Num = Warning.Gorm_Read_Company_Warning_List(T_pids, tpList, T_name, T_handle, Time_start, Time_end, page, page_z, c.T_pid, countRedisKey)
  416. r_jsons.Page = page
  417. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  418. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  419. c.ServeJSON()
  420. return
  421. }
  422. // 查询告警 暂时没有用到
  423. func (c *DeviceController) DeviceWarning_Get() {
  424. id, _ := c.GetInt("T_id")
  425. T_time := c.GetString("T_time")
  426. if len(T_time) == 0 {
  427. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  428. c.ServeJSON()
  429. return
  430. }
  431. var T_year, T_month string
  432. date, err := time.Parse("2006-01-02 15:04:05", T_time)
  433. if err != nil {
  434. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  435. c.ServeJSON()
  436. return
  437. }
  438. T_year, T_month = date.Format("2006"), date.Format("01")
  439. Wtab := "warning"
  440. Wtab += "_" + T_year + "_" + T_month
  441. T, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
  442. if err != nil {
  443. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  444. c.ServeJSON()
  445. return
  446. }
  447. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.WarningToWarning_R(0, T)}
  448. c.ServeJSON()
  449. return
  450. }
  451. // 编辑告警(处理告警)
  452. func (c *DeviceController) DeviceWarning_Post() {
  453. id, _ := c.GetInt("T_id")
  454. T_Text := c.GetString("T_Text")
  455. T_time := c.GetString("T_time")
  456. T_history, _ := c.GetInt("T_history") // TODO 后期废弃
  457. if len(T_time) == 0 && conf.UseNewWarningQuery {
  458. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  459. c.ServeJSON()
  460. return
  461. }
  462. var T_year, T_month string
  463. var warning Warning.Warning
  464. Wtab := "warning"
  465. if T_history == 1 {
  466. date, err := time.Parse("2006-01-02 15:04:05", T_time)
  467. if err != nil {
  468. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  469. c.ServeJSON()
  470. return
  471. }
  472. T_year, T_month = date.Format("2006"), date.Format("01")
  473. Wtab += "_" + T_year + "_" + T_month
  474. warning, err = Warning.Read_Warning_ById_Backups(id, T_year, T_month)
  475. if err != nil {
  476. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  477. c.ServeJSON()
  478. return
  479. }
  480. warning.T_Text = T_Text
  481. warning.T_State = 2
  482. warning.UpdateTime = time.Now()
  483. if is := Warning.Update_Warning_Backups(warning, T_year, T_month); !is {
  484. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  485. c.ServeJSON()
  486. return
  487. }
  488. if !conf.UseNewWarningQuery {
  489. warning = Warning.Read_Warning_ById(int64(id))
  490. if warning.Id > 0 {
  491. warning.T_Text = T_Text
  492. warning.T_State = 2
  493. warning.UpdateTime = time.Now()
  494. if is := Warning.Update_Warning(warning, "T_Text", "T_State", "UpdateTime"); !is {
  495. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  496. c.ServeJSON()
  497. return
  498. }
  499. }
  500. }
  501. } else {
  502. // 使用新的查询方式后 warning表不做更新
  503. if !conf.UseNewWarningQuery {
  504. warning = Warning.Read_Warning_ById(int64(id))
  505. if warning.Id == 0 {
  506. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  507. c.ServeJSON()
  508. return
  509. }
  510. warning.T_Text = T_Text
  511. warning.T_State = 2
  512. warning.UpdateTime = time.Now()
  513. if is := Warning.Update_Warning(warning, "T_Text", "T_State", "UpdateTime"); !is {
  514. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  515. c.ServeJSON()
  516. return
  517. }
  518. }
  519. // 同步处理历史数据
  520. date := warning.T_Ut
  521. T_year, T_month = date.Format("2006"), date.Format("01")
  522. Wtab += "_" + T_year + "_" + T_month
  523. warningBackups, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
  524. if err != nil && err.Error() != "<QuerySeter> no row found" {
  525. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  526. c.ServeJSON()
  527. return
  528. }
  529. if warningBackups.Id > 0 {
  530. warningBackups.T_Text = T_Text
  531. warningBackups.T_State = 2
  532. warningBackups.UpdateTime = time.Now()
  533. if is := Warning.Update_Warning_Backups(warningBackups, T_year, T_month); !is {
  534. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  535. c.ServeJSON()
  536. return
  537. }
  538. }
  539. }
  540. // 删除报警列表统计的缓存
  541. Warning.Redis_WarningCount_DelK(c.T_pid)
  542. Warning.Redis_WarningCount_DelK(c.Admin_r.Id)
  543. System.Add_UserLogs(c.Admin_r.T_uuid, "设备管理", "报警处理操作", Wtab+":"+strconv.Itoa(id)+"->"+T_Text)
  544. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  545. c.ServeJSON()
  546. return
  547. }
  548. // 删除告警
  549. func (c *DeviceController) DeviceWarning_Del() {
  550. id, _ := c.GetInt("T_id")
  551. T_time := c.GetString("T_time")
  552. T_history, _ := c.GetInt("T_history") // TODO 后期废弃
  553. if len(T_time) == 0 && conf.UseNewWarningQuery {
  554. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  555. c.ServeJSON()
  556. return
  557. }
  558. var T_year, T_month string
  559. var warning Warning.Warning
  560. Wtab := "warning"
  561. if T_history == 1 {
  562. date, err := time.Parse("2006-01-02 15:04:05", T_time)
  563. if err != nil {
  564. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  565. c.ServeJSON()
  566. return
  567. }
  568. T_year, T_month = date.Format("2006"), date.Format("01")
  569. Wtab += "_" + T_year + "_" + T_month
  570. warning, err = Warning.Read_Warning_ById_Backups(id, T_year, T_month)
  571. if err != nil {
  572. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  573. c.ServeJSON()
  574. return
  575. }
  576. warning.T_State = 0
  577. warning.UpdateTime = time.Now()
  578. if is := Warning.Update_Warning_Backups(warning, T_year, T_month); !is {
  579. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  580. c.ServeJSON()
  581. return
  582. }
  583. warning = Warning.Read_Warning_ById(int64(id))
  584. if warning.Id > 0 {
  585. warning.T_State = 0
  586. if is := Warning.Update_Warning(warning, "T_State"); !is {
  587. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  588. c.ServeJSON()
  589. return
  590. }
  591. }
  592. } else {
  593. warning = Warning.Read_Warning_ById(int64(id))
  594. if warning.Id == 0 {
  595. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  596. c.ServeJSON()
  597. return
  598. }
  599. warning.T_State = 0
  600. if is := Warning.Update_Warning(warning, "T_State"); !is {
  601. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  602. c.ServeJSON()
  603. return
  604. }
  605. date := warning.T_Ut
  606. T_year, T_month = date.Format("2006"), date.Format("01")
  607. Wtab += "_" + T_year + "_" + T_month
  608. warningBackups, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
  609. if err != nil && err.Error() != "<QuerySeter> no row found" {
  610. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
  611. c.ServeJSON()
  612. return
  613. }
  614. if warningBackups.Id > 0 {
  615. warningBackups.T_State = 0
  616. warningBackups.UpdateTime = time.Now()
  617. if is := Warning.Update_Warning_Backups(warningBackups, T_year, T_month); !is {
  618. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  619. c.ServeJSON()
  620. return
  621. }
  622. }
  623. }
  624. // 删除报警列表统计的缓存
  625. Warning.Redis_WarningCount_DelK(c.T_pid)
  626. Warning.Redis_WarningCount_DelK(c.Admin_r.Id)
  627. System.Add_UserLogs(c.Admin_r.T_uuid, "设备管理", "报警删除操作", Wtab+":"+strconv.Itoa(id))
  628. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  629. c.ServeJSON()
  630. return
  631. }
  632. // 导出告警
  633. func (c *DeviceController) DeviceWarning_Data_Excel() {
  634. T_name := c.GetString("T_name")
  635. T_tp := c.GetString("T_tp")
  636. Time_start := c.GetString("Time_start")
  637. Time_end := c.GetString("Time_end")
  638. T_handle, _ := c.GetInt("T_handle")
  639. T_admin, _ := c.GetInt("T_admin")
  640. T_history, _ := c.GetInt("T_history")
  641. tpCount := Warning.Read_WarningType_Count()
  642. var tpList []string
  643. if len(T_tp) > 0 {
  644. tpList = lib.SplitStringIds(T_tp, "T")
  645. if len(tpList) == int(tpCount) {
  646. // 类型全选,替换为空,不进行查询筛选(查询全部)
  647. tpList = []string{}
  648. }
  649. } else {
  650. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  651. if power.T_warning != "*" {
  652. tpList = lib.SplitStringIds(power.T_warning, "W")
  653. }
  654. }
  655. countRedisKey := fmt.Sprintf("Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0, T_history)
  656. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  657. if err != nil {
  658. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  659. c.ServeJSON()
  660. return
  661. }
  662. var T_year, T_month string
  663. if T_history == 1 {
  664. date, err := time.Parse("2006-01-02 15:04:05", Time_start)
  665. if err != nil {
  666. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  667. c.ServeJSON()
  668. return
  669. }
  670. T_year, T_month = date.Format("2006"), date.Format("01")
  671. }
  672. var Device_data []Warning.Warning_R
  673. // 管理员界面
  674. if T_admin == 1 {
  675. if T_history == 1 {
  676. // 获取备份
  677. Device_data, _ = Warning.Read_Admin_Warning_Backups(*c.Admin_r, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  678. } else {
  679. // 获取最新
  680. Device_data, _ = Warning.Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  681. }
  682. }
  683. if T_history == 1 {
  684. // 获取备份
  685. Device_data, _ = Warning.Read_Warning_Backups(c.T_pid, bindSN, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  686. } else {
  687. // 获取最新
  688. Device_data, _ = Warning.Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  689. }
  690. f := excelize.NewFile() // 设置单元格的值
  691. // 这里设置表头ÒÒ
  692. f.SetCellValue("Sheet1", "A1", "报警类型")
  693. f.SetCellValue("Sheet1", "B1", "Sn")
  694. f.SetCellValue("Sheet1", "C1", "设备名称")
  695. f.SetCellValue("Sheet1", "D1", "传感器")
  696. f.SetCellValue("Sheet1", "E1", "报警内容")
  697. f.SetCellValue("Sheet1", "F1", "记录时间")
  698. f.SetCellValue("Sheet1", "G1", "处理")
  699. f.SetCellValue("Sheet1", "H1", "处理时间")
  700. // 设置列宽
  701. f.SetColWidth("Sheet1", "A", "A", 20)
  702. f.SetColWidth("Sheet1", "B", "B", 25)
  703. f.SetColWidth("Sheet1", "C", "C", 30)
  704. f.SetColWidth("Sheet1", "D", "D", 30)
  705. f.SetColWidth("Sheet1", "G", "E", 30)
  706. f.SetColWidth("Sheet1", "H", "F", 15)
  707. f.SetColWidth("Sheet1", "I", "G", 30)
  708. f.SetColWidth("Sheet1", "J", "H", 15)
  709. line := 1
  710. // 循环写入数据
  711. for _, v := range Device_data {
  712. line++
  713. f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
  714. f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
  715. f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), v.T_D_name)
  716. f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_DS_name)
  717. f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_Remark)
  718. f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Ut)
  719. f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Text)
  720. f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.UpdateTime)
  721. }
  722. timeStr := time.Now().Format("20060102150405")
  723. // 保存文件
  724. if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
  725. fmt.Println(err)
  726. }
  727. url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
  728. if !is {
  729. c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
  730. c.ServeJSON()
  731. return
  732. }
  733. // 删除文件
  734. err = os.Remove("ofile/" + timeStr + ".xlsx")
  735. if err != nil {
  736. fmt.Println(err)
  737. }
  738. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
  739. c.ServeJSON()
  740. return
  741. }
  742. func (c *DeviceController) DeviceWarning_Data_Excel2() {
  743. T_name := c.GetString("T_name")
  744. T_tp := c.GetString("T_tp")
  745. Time_start := c.GetString("Time_start")
  746. Time_end := c.GetString("Time_end")
  747. T_handle, _ := c.GetInt("T_handle")
  748. T_admin, _ := c.GetInt("T_admin")
  749. tpCount := Warning.Read_WarningType_Count()
  750. var tpList []string
  751. if len(T_tp) > 0 {
  752. tpList = lib.SplitStringIds(T_tp, "T")
  753. if len(tpList) == int(tpCount) {
  754. // 类型全选,替换为空,不进行查询筛选(查询全部)
  755. tpList = []string{}
  756. }
  757. } else {
  758. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  759. if power.T_warning != "*" {
  760. tpList = lib.SplitStringIds(power.T_warning, "W")
  761. }
  762. }
  763. countRedisKey := fmt.Sprintf("Warning_Count_%d_%s_%s_%s_%s_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0)
  764. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  765. if err != nil {
  766. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  767. c.ServeJSON()
  768. return
  769. }
  770. var Device_data []Warning.Warning_R
  771. // 管理员界面
  772. if T_admin == 1 {
  773. // 获取最新
  774. //Device_data, _ = Warning.Gorm_Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  775. Device_data, _ = Warning.Gorm_Read_Admin_Warning_List_Excel(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  776. } else {
  777. // 获取最新
  778. //Device_data, _ = Warning.Gorm_Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  779. Device_data, _ = Warning.Gorm_Read_Warning_List_Excel(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
  780. }
  781. f := excelize.NewFile() // 设置单元格的值
  782. // 这里设置表头ÒÒ
  783. f.SetCellValue("Sheet1", "A1", "报警类型")
  784. f.SetCellValue("Sheet1", "B1", "Sn")
  785. f.SetCellValue("Sheet1", "C1", "设备名称")
  786. f.SetCellValue("Sheet1", "D1", "传感器")
  787. f.SetCellValue("Sheet1", "E1", "报警内容")
  788. f.SetCellValue("Sheet1", "F1", "记录时间")
  789. f.SetCellValue("Sheet1", "G1", "处理")
  790. f.SetCellValue("Sheet1", "H1", "处理时间")
  791. // 设置列宽
  792. f.SetColWidth("Sheet1", "A", "A", 20)
  793. f.SetColWidth("Sheet1", "B", "B", 25)
  794. f.SetColWidth("Sheet1", "C", "C", 30)
  795. f.SetColWidth("Sheet1", "D", "D", 30)
  796. f.SetColWidth("Sheet1", "G", "E", 30)
  797. f.SetColWidth("Sheet1", "H", "F", 15)
  798. f.SetColWidth("Sheet1", "I", "G", 30)
  799. f.SetColWidth("Sheet1", "J", "H", 15)
  800. line := 1
  801. // 循环写入数据
  802. for _, v := range Device_data {
  803. line++
  804. f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
  805. f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
  806. f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), v.T_D_name)
  807. f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_DS_name)
  808. f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_Remark)
  809. f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Ut)
  810. f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Text)
  811. f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.UpdateTime)
  812. }
  813. timeStr := time.Now().Format("20060102150405")
  814. // 保存文件
  815. if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
  816. fmt.Println(err)
  817. }
  818. url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
  819. if !is {
  820. c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
  821. c.ServeJSON()
  822. return
  823. }
  824. // 删除文件
  825. err = os.Remove("ofile/" + timeStr + ".xlsx")
  826. if err != nil {
  827. fmt.Println(err)
  828. }
  829. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
  830. c.ServeJSON()
  831. return
  832. }
  833. // 公司管理 - 导出告警
  834. func (c *DeviceController) CompanyWarning_Data_Excel() {
  835. T_name := c.GetString("T_name")
  836. T_tp := c.GetString("T_tp")
  837. Time_start := c.GetString("Time_start")
  838. Time_end := c.GetString("Time_end")
  839. T_handle, _ := c.GetInt("T_handle")
  840. T_history, _ := c.GetInt("T_history")
  841. tpCount := Warning.Read_WarningType_Count()
  842. var tpList []string
  843. if len(T_tp) > 0 {
  844. tpList = lib.SplitStringIds(T_tp, "T")
  845. if len(tpList) == int(tpCount) {
  846. // 类型全选,替换为空,不进行查询筛选(查询全部)
  847. tpList = []string{}
  848. }
  849. } else {
  850. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  851. if power.T_warning != "*" {
  852. tpList = lib.SplitStringIds(power.T_warning, "W")
  853. }
  854. }
  855. countRedisKey := fmt.Sprintf("Company_Warning_Count_%d_%s_%s_%s_%s_%d_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0, T_history)
  856. Account.Read_Company_All_Maps()
  857. Company_r, err := Account.Read_Company_ById(c.T_pid)
  858. if err != nil {
  859. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
  860. c.ServeJSON()
  861. return
  862. }
  863. T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
  864. var T_year, T_month string
  865. if T_history == 1 {
  866. date, err := time.Parse("2006-01-02 15:04:05", Time_start)
  867. if err != nil {
  868. c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
  869. c.ServeJSON()
  870. return
  871. }
  872. T_year, T_month = date.Format("2006"), date.Format("01")
  873. }
  874. var Device_data []Warning.CompanyWarning_R
  875. if T_history == 1 {
  876. // 获取备份
  877. Device_data, _ = Warning.Read_Company_Warning_Backups(T_pids, T_year, T_month, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, c.T_pid, countRedisKey)
  878. } else {
  879. // 获取最新
  880. Device_data, _ = Warning.Read_Company_Warning_List(T_pids, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, c.T_pid, countRedisKey)
  881. }
  882. f := excelize.NewFile() // 设置单元格的值
  883. // 这里设置表头
  884. f.SetCellValue("Sheet1", "A1", "报警类型")
  885. f.SetCellValue("Sheet1", "A1", "公司名称")
  886. f.SetCellValue("Sheet1", "B1", "Sn")
  887. f.SetCellValue("Sheet1", "C1", "设备名称")
  888. f.SetCellValue("Sheet1", "D1", "传感器")
  889. f.SetCellValue("Sheet1", "E1", "报警内容")
  890. f.SetCellValue("Sheet1", "F1", "记录时间")
  891. f.SetCellValue("Sheet1", "G1", "处理")
  892. f.SetCellValue("Sheet1", "H1", "处理时间")
  893. // 设置列宽
  894. f.SetColWidth("Sheet1", "A", "A", 20)
  895. f.SetColWidth("Sheet1", "B", "B", 40)
  896. f.SetColWidth("Sheet1", "C", "C", 25)
  897. f.SetColWidth("Sheet1", "D", "D", 30)
  898. f.SetColWidth("Sheet1", "E", "E", 30)
  899. f.SetColWidth("Sheet1", "F", "F", 30)
  900. f.SetColWidth("Sheet1", "J", "J", 15)
  901. f.SetColWidth("Sheet1", "H", "H", 30)
  902. f.SetColWidth("Sheet1", "I", "I", 15)
  903. line := 1
  904. // 循环写入数据
  905. for _, v := range Device_data {
  906. line++
  907. f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
  908. f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s", v.T_pid_name))
  909. f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
  910. f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_D_name)
  911. f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_DS_name)
  912. f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Remark)
  913. f.SetCellValue("Sheet1", fmt.Sprintf("J%d", line), v.T_Ut)
  914. f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.T_Text)
  915. f.SetCellValue("Sheet1", fmt.Sprintf("I%d", line), v.UpdateTime)
  916. }
  917. timeStr := time.Now().Format("20060102150405")
  918. // 保存文件
  919. if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
  920. fmt.Println(err)
  921. }
  922. url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
  923. if !is {
  924. c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
  925. c.ServeJSON()
  926. return
  927. }
  928. // 删除文件
  929. err = os.Remove("ofile/" + timeStr + ".xlsx")
  930. if err != nil {
  931. fmt.Println(err)
  932. }
  933. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
  934. c.ServeJSON()
  935. return
  936. }
  937. func (c *DeviceController) CompanyWarning_Data_Excel2() {
  938. T_name := c.GetString("T_name")
  939. T_tp := c.GetString("T_tp")
  940. Time_start := c.GetString("Time_start")
  941. Time_end := c.GetString("Time_end")
  942. T_handle, _ := c.GetInt("T_handle")
  943. tpCount := Warning.Read_WarningType_Count()
  944. var tpList []string
  945. if len(T_tp) > 0 {
  946. tpList = lib.SplitStringIds(T_tp, "T")
  947. if len(tpList) == int(tpCount) {
  948. // 类型全选,替换为空,不进行查询筛选(查询全部)
  949. tpList = []string{}
  950. }
  951. } else {
  952. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  953. if power.T_warning != "*" {
  954. tpList = lib.SplitStringIds(power.T_warning, "W")
  955. }
  956. }
  957. countRedisKey := fmt.Sprintf("Company_Warning_Count_%d_%s_%s_%s_%s_%d_%d", c.T_pid, T_name, T_tp, Time_start, Time_end, T_handle, 0)
  958. Account.Read_Company_All_Maps()
  959. Company_r, err := Account.Read_Company_ById(c.T_pid)
  960. if err != nil {
  961. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
  962. c.ServeJSON()
  963. return
  964. }
  965. T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
  966. var Device_data []Warning.CompanyWarning_R
  967. // 获取最新
  968. Device_data, _ = Warning.Gorm_Read_Company_Warning_List_Excel(T_pids, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, c.T_pid, countRedisKey)
  969. f := excelize.NewFile() // 设置单元格的值
  970. // 这里设置表头
  971. f.SetCellValue("Sheet1", "A1", "报警类型")
  972. f.SetCellValue("Sheet1", "B1", "公司名称")
  973. f.SetCellValue("Sheet1", "C1", "Sn")
  974. f.SetCellValue("Sheet1", "D1", "设备名称")
  975. f.SetCellValue("Sheet1", "E1", "传感器")
  976. f.SetCellValue("Sheet1", "F1", "报警内容")
  977. f.SetCellValue("Sheet1", "G1", "记录时间")
  978. f.SetCellValue("Sheet1", "H1", "处理")
  979. f.SetCellValue("Sheet1", "I1", "处理时间")
  980. // 设置列宽
  981. f.SetColWidth("Sheet1", "A", "A", 20)
  982. f.SetColWidth("Sheet1", "B", "B", 40)
  983. f.SetColWidth("Sheet1", "C", "C", 25)
  984. f.SetColWidth("Sheet1", "D", "D", 30)
  985. f.SetColWidth("Sheet1", "E", "E", 30)
  986. f.SetColWidth("Sheet1", "F", "F", 30)
  987. f.SetColWidth("Sheet1", "J", "J", 15)
  988. f.SetColWidth("Sheet1", "H", "H", 30)
  989. f.SetColWidth("Sheet1", "I", "I", 15)
  990. line := 1
  991. // 循环写入数据
  992. for _, v := range Device_data {
  993. line++
  994. f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
  995. f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s", v.T_pid_name))
  996. f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
  997. f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_D_name)
  998. f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_DS_name)
  999. f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Remark)
  1000. f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Ut)
  1001. f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.T_Text)
  1002. f.SetCellValue("Sheet1", fmt.Sprintf("I%d", line), v.UpdateTime)
  1003. }
  1004. timeStr := time.Now().Format("20060102150405")
  1005. // 保存文件
  1006. if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
  1007. fmt.Println(err)
  1008. }
  1009. url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
  1010. if !is {
  1011. c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
  1012. c.ServeJSON()
  1013. return
  1014. }
  1015. // 删除文件
  1016. err = os.Remove("ofile/" + timeStr + ".xlsx")
  1017. if err != nil {
  1018. fmt.Println(err)
  1019. }
  1020. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
  1021. c.ServeJSON()
  1022. return
  1023. }
  1024. // 告警提示列表
  1025. func (c *DeviceController) DeviceWarningList_T_Tips() {
  1026. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_All_T_Notice_mechanism()}
  1027. c.ServeJSON()
  1028. return
  1029. }
  1030. // 告警类型列表
  1031. func (c *DeviceController) WarningType_List_All() {
  1032. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_All()}
  1033. c.ServeJSON()
  1034. return
  1035. }
  1036. // 告警类型列表 - 权限关联列表
  1037. func (c *DeviceController) WarningType_Power_List_All() {
  1038. power, err := Account.Read_Power_ById(c.Admin_r.T_power)
  1039. if err != nil {
  1040. c.Data["json"] = lib.JSONS{Code: 200, Msg: "获取菜单失败"}
  1041. c.ServeJSON()
  1042. return
  1043. }
  1044. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_Power_All(power.T_warning)}
  1045. c.ServeJSON()
  1046. return
  1047. }
  1048. // 通过传感器类型获取报警列表
  1049. func (c *DeviceController) Read_Warning_List_By_DS_T_type() {
  1050. var r_jsons lib.R_JSONS
  1051. page, _ := c.GetInt("page")
  1052. if page < 1 {
  1053. page = 1
  1054. }
  1055. page_z, _ := c.GetInt("page_z")
  1056. if page_z < 1 {
  1057. page_z = conf.Page_size
  1058. }
  1059. T_type, _ := c.GetInt("T_type")
  1060. T_name := c.GetString("T_name")
  1061. bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
  1062. if err != nil {
  1063. c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
  1064. c.ServeJSON()
  1065. return
  1066. }
  1067. var T_pids []int
  1068. if c.Admin_r.T_pid > 0 {
  1069. // 查询公司
  1070. Company_r, err := Account.Read_Company_ById(c.Admin_r.T_pid)
  1071. if err != nil {
  1072. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
  1073. c.ServeJSON()
  1074. return
  1075. }
  1076. // 查询公司下面所有子公司id
  1077. T_pids = Account.ReadCompanyIds_T_path(Company_r.T_path)
  1078. }
  1079. var tpList []string
  1080. power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
  1081. if power.T_warning != "*" {
  1082. tpList = lib.SplitStringIds(power.T_warning, "W")
  1083. }
  1084. countRedisKey := fmt.Sprintf("Company_Warning_Count_By_DS_T_type_%d_%s_%d_%d", c.T_pid, T_name, T_type, c.Admin_r.T_pid)
  1085. r_jsons.Data, r_jsons.Num = Warning.Read_Warning_List_By_DS_T_type(c.Admin_r, c.T_pid, bindSN, T_pids, T_type, T_name, tpList, page, page_z, countRedisKey)
  1086. r_jsons.Page = page
  1087. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  1088. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  1089. c.ServeJSON()
  1090. return
  1091. }
  1092. func (c *DeviceController) Read_WarningSend_List() {
  1093. var r_jsons lib.R_JSONS
  1094. page, _ := c.GetInt("page")
  1095. if page < 1 {
  1096. page = 1
  1097. }
  1098. page_z, _ := c.GetInt("page_z")
  1099. if page_z < 1 {
  1100. page_z = conf.Page_size
  1101. }
  1102. T_ntype, _ := c.GetInt("T_ntype")
  1103. r_jsons.Data, r_jsons.Num = Warning.Read_WarningSand_List(c.T_pid, T_ntype, page, page_z)
  1104. r_jsons.Page = page
  1105. r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
  1106. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  1107. c.ServeJSON()
  1108. return
  1109. }
  1110. func (c *DeviceController) Read_WarningRate_List() {
  1111. page, _ := c.GetInt("page")
  1112. if page < 1 {
  1113. page = 1
  1114. }
  1115. page_z, _ := c.GetInt("page_z")
  1116. if page_z < 1 {
  1117. page_z = conf.Page_size
  1118. }
  1119. T_uuid := c.GetString("T_uuid")
  1120. T_year, _ := c.GetInt("T_year")
  1121. T_month, _ := c.GetInt("T_month")
  1122. var startTime, endTime string
  1123. now := time.Now()
  1124. var month time.Month
  1125. // 默认本月
  1126. if T_year == 0 {
  1127. T_year = now.Year()
  1128. }
  1129. if T_month == 0 {
  1130. month = now.Month()
  1131. } else {
  1132. month = time.Month(T_month)
  1133. }
  1134. T_type := Warning.WarningRateDay
  1135. if T_month == 0 {
  1136. T_type = Warning.WarningRateMonth
  1137. }
  1138. if T_type == Warning.WarningRateDay {
  1139. startTime = time.Date(T_year, month, 1, 0, 0, 0, 0, time.Local).Format("2006-01-02")
  1140. endTime = time.Date(T_year, month+1, 0, 0, 0, 0, 0, time.Local).Format("2006-01-02")
  1141. }
  1142. // 默认本年
  1143. if T_type == Warning.WarningRateMonth {
  1144. startTime = time.Date(T_year, 1, 1, 0, 0, 0, 0, time.Local).Format("2006-01")
  1145. endTime = time.Date(T_year, 12, 31, 0, 0, 0, 0, time.Local).Format("2006-01")
  1146. }
  1147. if len(T_uuid) == 0 {
  1148. T_uuid = c.Admin_r.T_uuid
  1149. }
  1150. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningRate_List(T_uuid, T_type, startTime, endTime)}
  1151. c.ServeJSON()
  1152. return
  1153. }
  1154. func (c *DeviceController) Read_WarningHandle_List() {
  1155. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningHandle_List()}
  1156. c.ServeJSON()
  1157. return
  1158. }
  1159. func Cron_WarningRate() {
  1160. // 创建一个定时任务对象
  1161. c := cron.New(cron.WithSeconds())
  1162. // 给对象增加定时任务
  1163. // @daily 每日运行一次
  1164. // c.AddFunc("0 */1 * * * ?", Cron_WarningRateDay_Add)
  1165. // c.AddFunc("0 */1 * * * ?", Cron_WarningRateMonth_Add)
  1166. c.AddFunc("@daily", Cron_WarningRateDay_Add)
  1167. c.AddFunc("@monthly", Cron_WarningRateMonth_Add)
  1168. // 启动定时任务
  1169. c.Start()
  1170. defer c.Stop()
  1171. // 查询语句,阻塞,让main函数不退出,保持程序运行
  1172. select {}
  1173. }
  1174. // 保存每天的设备报警率
  1175. func Cron_WarningRateDay_Add() {
  1176. T_date := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
  1177. logs.Info("开始统计" + T_date + "设备报警率")
  1178. // 获取管理员用户列表
  1179. adminList := Account.Read_Admin_List_All()
  1180. // 获取内部用户关联的pid
  1181. for _, admin := range adminList {
  1182. // 内部用户已绑定公司,* 绑定所有公司
  1183. if len(admin.T_pids) == 0 || admin.T_pids == "*" {
  1184. continue
  1185. }
  1186. // 获取排除后的pid
  1187. T_pids := lib.SplitStringToIntIds(admin.T_pids, "P")
  1188. if len(T_pids) == 0 {
  1189. continue
  1190. }
  1191. pids := Account.ReadCompanyWarningIds_T_pids(T_pids)
  1192. if len(pids) == 0 {
  1193. continue
  1194. }
  1195. // 获取探头数量
  1196. dsCnt := Device.Read_DeviceSensorCount_ByT_pids(pids)
  1197. // 获取当天报警数量
  1198. warningCount := Warning.Read_WarningCount_byDay(pids)
  1199. T_rate, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(warningCount)/float64(dsCnt)), 64) // 保留2位小数
  1200. // 保存设备报警率
  1201. warningRate := Warning.WarningRate{
  1202. T_uuid: admin.T_uuid,
  1203. T_date: T_date,
  1204. T_type: Warning.WarningRateDay,
  1205. T_device_num: dsCnt,
  1206. T_warning_num: warningCount,
  1207. T_rate: float32(T_rate),
  1208. }
  1209. Warning.Add_WarningRate(warningRate)
  1210. }
  1211. }
  1212. // 保存每月的设备报警率
  1213. func Cron_WarningRateMonth_Add() {
  1214. now := time.Now()
  1215. // fixme 测试
  1216. //now := time.Now().AddDate(0, 1, 0)
  1217. T_date := now.AddDate(0, -1, 0).Format("2006-01")
  1218. firstOfMonth := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, time.Local)
  1219. startTime := firstOfMonth.AddDate(0, -1, 0).Format("2006-01-02") // 上月第一天
  1220. endTime := firstOfMonth.AddDate(0, 0, -1).Format("2006-01-02") // 上月最后一天
  1221. logs.Info("开始统计" + T_date + "设备报警率")
  1222. logs.Info("开始时间" + startTime)
  1223. logs.Info("结束时间" + endTime)
  1224. rateAvgList := Warning.Read_WarningRateAvg_Month(startTime, endTime)
  1225. // 获取内部用户关联的pid
  1226. for _, rate := range rateAvgList {
  1227. T_rate, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(rate.T_warning_num)/float64(rate.T_device_num)), 64) // 保留2位小数
  1228. // 保存设备报警率
  1229. warningRate := Warning.WarningRate{
  1230. T_uuid: rate.T_uuid,
  1231. T_date: T_date,
  1232. T_type: Warning.WarningRateMonth,
  1233. T_device_num: rate.T_device_num,
  1234. T_warning_num: rate.T_warning_num,
  1235. T_rate: float32(T_rate),
  1236. }
  1237. Warning.Add_WarningRate(warningRate)
  1238. }
  1239. }