Device.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. package Device
  2. import (
  3. "fmt"
  4. _ "github.com/astaxie/beego/cache/redis"
  5. "github.com/beego/beego/v2/adapter/orm"
  6. _ "github.com/go-sql-driver/mysql"
  7. "time"
  8. )
  9. // 模板
  10. type Device struct {
  11. Id int `orm:"column(ID);size(11);auto;pk"`
  12. T_pid int `orm:"index;size(256);"` // Account.Company 绑定公司
  13. T_sn string `orm:"size(256);"` // 设备序列号
  14. T_tab string `orm:"size(256);"` // 设备标签
  15. CreateTime time.Time `orm:"column(create_time);type(timestamp);auto_now_add"` //auto_now_add 第一次保存时才设置时间
  16. UpdateTime time.Time `orm:"column(update_time);type(timestamp);auto_now"` //auto_now 每次 model 保存时都会对时间自动更新
  17. }
  18. type Device_R struct {
  19. T_tab string // 设备标签
  20. T_sn []string // 设备序列号
  21. }
  22. func (t *Device) TableName() string {
  23. return "device" // 数据库名称 // ************** 替换 FormulaList **************
  24. }
  25. func init() {
  26. //注册模型
  27. orm.RegisterModel(new(Device))
  28. }
  29. // ---------------- 特殊方法 -------------------
  30. // 获取 ById
  31. func Read_Device_ById(id int) (r Device, err error) {
  32. o := orm.NewOrm()
  33. r = Device{Id: id}
  34. err = o.Read(&r)
  35. if err != nil {
  36. fmt.Println(err)
  37. }
  38. return r, err
  39. }
  40. // 添加
  41. func Add_Device(m Device) (id int64, err error) {
  42. o := orm.NewOrm()
  43. id, err = o.Insert(&m)
  44. if err != nil {
  45. fmt.Println(err)
  46. }
  47. return id, err
  48. }
  49. // 修改
  50. func Update_Device(r Device, cols ...string) bool {
  51. o := orm.NewOrm()
  52. if num, err := o.Update(&r, cols...); err == nil {
  53. fmt.Println("Number of records updated in database:", num)
  54. return true
  55. }
  56. return false
  57. }
  58. // 删除
  59. func Delete_Device(m Device) bool {
  60. o := orm.NewOrm()
  61. if num, err := o.Delete(&m); err == nil {
  62. fmt.Println("Number of records deleted in database:", num)
  63. return true
  64. }
  65. return false
  66. }
  67. // 获取列表
  68. func Read_Device_BT_pid(T_pid int) []Device_R {
  69. o := orm.NewOrm()
  70. // 也可以直接使用 Model 结构体作为表名
  71. qs := o.QueryTable(new(Device))
  72. var r []Device
  73. qs.Filter("T_pid", T_pid).OrderBy("T_tab").All(&r)
  74. var Device_list map[string][]string // 泛型
  75. Device_list = make(map[string][]string)
  76. for _, v := range r {
  77. Device_list[v.T_tab] = append(Device_list[v.T_tab], v.T_sn)
  78. }
  79. Device_R_list := []Device_R{}
  80. for k, v := range Device_list {
  81. Device_R_list = append(Device_R_list, Device_R{T_tab: k, T_sn: v})
  82. }
  83. return Device_R_list
  84. }