Cold_Api/models/Warning/Warning.go

package Warning

import (
	"Cold_Api/conf"
	"Cold_Api/controllers/lib"
	"fmt"
	"github.com/beego/beego/v2/adapter/orm"
	orm2 "github.com/beego/beego/v2/client/orm"
	_ "github.com/go-sql-driver/mysql"
	"strconv"
	"time"
)

// 模板
type Warning struct {
	Id        int       `orm:"column(ID);size(11);auto;pk"`
	T_pid     int       `orm:"index;size(256);null"`        // Account.Company 绑定公司
	T_tp      int       `orm:"index;size(200);null"`        // 报警类型   ->WarningList
	T_sn      string    `orm:"index;size(256);null"`        // 设备序列号
	T_D_name  string    `orm:"size(256);null"`              // 设备名称
	T_id      int       `orm:"index;size(200);null"`        // 传感器 ID
	T_DS_name string    `orm:"size(256);null"`              // 传感器名称
	T_Remark  string    `orm:"type(text);null"`             // 采集内容
	T_Ut      time.Time `orm:"index;type(timestamp);null;"` // 采集时间

	T_Text string `orm:"type(text);null"` // 处理备注
	T_Log  string `orm:"type(text);null"` // 通知日志
	T_Msid int64  `orm:"size(256);null"`  // 消息ID

	T_State    int       `orm:"size(2);1"`                                             // 0 删除   1 正常   2 未处理
	CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间
	UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`     //auto_now 每次 model 保存时都会对时间自动更新
}

/*
分表机制
|--------30天--------|
|-----8月---||--------7月---------||--------6月---------||--------5月---------|

每天备份 到对应 月份

*/

// 模板
type Warning_R struct {
	Id         int
	T_pid      int    // Account.Company 绑定公司
	T_tp       int    // 报警类型   ->WarningList
	T_sn       string // 设备序列号
	T_D_name   string // 设备名称
	T_id       int    // 传感器 ID
	T_DS_name  string // 传感器名称
	T_Remark   string // 采集内容
	T_Ut       string // 采集时间
	T_Text     string // 处理备注
	T_Log      string // 通知日志
	T_Msid     int64  // 消息ID
	T_State    int    // 0 删除   1 未处理   2 已处理
	CreateTime string // 创建时间
}

func (t *Warning) TableName() string {
	return "warning" // 数据库名称   // ************** 替换 FormulaList **************
}

func init() {
	//注册模型
	orm.RegisterModel(new(Warning))
	orm.Debug = true
}

// ---------------- 特殊方法 -------------------
func WarningToWarning_R(t Warning) (r Warning_R) {
	r.Id = t.Id
	r.T_pid = t.T_pid
	r.T_tp = t.T_tp
	r.T_sn = t.T_sn
	r.T_D_name = t.T_D_name
	r.T_id = t.T_id
	r.T_DS_name = t.T_DS_name
	r.T_Remark = t.T_Remark
	r.T_Ut = t.T_Ut.Format("2006-01-02 15:04:05")
	r.T_Text = t.T_Text
	r.T_Log = t.T_Log
	r.T_Msid = t.T_Msid
	r.T_State = t.T_State
	r.CreateTime = t.CreateTime.Format("2006-01-02 15:04:05")
	return r
}

// 获取 ById
func Read_Warning_ById(id int) (r Warning) {
	o := orm.NewOrm()
	r = Warning{Id: id}
	err := o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		fmt.Println(err)
	}
	return r
}

// 添加
func Add_Warning(m Warning) (id int64, err error) {
	o := orm.NewOrm()

	id, err = o.Insert(&m)
	return id, err
}

// 修改
func Update_Warning(r Warning, cols ...string) bool {
	o := orm.NewOrm()
	if num, err := o.Update(&r, cols...); err == nil {
		fmt.Println("Number of records updated in database:", num)
		return true
	}
	return false
}

// 修改
func Update_Warning_ById(m Warning) (err error) {
	o := orm.NewOrm()
	v := Warning{Id: m.Id}
	// ascertain id exists in the database
	if err = o.Read(&v); err == nil {
		var num int64
		m.T_State = 2
		if num, err = o.Update(&m, "T_Text"); err == nil {
			fmt.Println("Number of records updated in database:", num)
		}
	}
	return
}

// 修改
func Update_Warning_Delete(m Warning) (err error) {
	o := orm.NewOrm()
	v := Warning{Id: m.Id}
	// ascertain id exists in the database
	if err = o.Read(&v); err == nil {
		v.T_State = 0
		if num, err := o.Update(&v, "T_State"); err == nil {
			fmt.Println("Number of records updated in database:", num)
		}
	}
	return
}

// 获取列表
func Read_Warning_1(T_pid int, page int, T_sn string, T_Name string, Time_start_ string, Time_end_ string) (r []Warning, cnt int64) {

	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名

	qs := o.QueryTable(new(Warning))
	var offset int64
	var page_z int64
	page_z = int64(conf.Page_size)
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1)) * page_z
	}
	if page == 9999 {
		page_z = 9999
		offset = 0
	}

	cond := orm.NewCondition()

	cond1 := cond.And("T_State__gt", 0)

	if T_pid > 0 {
		cond1 = cond.AndCond(cond1).And("T_pid", T_pid)
	}
	if len(T_Name) > 1 {
		cond1 = cond.AndCond(cond1).AndCond(cond.Or("T_Name__icontains", T_Name).Or("T_Title__icontains", T_Name).Or("T_Addr__icontains", T_Name).Or("T_sn__icontains", T_Name))
	}

	if len(T_sn) > 1 {
		cond1 = cond.AndCond(cond1).And("T_sn__icontains", T_sn)
	}

	if len(Time_start_) > 1 {
		cond1 = cond.AndCond(cond1).And("T_Ut__gte", Time_start_)
	}
	if len(Time_end_) > 1 {
		cond1 = cond.AndCond(cond1).And("T_Ut__lte", Time_end_)
	}

	qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&r)
	cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
	return r, cnt

	//if len(Time_start_) > 1 && len(Time_end_) > 1 {
	//	fmt.Println(Time_start_, Time_end_)
	//	qs.Limit(page_z, offset).Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__lte", Time_end_).Filter("T_Ut__gte", Time_start_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).OrderBy("-Id").All(&r)
	//	cnt, _ = qs.Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__lte", Time_end_).Filter("T_Ut__gte", Time_start_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).Count()
	//	return r, cnt
	//}
	//
	//if len(Time_start_) > 1 {
	//	qs.Limit(page_z, offset).Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__gte", Time_start_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).OrderBy("-Id").All(&r)
	//	cnt, _ = qs.Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__gte", Time_start_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).Count()
	//	return r, cnt
	//} else if len(Time_end_) > 1 {
	//	qs.Limit(page_z, offset).Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__lte", Time_end_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).OrderBy("-Id").All(&r)
	//	cnt, _ = qs.Filter("T_Bind__icontains", T_Bind).Filter("T_Ut__lte", Time_end_).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).Count()
	//	return r, cnt
	//}
	//
	//qs.Limit(page_z, offset).Filter("T_Bind__icontains", T_Bind).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).OrderBy("-Id").All(&r)
	//cnt, _ = qs.Filter("T_Bind__icontains", T_Bind).Filter("T_Name__icontains", T_Name).Filter("T_Addr__icontains", T_Name).Filter("T_sn__icontains", T_Name).Filter("T_Title__icontains", T_Name).Count()
	//
	//return r, cnt
}

// 获取列表
func Read_Warning(T_pid int, T_sn string, T_id string, T_title string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int) (r []Warning_R, cnt int64) {

	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名
	var map_r []Warning
	qs := o.QueryTable(new(Warning))
	var offset int64
	if page_z == 0 {
		page_z = conf.Page_size
	}
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * page_z)
	}

	cond := orm.NewCondition()

	cond1 := cond.And("T_State__gt", 0)

	if T_pid > 0 {
		cond1 = cond.AndCond(cond1).And("T_pid", T_pid)
	}
	if len(T_title) > 0 {
		cond1 = cond1.And("T_Title", T_title)
	}

	if len(T_sn) > 0 {
		cond1 = cond1.And("T_sn", T_sn)
	}
	if len(T_id) > 0 {
		cond1 = cond1.And("T_Id", T_id)
	}

	if len(Time_start_) > 0 {
		cond1 = cond1.And("T_Ut__gte", Time_start_)
	}
	if len(Time_end_) > 0 {
		cond1 = cond1.And("T_Ut__lte", Time_end_)
	}
	//不填或0：所有      1：已处理     2：未处理
	if T_handle == 1 {
		cond1 = cond1.And("T_Text__isnull", true)
	}
	if T_handle == 2 {
		cond1 = cond1.And("T_Text__isnull", false).And("T_State", 2)
	}

	qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_Ut").All(&map_r)
	cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()

	for _, v := range map_r {
		r = append(r, WarningToWarning_R(v))
	}

	return r, cnt

}

// 获取列表
func Read_Warning_Backups(T_pid int, T_year string, T_month string, T_sn string, T_id string, T_title string, T_handle int, Time_start_ string, Time_end_ string, page int, page_z int) (r []Warning_R, cnt int64) {

	o := orm.NewOrm()
	var maps []Warning
	var maps_z []orm2.ParamsList
	// 也可以直接使用 Model 结构体作为表名
	// Warning_2022_07
	if len(T_month) == 1 {
		T_month = "0" + T_month
	}
	Wtab := "warning_" + T_year + "_" + T_month

	var offset int
	if page_z == 0 {
		page_z = conf.Page_size
	}
	if page <= 1 {
		offset = 0
	} else {
		offset = (page - 1) * page_z
	}

	sql_WHERE := ""

	cond := orm.NewCondition()

	cond1 := cond.And("T_State__gt", 0)
	sql_WHERE += " t__state > 0 "

	if T_pid > 0 {
		sql_WHERE += " AND t_pid = " + strconv.Itoa(T_pid)
	}
	if len(T_title) > 0 {
		//cond1 = cond1.And("T_Title", T_title)
		T_title = lib.ReplaceSQL(T_title)
		sql_WHERE += " AND t__title = '" + T_title + "'"
	}

	if len(T_sn) > 0 {
		//cond1 = cond1.And("T_sn", T_sn)
		T_sn = lib.ReplaceSQL(T_sn)
		sql_WHERE += " AND t_sn = '" + T_sn + "'"
	}
	if len(T_id) > 0 {
		//cond1 = cond1.And("T_Id", T_id)
		T_id = lib.ReplaceSQL(T_id)
		sql_WHERE += " AND t__id = " + T_id
	}

	if len(Time_start_) > 0 {
		//cond1 = cond1.And("T_Ut__gte", Time_start_)
		Time_start_ = lib.ReplaceSQL(Time_start_)
		sql_WHERE += " AND t__ut >= " + Time_start_
	}
	if len(Time_end_) > 0 {
		//cond1 = cond1.And("T_Ut__lte", Time_end_)
		Time_end_ = lib.ReplaceSQL(Time_end_)
		sql_WHERE += " AND t__ut <= " + Time_end_
	}
	//不填或0：所有      1：已处理     2：未处理
	if T_handle == 1 {
		//cond1 = cond1.And("T_Text__isnull", true)
		sql_WHERE += " AND t__text IS NULL"
	}
	if T_handle == 2 {
		cond1 = cond1.And("T_Text__isnull", false).And("T_State", 2)
		sql_WHERE += " AND t__text IS NOT NULL AND t__state = 2 "
	}

	// -------------

	sql := "SELECT COUNT(ID) FROM " + Wtab + " WHERE " + sql_WHERE
	fmt.Println(sql)
	_, err := o.Raw(sql).ValuesList(&maps_z)
	if err != nil {
		return r, 0
	}
	if len(maps_z) == 0 {
		return r, 0
	}
	//fmt.Println("maps_z；",maps_z[0][0])
	sql = "SELECT ID ,t_sn ,t__id ,t__name ,t__msid ,t__t ,t__r_h ,t__title ,t__addr ,t__remark ,t__ut ,t__state ,create_time ,update_time ,t__bind ,t__text ,t__handel ,t__log " +
		"FROM " + Wtab + " WHERE " + sql_WHERE + " ORDER BY t__ut DESC "
	if page_z != 9999 {
		sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(page_z)
	}

	fmt.Println(sql)
	_, err = o.Raw(sql).QueryRows(&maps)

	//value, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", cnt), 64)
	//key, _ := strconv.Atoi(maps_z[0][0].(string))
	key, _ := strconv.ParseInt(maps_z[0][0].(string), 10, 64)

	for _, v := range maps {
		r = append(r, WarningToWarning_R(v))
	}

	return r, key

}

func Read_Warning_ALL_T_Bind_TIME_1d_Count(T_pid int, T_sn string) int {
	o := orm.NewOrm()
	var maps_z []orm2.ParamsList

	now := time.Now()
	// 一天前
	//d, _ := time.ParseDuration("-24h")
	//now = now.Add(d)

	timeStr := now.Format("2006-01-02") + " 00:00:00"
	timeStr1 := now.Format("2006-01-02") + " 23:59:59"

	sql := "SELECT * FROM warning WHERE t__title NOT LIKE '%恢复%' AND t__title NOT LIKE '%任务%' AND t__ut > '" + timeStr + "' AND t__ut < '" + timeStr1 + "' AND t_pid = '" + strconv.Itoa(T_pid) + "%' AND t_sn  LIKE '%" + T_sn + "%' group by t_sn"
	fmt.Println(sql)
	_, err := o.Raw(sql).ValuesList(&maps_z)
	if err != nil {
		return 0
	}
	if len(maps_z) == 0 {
		return 0
	}

	//value, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", cnt), 64)

	return len(maps_z)
}