ColdVerify_server/models/VerifyTemplate/VerifyTemplate.go

package VerifyTemplate

import (
	"ColdVerify_server/lib"
	"ColdVerify_server/logs"
	"encoding/json"
	"github.com/beego/beego/v2/adapter/orm"
	orm2 "github.com/beego/beego/v2/client/orm"
	_ "github.com/go-sql-driver/mysql"
	"strings"
	"time"
)

// 模版
type VerifyTemplate struct {
	Id      int `orm:"column(ID);size(11);auto;pk"`
	T_class int `orm:"size(8);default(0)"` // 分类

	T_VerifyTemplate_id string `orm:"size(256);null"`     // 标题
	T_name              string `orm:"size(256);null"`     // 标题
	T_sort              int    `orm:"size(2);default(1)"` // 排

	T_scheme    string `orm:"size(256);null"` // 方案模版
	T_reporting string `orm:"size(256);null"` // 报告模版
	T_marking   string `orm:"size(256);null"` // 标识模版
	T_inspect   string `orm:"size(256);null"` // 自检
	T_cover     string `orm:"size(256);null"` // 封面

	T_deploy string `orm:"type(text);null"` // 布点选项  格式 布点内容1/布点范围1|布点内容2|布点内容3

	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 保存时都会对时间自动更新
}

// 模版布点
type VerifyTemplateDeploy struct {
	T_name  string // 布点内容
	T_scope string // 布点范围
}

type VerifyTemplate_R struct {
	T_VerifyTemplate_id string `orm:"size(256);null"` // 标题
	T_name              string `orm:"size(256);null"` // 标题
	T_sort              int    `orm:"size(200);null"` // 分类

	T_scheme      string                 // 方案模版
	T_reporting   string                 // 报告模版
	T_inspect     string                 // 自检
	T_marking     string                 // 标识模版
	T_cover       string                 // 封面
	T_deploy      string                 // 布点选项  格式 布点内容1|布点内容2|布点内容3
	T_deploy_list []VerifyTemplateDeploy // 布点选项  格式 布点内容1|布点内容2|布点内容3
}

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

func init() {

	//注册模型
	orm.RegisterModel(new(VerifyTemplate))

}

// -------------------------------------------------------------
func VerifyTemplateToVerifyTemplate_R(T VerifyTemplate) (T_r VerifyTemplate_R) {
	T_r.T_VerifyTemplate_id = T.T_VerifyTemplate_id
	T_r.T_name = T.T_name
	T_r.T_sort = T.T_sort
	T_r.T_scheme = T.T_scheme
	T_r.T_reporting = T.T_reporting
	T_r.T_inspect = T.T_inspect
	T_r.T_marking = T.T_marking
	T_r.T_cover = T.T_cover
	var deployList []VerifyTemplateDeploy
	if len(T.T_deploy) > 0 {
		if strings.Contains(strings.Trim(T.T_deploy, "|"), "|") {
			deploys := strings.Split(T.T_deploy, "|")
			for _, v := range deploys {
				deployList = append(deployList, VerifyTemplateDeploy{T_name: v})
			}
		} else {
			err := json.Unmarshal([]byte(T.T_deploy), &deployList)
			if err != nil {
				logs.Error("布点 JSON 反序列化失败:", err)
			}
		}
	}
	for _, deploy := range deployList {
		T_r.T_deploy += deploy.T_name + "|"
	}
	T_r.T_deploy_list = deployList

	return T_r
}

// ---------------- 特殊方法 -------------------

// 获取 ById
func Read_VerifyTemplate_ById(id int) (r VerifyTemplate, is bool) {
	o := orm.NewOrm()
	r = VerifyTemplate{Id: id}
	err := o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return r, false
	}
	return r, true
}

// 获取 By
func Read_VerifyTemplate(T_VerifyTemplate_id string) (r VerifyTemplate, is bool) {

	o := orm.NewOrm()
	qs := o.QueryTable(new(VerifyTemplate))
	err := qs.Filter("T_VerifyTemplate_id", T_VerifyTemplate_id).One(&r)
	if err != nil {
		return r, false
	}

	return r, true
}

func Read_VerifyTemplate_T_name(T_name string) (r VerifyTemplate, is bool) {

	o := orm.NewOrm()
	qs := o.QueryTable(new(VerifyTemplate))
	err := qs.Filter("T_name", T_name).One(&r)
	if err != nil {
		return r, false
	}

	return r, true
}

// 添加
func Add_VerifyTemplate(r VerifyTemplate) (T_VerifyTemplate_id string, is bool) {
	o := orm.NewOrm()

	// 生成编号
	rand_x := 0
	for true {
		r.T_VerifyTemplate_id = lib.GetRandstring(4, "", int64(rand_x)) // 1,336,336
		err := o.Read(&r, "T_VerifyTemplate_id")                        // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
		if err != nil {
			break
		}
		rand_x += 1
	}

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

	return r.T_VerifyTemplate_id, true
}

// 删除
func Delete_VerifyTemplate(v VerifyTemplate) bool {
	o := orm.NewOrm()
	if num, err := o.Delete(&v); err == nil {
		logs.Println("Number of records deleted in database:", num)
	} else {
		return false
	}

	return true
}

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

// 获取列表
func Read_VerifyTemplate_List(T_class int, T_name string, page int, page_z int) ([]VerifyTemplate_R, int64) {

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

	cond := orm.NewCondition()
	cond1 := cond.And("T_class", T_class).And("T_name__icontains", T_name) // .AndNot("status__in", 1).Or("profile__age__gt", 2000)

	qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("T_name").All(&r)
	cnt, _ := qs.SetCond((*orm2.Condition)(cond1)).Count()
	// 转换
	var VerifyTemplate_r []VerifyTemplate_R
	for _, v := range r {
		VerifyTemplate_r = append(VerifyTemplate_r, VerifyTemplateToVerifyTemplate_R(v))
	}

	return VerifyTemplate_r, cnt
}