package InfoCollection import ( "ColdVerify_server/lib" "ColdVerify_server/logs" _ "github.com/astaxie/beego/cache/redis" "github.com/beego/beego/v2/adapter/orm" orm2 "github.com/beego/beego/v2/client/orm" _ "github.com/go-sql-driver/mysql" "strconv" ) // 模板 type InfoTemplateClass struct { Id int `orm:"column(ID);size(11);auto;pk"` T_fid int `orm:"size(200);null"` // 父级分类ID,主类 为 0 T_name string `orm:"size(256);null"` // 标题 T_path string `orm:"size(256);null"` // 路径 T_State int `orm:"size(2);default(1)"` // 0 删除(伪删除) 1 正常 Children []InfoTemplateClass `orm:"-"` } func (t *InfoTemplateClass) TableName() string { return "info_template_class" // 数据库名称 // ************** 替换 FormulaList ************** } func init() { //注册模型 orm.RegisterModel(new(InfoTemplateClass)) } type InfoTemplateClass_R struct { Id int T_fid int // 项目 id, InfoTemplateClass.ID T_name string // 名称 T_State int // 0 删除(伪删除) 1 正常(已交付) 2 开发中 Children []InfoTemplateClass `orm:"-"` } func Read_InfoTemplateClass_List_ALL() (maps []InfoTemplateClass) { o := orm.NewOrm() qs := o.QueryTable(new(InfoTemplateClass)) qs.OrderBy("Id").All(&maps) return maps } func InfoTemplateClassListToMap(A []InfoTemplateClass) map[string]string { maps := make(map[string]string, len(A)) for _, v := range A { maps[strconv.Itoa(v.Id)] = v.T_name } return maps } // 获取 ById func Read_InfoTemplateClass_ById(id int) (r InfoTemplateClass, err error) { o := orm.NewOrm() r = InfoTemplateClass{Id: id, T_State: 1} err = o.Read(&r, "Id", "T_State") if err != nil { logs.Error(lib.FuncName(), err) return r, err } return r, err } // 添加 func Add_InfoTemplateClass(var_ InfoTemplateClass) (id int64, err error) { o := orm.NewOrm() o.Begin() id, err = o.Insert(&var_) if err != nil { o.Rollback() return id, err } T_Path := "/0/" + strconv.Itoa(int(id)) + "/" if var_.T_fid != 0 { class_r, err := Read_InfoTemplateClass_ById(var_.T_fid) if err != nil { o.Rollback() return id, err } T_Path = class_r.T_path + strconv.Itoa(int(id)) + "/" } var_.Id = int(id) var_.T_path = T_Path _, err = o.Update(&var_, "T_path") if err != nil { o.Rollback() return id, err } o.Commit() return id, err } // 修改 func Update_InfoTemplateClass(r InfoTemplateClass, cols ...string) bool { o := orm.NewOrm() if num, err := o.Update(&r, cols...); err == nil { logs.Println("Number of records updated in database:", num) return true } return false } // 批量删除 func Delete_InfoTemplateClass_ByIds(ids []int) bool { o := orm.NewOrm() o.Begin() for _, id := range ids { v := InfoTemplateClass{Id: id, T_State: 0} if _, err := o.Update(&v, "T_State"); err != nil { o.Rollback() return false } } o.Commit() return true } // 获取列表 func Read_InfoTemplateClass_List() (r []InfoTemplateClass) { o := orm.NewOrm() // 也可以直接使用 Model 结构体作为表名 var map_r []InfoTemplateClass qs := o.QueryTable(new(InfoTemplateClass)) cond := orm.NewCondition() cond1 := cond.And("T_State", 1) qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&map_r) r = InfoTemplateClass_Call(map_r, 0) return r } func InfoTemplateClass_Call(list []InfoTemplateClass, parentId int) []InfoTemplateClass { res := make([]InfoTemplateClass, 0) for _, v := range list { if v.T_fid == parentId { v.Children = InfoTemplateClass_Call(list, v.Id) res = append(res, v) } } return res } // 通过T_pid查询所有子id func ReadInfoTemplateClassIds_T_path(T_path string) (InfoTemplateClassIds []int) { o := orm.NewOrm() qs := o.QueryTable(new(InfoTemplateClass)) var InfoTemplateClassList []InfoTemplateClass qs.Filter("T_path__startswith", T_path).Filter("T_State", 1).All(&InfoTemplateClassList) for _, v := range InfoTemplateClassList { InfoTemplateClassIds = append(InfoTemplateClassIds, v.Id) } return InfoTemplateClassIds }