Warning.go 56 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046
  1. package Warning
  2. import (
  3. "Cold_Api/conf"
  4. "Cold_Api/controllers/lib"
  5. db "Cold_Api/initialize"
  6. "Cold_Api/models/Account"
  7. "Cold_Api/models/System"
  8. "encoding/json"
  9. "errors"
  10. "fmt"
  11. "sort"
  12. "strconv"
  13. "strings"
  14. "sync"
  15. "time"
  16. "github.com/astaxie/beego/cache"
  17. _ "github.com/astaxie/beego/cache/redis"
  18. "github.com/beego/beego/v2/adapter/orm"
  19. orm2 "github.com/beego/beego/v2/client/orm"
  20. "github.com/beego/beego/v2/core/logs"
  21. _ "github.com/go-sql-driver/mysql"
  22. "github.com/gomodule/redigo/redis"
  23. "gorm.io/gorm"
  24. )
  25. var (
  26. CompanyCallthepolice = "companyCallthepolice" // 公司管理-设备报警
  27. Callthepolice = "callthepolice" // 公司管理-设备报警
  28. )
  29. // 模板
  30. type Warning struct {
  31. Id int64 `orm:"column(ID);size(11);auto;pk" gorm:"column:ID"`
  32. T_pid int `orm:"size(256);null" gorm:"column:t_pid"` // Account.Company 绑定公司
  33. T_tp int `orm:"size(200);null" gorm:"column:t_tp"` // 报警类型 ->WarningList
  34. T_sn string `orm:"index;size(256);null" gorm:"column:t_sn"` // 设备序列号
  35. T_D_name string `orm:"size(256);null" gorm:"column:t__d_name"` // 设备名称
  36. T_id int `orm:"size(200);null" gorm:"column:t_id"` // 传感器 ID
  37. T_DS_name string `orm:"size(256);null" gorm:"column:t__d_s_name"` // 传感器名称
  38. T_Remark string `orm:"type(text);null" gorm:"column:t__remark"` // 采集内容
  39. T_Ut time.Time `orm:"type(timestamp);null;" gorm:"column:t__ut"` // 采集时间
  40. T_fUt time.Time `orm:"type(timestamp);null;" gorm:"column:t_f_ut"` // 首次采集时间
  41. T_Text string `orm:"type(text);null" gorm:"column:t__text"` // 处理备注
  42. T_Log string `orm:"type(text);null" gorm:"column:t__log"` // 通知日志
  43. T_Msid int64 `orm:"size(256);null" gorm:"column:t__msid"` // 消息ID
  44. T_State int `orm:"size(2);default(2)" gorm:"column:t__state"` // 0 删除 1 不处理 2 已处理 3 未处理
  45. CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间
  46. UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"` //auto_now 每次 model 保存时都会对时间自动更新
  47. }
  48. // 多字段索引
  49. func (u *Warning) TableIndex() [][]string {
  50. return [][]string{
  51. []string{"T_pid", "T_tp", "T_State"},
  52. }
  53. }
  54. // 附件数据结构
  55. type Attachment struct {
  56. Name string `json:"name"` // 附件名称
  57. Url string `json:"url"` // 附件链接
  58. }
  59. // 处理记录
  60. type HandlingRecord struct {
  61. Remark string `json:"remark"` // 处理备注
  62. Handler string `json:"handler"` // 处理人
  63. Handler_uuid string `json:"handler_uuid"` // 处理人uuid
  64. HandledAt time.Time `json:"handledAt"` // 处理时间
  65. Attachments []Attachment `json:"attachments"` // 处理凭证(图片链接)
  66. }
  67. /*
  68. 分表机制
  69. |--------30天--------|
  70. |-----8月---||--------7月---------||--------6月---------||--------5月---------|
  71. 每天备份 到对应 月份
  72. */
  73. // 模板
  74. type Warning_R struct {
  75. Id int64
  76. T_pid int // Account.Company 绑定公司
  77. T_pid_name string // Account.Company 公司名称
  78. T_tp int // 报警类型 ->WarningList
  79. T_tp_name string // 报警类型名称
  80. T_sn string // 设备序列号
  81. T_D_name string // 设备名称
  82. T_id int // 传感器 ID
  83. T_DS_name string // 传感器名称
  84. T_Remark string // 采集内容
  85. T_Ut string // 采集时间
  86. T_fUt string // 首次采集时间
  87. T_Text string // 处理备注
  88. T_handler string // 处理人
  89. T_Log []string // 处理日志
  90. T_handling_record []HandlingRecord // 处理记录详情(包含附件)
  91. T_attachments []Attachment // 最新一条处理凭证信息
  92. T_Msid int64 // 消息ID
  93. T_State int // 0 删除 1 未处理 2 已处理
  94. T_history int // 0 40天 1 历史数据
  95. CreateTime string // 创建时间
  96. UpdateTime string // 更新时间
  97. }
  98. type CompanyWarning_R struct {
  99. Id int64
  100. T_pid int // Account.Company 绑定公司
  101. T_pid_name string // Account.Company 公司名称
  102. T_tp int // 报警类型 ->WarningList
  103. T_tp_name string // 报警类型名称
  104. T_sn string // 设备序列号
  105. T_D_name string // 设备名称
  106. T_id int // 传感器 ID
  107. T_DS_name string // 传感器名称
  108. T_Remark string // 采集内容
  109. T_Ut string // 采集时间
  110. T_fUt string // 首次采集时间
  111. T_Text string // 处理备注
  112. T_handler string // 处理人
  113. T_Log []string // 处理日志
  114. T_Msid int64 // 消息ID
  115. T_State int // 0 删除 1 未处理 2 已处理
  116. T_history int // 0 40天 1 历史数据
  117. T_attachments []Attachment // 处理凭证(图片链接)
  118. CreateTime string // 创建时间
  119. UpdateTime string // 更新时间
  120. }
  121. type Warning_Applet struct {
  122. Id int64
  123. T_pid int // Account.Company 绑定公司
  124. T_company_name string // 公司名称
  125. T_tp int // 报警类型 ->WarningList
  126. T_tp_name string // 报警类型名称
  127. T_sn string // 设备序列号
  128. T_D_name string // 设备名称
  129. T_id int // 传感器 ID
  130. T_DS_name string // 传感器名称
  131. T_Remark string // 采集内容
  132. T_Ut string // 采集时间
  133. T_fUt string // 首次采集时间
  134. T_Text string // 处理备注
  135. T_handler string // 处理人
  136. T_Log []string // 处理日志
  137. T_Msid int64 // 消息ID
  138. T_State int // 0 删除 1 未处理 2 已处理
  139. T_attachments []Attachment // 处理凭证(图片链接)
  140. CreateTime string // 创建时间
  141. }
  142. type Warning_Count struct {
  143. Key string
  144. Count int64
  145. TableCount map[string]int64
  146. }
  147. func (t *Warning) TableName() string {
  148. return "warning" // 数据库名称 // ************** 替换 FormulaList **************
  149. }
  150. var redis_Warning_Count cache.Cache
  151. func init() {
  152. //注册模型
  153. orm.RegisterModel(new(Warning))
  154. config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
  155. "redis_Warning_Count", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
  156. fmt.Println(config)
  157. var err error
  158. redis_Warning_Count, err = cache.NewCache("redis", config)
  159. if err != nil || redis_Warning_Count == nil {
  160. errMsg := "failed to init redis"
  161. logs.Error(errMsg, err)
  162. panic(errMsg)
  163. }
  164. }
  165. func Redis_WarningCount_Set(id int, key string, count int64, TableCount map[string]int64) (err error) {
  166. //json序列化
  167. idKey := strconv.Itoa(id)
  168. TableCount["SaveTime"] = time.Now().Unix()
  169. warningCount, err := Redis_WarningCount_Get(idKey)
  170. warningCount = append(warningCount, Warning_Count{
  171. Key: key,
  172. Count: count,
  173. TableCount: TableCount,
  174. })
  175. str, err := json.Marshal(warningCount)
  176. if err != nil {
  177. logs.Error(lib.FuncName(), err)
  178. return
  179. }
  180. err = redis_Warning_Count.Put(idKey, str, 30*time.Minute)
  181. if err != nil {
  182. logs.Error("Redis_Admin_Set", "set key:", idKey, ",value:", str, err)
  183. }
  184. return
  185. }
  186. // if r,is :=Redis_Get(T_sn);is{
  187. // return r,nil
  188. // }
  189. func Redis_WarningCount_Get(userId string) (count []Warning_Count, err error) {
  190. if redis_Warning_Count.IsExist(userId) {
  191. //println("找到key:",key)
  192. v := redis_Warning_Count.Get(userId)
  193. err = json.Unmarshal(v.([]byte), &count)
  194. if err != nil {
  195. logs.Error(lib.FuncName(), err)
  196. return count, err
  197. }
  198. return count, nil
  199. }
  200. return count, redis.ErrNil
  201. }
  202. func Redis_WarningCount_GetOne(id int, key string) (count int64, tableCount map[string]int64, err error) {
  203. countList := []Warning_Count{}
  204. tableCount = make(map[string]int64)
  205. idKey := strconv.Itoa(id)
  206. if !redis_Warning_Count.IsExist(idKey) {
  207. return count, tableCount, redis.ErrNil
  208. }
  209. //println("找到key:",key)
  210. v := redis_Warning_Count.Get(idKey)
  211. err = json.Unmarshal(v.([]byte), &countList)
  212. if err != nil {
  213. logs.Error(lib.FuncName(), err)
  214. return count, tableCount, err
  215. }
  216. for _, c := range countList {
  217. if c.Key == key {
  218. return c.Count, c.TableCount, nil
  219. }
  220. }
  221. return count, tableCount, redis.ErrNil
  222. }
  223. func Redis_WarningCount_DelK(id int) (err error) {
  224. idKey := strconv.Itoa(id)
  225. err = redis_Warning_Count.Delete(idKey)
  226. if err != nil {
  227. logs.Error(lib.FuncName(), err)
  228. }
  229. return
  230. }
  231. // ---------------- 特殊方法 -------------------
  232. // T_history 0 40天 1 历史数据
  233. func WarningToWarning_R(T_history int, t Warning) (r Warning_R) {
  234. r.Id = t.Id
  235. r.T_pid = t.T_pid
  236. r.T_tp = t.T_tp
  237. r.T_tp_name = Read_WarningType_Get(t.T_tp)
  238. r.T_sn = t.T_sn
  239. r.T_D_name = t.T_D_name
  240. r.T_id = t.T_id
  241. r.T_DS_name = t.T_DS_name
  242. r.T_Remark = t.T_Remark
  243. r.T_Ut = t.T_Ut.Format("2006-01-02 15:04:05")
  244. if !t.T_fUt.IsZero() {
  245. r.T_fUt = t.T_fUt.Format("2006-01-02 15:04:05")
  246. }
  247. //r.T_Text = t.T_Text
  248. r.T_Text, r.T_attachments = GetLatestRemark(t)
  249. r.T_handler = GetHandlers(t)
  250. if len(t.T_Log) > 0 {
  251. r.T_Log = strings.Split(strings.TrimRight(t.T_Log, "\n"), "\n")
  252. }
  253. // 解析处理记录详情(包含附件)
  254. handlingRecords, _ := GetHandlingRecords(t)
  255. if handlingRecords != nil {
  256. r.T_handling_record = handlingRecords
  257. } else {
  258. r.T_handling_record = []HandlingRecord{}
  259. }
  260. r.T_Msid = t.T_Msid
  261. r.T_State = t.T_State
  262. r.T_history = T_history
  263. r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
  264. r.UpdateTime = t.UpdateTime.Format("2006-01-02 15:04:05")
  265. return r
  266. }
  267. func WarningToAdminWarning_R(T_history int, t Warning) (r Warning_R) {
  268. r.Id = t.Id
  269. r.T_pid = t.T_pid
  270. r.T_pid_name = Account.Read_Company_Get(t.T_pid)
  271. r.T_tp = t.T_tp
  272. r.T_tp_name = Read_WarningType_Get(t.T_tp)
  273. r.T_sn = t.T_sn
  274. r.T_D_name = t.T_D_name
  275. r.T_id = t.T_id
  276. r.T_DS_name = t.T_DS_name
  277. r.T_Remark = t.T_Remark
  278. r.T_Ut = t.T_Ut.Format("2006-01-02 15:04:05")
  279. if !t.T_fUt.IsZero() {
  280. r.T_fUt = t.T_fUt.Format("2006-01-02 15:04:05")
  281. }
  282. r.T_Text, r.T_attachments = GetLatestRemark(t)
  283. r.T_handler = GetHandlers(t)
  284. if len(t.T_Log) > 0 {
  285. r.T_Log = strings.Split(strings.TrimRight(t.T_Log, "\n"), "\n")
  286. }
  287. r.T_Msid = t.T_Msid
  288. r.T_State = t.T_State
  289. r.T_history = T_history
  290. r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
  291. r.UpdateTime = t.UpdateTime.Format("2006-01-02 15:04:05")
  292. return r
  293. }
  294. func WarningToCompanyWarning_R(T_history int, t Warning) (r CompanyWarning_R) {
  295. r.Id = t.Id
  296. r.T_pid = t.T_pid
  297. r.T_pid_name = Account.Read_Company_Get(t.T_pid)
  298. r.T_tp = t.T_tp
  299. r.T_tp_name = Read_WarningType_Get(t.T_tp)
  300. r.T_sn = t.T_sn
  301. r.T_D_name = t.T_D_name
  302. r.T_id = t.T_id
  303. r.T_DS_name = t.T_DS_name
  304. r.T_Remark = t.T_Remark
  305. r.T_Ut = t.T_Ut.Format("2006-01-02 15:04:05")
  306. if !t.T_fUt.IsZero() {
  307. r.T_fUt = t.T_fUt.Format("2006-01-02 15:04:05")
  308. }
  309. r.T_Text, r.T_attachments = GetLatestRemark(t)
  310. r.T_handler = GetHandlers(t)
  311. if len(t.T_Log) > 0 {
  312. r.T_Log = strings.Split(strings.TrimRight(t.T_Log, "\n"), "\n")
  313. }
  314. r.T_Msid = t.T_Msid
  315. r.T_State = t.T_State
  316. r.T_history = T_history
  317. r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
  318. r.UpdateTime = t.UpdateTime.Format("2006-01-02 15:04:05")
  319. return r
  320. }
  321. func WarningToWarning_Applet(t Warning) (r Warning_Applet) {
  322. r.Id = t.Id
  323. r.T_pid = t.T_pid
  324. company, _ := Account.Read_Company_ById(t.T_pid)
  325. r.T_company_name = company.T_name
  326. r.T_tp = t.T_tp
  327. r.T_tp_name = Read_WarningType_Get(t.T_tp)
  328. r.T_sn = t.T_sn
  329. r.T_D_name = t.T_D_name
  330. r.T_id = t.T_id
  331. r.T_DS_name = t.T_DS_name
  332. r.T_Remark = t.T_Remark
  333. r.T_Ut = t.T_Ut.Format("2006-01-02 15:04:05")
  334. if !t.T_fUt.IsZero() {
  335. r.T_fUt = t.T_fUt.Format("2006-01-02 15:04:05")
  336. }
  337. r.T_Text, r.T_attachments = GetLatestRemark(t)
  338. r.T_handler = GetHandlers(t)
  339. if len(t.T_Log) > 0 {
  340. r.T_Log = strings.Split(strings.TrimRight(t.T_Log, "\n"), "\n")
  341. }
  342. r.T_Msid = t.T_Msid
  343. r.T_State = t.T_State
  344. r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
  345. return r
  346. }
  347. // 获取 ById
  348. func Read_Warning_ById(id int64) (r Warning) {
  349. o := orm.NewOrm()
  350. r = Warning{Id: id}
  351. err := o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  352. if err != nil {
  353. logs.Error(lib.FuncName(), err)
  354. }
  355. return r
  356. }
  357. // 获取 ById
  358. func Read_Warning_ById_Backups(id int, T_year string, T_month string) (r Warning, err error) {
  359. o := orm.NewOrm()
  360. var maps []Warning
  361. // Warning_2022_07
  362. Wtab := "warning_" + T_year + "_" + T_month
  363. sql_WHERE := fmt.Sprintf(" t__state > 0 AND `ID`= %d", id)
  364. sql := "SELECT ID,t_pid,t_tp,t_sn,t__d_name,t_id,t__d_s_name,t__remark,t__ut,t__text,t__log,t__msid,t__state,create_time,update_time " +
  365. "FROM " + Wtab + " WHERE" + sql_WHERE
  366. logs.Debug(sql)
  367. num, err := o.Raw(sql).QueryRows(&maps)
  368. if err != nil {
  369. logs.Error(lib.FuncName(), err)
  370. return
  371. }
  372. if num == 0 {
  373. return r, orm.ErrNoRows
  374. }
  375. return maps[0], nil
  376. }
  377. // 添加
  378. func Add_Warning(m Warning) (id int64, err error) {
  379. o := orm.NewOrm()
  380. id, err = o.Insert(&m)
  381. if err != nil {
  382. logs.Error(lib.FuncName(), err)
  383. }
  384. return id, err
  385. }
  386. // 修改
  387. func Update_Warning(r Warning, cols ...string) bool {
  388. o := orm.NewOrm()
  389. num, err := o.Update(&r, cols...)
  390. if err != nil {
  391. logs.Error(lib.FuncName(), err)
  392. return false
  393. }
  394. logs.Info("Number of records updated in database:", num)
  395. return true
  396. }
  397. // 修改
  398. func Update_Warning_Backups(r Warning, T_year string, T_month string) bool {
  399. o := orm.NewOrm()
  400. Wtab := "warning_" + T_year + "_" + T_month
  401. num, err := o.Raw("UPDATE "+Wtab+" SET `t__text` = ?,`t__state` = ?, `update_time` = ? WHERE `ID` = ?", r.T_Text, r.T_State, r.UpdateTime, r.Id).Exec()
  402. if err != nil {
  403. logs.Error(lib.FuncName(), err)
  404. return false
  405. }
  406. logs.Info("Number of records updated in database:", num)
  407. return true
  408. }
  409. // 获取列表
  410. func Read_Warning_List(T_pid int, bindSN, tpList []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int,
  411. countRedisKey string) (r []Warning_R, cnt int64) {
  412. o := orm.NewOrm()
  413. // 也可以直接使用 Model 结构体作为表名
  414. var map_r []Warning
  415. qs := o.QueryTable(new(Warning))
  416. var offset int64
  417. if page <= 1 {
  418. offset = 0
  419. } else {
  420. offset = int64((page - 1) * page_z)
  421. }
  422. cond := orm.NewCondition()
  423. cond1 := orm.NewCondition()
  424. if len(bindSN) > 0 {
  425. cond1 = cond1.And("T_sn__in", bindSN)
  426. }
  427. if T_pid > 0 {
  428. cond1 = cond1.And("T_pid", T_pid)
  429. }
  430. if len(tpList) > 0 {
  431. cond1 = cond1.And("T_tp__in", tpList)
  432. }
  433. if len(T_name) > 0 {
  434. if len(T_name) == 16 {
  435. cond1 = cond1.And("T_sn", T_name)
  436. } else {
  437. cond1 = cond1.AndCond(cond.Or("T_sn__icontains", T_name).
  438. Or("T_D_name__icontains", T_name).
  439. Or("T_id__icontains", T_name).
  440. Or("T_DS_name__icontains", T_name))
  441. }
  442. }
  443. if len(Time_start_) > 0 {
  444. cond1 = cond1.And("T_Ut__gte", Time_start_)
  445. }
  446. if len(Time_end_) > 0 {
  447. cond1 = cond1.And("T_Ut__lte", Time_end_)
  448. }
  449. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  450. // T_handle 1:默认 2:全部记录 3 未处理
  451. if T_handle == 1 {
  452. cond1 = cond1.And("T_State__gt", 1)
  453. } else if T_handle == 3 {
  454. cond1 = cond1.And("T_State", 3)
  455. } else {
  456. cond1 = cond1.And("T_State__gt", 0)
  457. }
  458. var err error
  459. // 从redis获取报警统计数量
  460. cnt, _, err = Redis_WarningCount_GetOne(T_pid, countRedisKey)
  461. if err != nil {
  462. cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
  463. if err != nil {
  464. logs.Error(lib.FuncName(), err)
  465. return
  466. }
  467. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, map[string]int64{})
  468. }
  469. if cnt == 0 {
  470. return r, 0
  471. }
  472. if page_z == 9999 {
  473. // 获取全部
  474. _, err = qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&map_r)
  475. } else {
  476. _, err = qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&map_r)
  477. }
  478. if err != nil {
  479. logs.Error(lib.FuncName(), err)
  480. return
  481. }
  482. for _, v := range map_r {
  483. r = append(r, WarningToWarning_R(0, v))
  484. }
  485. return r, cnt
  486. }
  487. // 管理员报警列表
  488. func Read_Admin_Warning_List(admin Account.Admin, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int,
  489. countRedisKey string) (r []Warning_R, cnt int64) {
  490. if len(admin.T_pids) == 0 {
  491. return r, cnt
  492. }
  493. o := orm.NewOrm()
  494. // 也可以直接使用 Model 结构体作为表名
  495. var map_r []Warning
  496. qs := o.QueryTable(new(Warning))
  497. var offset int64
  498. if page_z == 0 {
  499. page_z = conf.Page_size
  500. }
  501. if page <= 1 {
  502. offset = 0
  503. } else {
  504. offset = int64((page - 1) * page_z)
  505. }
  506. cond := orm.NewCondition()
  507. cond1 := orm.NewCondition()
  508. if admin.T_pids != "*" && len(admin.T_pids) > 0 {
  509. list := lib.SplitStringIds(admin.T_pids, "P")
  510. cond1 = cond1.And("T_pid__in", list)
  511. }
  512. if len(T_tp) > 0 {
  513. cond1 = cond1.And("T_tp__in", T_tp)
  514. }
  515. if len(T_name) > 0 {
  516. if len(T_name) == 16 {
  517. cond1 = cond1.And("T_sn", T_name)
  518. } else {
  519. cond1 = cond1.AndCond(cond.Or("T_sn__icontains", T_name).
  520. Or("T_D_name__icontains", T_name).
  521. Or("T_id__icontains", T_name).
  522. Or("T_DS_name__icontains", T_name))
  523. }
  524. }
  525. if len(Time_start_) > 0 {
  526. cond1 = cond1.And("T_Ut__gte", Time_start_)
  527. }
  528. if len(Time_end_) > 0 {
  529. cond1 = cond1.And("T_Ut__lte", Time_end_)
  530. }
  531. //1:默认 2:全部记录 3 未处理
  532. if T_handle == 1 {
  533. cond1 = cond1.And("T_State__gt", 1)
  534. } else if T_handle == 3 {
  535. cond1 = cond1.And("T_State", 3)
  536. } else {
  537. cond1 = cond1.And("T_State__gt", 0)
  538. }
  539. var err error
  540. // 从redis获取报警统计数量
  541. cnt, _, err = Redis_WarningCount_GetOne(admin.Id, countRedisKey)
  542. if err != nil {
  543. cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
  544. if err != nil {
  545. logs.Error(lib.FuncName(), err)
  546. return
  547. }
  548. Redis_WarningCount_Set(admin.Id, countRedisKey, cnt, map[string]int64{})
  549. }
  550. if cnt == 0 {
  551. return r, 0
  552. }
  553. _, err = qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&map_r)
  554. if err != nil {
  555. logs.Error(lib.FuncName(), err)
  556. }
  557. for _, v := range map_r {
  558. r = append(r, WarningToWarning_R(0, v))
  559. }
  560. return r, cnt
  561. }
  562. // 获取列表备份
  563. func Read_Warning_Backups(T_pid int, bindSN []string, T_year string, T_month string, tpList []string, T_name string, T_handle int, Time_start_ string,
  564. Time_end_ string, page int, page_z int, countRedisKey string) (r []Warning_R, cnt int64) {
  565. o := orm.NewOrm()
  566. var maps []Warning
  567. var maps_z []orm2.ParamsList
  568. // 也可以直接使用 Model 结构体作为表名
  569. // Warning_2022_07
  570. Wtab := "warning_" + T_year + "_" + T_month
  571. var offset int
  572. if page <= 1 {
  573. offset = 0
  574. } else {
  575. offset = (page - 1) * page_z
  576. }
  577. sql_WHERE := ""
  578. if len(bindSN) > 0 {
  579. sql_WHERE += fmt.Sprintf(" AND t_sn in (%s)", lib.StringListToQuotesDotStr(bindSN))
  580. }
  581. if T_pid > 0 {
  582. sql_WHERE += " AND t_pid = " + strconv.Itoa(T_pid)
  583. }
  584. if len(tpList) > 0 {
  585. tp := lib.StringListToDotStr(tpList)
  586. sql_WHERE += fmt.Sprintf(" AND t_tp in (%s)", tp)
  587. }
  588. if len(T_name) > 0 {
  589. sql_WHERE += " AND (t_sn like '%" + T_name + "%' OR t__d_name like '%" + T_name + "%' OR t_id like '%" + T_name + "%' OR t__d_s_name like '%" + T_name + "%')"
  590. }
  591. if len(Time_start_) > 0 {
  592. //cond1 = cond1.And("T_Ut__gte", Time_start_)
  593. Time_start_ = lib.ReplaceSQL(Time_start_)
  594. sql_WHERE += fmt.Sprintf(" AND t__ut >= '%s'", Time_start_)
  595. }
  596. if len(Time_end_) > 0 {
  597. //cond1 = cond1.And("T_Ut__lte", Time_end_)
  598. Time_end_ = lib.ReplaceSQL(Time_end_)
  599. sql_WHERE += fmt.Sprintf(" AND t__ut <= '%s'", Time_end_)
  600. }
  601. //1:默认 2:全部记录 3 未处理
  602. if T_handle == 1 {
  603. sql_WHERE += " AND t__state > 1"
  604. } else if T_handle == 3 {
  605. sql_WHERE += " AND t__state = 3"
  606. } else {
  607. sql_WHERE += " AND t__state > 0"
  608. }
  609. // -------------
  610. sql_WHERE = strings.TrimLeft(sql_WHERE, " AND")
  611. // 从redis获取报警统计数量
  612. cnt, _, err := Redis_WarningCount_GetOne(T_pid, countRedisKey)
  613. if err != nil {
  614. sql := "SELECT COUNT(*) FROM " + Wtab + " WHERE " + sql_WHERE
  615. logs.Debug(sql)
  616. _, err = o.Raw(sql).ValuesList(&maps_z)
  617. if err != nil {
  618. logs.Error(lib.FuncName(), err)
  619. return r, 0
  620. }
  621. if len(maps_z) == 0 {
  622. return r, 0
  623. }
  624. cnt, _ = strconv.ParseInt(maps_z[0][0].(string), 10, 64)
  625. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, map[string]int64{})
  626. }
  627. if cnt == 0 {
  628. return r, 0
  629. }
  630. sql := "SELECT ID,t_pid,t_tp,t_sn,t__d_name,t_id,t__d_s_name,t__remark,t__ut,t__text,t__log,t__msid,t__state,create_time,update_time " +
  631. "FROM " + Wtab + " WHERE " + sql_WHERE + " ORDER BY t__ut DESC"
  632. if page_z != 9999 {
  633. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(page_z)
  634. }
  635. logs.Debug(sql)
  636. _, err = o.Raw(sql).QueryRows(&maps)
  637. if err != nil {
  638. logs.Error(lib.FuncName(), err)
  639. return
  640. }
  641. for _, v := range maps {
  642. r = append(r, WarningToWarning_R(1, v))
  643. }
  644. return r, cnt
  645. }
  646. // 获取管理员列表备份
  647. func Read_Admin_Warning_Backups(admin Account.Admin, T_year string, T_month string, T_tp []string, T_name string, T_handle int,
  648. Time_start_ string, Time_end_ string, page int, page_z int, countRedisKey string) (r []Warning_R, cnt int64) {
  649. if len(admin.T_pids) == 0 {
  650. return r, cnt
  651. }
  652. o := orm.NewOrm()
  653. var maps []Warning
  654. var maps_z []orm2.ParamsList
  655. // 也可以直接使用 Model 结构体作为表名
  656. // Warning_2022_07
  657. Wtab := "warning_" + T_year + "_" + T_month
  658. var offset int
  659. if page_z == 0 {
  660. page_z = conf.Page_size
  661. }
  662. if page <= 1 {
  663. offset = 0
  664. } else {
  665. offset = (page - 1) * page_z
  666. }
  667. sql_WHERE := ""
  668. if admin.T_pids != "*" {
  669. list := lib.SplitStringToDotStr(admin.T_pids, "P")
  670. sql_WHERE += fmt.Sprintf(" AND t_pid in (%s)", list)
  671. }
  672. if len(T_tp) > 0 {
  673. tp := lib.StringListToDotStr(T_tp)
  674. sql_WHERE += fmt.Sprintf(" AND t_tp in (%s)", tp)
  675. }
  676. if len(T_name) > 0 {
  677. sql_WHERE += " AND (t_sn like '%" + T_name + "%' OR t__d_name like '%" + T_name + "%' OR t_id like '%" + T_name + "%' OR t__d_s_name like '%" + T_name + "%')"
  678. }
  679. if len(Time_start_) > 0 {
  680. //cond1 = cond1.And("T_Ut__gte", Time_start_)
  681. Time_start_ = lib.ReplaceSQL(Time_start_)
  682. sql_WHERE += fmt.Sprintf(" AND t__ut >= '%s'", Time_start_)
  683. }
  684. if len(Time_end_) > 0 {
  685. //cond1 = cond1.And("T_Ut__lte", Time_end_)
  686. Time_end_ = lib.ReplaceSQL(Time_end_)
  687. sql_WHERE += fmt.Sprintf(" AND t__ut <= '%s'", Time_end_)
  688. }
  689. //1:默认 2:全部记录 3 未处理
  690. if T_handle == 1 {
  691. sql_WHERE += " AND t__state > 1"
  692. } else if T_handle == 3 {
  693. sql_WHERE += " AND t__state = 3"
  694. } else {
  695. sql_WHERE += " AND t__state > 0"
  696. }
  697. sql_WHERE = strings.TrimLeft(sql_WHERE, " AND")
  698. // 从redis获取报警统计数量
  699. cnt, _, err := Redis_WarningCount_GetOne(admin.Id, countRedisKey)
  700. if err != nil {
  701. sql := "SELECT COUNT(*) FROM " + Wtab + " WHERE " + sql_WHERE
  702. logs.Debug(sql)
  703. _, err = o.Raw(sql).ValuesList(&maps_z)
  704. if err != nil {
  705. logs.Error(lib.FuncName(), err)
  706. return r, 0
  707. }
  708. if len(maps_z) == 0 {
  709. return r, 0
  710. }
  711. cnt, _ = strconv.ParseInt(maps_z[0][0].(string), 10, 64)
  712. Redis_WarningCount_Set(admin.Id, countRedisKey, cnt, map[string]int64{})
  713. }
  714. if cnt == 0 {
  715. return r, 0
  716. }
  717. sql := "SELECT ID,t_pid,t_tp,t_sn,t__d_name,t_id,t__d_s_name,t__remark,t__ut,t__text,t__log,t__msid,t__state,create_time,update_time " +
  718. "FROM " + Wtab + " WHERE " + sql_WHERE + " ORDER BY t__ut DESC"
  719. if page_z != 9999 {
  720. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(page_z)
  721. }
  722. logs.Debug(sql)
  723. _, err = o.Raw(sql).QueryRows(&maps)
  724. if err != nil {
  725. logs.Error(lib.FuncName(), err)
  726. return
  727. }
  728. key, _ := strconv.ParseInt(maps_z[0][0].(string), 10, 64)
  729. for _, v := range maps {
  730. r = append(r, WarningToWarning_R(1, v))
  731. }
  732. return r, key
  733. }
  734. // 用户首页统计
  735. func Read_Warning_ALL_T_State_Count2(T_pid int, T_pids []int, T_handle int, bindSN []string, T_Warning string, Is_Today bool) (cnt int64) {
  736. o := orm.NewOrm()
  737. qs := o.QueryTable(new(Warning))
  738. cond := orm.NewCondition()
  739. cond1 := cond.And("T_State__gt", 0)
  740. if T_pid > 0 {
  741. cond1 = cond1.And("T_pid", T_pid)
  742. }
  743. if T_pid == 0 {
  744. cond1 = cond1.And("T_pid__in", T_pids)
  745. }
  746. if len(bindSN) > 0 {
  747. cond1 = cond1.And("T_sn__in", bindSN)
  748. }
  749. if Is_Today {
  750. startTime := time.Now().Format("2006-01-02") + " 00:00:00"
  751. endTime := time.Now().Format("2006-01-02") + " 23:59:59"
  752. cond1 = cond1.And("T_Ut__gte", startTime).And("T_Ut__lte", endTime)
  753. }
  754. if len(T_Warning) > 0 && T_Warning != "*" {
  755. list := lib.SplitStringIds(T_Warning, "W")
  756. cond1 = cond1.And("T_tp__in", list)
  757. }
  758. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  759. // T_handle 1:默认 2:全部记录 3 未处理
  760. if T_handle == 1 {
  761. cond1 = cond1.And("T_State__gt", 1)
  762. }
  763. if T_handle == 3 {
  764. cond1 = cond1.And("T_State", 3)
  765. }
  766. cnt, err := qs.SetCond((*orm2.Condition)(cond1)).Count()
  767. if err != nil {
  768. logs.Error(lib.FuncName(), err)
  769. }
  770. return cnt
  771. }
  772. // 用户首页统计(查询最近40天),使用gorm查询
  773. func Read_Warning_ALL_T_State_Count(T_pid int, T_pids []int, T_handle int, bindSN []string, T_Warning string, Is_Today bool) (cnt int64) {
  774. tableName := "warning"
  775. gormDB := db.DB.Table(tableName)
  776. daysAgo40 := time.Now().AddDate(0, 0, -40).Format("2006-01-02")
  777. gormDB.Where("t__state > 0 ").Where("t__ut > ?", daysAgo40+" 00:00:00")
  778. if T_pid > 0 {
  779. gormDB.Where("t_pid = ?", T_pid)
  780. }
  781. if T_pid == 0 {
  782. gormDB.Where("t_pid in (?)", T_pids)
  783. }
  784. if len(bindSN) > 0 {
  785. gormDB.Where("t_sn in (?)", bindSN)
  786. }
  787. if Is_Today {
  788. startTime := time.Now().Format("2006-01-02") + " 00:00:00"
  789. endTime := time.Now().Format("2006-01-02") + " 23:59:59"
  790. gormDB.Where("t__ut between ? and ?", startTime, endTime)
  791. }
  792. if len(T_Warning) > 0 && T_Warning != "*" {
  793. list := lib.SplitStringIds(T_Warning, "W")
  794. gormDB.Where("t_tp in (?)", list)
  795. }
  796. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  797. // T_handle 1:默认 2:全部记录 3 未处理
  798. if T_handle == 1 {
  799. gormDB.Where("t__state > ?", 1)
  800. }
  801. if T_handle == 3 {
  802. gormDB.Where("t__state = ?", 3)
  803. }
  804. err := gormDB.Count(&cnt).Error
  805. if err != nil {
  806. logs.Error(lib.FuncName(), err)
  807. }
  808. return cnt
  809. }
  810. // 设备日志
  811. type DeviceLogs struct {
  812. Id int64
  813. T_sn string //
  814. Logs_Txt string // 详情
  815. CreateTime string
  816. }
  817. func WarningToDeviceLogs(t Warning) (r DeviceLogs) {
  818. r.Id = t.Id
  819. r.T_sn = t.T_sn
  820. r.Logs_Txt = t.T_Remark
  821. r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
  822. return r
  823. }
  824. // 获取列表
  825. func Read_DeviceLogs_List(T_sn string, page, page_z int) (r []DeviceLogs, cnt int64) {
  826. o := orm.NewOrm()
  827. // 也可以直接使用 Model 结构体作为表名
  828. qs := o.QueryTable(new(Warning))
  829. var maps []Warning
  830. var offset int64
  831. if page <= 1 {
  832. offset = 0
  833. } else {
  834. offset = int64((page - 1) * page_z)
  835. }
  836. _, err := qs.Limit(page_z, offset).Filter("T_sn", T_sn).Filter("T_tp__in", DeviceLogType).OrderBy("-Id").All(&maps)
  837. if err != nil {
  838. logs.Error(lib.FuncName(), err)
  839. return
  840. }
  841. cnt, err = qs.Filter("T_sn", T_sn).Filter("T_tp__in", DeviceLogType).Count()
  842. if err != nil {
  843. logs.Error(lib.FuncName(), err)
  844. return
  845. }
  846. for _, v := range maps {
  847. r = append(r, WarningToDeviceLogs(v))
  848. }
  849. return r, cnt
  850. }
  851. // 通过传感器类型获取报警列表
  852. func Read_Warning_List_By_DS_T_type(admin *Account.Admin, T_pid int, bindSN []string, T_pids []int, DS_T_type int, T_name string, T_tp []string,
  853. page int, page_z int, countRedisKey string) (r []Warning_Applet, cnt int64) {
  854. o := orm.NewOrm()
  855. // 也可以直接使用 Model 结构体作为表名
  856. var maps []Warning
  857. var maps_z []orm2.ParamsList
  858. var offset int
  859. if page <= 1 {
  860. offset = 0
  861. } else {
  862. offset = (page - 1) * page_z
  863. }
  864. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  865. sql_WHERE := "1 < t__state"
  866. if admin.T_pid > 0 {
  867. if len(T_pids) > 0 {
  868. sql_WHERE += fmt.Sprintf(" AND t_pid in (%s) ", lib.IntListToDotStr(T_pids))
  869. }
  870. } else {
  871. sql_WHERE += " AND t_pid = " + strconv.Itoa(T_pid)
  872. }
  873. if len(T_tp) > 0 {
  874. tp := lib.StringListToDotStr(T_tp)
  875. sql_WHERE += fmt.Sprintf(" AND w.t_tp in (%s)", tp)
  876. }
  877. if len(bindSN) > 0 {
  878. sql_WHERE += fmt.Sprintf(" AND t_sn in (%s)", lib.StringListToQuotesDotStr(bindSN))
  879. }
  880. if DS_T_type > 0 {
  881. sql_WHERE += " AND ds.t_type = " + strconv.Itoa(DS_T_type)
  882. }
  883. if len(T_name) > 0 {
  884. sql_WHERE += " AND (t_sn like '%" + T_name + "%' OR t__d_name like '%" + T_name + "%' OR t_id like '%" + T_name + "%' OR t__d_s_name like '%" + T_name + "%')"
  885. }
  886. sql := "SELECT COUNT(w.ID) FROM warning w " +
  887. "LEFT JOIN ( SELECT t_sn AS tsn,t_id AS tid,t_type FROM device_sensor) AS ds " +
  888. "ON w.t_sn = ds.tsn AND w.t_id = ds.tid " +
  889. "WHERE " + sql_WHERE
  890. cnt, _, err := Redis_WarningCount_GetOne(T_pid, countRedisKey)
  891. if err != nil {
  892. logs.Debug(sql)
  893. _, err = o.Raw(sql).ValuesList(&maps_z)
  894. if err != nil {
  895. logs.Error(lib.FuncName(), err)
  896. return r, 0
  897. }
  898. if len(maps_z) == 0 {
  899. return r, 0
  900. }
  901. cnt, _ = strconv.ParseInt(maps_z[0][0].(string), 10, 64)
  902. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, map[string]int64{})
  903. }
  904. if cnt == 0 {
  905. return r, 0
  906. }
  907. sql = "SELECT w.*, ds.tsn, ds.tid, ds.t_type FROM warning w " +
  908. "LEFT JOIN ( SELECT t_sn AS tsn,t_id AS tid,t_type FROM device_sensor) AS ds " +
  909. "ON w.t_sn = ds.tsn AND w.t_id = ds.tid " +
  910. "WHERE " + sql_WHERE + " ORDER BY t__ut DESC"
  911. if page_z != 9999 {
  912. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(page_z)
  913. }
  914. logs.Debug(sql)
  915. _, err = o.Raw(sql).QueryRows(&maps)
  916. if err != nil {
  917. logs.Error(lib.FuncName(), err)
  918. return
  919. }
  920. for _, v := range maps {
  921. r = append(r, WarningToWarning_Applet(v))
  922. }
  923. return r, cnt
  924. }
  925. // 获取当天设备报警总数
  926. func Read_WarningCount_byDay(T_pids []int) (cnt int64) {
  927. o := orm.NewOrm()
  928. qs := o.QueryTable(new(Warning))
  929. cond := orm.NewCondition()
  930. yesterday := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
  931. cond = cond.And("T_State__gt", 0).And("T_tp__in", RateType).And("CreateTime__gte", yesterday+" 00:00:00").And("CreateTime__lte", yesterday+" 23:59:59")
  932. if len(T_pids) == 1 {
  933. cond = cond.And("T_pid", T_pids[0])
  934. } else if len(T_pids) > 1 {
  935. cond = cond.And("T_pid__in", T_pids)
  936. }
  937. cnt, err := qs.SetCond((*orm2.Condition)(cond)).Count()
  938. if err != nil {
  939. logs.Error(lib.FuncName(), err)
  940. }
  941. return cnt
  942. }
  943. // 公司管理报警列表
  944. func Read_Company_Warning_List(T_pids []int, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int,
  945. T_pid int, countRedisKey string) (r []CompanyWarning_R, cnt int64) {
  946. o := orm.NewOrm()
  947. // 也可以直接使用 Model 结构体作为表名
  948. var map_r []Warning
  949. qs := o.QueryTable(new(Warning))
  950. var offset int64
  951. if page_z == 0 {
  952. page_z = conf.Page_size
  953. }
  954. if page <= 1 {
  955. offset = 0
  956. } else {
  957. offset = int64((page - 1) * page_z)
  958. }
  959. cond := orm.NewCondition()
  960. cond1 := orm.NewCondition()
  961. if len(T_pids) == 1 {
  962. cond1 = cond1.And("T_pid", T_pids[0])
  963. } else if len(T_pids) > 1 {
  964. cond1 = cond1.And("T_pid__in", T_pids)
  965. }
  966. if len(T_tp) > 0 {
  967. cond1 = cond1.And("T_tp__in", T_tp)
  968. }
  969. if len(T_name) > 0 {
  970. if len(T_name) == 16 {
  971. cond1 = cond1.And("T_sn", T_name)
  972. } else {
  973. cond1 = cond1.AndCond(cond.Or("T_sn__icontains", T_name).
  974. Or("T_D_name__icontains", T_name).
  975. Or("T_id__icontains", T_name).
  976. Or("T_DS_name__icontains", T_name))
  977. }
  978. }
  979. if len(Time_start_) > 0 {
  980. cond1 = cond1.And("T_Ut__gte", Time_start_)
  981. }
  982. if len(Time_end_) > 0 {
  983. cond1 = cond1.And("T_Ut__lte", Time_end_)
  984. }
  985. //1:默认 2:全部记录 3 未处理
  986. if T_handle == 1 {
  987. cond1 = cond1.And("T_State__gt", 1)
  988. } else if T_handle == 3 {
  989. cond1 = cond1.And("T_State", 3)
  990. } else {
  991. cond1 = cond1.And("T_State__gt", 0)
  992. }
  993. var err error
  994. // 从redis获取报警统计数量
  995. cnt, _, err = Redis_WarningCount_GetOne(T_pid, countRedisKey)
  996. if err != nil {
  997. cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
  998. if err != nil {
  999. logs.Error(lib.FuncName(), err)
  1000. return
  1001. }
  1002. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, map[string]int64{})
  1003. }
  1004. if cnt == 0 {
  1005. return r, 0
  1006. }
  1007. _, err = qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&map_r)
  1008. if err != nil {
  1009. logs.Error(lib.FuncName(), err)
  1010. }
  1011. for _, v := range map_r {
  1012. r = append(r, WarningToCompanyWarning_R(0, v))
  1013. }
  1014. return r, cnt
  1015. }
  1016. // 获取公司管理列表备份
  1017. func Read_Company_Warning_Backups(T_pids []int, T_year string, T_month string, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string,
  1018. page int, page_z int, T_pid int, countRedisKey string) (r []CompanyWarning_R, cnt int64) {
  1019. if len(T_pids) == 0 {
  1020. return r, cnt
  1021. }
  1022. o := orm.NewOrm()
  1023. var maps []Warning
  1024. var maps_z []orm2.ParamsList
  1025. // 也可以直接使用 Model 结构体作为表名
  1026. // Warning_2022_07
  1027. Wtab := "warning_" + T_year + "_" + T_month
  1028. var offset int
  1029. if page_z == 0 {
  1030. page_z = conf.Page_size
  1031. }
  1032. if page <= 1 {
  1033. offset = 0
  1034. } else {
  1035. offset = (page - 1) * page_z
  1036. }
  1037. sql_WHERE := ""
  1038. if len(T_pids) == 1 {
  1039. sql_WHERE += fmt.Sprintf(" AND t_pid = %d", T_pids[0])
  1040. } else if len(T_pids) > 1 {
  1041. sql_WHERE += fmt.Sprintf(" AND t_pid in (%s)", lib.IntListToDotStr(T_pids))
  1042. }
  1043. if len(T_tp) > 0 {
  1044. tp := lib.StringListToDotStr(T_tp)
  1045. sql_WHERE += fmt.Sprintf(" AND t_tp in (%s)", tp)
  1046. }
  1047. if len(T_name) > 0 {
  1048. sql_WHERE += " AND (t_sn like '%" + T_name + "%' OR t__d_name like '%" + T_name + "%' OR t_id like '%" + T_name + "%' OR t__d_s_name like '%" + T_name + "%')"
  1049. }
  1050. if len(Time_start_) > 0 {
  1051. //cond1 = cond1.And("T_Ut__gte", Time_start_)
  1052. Time_start_ = lib.ReplaceSQL(Time_start_)
  1053. sql_WHERE += fmt.Sprintf(" AND t__ut >= '%s'", Time_start_)
  1054. }
  1055. if len(Time_end_) > 0 {
  1056. //cond1 = cond1.And("T_Ut__lte", Time_end_)
  1057. Time_end_ = lib.ReplaceSQL(Time_end_)
  1058. sql_WHERE += fmt.Sprintf(" AND t__ut <= '%s'", Time_end_)
  1059. }
  1060. //1:默认 2:全部记录 3 未处理
  1061. if T_handle == 1 {
  1062. sql_WHERE += " AND t__state > 1"
  1063. } else if T_handle == 3 {
  1064. sql_WHERE += " AND t__state = 3"
  1065. } else {
  1066. sql_WHERE += " AND t__state > 0"
  1067. }
  1068. sql_WHERE = strings.TrimLeft(sql_WHERE, " AND")
  1069. // 从redis获取报警统计数量
  1070. cnt, _, err := Redis_WarningCount_GetOne(T_pid, countRedisKey)
  1071. if err != nil {
  1072. sql := "SELECT COUNT(*) FROM " + Wtab + " WHERE " + sql_WHERE
  1073. logs.Debug(sql)
  1074. _, err = o.Raw(sql).ValuesList(&maps_z)
  1075. if err != nil {
  1076. logs.Error(lib.FuncName(), err)
  1077. return r, 0
  1078. }
  1079. if len(maps_z) == 0 {
  1080. return r, 0
  1081. }
  1082. cnt, _ = strconv.ParseInt(maps_z[0][0].(string), 10, 64)
  1083. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, map[string]int64{})
  1084. }
  1085. if cnt == 0 {
  1086. return r, 0
  1087. }
  1088. sql := "SELECT ID,t_pid,t_tp,t_sn,t__d_name,t_id,t__d_s_name,t__remark,t__ut,t__text,t__log,t__msid,t__state,create_time,update_time " +
  1089. "FROM " + Wtab + " WHERE " + sql_WHERE + " ORDER BY t__ut DESC"
  1090. if page_z != 9999 {
  1091. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(page_z)
  1092. }
  1093. logs.Debug(sql)
  1094. _, err = o.Raw(sql).QueryRows(&maps)
  1095. if err != nil {
  1096. logs.Error(lib.FuncName(), err)
  1097. return
  1098. }
  1099. for _, v := range maps {
  1100. r = append(r, WarningToCompanyWarning_R(1, v))
  1101. }
  1102. return r, cnt
  1103. }
  1104. func GormReadWarningListScopes(T_pid int, bindSN, tpList []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string) func(db *gorm.DB) *gorm.DB {
  1105. return func(db *gorm.DB) *gorm.DB {
  1106. if T_pid > 0 {
  1107. db.Where("t_pid = ?", T_pid)
  1108. }
  1109. if T_abandon == 1 {
  1110. // 弃用
  1111. if len(abandonSN) > 0 {
  1112. db.Where("t_sn in (?)", abandonSN)
  1113. } else {
  1114. db.Where("1=0")
  1115. return db
  1116. }
  1117. } else if conf.UseNewWarningQuery {
  1118. if len(bindSN) > 0 {
  1119. // 将绑定sn内非弃用sn筛选出来
  1120. bindSNWithOutAbandon := lib.FindUniqueElements(bindSN, abandonSN)
  1121. if len(bindSNWithOutAbandon) > 0 {
  1122. db.Where("t_sn in (?)", bindSNWithOutAbandon)
  1123. } else {
  1124. db.Where("1=0")
  1125. return db
  1126. }
  1127. } else {
  1128. if len(abandonSN) > 0 {
  1129. db.Where("t_sn not in (?)", abandonSN)
  1130. }
  1131. }
  1132. }
  1133. if len(tpList) > 0 {
  1134. min, max, is := checkStringListContinuous(tpList)
  1135. if is {
  1136. db.Where("t_tp between ? and ?", min, max)
  1137. } else {
  1138. db.Where("t_tp in (?)", tpList)
  1139. }
  1140. }
  1141. if len(T_name) > 0 {
  1142. if len(T_name) == 16 {
  1143. db.Where("t_sn = ?", T_name)
  1144. } else {
  1145. likeName := "%" + T_name + "%"
  1146. db.Where("t_sn like ? or t__d_name like ? or t_id like ? or t__d_s_name like ? ", likeName, likeName, likeName, likeName)
  1147. }
  1148. }
  1149. if len(Time_start_) > 0 {
  1150. db.Where("t__ut >= ?", Time_start_)
  1151. }
  1152. if len(Time_end_) > 0 {
  1153. db.Where("t__ut <= ?", Time_end_)
  1154. }
  1155. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  1156. // T_handle 1:默认 2:全部记录 3 未处理
  1157. if T_handle == 1 {
  1158. db.Where("t__state > ?", 1)
  1159. } else if T_handle == 3 {
  1160. db.Where("t__state = ?", 3)
  1161. } else {
  1162. db.Where("t__state > ?", 0)
  1163. }
  1164. return db
  1165. }
  1166. }
  1167. // 获取列表
  1168. func Gorm_Read_Warning_List(T_pid int, bindSN, tpList []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string, page int, page_z int,
  1169. countRedisKey string) (r []Warning_R, cnt int64) {
  1170. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1171. var err error
  1172. var warnings []Warning
  1173. tableCount := make(map[string]int64)
  1174. gormReadWarningListScopes := GormReadWarningListScopes(T_pid, bindSN, tpList, T_name, T_handle, Time_start_, Time_end_, T_abandon, abandonSN)
  1175. // 从redis获取报警统计数量
  1176. cnt, tableCount, err = Redis_WarningCount_GetOne(T_pid, countRedisKey)
  1177. if err != nil {
  1178. for _, tableName := range tableNames {
  1179. tableDB := db.DB.Table(tableName)
  1180. var count int64
  1181. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  1182. tableCount[tableName] = count
  1183. cnt += count
  1184. }
  1185. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, tableCount)
  1186. }
  1187. if cnt == 0 {
  1188. return r, 0
  1189. }
  1190. if page_z == 9999 {
  1191. for _, tableName := range tableNames {
  1192. var tableWarning []Warning
  1193. tableDB := db.DB.Table(tableName)
  1194. tableDB.Scopes(gormReadWarningListScopes).
  1195. Order("t__ut desc").Offset(-1).Limit(-1).Find(&tableWarning)
  1196. warnings = append(warnings, tableWarning...)
  1197. }
  1198. for _, v := range warnings {
  1199. r = append(r, WarningToWarning_R(0, v))
  1200. }
  1201. return r, cnt
  1202. }
  1203. offset := (page - 1) * page_z
  1204. remaining := offset
  1205. for _, tableName := range tableNames {
  1206. var tableWarning []Warning
  1207. tableDB := db.DB.Table(tableName)
  1208. var count int64
  1209. if tableName == fmt.Sprintf("warning_%d_%02d", time.Now().Year(), time.Now().Month()) {
  1210. if saveTime, ok := tableCount["SaveTime"]; ok {
  1211. t1 := time.Unix(saveTime, 0)
  1212. if time.Now().Sub(t1) < 5*time.Minute {
  1213. if count, ok = tableCount[tableName]; !ok {
  1214. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  1215. tableCount[tableName] = count
  1216. }
  1217. } else {
  1218. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  1219. tableCount[tableName] = count
  1220. }
  1221. }
  1222. } else {
  1223. var ok bool
  1224. if count, ok = tableCount[tableName]; !ok {
  1225. tableDB.Scopes(gormReadWarningListScopes).Count(&count)
  1226. tableCount[tableName] = count
  1227. }
  1228. }
  1229. if remaining >= int(count) {
  1230. remaining -= int(count)
  1231. continue
  1232. }
  1233. tableDB2 := db.DB.Table(tableName)
  1234. tableDB2.Scopes(gormReadWarningListScopes).Order("t__ut desc").Offset(remaining).Limit(page_z - len(warnings)).Find(&tableWarning)
  1235. warnings = append(warnings, tableWarning...)
  1236. if len(warnings) >= page_z {
  1237. break
  1238. }
  1239. remaining = 0
  1240. }
  1241. for _, v := range warnings {
  1242. r = append(r, WarningToWarning_R(0, v))
  1243. }
  1244. return r, cnt
  1245. }
  1246. func Gorm_Read_Warning_List_Excel(T_pid int, bindSN, tpList []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string, page int, page_z int,
  1247. countRedisKey string) (r []Warning_R, cnt int64) {
  1248. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1249. gormReadWarningListScopes := GormReadWarningListScopes(T_pid, bindSN, tpList, T_name, T_handle, Time_start_, Time_end_, T_abandon, abandonSN)
  1250. var warnings []Warning
  1251. var wg sync.WaitGroup
  1252. for _, tableName := range tableNames {
  1253. wg.Add(1)
  1254. go func(table string) {
  1255. defer wg.Done()
  1256. offset := 0
  1257. limit := 10000
  1258. for {
  1259. var tableWarning []Warning
  1260. tableDB := db.DB.Table(table)
  1261. tableDB.Scopes(gormReadWarningListScopes).
  1262. Order("t__ut desc").Offset(offset).Limit(limit).Find(&tableWarning)
  1263. if len(tableWarning) == 0 {
  1264. break // 如果没有数据则结束循环
  1265. }
  1266. offset += limit // 更新偏移量
  1267. warnings = append(warnings, tableWarning...)
  1268. }
  1269. }(tableName)
  1270. }
  1271. // 等待所有协程完成
  1272. wg.Wait()
  1273. // 倒序
  1274. sort.Slice(warnings, func(i, j int) bool {
  1275. if warnings[i].T_Ut.After(warnings[j].T_Ut) {
  1276. return true
  1277. }
  1278. return false
  1279. })
  1280. for _, v := range warnings {
  1281. r = append(r, WarningToWarning_R(0, v))
  1282. }
  1283. return r, cnt
  1284. }
  1285. func GormReadAdminWarningListScopes(admin Account.Admin, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string) func(db *gorm.DB) *gorm.DB {
  1286. return func(db *gorm.DB) *gorm.DB {
  1287. if admin.T_pids != "*" && len(admin.T_pids) > 0 {
  1288. list := lib.SplitStringIds(admin.T_pids, "P")
  1289. db.Where("t_pid in (?)", list)
  1290. }
  1291. if T_abandon == 1 {
  1292. // 查询弃用设备
  1293. if len(abandonSN) > 0 {
  1294. db.Where("t_sn in (?)", abandonSN)
  1295. } else {
  1296. db.Where("1=0")
  1297. return db
  1298. }
  1299. } else if conf.UseNewWarningQuery {
  1300. if len(abandonSN) > 0 {
  1301. db.Where("t_sn not in (?)", abandonSN)
  1302. }
  1303. }
  1304. if len(T_tp) > 0 {
  1305. min, max, is := checkStringListContinuous(T_tp)
  1306. if is {
  1307. db.Where("t_tp between ? and ?", min, max)
  1308. } else {
  1309. db.Where("t_tp in (?)", T_tp)
  1310. }
  1311. }
  1312. if len(T_name) > 0 {
  1313. if len(T_name) == 16 {
  1314. db.Where("t_sn = ?", T_name)
  1315. } else {
  1316. likeName := "%" + T_name + "%"
  1317. db.Where("t_sn like ? or t__d_name like ? or t_id like ? or t__d_s_name like ? ", likeName, likeName, likeName, likeName)
  1318. }
  1319. }
  1320. if len(Time_start_) > 0 {
  1321. db.Where("t__ut >= ?", Time_start_)
  1322. }
  1323. if len(Time_end_) > 0 {
  1324. db.Where("t__ut <= ?", Time_end_)
  1325. }
  1326. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  1327. // T_handle 1:默认 2:全部记录 3 未处理
  1328. if T_handle == 1 {
  1329. db.Where("t__state > ?", 1)
  1330. } else if T_handle == 3 {
  1331. db.Where("t__state = ?", 3)
  1332. } else {
  1333. db.Where("t__state > ?", 0)
  1334. }
  1335. return db
  1336. }
  1337. }
  1338. // 管理员报警列表
  1339. func Gorm_Read_Admin_Warning_List(admin Account.Admin, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string, page int, page_z int,
  1340. countRedisKey string) (r []Warning_R, cnt int64) {
  1341. if len(admin.T_pids) == 0 {
  1342. return r, cnt
  1343. }
  1344. gormReadAdminWarningListScopes := GormReadAdminWarningListScopes(admin, T_tp, T_name, T_handle, Time_start_, Time_end_, T_abandon, abandonSN)
  1345. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1346. var err error
  1347. var warnings []Warning
  1348. tableCount := make(map[string]int64)
  1349. // 从redis获取报警统计数量
  1350. cnt, tableCount, err = Redis_WarningCount_GetOne(admin.Id, countRedisKey)
  1351. if err != nil {
  1352. for _, tableName := range tableNames {
  1353. tableDB := db.DB.Table(tableName)
  1354. var count int64
  1355. tableDB.Scopes(gormReadAdminWarningListScopes).Count(&count)
  1356. tableCount[tableName] = count
  1357. cnt += count
  1358. }
  1359. Redis_WarningCount_Set(admin.Id, countRedisKey, cnt, tableCount)
  1360. }
  1361. if cnt == 0 {
  1362. return r, 0
  1363. }
  1364. // page_z == 9999 获取全部
  1365. if page_z == 9999 {
  1366. for _, tableName := range tableNames {
  1367. var tableWarning []Warning
  1368. tableDB := db.DB.Table(tableName)
  1369. tableDB.Scopes(gormReadAdminWarningListScopes).
  1370. Order("t__ut desc").Offset(-1).Limit(-1).Find(&tableWarning)
  1371. warnings = append(warnings, tableWarning...)
  1372. }
  1373. for _, v := range warnings {
  1374. r = append(r, WarningToAdminWarning_R(0, v))
  1375. }
  1376. return r, cnt
  1377. }
  1378. offset := (page - 1) * page_z
  1379. remaining := offset
  1380. for _, tableName := range tableNames {
  1381. var tableWarning []Warning
  1382. tableDB := db.DB.Table(tableName)
  1383. count := tableCount[tableName]
  1384. if remaining >= int(count) {
  1385. remaining -= int(count)
  1386. continue
  1387. }
  1388. tableDB.Scopes(gormReadAdminWarningListScopes).
  1389. Order("t__ut desc").Offset(remaining).Limit(page_z - len(warnings)).Find(&tableWarning)
  1390. warnings = append(warnings, tableWarning...)
  1391. if len(warnings) >= page_z {
  1392. break
  1393. }
  1394. remaining = 0
  1395. }
  1396. for _, v := range warnings {
  1397. r = append(r, WarningToAdminWarning_R(0, v))
  1398. }
  1399. return r, cnt
  1400. }
  1401. func Gorm_Read_Admin_Warning_List_Excel(admin Account.Admin, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, T_abandon int, abandonSN []string, page int, page_z int,
  1402. countRedisKey string) (r []Warning_R, cnt int64) {
  1403. if len(admin.T_pids) == 0 {
  1404. return r, cnt
  1405. }
  1406. gormReadAdminWarningListScopes := GormReadAdminWarningListScopes(admin, T_tp, T_name, T_handle, Time_start_, Time_end_, T_abandon, abandonSN)
  1407. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1408. var warnings []Warning
  1409. var wg sync.WaitGroup
  1410. for _, tableName := range tableNames {
  1411. wg.Add(1)
  1412. go func(table string) {
  1413. defer wg.Done()
  1414. offset := 0
  1415. limit := 10000
  1416. for {
  1417. var tableWarning []Warning
  1418. tableDB := db.DB.Table(table)
  1419. tableDB.Scopes(gormReadAdminWarningListScopes).
  1420. Order("t__ut desc").Offset(offset).Limit(limit).Find(&tableWarning)
  1421. if len(tableWarning) == 0 {
  1422. break // 如果没有数据则结束循环
  1423. }
  1424. offset += limit // 更新偏移量
  1425. warnings = append(warnings, tableWarning...)
  1426. }
  1427. }(tableName)
  1428. }
  1429. // 等待所有协程完成
  1430. wg.Wait()
  1431. // 倒序
  1432. sort.Slice(warnings, func(i, j int) bool {
  1433. if warnings[i].T_Ut.After(warnings[j].T_Ut) {
  1434. return true
  1435. }
  1436. return false
  1437. })
  1438. for _, v := range warnings {
  1439. r = append(r, WarningToWarning_R(0, v))
  1440. }
  1441. return r, cnt
  1442. }
  1443. func GormReadCompanyWarningListScopes(T_pids []int, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string) func(db *gorm.DB) *gorm.DB {
  1444. return func(db *gorm.DB) *gorm.DB {
  1445. if len(T_pids) > 0 {
  1446. db.Where("t_pid in (?)", T_pids)
  1447. }
  1448. if len(T_tp) > 0 {
  1449. min, max, is := checkStringListContinuous(T_tp)
  1450. if is {
  1451. db.Where("t_tp between ? and ?", min, max)
  1452. } else {
  1453. db.Where("t_tp in (?)", T_tp)
  1454. }
  1455. }
  1456. if len(T_name) > 0 {
  1457. if len(T_name) == 16 {
  1458. db.Where("t_sn = ?", T_name)
  1459. } else {
  1460. likeName := "%" + T_name + "%"
  1461. db.Where("t_sn like ? or t__d_name like ? or t_id like ? or t__d_s_name like ?", likeName, likeName, likeName, likeName)
  1462. }
  1463. }
  1464. if len(Time_start_) > 0 {
  1465. db.Where("t__ut >= ?", Time_start_)
  1466. }
  1467. if len(Time_end_) > 0 {
  1468. db.Where("t__ut <= ?", Time_end_)
  1469. }
  1470. // T_State 0 删除 1 不处理 2 已处理 3 未处理
  1471. // T_handle 1:默认 2:全部记录 3 未处理
  1472. if T_handle == 1 {
  1473. db.Where("t__state > ?", 1)
  1474. } else if T_handle == 3 {
  1475. db.Where("t__state = ?", 3)
  1476. } else {
  1477. db.Where("t__state > ?", 0)
  1478. }
  1479. return db
  1480. }
  1481. }
  1482. // 公司管理报警列表
  1483. func Gorm_Read_Company_Warning_List(T_pids []int, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int,
  1484. T_pid int, countRedisKey string) (r []CompanyWarning_R, cnt int64) {
  1485. gormReadCompanyWarningListScopes := GormReadCompanyWarningListScopes(T_pids, T_tp, T_name, T_handle, Time_start_, Time_end_)
  1486. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1487. var err error
  1488. var warnings []Warning
  1489. tableCount := make(map[string]int64)
  1490. // 从redis获取报警统计数量
  1491. cnt, tableCount, err = Redis_WarningCount_GetOne(T_pid, countRedisKey)
  1492. if err != nil {
  1493. for _, tableName := range tableNames {
  1494. tableDB := db.DB.Table(tableName)
  1495. var count int64
  1496. tableDB.Scopes(gormReadCompanyWarningListScopes).Count(&count)
  1497. tableCount[tableName] = count
  1498. cnt += count
  1499. }
  1500. Redis_WarningCount_Set(T_pid, countRedisKey, cnt, tableCount)
  1501. }
  1502. if cnt == 0 {
  1503. return r, 0
  1504. }
  1505. offset := (page - 1) * page_z
  1506. remaining := offset
  1507. for _, tableName := range tableNames {
  1508. var tableWarning []Warning
  1509. tableDB := db.DB.Table(tableName)
  1510. count := tableCount[tableName]
  1511. if remaining >= int(count) {
  1512. remaining -= int(count)
  1513. continue
  1514. }
  1515. tableDB.Scopes(gormReadCompanyWarningListScopes).
  1516. Order("t__ut desc").Offset(remaining).Limit(page_z - len(warnings)).Find(&tableWarning)
  1517. warnings = append(warnings, tableWarning...)
  1518. if len(warnings) >= page_z {
  1519. break
  1520. }
  1521. remaining = 0
  1522. }
  1523. for _, v := range warnings {
  1524. r = append(r, WarningToCompanyWarning_R(0, v))
  1525. }
  1526. return r, cnt
  1527. }
  1528. func Gorm_Read_Company_Warning_List_Excel(T_pids []int, T_tp []string, T_name string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int,
  1529. T_pid int, countRedisKey string) (r []CompanyWarning_R, cnt int64) {
  1530. gormReadCompanyWarningListScopes := GormReadCompanyWarningListScopes(T_pids, T_tp, T_name, T_handle, Time_start_, Time_end_)
  1531. tableNames := GenerateWarningTableNames(Time_start_, Time_end_)
  1532. var warnings []Warning
  1533. var wg sync.WaitGroup
  1534. for _, tableName := range tableNames {
  1535. wg.Add(1)
  1536. go func(table string) {
  1537. defer wg.Done()
  1538. offset := 0
  1539. limit := 10000
  1540. for {
  1541. var tableWarning []Warning
  1542. tableDB := db.DB.Table(table)
  1543. tableDB.Scopes(gormReadCompanyWarningListScopes).
  1544. Order("t__ut desc").Offset(offset).Limit(limit).Find(&tableWarning)
  1545. if len(tableWarning) == 0 {
  1546. break // 如果没有数据则结束循环
  1547. }
  1548. offset += limit // 更新偏移量
  1549. warnings = append(warnings, tableWarning...)
  1550. }
  1551. }(tableName)
  1552. }
  1553. // 等待所有协程完成
  1554. wg.Wait()
  1555. // 倒序
  1556. sort.Slice(warnings, func(i, j int) bool {
  1557. if warnings[i].T_Ut.After(warnings[j].T_Ut) {
  1558. return true
  1559. }
  1560. return false
  1561. })
  1562. for _, v := range warnings {
  1563. r = append(r, WarningToCompanyWarning_R(0, v))
  1564. }
  1565. return r, cnt
  1566. }
  1567. func GenerateWarningTableNames(startDateStr string, endDateStr string) []string {
  1568. if len(startDateStr) == 0 || len(endDateStr) == 0 {
  1569. return []string{}
  1570. }
  1571. startDate, _ := lib.TimeStrToTime(startDateStr)
  1572. endDate, _ := lib.TimeStrToTime(endDateStr)
  1573. monthsArray := []string{}
  1574. for startDate.Before(endDate) || startDate.Equal(endDate) {
  1575. monthsArray = append(monthsArray, fmt.Sprintf("warning_%d_%02d", startDate.Year(), startDate.Month()))
  1576. startDate = time.Date(startDate.Year(), startDate.Month()+1, 1, 0, 0, 0, 0, time.Local)
  1577. }
  1578. sort.Stable(sort.Reverse(sort.StringSlice(monthsArray)))
  1579. return monthsArray
  1580. }
  1581. // 检查数据列表是否连续 返回最小值 最大值
  1582. func checkStringListContinuous(nums []string) (int, int, bool) {
  1583. intNums := make([]int, len(nums))
  1584. for i, num := range nums {
  1585. parsedNum, err := strconv.Atoi(num)
  1586. if err != nil {
  1587. logs.Error("Error converting string to int:", err)
  1588. return 0, 0, false
  1589. }
  1590. intNums[i] = parsedNum
  1591. }
  1592. sort.Ints(intNums)
  1593. for i := 1; i < len(intNums); i++ {
  1594. if intNums[i] != intNums[i-1]+1 {
  1595. return 0, 0, false
  1596. }
  1597. }
  1598. return intNums[0], intNums[len(intNums)-1], true
  1599. }
  1600. // BatchHandle 处理本公司指定类型40天内的所有报警数据
  1601. func BatchHandle(Pid, T_tp int, T_Text, T_name, T_uuid string) bool {
  1602. var warings []Warning
  1603. var err error
  1604. begin := db.DB.Begin()
  1605. defer func() {
  1606. if err != nil {
  1607. begin.Rollback()
  1608. } else {
  1609. begin.Commit()
  1610. }
  1611. }()
  1612. err = begin.Model(&Warning{}).Where("t_pid =?", Pid).
  1613. Where("t__state != ?", 2).
  1614. Where("t_tp = ?", T_tp).Find(&warings).Error
  1615. if err != nil {
  1616. logs.Error("查询报警记录信息失败", err)
  1617. return false
  1618. }
  1619. for _, waring := range warings {
  1620. text := AddHandlingRecord(waring, T_Text, T_name, T_uuid)
  1621. err = begin.Model(&waring).Updates(map[string]any{
  1622. "t__text": text,
  1623. "t__state": 2,
  1624. }).Error
  1625. if err != nil {
  1626. logs.Error("报警记录信息更新失败", err)
  1627. return false
  1628. }
  1629. Wtab := "warning"
  1630. T_year, T_month := waring.CreateTime.Format("2006"), waring.CreateTime.Format("01")
  1631. Wtab += "_" + T_year + "_" + T_month
  1632. waring.T_State = 2
  1633. waring.UpdateTime = time.Now()
  1634. waring.T_Text = text
  1635. backups := Update_Warning_Backups(waring, T_year, T_month)
  1636. if !backups {
  1637. err = errors.New("更新报警信息失败")
  1638. return false
  1639. }
  1640. System.Add_UserLogs(T_uuid, "设备管理", "40天内报警处理操作", Wtab+":"+strconv.Itoa(int(waring.Id))+"->"+T_Text)
  1641. }
  1642. return true
  1643. }
  1644. // 批量处理报警信息(带附件)
  1645. func BatchHandleWithAttachments(Pid, T_tp int, T_Text, T_name, T_uuid string, attachments []Attachment) bool {
  1646. var warings []Warning
  1647. var err error
  1648. begin := db.DB.Begin()
  1649. defer func() {
  1650. if err != nil {
  1651. begin.Rollback()
  1652. } else {
  1653. begin.Commit()
  1654. }
  1655. }()
  1656. err = begin.Model(&Warning{}).Where("t_pid =?", Pid).
  1657. Where("t__state != ?", 2).
  1658. Where("t_tp = ?", T_tp).Find(&warings).Error
  1659. if err != nil {
  1660. logs.Error("查询报警记录信息失败", err)
  1661. return false
  1662. }
  1663. for _, waring := range warings {
  1664. // 使用带附件的处理记录方法
  1665. text := AddHandlingRecordWithAttachments(waring, T_Text, T_name, T_uuid, attachments)
  1666. err = begin.Model(&waring).Updates(map[string]any{
  1667. "t__text": text,
  1668. "t__state": 2,
  1669. }).Error
  1670. if err != nil {
  1671. logs.Error("报警记录信息更新失败", err)
  1672. return false
  1673. }
  1674. Wtab := "warning"
  1675. T_year, T_month := waring.CreateTime.Format("2006"), waring.CreateTime.Format("01")
  1676. Wtab += "_" + T_year + "_" + T_month
  1677. waring.T_State = 2
  1678. waring.UpdateTime = time.Now()
  1679. waring.T_Text = text
  1680. backups := Update_Warning_Backups(waring, T_year, T_month)
  1681. if !backups {
  1682. err = errors.New("更新报警信息失败")
  1683. return false
  1684. }
  1685. // 记录操作日志,包含附件信息
  1686. logMsg := Wtab + ":" + strconv.Itoa(int(waring.Id)) + "->" + T_Text
  1687. if len(attachments) > 0 {
  1688. logMsg += fmt.Sprintf(", 附件数量: %d", len(attachments))
  1689. }
  1690. System.Add_UserLogs(T_uuid, "设备管理", "40天内报警处理操作", logMsg)
  1691. }
  1692. return true
  1693. }
  1694. func GetHandlingRecords(w Warning) ([]HandlingRecord, error) {
  1695. if len(w.T_Text) == 0 {
  1696. return nil, nil
  1697. }
  1698. // 尝试解析为新格式
  1699. var records []HandlingRecord
  1700. if err := json.Unmarshal([]byte(w.T_Text), &records); err == nil {
  1701. return records, nil
  1702. }
  1703. // 如果都不是JSON格式,直接作为字符串处理
  1704. return []HandlingRecord{
  1705. {
  1706. Remark: w.T_Text,
  1707. Handler: "",
  1708. HandledAt: w.UpdateTime,
  1709. },
  1710. }, nil
  1711. }
  1712. func AddHandlingRecord(w Warning, remark, handler, handler_uuid string) string {
  1713. // 获取现有记录
  1714. existingRecords, _ := GetHandlingRecords(w)
  1715. if existingRecords == nil {
  1716. existingRecords = []HandlingRecord{}
  1717. }
  1718. // 添加新记录
  1719. newRecord := HandlingRecord{
  1720. Remark: remark,
  1721. Handler: handler,
  1722. Handler_uuid: handler_uuid,
  1723. HandledAt: time.Now(),
  1724. Attachments: []Attachment{}, // 初始化为空切片
  1725. }
  1726. // 创建新格式数据
  1727. records := append(existingRecords, newRecord)
  1728. // 序列化为JSON
  1729. jsonData, err := json.Marshal(records)
  1730. if err != nil {
  1731. return ""
  1732. }
  1733. return string(jsonData)
  1734. }
  1735. // 添加带附件的处理记录
  1736. func AddHandlingRecordWithAttachments(w Warning, remark, handler, handler_uuid string, attachments []Attachment) string {
  1737. // 获取现有记录
  1738. existingRecords, _ := GetHandlingRecords(w)
  1739. if existingRecords == nil {
  1740. existingRecords = []HandlingRecord{}
  1741. }
  1742. // 添加新记录
  1743. newRecord := HandlingRecord{
  1744. Remark: remark,
  1745. Handler: handler,
  1746. Handler_uuid: handler_uuid,
  1747. HandledAt: time.Now(),
  1748. Attachments: attachments, // 使用传入的附件列表
  1749. }
  1750. // 创建新格式数据
  1751. records := append(existingRecords, newRecord)
  1752. // 序列化为JSON
  1753. jsonData, err := json.Marshal(records)
  1754. if err != nil {
  1755. return ""
  1756. }
  1757. return string(jsonData)
  1758. }
  1759. func GetLatestRemark(w Warning) (string, []Attachment) {
  1760. records, err := GetHandlingRecords(w)
  1761. if err != nil || len(records) == 0 {
  1762. // 如果是旧数据,直接返回原始文本
  1763. if len(w.T_Text) > 0 && w.T_Text[0] != '{' {
  1764. return w.T_Text, []Attachment{}
  1765. }
  1766. return "", []Attachment{}
  1767. }
  1768. // 返回最新的一条备注
  1769. return records[len(records)-1].Remark, records[len(records)-1].Attachments
  1770. }
  1771. // 获取处理人信息
  1772. func GetHandlers(w Warning) string {
  1773. records, err := GetHandlingRecords(w)
  1774. if err != nil || len(records) == 0 {
  1775. return ""
  1776. }
  1777. // 返回最新的一条处理人
  1778. return records[len(records)-1].Handler
  1779. }
  1780. func IsWarn(Pid int) bool {
  1781. var warings []Warning
  1782. var err error
  1783. begin := db.DB.Begin()
  1784. defer func() {
  1785. if err != nil {
  1786. begin.Rollback()
  1787. } else {
  1788. begin.Commit()
  1789. }
  1790. }()
  1791. err = begin.Model(&Warning{}).Where("t_pid =?", Pid).
  1792. Where("t__state != ?", 2).Find(&warings).Error
  1793. if err != nil {
  1794. logs.Error("报警记录信息更新失败", err)
  1795. return false
  1796. }
  1797. return true
  1798. }