123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437 |
- package controllers
- import (
- "Cold_Api/Nats/NatsServer"
- "Cold_Api/conf"
- "Cold_Api/controllers/lib"
- db "Cold_Api/initialize"
- "Cold_Api/models/Account"
- "Cold_Api/models/Device"
- "Cold_Api/models/System"
- "Cold_Api/models/Warning"
- "encoding/json"
- "fmt"
- "github.com/beego/beego/v2/core/logs"
- "github.com/robfig/cron/v3"
- "github.com/xuri/excelize/v2"
- "math"
- "net/http"
- "os"
- "strconv"
- "time"
- )
- // 设备告警 ------------------------------------------
- // 告警列表
- func (c *DeviceController) DeviceWarning_List() {
- type R_JSONS struct {
- //必须的大写开头
- Data []Warning.Warning_R
- Num int64
- Page int
- Page_size int
- }
- var r_jsons R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_admin, _ := c.GetInt("T_admin")
- T_history, _ := c.GetInt("T_history")
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- var T_year, T_month string
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", Time_start)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- }
- // 管理员界面
- if T_admin == 1 {
- if T_history == 1 {
- // 获取备份
- 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)
- } else {
- // 获取最新
- 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)
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- // 公司报警界面
- if T_history == 1 {
- 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)
- } else {
- // 获取最新
- 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)
- }
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) DeviceWarning_List2() {
- type R_JSONS struct {
- //必须的大写开头
- Data []Warning.Warning_R
- Num int64
- Page int
- Page_size int
- }
- var r_jsons R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_admin, _ := c.GetInt("T_admin")
- T_history, _ := c.GetInt("T_history")
- if len(Time_start) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "开始时间不能为空!"}
- c.ServeJSON()
- return
- }
- if len(Time_end) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "结束时间不能为空!"}
- c.ServeJSON()
- return
- }
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- // 管理员界面
- if T_admin == 1 {
- // 获取最新
- 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)
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- 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)
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) DeviceWarning_List_Sse() {
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_admin, _ := c.GetInt("T_admin")
- tpCount := Warning.Read_WarningType_Count()
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- tableNames := Warning.GenerateWarningTableNames(Time_start, Time_end)
- // 设置响应头
- c.Ctx.ResponseWriter.Header().Set("Content-Type", "text/event-stream")
- c.Ctx.ResponseWriter.Header().Set("Cache-Control", "no-cache")
- c.Ctx.ResponseWriter.Header().Set("Connection", "keep-alive")
- c.Ctx.ResponseWriter.WriteHeader(http.StatusOK)
- var cnt int64
- tableCount := make(map[string]int64)
- // 管理员界面
- if T_admin == 1 {
- gormReadWarningListScopes := Warning.GormReadAdminWarningListScopes(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end)
- cnt, tableCount, err = Warning.Redis_WarningCount_GetOne(c.Admin_r.Id, countRedisKey)
- offset := (page - 1) * page_z
- remaining := offset
- var warningsCount int
- for _, tableName := range tableNames {
- tableDB := db.DB.Table(tableName)
- var count int64
- if tableName == fmt.Sprintf("warning_%d_%02d", time.Now().Year(), time.Now().Month()) {
- if saveTime, ok := tableCount["SaveTime"]; ok {
- t1 := time.Unix(saveTime, 0)
- if time.Now().Sub(t1) < 5*time.Minute {
- if count, ok = tableCount[tableName]; !ok {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- } else {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- }
- } else {
- var ok bool
- if count, ok = tableCount[tableName]; !ok {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- }
- cnt += count
- if remaining >= int(count) {
- remaining -= int(count)
- continue
- }
- var tableWarning []Warning.Warning
- if warningsCount < page_z {
- tableDB.Scopes(gormReadWarningListScopes).
- Order("t__ut desc").Offset(remaining).Limit(page_z - warningsCount).Find(&tableWarning)
- }
- warningsCount += len(tableWarning)
- remaining = 0
- var r_jsons lib.R_JSONS
- r_jsons.Num = cnt
- r_jsons.Data = tableWarning
- json_b, _ := json.Marshal(r_jsons)
- c.Ctx.ResponseWriter.Write(json_b)
- c.Ctx.ResponseWriter.Flush()
- // 可以根据需要调整间隔时间
- time.Sleep(1 * time.Second)
- }
- Warning.Redis_WarningCount_Set(c.Admin_r.Id, countRedisKey, cnt, tableCount)
- return
- }
- //从redis获取报警统计数量
- gormReadWarningListScopes := Warning.GormReadWarningListScopes(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end)
- cnt, tableCount, err = Warning.Redis_WarningCount_GetOne(c.T_pid, countRedisKey)
- offset := (page - 1) * page_z
- remaining := offset
- var warningsCount int
- for _, tableName := range tableNames {
- tableDB := db.DB.Table(tableName)
- var count int64
- if tableName == fmt.Sprintf("warning_%d_%02d", time.Now().Year(), time.Now().Month()) {
- if saveTime, ok := tableCount["SaveTime"]; ok {
- t1 := time.Unix(saveTime, 0)
- if time.Now().Sub(t1) < 5*time.Minute {
- if count, ok = tableCount[tableName]; !ok {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- } else {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- }
- } else {
- var ok bool
- if count, ok = tableCount[tableName]; !ok {
- tableDB.Scopes(gormReadWarningListScopes).Count(&count)
- tableCount[tableName] = count
- }
- }
- cnt += count
- if remaining >= int(count) {
- remaining -= int(count)
- continue
- }
- var tableWarning []Warning.Warning
- if warningsCount < page_z {
- tableDB.Scopes(gormReadWarningListScopes).
- Order("t__ut desc").Offset(remaining).Limit(page_z - warningsCount).Find(&tableWarning)
- }
- warningsCount += len(tableWarning)
- remaining = 0
- var r_jsons lib.R_JSONS
- r_jsons.Num = cnt
- r_jsons.Data = tableWarning
- json_b, _ := json.Marshal(r_jsons)
- c.Ctx.ResponseWriter.Write(json_b)
- c.Ctx.ResponseWriter.Flush()
- // 可以根据需要调整间隔时间
- time.Sleep(1 * time.Second)
- }
- Warning.Redis_WarningCount_Set(c.T_pid, countRedisKey, cnt, tableCount)
- }
- func (c *DeviceController) CompanyWarning_List() {
- var r_jsons lib.R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_history, _ := c.GetInt("T_history")
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- Company_r, err := Account.Read_Company_ById(c.T_pid)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
- c.ServeJSON()
- return
- }
- T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
- var T_year, T_month string
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", Time_start)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- }
- Account.Read_Company_All_Maps()
- if T_history == 1 {
- 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)
- } else {
- // 获取最新
- 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)
- }
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) CompanyWarning_List2() {
- var r_jsons lib.R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- if len(Time_start) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "开始时间不能为空!"}
- c.ServeJSON()
- return
- }
- if len(Time_end) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "结束时间不能为空!"}
- c.ServeJSON()
- return
- }
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- Company_r, err := Account.Read_Company_ById(c.T_pid)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
- c.ServeJSON()
- return
- }
- T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
- Account.Read_Company_All_Maps()
- // 获取最新
- 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)
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- // 查询告警 暂时没有用到
- func (c *DeviceController) DeviceWarning_Get() {
- id, _ := c.GetInt("T_id")
- T_time := c.GetString("T_time")
- if len(T_time) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- var T_year, T_month string
- date, err := time.Parse("2006-01-02 15:04:05", T_time)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- Wtab := "warning"
- Wtab += "_" + T_year + "_" + T_month
- T, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.WarningToWarning_R(0, T)}
- c.ServeJSON()
- return
- }
- // 编辑告警(处理告警)
- func (c *DeviceController) DeviceWarning_Post() {
- id, _ := c.GetInt("T_id")
- T_Text := c.GetString("T_Text")
- T_time := c.GetString("T_time")
- T_history, _ := c.GetInt("T_history") // TODO 后期废弃
- if len(T_time) == 0 && conf.UseNewWarningQuery {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- var T_year, T_month string
- var warning Warning.Warning
- Wtab := "warning"
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", T_time)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- Wtab += "_" + T_year + "_" + T_month
- warning, err = Warning.Read_Warning_ById_Backups(id, T_year, T_month)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- warning.T_Text = T_Text
- warning.T_State = 2
- warning.UpdateTime = time.Now()
- if is := Warning.Update_Warning_Backups(warning, T_year, T_month); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
- c.ServeJSON()
- return
- }
- if !conf.UseNewWarningQuery {
- warning = Warning.Read_Warning_ById(int64(id))
- if warning.Id > 0 {
- warning.T_Text = T_Text
- warning.T_State = 2
- warning.UpdateTime = time.Now()
- if is := Warning.Update_Warning(warning, "T_Text", "T_State", "UpdateTime"); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
- c.ServeJSON()
- return
- }
- }
- }
- } else {
- // 使用新的查询方式后 warning表不做更新
- if !conf.UseNewWarningQuery {
- warning = Warning.Read_Warning_ById(int64(id))
- if warning.Id == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- warning.T_Text = T_Text
- warning.T_State = 2
- warning.UpdateTime = time.Now()
- if is := Warning.Update_Warning(warning, "T_Text", "T_State", "UpdateTime"); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
- c.ServeJSON()
- return
- }
- }
- // 同步处理历史数据
- date := warning.T_Ut
- T_year, T_month = date.Format("2006"), date.Format("01")
- Wtab += "_" + T_year + "_" + T_month
- warningBackups, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
- if err != nil && err.Error() != "<QuerySeter> no row found" {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- if warningBackups.Id > 0 {
- warningBackups.T_Text = T_Text
- warningBackups.T_State = 2
- warningBackups.UpdateTime = time.Now()
- if is := Warning.Update_Warning_Backups(warningBackups, T_year, T_month); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
- c.ServeJSON()
- return
- }
- }
- }
- // 删除报警列表统计的缓存
- Warning.Redis_WarningCount_DelK(c.T_pid)
- Warning.Redis_WarningCount_DelK(c.Admin_r.Id)
- System.Add_UserLogs(c.Admin_r.T_uuid, "设备管理", "报警处理操作", Wtab+":"+strconv.Itoa(id)+"->"+T_Text)
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
- c.ServeJSON()
- return
- }
- // 删除告警
- func (c *DeviceController) DeviceWarning_Del() {
- id, _ := c.GetInt("T_id")
- T_time := c.GetString("T_time")
- T_history, _ := c.GetInt("T_history") // TODO 后期废弃
- if len(T_time) == 0 && conf.UseNewWarningQuery {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- var T_year, T_month string
- var warning Warning.Warning
- Wtab := "warning"
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", T_time)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- Wtab += "_" + T_year + "_" + T_month
- warning, err = Warning.Read_Warning_ById_Backups(id, T_year, T_month)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- warning.T_State = 0
- warning.UpdateTime = time.Now()
- if is := Warning.Update_Warning_Backups(warning, T_year, T_month); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
- c.ServeJSON()
- return
- }
- warning = Warning.Read_Warning_ById(int64(id))
- if warning.Id > 0 {
- warning.T_State = 0
- if is := Warning.Update_Warning(warning, "T_State"); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
- c.ServeJSON()
- return
- }
- }
- } else {
- warning = Warning.Read_Warning_ById(int64(id))
- if warning.Id == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- warning.T_State = 0
- if is := Warning.Update_Warning(warning, "T_State"); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
- c.ServeJSON()
- return
- }
- date := warning.T_Ut
- T_year, T_month = date.Format("2006"), date.Format("01")
- Wtab += "_" + T_year + "_" + T_month
- warningBackups, err := Warning.Read_Warning_ById_Backups(id, T_year, T_month)
- if err != nil && err.Error() != "<QuerySeter> no row found" {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
- c.ServeJSON()
- return
- }
- if warningBackups.Id > 0 {
- warningBackups.T_State = 0
- warningBackups.UpdateTime = time.Now()
- if is := Warning.Update_Warning_Backups(warningBackups, T_year, T_month); !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
- c.ServeJSON()
- return
- }
- }
- }
- // 删除报警列表统计的缓存
- Warning.Redis_WarningCount_DelK(c.T_pid)
- Warning.Redis_WarningCount_DelK(c.Admin_r.Id)
- System.Add_UserLogs(c.Admin_r.T_uuid, "设备管理", "报警删除操作", Wtab+":"+strconv.Itoa(id))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
- c.ServeJSON()
- return
- }
- // 导出告警
- func (c *DeviceController) DeviceWarning_Data_Excel() {
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_admin, _ := c.GetInt("T_admin")
- T_history, _ := c.GetInt("T_history")
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- var T_year, T_month string
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", Time_start)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- }
- var Device_data []Warning.Warning_R
- // 管理员界面
- if T_admin == 1 {
- if T_history == 1 {
- // 获取备份
- 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)
- } else {
- // 获取最新
- Device_data, _ = Warning.Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- }
- }
- if T_history == 1 {
- // 获取备份
- 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)
- } else {
- // 获取最新
- Device_data, _ = Warning.Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- }
- f := excelize.NewFile() // 设置单元格的值
- // 这里设置表头ÒÒ
- f.SetCellValue("Sheet1", "A1", "报警类型")
- f.SetCellValue("Sheet1", "B1", "Sn")
- f.SetCellValue("Sheet1", "C1", "设备名称")
- f.SetCellValue("Sheet1", "D1", "传感器")
- f.SetCellValue("Sheet1", "E1", "报警内容")
- f.SetCellValue("Sheet1", "F1", "记录时间")
- f.SetCellValue("Sheet1", "G1", "处理")
- f.SetCellValue("Sheet1", "H1", "处理时间")
- // 设置列宽
- f.SetColWidth("Sheet1", "A", "A", 20)
- f.SetColWidth("Sheet1", "B", "B", 25)
- f.SetColWidth("Sheet1", "C", "C", 30)
- f.SetColWidth("Sheet1", "D", "D", 30)
- f.SetColWidth("Sheet1", "G", "E", 30)
- f.SetColWidth("Sheet1", "H", "F", 15)
- f.SetColWidth("Sheet1", "I", "G", 30)
- f.SetColWidth("Sheet1", "J", "H", 15)
- line := 1
- // 循环写入数据
- for _, v := range Device_data {
- line++
- f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
- f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
- f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), v.T_D_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_DS_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_Remark)
- f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Ut)
- f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Text)
- f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.UpdateTime)
- }
- timeStr := time.Now().Format("20060102150405")
- // 保存文件
- if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
- fmt.Println(err)
- }
- url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
- if !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
- c.ServeJSON()
- return
- }
- // 删除文件
- err = os.Remove("ofile/" + timeStr + ".xlsx")
- if err != nil {
- fmt.Println(err)
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) DeviceWarning_Data_Excel2() {
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- if len(Time_start) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "开始时间不能为空!"}
- c.ServeJSON()
- return
- }
- if len(Time_end) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "结束时间不能为空!"}
- c.ServeJSON()
- return
- }
- T_admin, _ := c.GetInt("T_admin")
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- var Device_data []Warning.Warning_R
- // 管理员界面
- if T_admin == 1 {
- // 获取最新
- //Device_data, _ = Warning.Gorm_Read_Admin_Warning_List(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- Device_data, _ = Warning.Gorm_Read_Admin_Warning_List_Excel(*c.Admin_r, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- } else {
- // 获取最新
- //Device_data, _ = Warning.Gorm_Read_Warning_List(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- Device_data, _ = Warning.Gorm_Read_Warning_List_Excel(c.T_pid, bindSN, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, countRedisKey)
- }
- f := excelize.NewFile() // 设置单元格的值
- // 这里设置表头ÒÒ
- f.SetCellValue("Sheet1", "A1", "报警类型")
- f.SetCellValue("Sheet1", "B1", "Sn")
- f.SetCellValue("Sheet1", "C1", "设备名称")
- f.SetCellValue("Sheet1", "D1", "传感器")
- f.SetCellValue("Sheet1", "E1", "报警内容")
- f.SetCellValue("Sheet1", "F1", "记录时间")
- f.SetCellValue("Sheet1", "G1", "处理")
- f.SetCellValue("Sheet1", "H1", "处理时间")
- // 设置列宽
- f.SetColWidth("Sheet1", "A", "A", 20)
- f.SetColWidth("Sheet1", "B", "B", 25)
- f.SetColWidth("Sheet1", "C", "C", 30)
- f.SetColWidth("Sheet1", "D", "D", 30)
- f.SetColWidth("Sheet1", "G", "E", 30)
- f.SetColWidth("Sheet1", "H", "F", 15)
- f.SetColWidth("Sheet1", "I", "G", 30)
- f.SetColWidth("Sheet1", "J", "H", 15)
- line := 1
- // 循环写入数据
- for _, v := range Device_data {
- line++
- f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
- f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
- f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), v.T_D_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_DS_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_Remark)
- f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Ut)
- f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Text)
- f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.UpdateTime)
- }
- timeStr := time.Now().Format("20060102150405")
- // 保存文件
- if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
- fmt.Println(err)
- }
- url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
- if !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
- c.ServeJSON()
- return
- }
- // 删除文件
- err = os.Remove("ofile/" + timeStr + ".xlsx")
- if err != nil {
- fmt.Println(err)
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
- c.ServeJSON()
- return
- }
- // 公司管理 - 导出告警
- func (c *DeviceController) CompanyWarning_Data_Excel() {
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- T_history, _ := c.GetInt("T_history")
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- Account.Read_Company_All_Maps()
- Company_r, err := Account.Read_Company_ById(c.T_pid)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
- c.ServeJSON()
- return
- }
- T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
- var T_year, T_month string
- if T_history == 1 {
- date, err := time.Parse("2006-01-02 15:04:05", Time_start)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式错误"}
- c.ServeJSON()
- return
- }
- T_year, T_month = date.Format("2006"), date.Format("01")
- }
- var Device_data []Warning.CompanyWarning_R
- if T_history == 1 {
- // 获取备份
- 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)
- } else {
- // 获取最新
- Device_data, _ = Warning.Read_Company_Warning_List(T_pids, tpList, T_name, T_handle, Time_start, Time_end, 0, 9999, c.T_pid, countRedisKey)
- }
- f := excelize.NewFile() // 设置单元格的值
- // 这里设置表头
- f.SetCellValue("Sheet1", "A1", "报警类型")
- f.SetCellValue("Sheet1", "A1", "公司名称")
- f.SetCellValue("Sheet1", "B1", "Sn")
- f.SetCellValue("Sheet1", "C1", "设备名称")
- f.SetCellValue("Sheet1", "D1", "传感器")
- f.SetCellValue("Sheet1", "E1", "报警内容")
- f.SetCellValue("Sheet1", "F1", "记录时间")
- f.SetCellValue("Sheet1", "G1", "处理")
- f.SetCellValue("Sheet1", "H1", "处理时间")
- // 设置列宽
- f.SetColWidth("Sheet1", "A", "A", 20)
- f.SetColWidth("Sheet1", "B", "B", 40)
- f.SetColWidth("Sheet1", "C", "C", 25)
- f.SetColWidth("Sheet1", "D", "D", 30)
- f.SetColWidth("Sheet1", "E", "E", 30)
- f.SetColWidth("Sheet1", "F", "F", 30)
- f.SetColWidth("Sheet1", "J", "J", 15)
- f.SetColWidth("Sheet1", "H", "H", 30)
- f.SetColWidth("Sheet1", "I", "I", 15)
- line := 1
- // 循环写入数据
- for _, v := range Device_data {
- line++
- f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
- f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s", v.T_pid_name))
- f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
- f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_D_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_DS_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Remark)
- f.SetCellValue("Sheet1", fmt.Sprintf("J%d", line), v.T_Ut)
- f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.T_Text)
- f.SetCellValue("Sheet1", fmt.Sprintf("I%d", line), v.UpdateTime)
- }
- timeStr := time.Now().Format("20060102150405")
- // 保存文件
- if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
- fmt.Println(err)
- }
- url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
- if !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
- c.ServeJSON()
- return
- }
- // 删除文件
- err = os.Remove("ofile/" + timeStr + ".xlsx")
- if err != nil {
- fmt.Println(err)
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) CompanyWarning_Data_Excel2() {
- T_name := c.GetString("T_name")
- T_tp := c.GetString("T_tp")
- Time_start := c.GetString("Time_start")
- Time_end := c.GetString("Time_end")
- T_handle, _ := c.GetInt("T_handle")
- if len(Time_start) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "开始时间不能为空!"}
- c.ServeJSON()
- return
- }
- if len(Time_end) == 0 {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "结束时间不能为空!"}
- c.ServeJSON()
- return
- }
- tpCount := Warning.Read_WarningType_Count()
- var tpList []string
- if len(T_tp) > 0 {
- tpList = lib.SplitStringIds(T_tp, "T")
- if len(tpList) == int(tpCount) {
- // 类型全选,替换为空,不进行查询筛选(查询全部)
- tpList = []string{}
- }
- } else {
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- }
- 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)
- Account.Read_Company_All_Maps()
- Company_r, err := Account.Read_Company_ById(c.T_pid)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
- c.ServeJSON()
- return
- }
- T_pids := Account.ReadCompanyIds_T_path(Company_r.T_path)
- var Device_data []Warning.CompanyWarning_R
- // 获取最新
- 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)
- f := excelize.NewFile() // 设置单元格的值
- // 这里设置表头
- f.SetCellValue("Sheet1", "A1", "报警类型")
- f.SetCellValue("Sheet1", "B1", "公司名称")
- f.SetCellValue("Sheet1", "C1", "Sn")
- f.SetCellValue("Sheet1", "D1", "设备名称")
- f.SetCellValue("Sheet1", "E1", "传感器")
- f.SetCellValue("Sheet1", "F1", "报警内容")
- f.SetCellValue("Sheet1", "G1", "记录时间")
- f.SetCellValue("Sheet1", "H1", "处理")
- f.SetCellValue("Sheet1", "I1", "处理时间")
- // 设置列宽
- f.SetColWidth("Sheet1", "A", "A", 20)
- f.SetColWidth("Sheet1", "B", "B", 40)
- f.SetColWidth("Sheet1", "C", "C", 25)
- f.SetColWidth("Sheet1", "D", "D", 30)
- f.SetColWidth("Sheet1", "E", "E", 30)
- f.SetColWidth("Sheet1", "F", "F", 30)
- f.SetColWidth("Sheet1", "J", "J", 15)
- f.SetColWidth("Sheet1", "H", "H", 30)
- f.SetColWidth("Sheet1", "I", "I", 15)
- line := 1
- // 循环写入数据
- for _, v := range Device_data {
- line++
- f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), Warning.Read_WarningType_Get(v.T_tp))
- f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), fmt.Sprintf("%s", v.T_pid_name))
- f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), fmt.Sprintf("%s[%d]", v.T_sn, v.T_id))
- f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_D_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_DS_name)
- f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_Remark)
- f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), v.T_Ut)
- f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.T_Text)
- f.SetCellValue("Sheet1", fmt.Sprintf("I%d", line), v.UpdateTime)
- }
- timeStr := time.Now().Format("20060102150405")
- // 保存文件
- if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
- fmt.Println(err)
- }
- url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx")
- if !is {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "oss!"}
- c.ServeJSON()
- return
- }
- // 删除文件
- err = os.Remove("ofile/" + timeStr + ".xlsx")
- if err != nil {
- fmt.Println(err)
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
- c.ServeJSON()
- return
- }
- // 告警提示列表
- func (c *DeviceController) DeviceWarningList_T_Tips() {
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_All_T_Notice_mechanism()}
- c.ServeJSON()
- return
- }
- // 告警类型列表
- func (c *DeviceController) WarningType_List_All() {
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_All()}
- c.ServeJSON()
- return
- }
- // 告警类型列表 - 权限关联列表
- func (c *DeviceController) WarningType_Power_List_All() {
- power, err := Account.Read_Power_ById(c.Admin_r.T_power)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "获取菜单失败"}
- c.ServeJSON()
- return
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningType_Power_All(power.T_warning)}
- c.ServeJSON()
- return
- }
- // 通过传感器类型获取报警列表
- func (c *DeviceController) Read_Warning_List_By_DS_T_type() {
- var r_jsons lib.R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_type, _ := c.GetInt("T_type")
- T_name := c.GetString("T_name")
- bindSN, err := Account.Read_UserDevice_List(*c.Admin_r)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
- c.ServeJSON()
- return
- }
- var T_pids []int
- if c.Admin_r.T_pid > 0 {
- // 查询公司
- Company_r, err := Account.Read_Company_ById(c.Admin_r.T_pid)
- if err != nil {
- c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id Err!"}
- c.ServeJSON()
- return
- }
- // 查询公司下面所有子公司id
- T_pids = Account.ReadCompanyIds_T_path(Company_r.T_path)
- }
- var tpList []string
- power, _ := Account.Read_Power_ById(c.Admin_r.T_power)
- if power.T_warning != "*" {
- tpList = lib.SplitStringIds(power.T_warning, "W")
- }
- 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)
- 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)
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) Read_WarningSend_List() {
- var r_jsons lib.R_JSONS
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_ntype, _ := c.GetInt("T_ntype")
- r_jsons.Data, r_jsons.Num = Warning.Read_WarningSand_List(c.T_pid, T_ntype, page, page_z)
- r_jsons.Page = page
- r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) Read_WarningRate_List() {
- page, _ := c.GetInt("page")
- if page < 1 {
- page = 1
- }
- page_z, _ := c.GetInt("page_z")
- if page_z < 1 {
- page_z = conf.Page_size
- }
- T_uuid := c.GetString("T_uuid")
- T_year, _ := c.GetInt("T_year")
- T_month, _ := c.GetInt("T_month")
- var startTime, endTime string
- now := time.Now()
- var month time.Month
- // 默认本月
- if T_year == 0 {
- T_year = now.Year()
- }
- if T_month == 0 {
- month = now.Month()
- } else {
- month = time.Month(T_month)
- }
- T_type := Warning.WarningRateDay
- if T_month == 0 {
- T_type = Warning.WarningRateMonth
- }
- if T_type == Warning.WarningRateDay {
- startTime = time.Date(T_year, month, 1, 0, 0, 0, 0, time.Local).Format("2006-01-02")
- endTime = time.Date(T_year, month+1, 0, 0, 0, 0, 0, time.Local).Format("2006-01-02")
- }
- // 默认本年
- if T_type == Warning.WarningRateMonth {
- startTime = time.Date(T_year, 1, 1, 0, 0, 0, 0, time.Local).Format("2006-01")
- endTime = time.Date(T_year, 12, 31, 0, 0, 0, 0, time.Local).Format("2006-01")
- }
- if len(T_uuid) == 0 {
- T_uuid = c.Admin_r.T_uuid
- }
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningRate_List(T_uuid, T_type, startTime, endTime)}
- c.ServeJSON()
- return
- }
- func (c *DeviceController) Read_WarningHandle_List() {
- c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Warning.Read_WarningHandle_List()}
- c.ServeJSON()
- return
- }
- func Cron_WarningRate() {
- // 创建一个定时任务对象
- c := cron.New(cron.WithSeconds())
- // 给对象增加定时任务
- // @daily 每日运行一次
- // c.AddFunc("0 */1 * * * ?", Cron_WarningRateDay_Add)
- // c.AddFunc("0 */1 * * * ?", Cron_WarningRateMonth_Add)
- c.AddFunc("@daily", Cron_WarningRateDay_Add)
- c.AddFunc("@monthly", Cron_WarningRateMonth_Add)
- // 启动定时任务
- c.Start()
- defer c.Stop()
- // 查询语句,阻塞,让main函数不退出,保持程序运行
- select {}
- }
- // 保存每天的设备报警率
- func Cron_WarningRateDay_Add() {
- T_date := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
- logs.Info("开始统计" + T_date + "设备报警率")
- // 获取管理员用户列表
- adminList := Account.Read_Admin_List_All()
- // 获取内部用户关联的pid
- for _, admin := range adminList {
- // 内部用户已绑定公司,* 绑定所有公司
- if len(admin.T_pids) == 0 || admin.T_pids == "*" {
- continue
- }
- // 获取排除后的pid
- T_pids := lib.SplitStringToIntIds(admin.T_pids, "P")
- if len(T_pids) == 0 {
- continue
- }
- pids := Account.ReadCompanyWarningIds_T_pids(T_pids)
- if len(pids) == 0 {
- continue
- }
- // 获取探头数量
- dsCnt := Device.Read_DeviceSensorCount_ByT_pids(pids)
- // 获取当天报警数量
- warningCount := Warning.Read_WarningCount_byDay(pids)
- T_rate, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(warningCount)/float64(dsCnt)), 64) // 保留2位小数
- // 保存设备报警率
- warningRate := Warning.WarningRate{
- T_uuid: admin.T_uuid,
- T_date: T_date,
- T_type: Warning.WarningRateDay,
- T_device_num: dsCnt,
- T_warning_num: warningCount,
- T_rate: float32(T_rate),
- }
- Warning.Add_WarningRate(warningRate)
- }
- }
- // 保存每月的设备报警率
- func Cron_WarningRateMonth_Add() {
- now := time.Now()
- // fixme 测试
- //now := time.Now().AddDate(0, 1, 0)
- T_date := now.AddDate(0, -1, 0).Format("2006-01")
- firstOfMonth := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, time.Local)
- startTime := firstOfMonth.AddDate(0, -1, 0).Format("2006-01-02") // 上月第一天
- endTime := firstOfMonth.AddDate(0, 0, -1).Format("2006-01-02") // 上月最后一天
- logs.Info("开始统计" + T_date + "设备报警率")
- logs.Info("开始时间" + startTime)
- logs.Info("结束时间" + endTime)
- rateAvgList := Warning.Read_WarningRateAvg_Month(startTime, endTime)
- // 获取内部用户关联的pid
- for _, rate := range rateAvgList {
- T_rate, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(rate.T_warning_num)/float64(rate.T_device_num)), 64) // 保留2位小数
- // 保存设备报警率
- warningRate := Warning.WarningRate{
- T_uuid: rate.T_uuid,
- T_date: T_date,
- T_type: Warning.WarningRateMonth,
- T_device_num: rate.T_device_num,
- T_warning_num: rate.T_warning_num,
- T_rate: float32(T_rate),
- }
- Warning.Add_WarningRate(warningRate)
- }
- }
|