Cold_WordOrder/models/WorkOrder/WorkOrder.go

package WorkOrder

import (
	"Cold_WorkOrder/conf"
	"Cold_WorkOrder/logs"
	"fmt"
	"github.com/beego/beego/v2/adapter/orm"
	orm2 "github.com/beego/beego/v2/client/orm"
	_ "github.com/go-sql-driver/mysql"
	"time"
)

// 模板
type WorkOrder struct {
	Id           int       `orm:"column(ID);size(11);auto;pk"`
	T_uuid       string    `orm:"size(50);null"`                                         // 提交人uuid
	T_Company_Id int       `orm:"size(11);null"`                                         // 提交人公司id
	T_title      string    `orm:"size(256);null"`                                        // 标题
	T_text       string    `orm:"type(text);null"`                                       // 详情
	T_Status     int       `orm:"size(2);default(1)"`                                    // 状态 1 未解决 2 已解决
	T_State      int       `orm:"size(2);default(1)"`                                    // 0 删除   1 正常
	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 WorkOrder_R struct {
	Id       int    // ID
	T_title  string //
	T_text   string // 详情
	T_Status int    // 状态 0 未解决 1 已解决
}

type WorkOrder_ struct {
	Id             int    // ID
	T_uuid         string // 用户uuid
	T_name         string // 用户名
	T_Company_name string // 公司名
	T_title        string // 标题
	T_text         string // 详情
	T_Status       int    // 状态 0 未解决 1 已解决
}

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

func init() {
	//注册模型
	orm.RegisterModel(new(WorkOrder))

}

func WorkOrderToWorkOrder_R(t WorkOrder) (r WorkOrder_R) {
	r.Id = t.Id
	r.T_title = t.T_title
	r.T_text = t.T_text
	r.T_text = t.T_text
	return
}

func WorkOrderToWorkOrder_(t WorkOrder, CompanyMap map[int]string, AdminMap map[string]string) (r WorkOrder_) {
	if name, ok := CompanyMap[t.T_Company_Id]; ok {
		r.T_Company_name = name
	}
	if name, ok := AdminMap[t.T_uuid]; ok {
		r.T_name = name
	}
	r.Id = t.Id
	r.T_uuid = t.T_uuid
	r.T_title = t.T_title
	r.T_text = t.T_text
	r.T_Status = t.T_Status
	return
}

// ---------------- 特殊方法 -------------------
// 获取 ById
func Read_WorkOrder_ById(id int) (r WorkOrder, err error) {
	o := orm.NewOrm()
	r = WorkOrder{Id: id}
	err = o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		logs.Println(err)
		return r, err

	}
	return r, nil
}

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

	id, err = o.Insert(&m)
	if err != nil {
		fmt.Println(err)
	}
	return id, err
}

// 修改
func Update_WorkOrder(r WorkOrder, 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 Delete_WorkOrder_ById(r WorkOrder) (err error) {
	r.T_State = 0
	Update_WorkOrder(r, "T_State")
	return
}

// 删除
func Delete_WorkOrder(m WorkOrder) (err error) {
	o := orm.NewOrm()
	if num, err := o.Delete(&m); err == nil {
		fmt.Println("Number of records deleted in database:", num)
	}
	return
}

// 获取列表
func Read_WorkOrder_List(T_Status int, page int, page_z int, CompanyIds []int, CompanyMap map[int]string, AdminMap map[string]string) (t []WorkOrder_, cnt int64) {
	o := orm.NewOrm()

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

	cond := orm.NewCondition()
	cond1 := cond.And("T_State", 1)
	if T_Status > 0 {
		cond1 = cond1.And("T_Status", T_Status)
	}
	if len(CompanyIds) > 0 {
		cond1 = cond1.And("T_Company_Id__in", CompanyIds)
	}

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

	for _, v := range r {
		t = append(t, WorkOrderToWorkOrder_(v, CompanyMap, AdminMap))
	}
	return t, cnt
}

// 获取用户列表
func Read_UserWorkOrder_List(T_uuid string, page int, page_z int) (t []WorkOrder_R, cnt int64) {
	o := orm.NewOrm()

	// 也可以直接使用 Model 结构体作为表名
	qs := o.QueryTable(new(WorkOrder))
	var r []WorkOrder
	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_uuid", T_uuid).And("T_State", 1)

	qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("Id").All(&r)
	cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
	for _, v := range r {
		t = append(t, WorkOrderToWorkOrder_R(v))
	}
	return t, cnt
}