WorkOrder.go 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
  1. package WorkOrder
  2. import (
  3. "Cold_WorkOrder/conf"
  4. "Cold_WorkOrder/lib"
  5. "Cold_WorkOrder/logs"
  6. "Cold_WorkOrder/models/Account"
  7. "fmt"
  8. "github.com/beego/beego/v2/adapter/orm"
  9. orm2 "github.com/beego/beego/v2/client/orm"
  10. _ "github.com/go-sql-driver/mysql"
  11. "time"
  12. )
  13. // 模板
  14. type WorkOrder struct {
  15. Id int `orm:"column(ID);size(11);auto;pk"`
  16. T_uuid string `orm:"size(50);null"` // 提交人uuid
  17. T_pid int `orm:"index;size(256);null"` // Account.Company 绑定公司
  18. T_title string `orm:"size(256);null"` // 标题
  19. T_text string `orm:"type(text);null"` // 详情
  20. T_State int `orm:"size(2);default(1)"` // 0 删除 1 已解决 2 未解决
  21. CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间
  22. UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"` //auto_now 每次 model 保存时都会对时间自动更新
  23. }
  24. type WorkOrder_R struct {
  25. Id int // ID
  26. T_title string // 标题
  27. T_text string // 详情
  28. T_State int // 0 删除 1 已解决 2 未解决
  29. }
  30. type WorkOrder_ struct {
  31. Id int // ID
  32. T_uuid string // 用户uuid
  33. T_name string // 用户名
  34. T_phone string // 电话
  35. T_Company_name string // 公司名
  36. T_title string // 标题
  37. T_text string // 详情
  38. T_State int // 状态 0 删除 1 已解决 2 未解决
  39. }
  40. func (t *WorkOrder) TableName() string {
  41. return "work_order" // 数据库名称 // ************** 替换 FormulaList **************
  42. }
  43. func init() {
  44. //注册模型
  45. orm.RegisterModel(new(WorkOrder))
  46. }
  47. func WorkOrderToWorkOrder_R(t WorkOrder) (r WorkOrder_R) {
  48. r.Id = t.Id
  49. r.T_title = t.T_title
  50. r.T_text = t.T_text
  51. r.T_State = t.T_State
  52. return
  53. }
  54. func WorkOrderToWorkOrder_(t WorkOrder) (r WorkOrder_) {
  55. r.T_Company_name = Account.Read_Company_T_name_Get(t.T_pid)
  56. r.T_name = Account.Read_User_T_name_Get(t.T_uuid)
  57. r.T_phone = Account.Read_User_T_phone_Get(t.T_uuid)
  58. r.Id = t.Id
  59. r.T_uuid = t.T_uuid
  60. r.T_title = t.T_title
  61. r.T_text = t.T_text
  62. r.T_State = t.T_State
  63. return
  64. }
  65. // ---------------- 特殊方法 -------------------
  66. // 获取 ById
  67. func Read_WorkOrder_ById(id int) (r WorkOrder, err error) {
  68. o := orm.NewOrm()
  69. r = WorkOrder{Id: id}
  70. err = o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  71. if err != nil {
  72. logs.Println(err)
  73. return r, err
  74. }
  75. if r.T_State == 0 {
  76. return r, orm.ErrNoRows
  77. }
  78. return r, nil
  79. }
  80. // 添加
  81. func Add_WorkOrder(m WorkOrder) (id int64, err error) {
  82. o := orm.NewOrm()
  83. id, err = o.Insert(&m)
  84. if err != nil {
  85. fmt.Println(err)
  86. }
  87. return id, err
  88. }
  89. // 修改
  90. func Update_WorkOrder(r WorkOrder, cols ...string) bool {
  91. o := orm.NewOrm()
  92. if num, err := o.Update(&r, cols...); err == nil {
  93. fmt.Println("Number of records updated in database:", num)
  94. return true
  95. }
  96. return false
  97. }
  98. // 删除
  99. func Delete_WorkOrder_ById(r WorkOrder) (err error) {
  100. r.T_State = 0
  101. Update_WorkOrder(r, "T_State")
  102. return
  103. }
  104. // 删除
  105. func Delete_WorkOrder(m WorkOrder) (err error) {
  106. o := orm.NewOrm()
  107. if num, err := o.Delete(&m); err == nil {
  108. fmt.Println("Number of records deleted in database:", num)
  109. }
  110. return
  111. }
  112. // 获取列表
  113. func Read_WorkOrder_List(T_handle int, page int, page_z int, CompanyIds []int) (t []WorkOrder_, cnt int64) {
  114. o := orm.NewOrm()
  115. // 也可以直接使用 Model 结构体作为表名
  116. qs := o.QueryTable(new(WorkOrder))
  117. var offset int64
  118. if page_z == 0 {
  119. page_z = conf.Page_size
  120. }
  121. var r []WorkOrder
  122. if page <= 1 {
  123. offset = 0
  124. } else {
  125. offset = int64((page - 1) * page_z)
  126. }
  127. cond := orm.NewCondition()
  128. cond1 := cond.And("T_State__gt", 0)
  129. if T_handle == 1 {
  130. cond1 = cond1.And("T_State", 1)
  131. }
  132. if T_handle == 2 {
  133. cond1 = cond1.And("T_State", 2)
  134. }
  135. if len(CompanyIds) > 0 {
  136. cond1 = cond1.And("T_pid__in", CompanyIds)
  137. }
  138. qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("Id").All(&r)
  139. cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
  140. for _, v := range r {
  141. t = append(t, WorkOrderToWorkOrder_(v))
  142. }
  143. return t, cnt
  144. }
  145. // 获取用户列表
  146. func Read_UserWorkOrder_List(admin_r Account.User, T_pid int, page int, page_z int) (t []WorkOrder_R, cnt int64) {
  147. o := orm.NewOrm()
  148. // 也可以直接使用 Model 结构体作为表名
  149. qs := o.QueryTable(new(WorkOrder))
  150. var r []WorkOrder
  151. var offset int64
  152. if page_z == 0 {
  153. page_z = conf.Page_size
  154. }
  155. if page <= 1 {
  156. offset = 0
  157. } else {
  158. offset = int64((page - 1) * page_z)
  159. }
  160. cond := orm.NewCondition()
  161. cond1 := cond.And("T_State__gt", 0)
  162. if admin_r.T_pid > 0 {
  163. // 用户进入
  164. cond1 = cond1.And("T_uuid", admin_r.T_uuid)
  165. } else {
  166. //管理员进入
  167. cond1 = cond1.And("T_pid", T_pid)
  168. }
  169. qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("Id").All(&r)
  170. cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
  171. for _, v := range r {
  172. t = append(t, WorkOrderToWorkOrder_R(v))
  173. }
  174. return t, cnt
  175. }
  176. func Read_WorkOrder_ALL_T_State_Count(T_pids string, T_handle int, Time_start string) (cnt int64) {
  177. o := orm.NewOrm()
  178. qs := o.QueryTable(new(WorkOrder))
  179. cond := orm.NewCondition()
  180. cond1 := cond.And("T_State__gt", 0)
  181. if len(Time_start) > 0 {
  182. cond1 = cond1.And("CreateTime__gte", Time_start)
  183. }
  184. if len(T_pids) > 0 && T_pids != "*" {
  185. list := lib.SplitStringIds(T_pids, "P")
  186. cond1 = cond1.And("T_pid__in", list)
  187. }
  188. //不填或0:所有 1:已处理 2:未处理
  189. if T_handle == 1 {
  190. cond1 = cond1.And("T_State", 1)
  191. }
  192. if T_handle == 2 {
  193. cond1 = cond1.And("T_State", 2)
  194. }
  195. cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
  196. return cnt
  197. }