ColdVerify_server/models/Task/TaskCopy.go

package Task

import (
	"ColdVerify_server/lib"
	"ColdVerify_server/logs"
	"github.com/beego/beego/v2/adapter/orm"
	orm2 "github.com/beego/beego/v2/client/orm"
	"time"
)

// 创建数据库  Device.CREATE_TaskData("")
func CREATE_TaskDataCopy(alias_name, T_copy_id string) bool {
	o := orm2.NewOrmUsingDB(alias_name)

	sql := "DROP TABLE IF EXISTS `z_task_data_copy_" + T_copy_id + "`"
	o.Raw(sql).Exec()

	sql = "CREATE TABLE IF NOT EXISTS `z_task_data_copy_" + T_copy_id + "` ( " +
		"  `ID` int(11) NOT NULL AUTO_INCREMENT," +
		"	`t_sn` varchar(256) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL," +
		"	`t_id` varchar(256) DEFAULT NULL," +
		"	`t_t` float(6, 1) NULL DEFAULT NULL," +
		"	`t_rh` float(6, 1) NULL DEFAULT NULL," +
		"	`t_time` datetime(0) NULL DEFAULT NULL," +
		"	PRIMARY KEY (`ID`) USING BTREE," +
		"   KEY `t_sn` (`t_sn`)," +
		"   KEY `t_id` (`t_id`)," +
		"   KEY `t_time` (`t_time`)" +
		") ENGINE = InnoDB AUTO_INCREMENT = 5 CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;"

	_, err := o.Raw(sql).Exec()
	if err != nil {
		return false
	}

	return true
}

type TaskCopy struct {
	Id         int       `orm:"column(ID);size(11);auto;pk"`
	T_task_id  string    `orm:"size(256);null"`                                        // 任务ID
	T_copy_id  string    `orm:"size(256);null"`                                        // 复制id
	T_time     string    `orm:"size(256);null"`                                        // 存档时间
	T_State    int       `orm:"size(2);default(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 保存时都会对时间自动更新
}

func init() {
	orm.RegisterModel(new(TaskCopy))
}

// 添加
func Add_TaskCopy(r TaskCopy) (string, bool) {
	o := orm.NewOrm()

	var taskCopy TaskCopy
	// 查询是否存在报告生成时的数据存档
	qs := o.QueryTable(new(TaskCopy))
	err := qs.Filter("T_task_id", r.T_task_id).Filter("T_State", 2).One(&taskCopy)
	if err != nil && err.Error() != orm.ErrNoRows.Error() {
		return "", false
	}
	// 删除之前的存档
	if taskCopy.Id > 0 {
		// 删除之前的存档记录和数据
		taskCopy.T_State = 0
		_, err = o.Update(&taskCopy, "T_State")
		if err != nil {
			logs.Error(lib.FuncName(), err)
			return "", false
		}

		sql := "DROP TABLE IF EXISTS `z_task_data_copy_" + taskCopy.T_copy_id + "`"
		o.Raw(sql).Exec()
	}

	// 生成编号
	rand_x := 0
	for true {
		r.T_copy_id = lib.GetRandstring(12, "abcdefghijklmnopqrstuvwxyz0123456789", int64(rand_x))
		err = o.Read(&r, "T_copy_id")
		if err != nil {
			break
		}
		rand_x += 1
	}

	_, err = o.Insert(&r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return "", false
	}

	return r.T_copy_id, true
}

// 数据存档
func TaskData_Arhiving(T_task_id, T_copy_id string) bool {

	o := orm.NewOrm()

	sql := "INSERT INTO `z_task_data_copy_" + T_copy_id + "`(`t_sn`, `t_id`, `t_t`, `t_rh`, `t_time`)" +
		" SELECT `t_sn`, `t_id`, `t_t`, `t_rh`, `t_time` FROM `z_task_data_" + T_task_id + "`"
	_, err := o.Raw(sql).Exec()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return false
	}
	return true
}