Admin.go 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  1. package Account
  2. import (
  3. "Cold_Api/conf"
  4. "Cold_Api/models"
  5. "encoding/json"
  6. "fmt"
  7. "github.com/astaxie/beego/cache"
  8. _ "github.com/astaxie/beego/cache/redis"
  9. "github.com/beego/beego/v2/adapter/orm"
  10. orm2 "github.com/beego/beego/v2/client/orm"
  11. _ "github.com/go-sql-driver/mysql"
  12. "time"
  13. )
  14. type Admin struct {
  15. Id int `orm:"column(ID);size(11);auto;pk"`
  16. T_uuid string `orm:"size(256);null"` // 用户编号
  17. T_pid int `orm:"size(200);null"` // 绑定公司 ( 只有创建公司用户时添加,内部人员 为0)
  18. T_pids string `orm:"size(200);null"` // 绑定公司管理 Pid| 如 P1|P2
  19. T_power int `orm:"size(2);default(0)"` // 权限 (关联权限表)
  20. T_user string `orm:"size(256);null"` // 用户名 (唯一)
  21. T_pass string `orm:"size(256);null"` // MD5
  22. T_name string `orm:"size(256);null"` // 姓名
  23. T_phone string `orm:"size(256);null"` // 电话
  24. T_mail string `orm:"size(200);null"` // 邮箱
  25. T_wx string `orm:"size(256);null"` // 微信
  26. T_State int `orm:"size(2);default(1)"` // 0删除 1 正常
  27. CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now 每次 model 保存时都会对时间自动更新
  28. UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"` //auto_now_add 第一次保存时才设置时间
  29. }
  30. type Admin_R struct {
  31. T_uuid string // 用户编号
  32. T_pid int // 绑定公司 ( 只有创建公司用户时添加,内部人员 为0)
  33. T_pids string // 绑定公司管理 Pid| 如 P1|P2
  34. T_power int // 权限 (关联权限表)
  35. T_user string // 用户名 (唯一)
  36. T_name string // 姓名
  37. T_phone string // 电话
  38. T_mail string // 邮箱
  39. T_wx string // 微信
  40. }
  41. type Admin_ struct {
  42. T_name string // 姓名
  43. T_phone string // 电话
  44. }
  45. func AdminToAdmin_R(r Admin) (v Admin_R) {
  46. v.T_uuid = r.T_uuid
  47. v.T_pid = r.T_pid
  48. v.T_pids = r.T_pids
  49. v.T_power = r.T_power
  50. v.T_user = r.T_user
  51. v.T_name = r.T_name
  52. v.T_phone = r.T_phone
  53. v.T_mail = r.T_mail
  54. v.T_wx = r.T_wx
  55. return v
  56. }
  57. func AdminToAdmin_(r Admin) (v Admin_) {
  58. v.T_name = r.T_name
  59. v.T_phone = r.T_phone
  60. return v
  61. }
  62. func (t *Admin) TableName() string {
  63. return "admin" // 数据库名称 // ************** 替换 FormulaList **************
  64. }
  65. var redisCache_Admin cache.Cache
  66. func init() {
  67. //注册模型
  68. orm.RegisterModel(new(Admin))
  69. config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
  70. "redis_Cold_User_Admin", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
  71. fmt.Println(config)
  72. var err error
  73. redisCache_Admin, err = cache.NewCache("redis", config)
  74. if err != nil || redisCache_Admin == nil {
  75. errMsg := "failed to init redis"
  76. fmt.Println(errMsg, err)
  77. }
  78. }
  79. // ---------------- Redis -------------------
  80. // Redis_Set(m.T_sn,m) // Redis 更新缓存
  81. func Redis_Admin_Set(r Admin) (err error) {
  82. //json序列化
  83. str, err := json.Marshal(r)
  84. if err != nil {
  85. fmt.Print(err)
  86. return
  87. }
  88. err = redisCache_Admin.Put(r.T_uuid, str, 24*time.Hour)
  89. if err != nil {
  90. fmt.Println("set key:", r.T_uuid, ",value:", str, err)
  91. }
  92. return
  93. }
  94. // if r,is :=Redis_Get(T_sn);is{
  95. // return r,nil
  96. // }
  97. func Redis_Admin_Get(key string) (r Admin, is bool) {
  98. if redisCache_Admin.IsExist(key) {
  99. //println("找到key:",key)
  100. v := redisCache_Admin.Get(key)
  101. json.Unmarshal(v.([]byte), &r)
  102. return r, true
  103. }
  104. //println("没有 找到key:",key)
  105. return Admin{}, false
  106. }
  107. func Redis_Admin_DelK(key string) (err error) {
  108. err = redisCache_Admin.Delete(key)
  109. return
  110. }
  111. // ---------------- 特殊方法 -------------------
  112. // 添加
  113. func Add_Admin(m Admin) (id int64, err error) {
  114. o := orm.NewOrm()
  115. id, err = o.Insert(&m)
  116. if err != nil {
  117. fmt.Println(err)
  118. }
  119. m.Id = int(id)
  120. Redis_Admin_Set(m)
  121. return id, err
  122. }
  123. //// 修改
  124. //func Update_Delete_Admin_ByT_uuid(T_uuid string) (err error) {
  125. // o := orm.NewOrm()
  126. // v := Admin{T_uuid: T_uuid}
  127. // // ascertain id exists in the database
  128. // if err = o.Read(&v, "T_uuid"); err == nil {
  129. // var num int64
  130. // v.T_State = 0
  131. //
  132. // if num, err = o.Update(&v, "T_State"); err == nil {
  133. // fmt.Println("Number of records updated in database:", num)
  134. // }
  135. //
  136. // Redis_Admin_Set(v)
  137. // }
  138. // return err
  139. //}
  140. func Read_Admin_T_user(T_user string) (Admin, error) {
  141. o := orm.NewOrm()
  142. r := Admin{T_user: T_user, T_State: 1}
  143. err := o.Read(&r, "T_user", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  144. if err != nil {
  145. fmt.Println(err)
  146. }
  147. return r, err
  148. }
  149. func Read_AdminLogin_verification(T_user string, T_pass string) (error, Admin) {
  150. o := orm.NewOrm()
  151. r := Admin{T_user: T_user, T_pass: T_pass, T_State: 1}
  152. err := o.Read(&r, "T_user", "T_pass", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  153. if err != nil {
  154. fmt.Println(err)
  155. }
  156. Redis_Admin_Set(r) // Redis 更新缓存
  157. return err, r
  158. }
  159. // 修改
  160. func Update_Admin(m Admin, cols ...string) bool {
  161. o := orm.NewOrm()
  162. if num, err := o.Update(&m, cols...); err == nil {
  163. fmt.Println("Number of records updated in database:", num)
  164. Redis_Admin_Set(m) // Redis 更新缓存
  165. return true
  166. }
  167. return false
  168. }
  169. // 删除
  170. func Delete_Admin(v Admin) bool {
  171. o := orm.NewOrm()
  172. v.T_State = 0
  173. if num, err := o.Update(&v, "T_State"); err == nil {
  174. fmt.Println("Number of records updated in database:", num)
  175. } else {
  176. return false
  177. }
  178. Redis_Admin_DelK(v.T_uuid) // Redis 更新缓存
  179. return true
  180. }
  181. // 获取 ById
  182. func Read_Admin_ById(Id int) (r Admin, err error) {
  183. o := orm.NewOrm()
  184. r = Admin{Id: Id, T_State: 1}
  185. err = o.Read(&r, "Id", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  186. if err != nil {
  187. fmt.Println(err)
  188. }
  189. return r, err
  190. }
  191. // 获取 ByT_uuid
  192. func Read_Admin_ByUuid(T_uuid string) (r Admin, err error) {
  193. if r, is := Redis_Admin_Get(T_uuid); is {
  194. //println("Redis_Get OK")
  195. return r, nil
  196. }
  197. o := orm.NewOrm()
  198. r = Admin{T_uuid: T_uuid, T_State: 1}
  199. err = o.Read(&r, "T_uuid", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  200. if err != nil {
  201. fmt.Println(err)
  202. }
  203. return r, err
  204. }
  205. // 获取列表
  206. func Read_Admin_List(T_pid int, T_name string, page int, page_z int) (AdminList []Admin_R, cnt int64) {
  207. o := orm.NewOrm()
  208. // 也可以直接使用 Model 结构体作为表名
  209. qs := o.QueryTable(new(Admin))
  210. var maps []Admin
  211. var offset int64
  212. if page_z == 0 {
  213. page_z = conf.Page_size
  214. }
  215. if page <= 1 {
  216. offset = 0
  217. } else {
  218. offset = int64((page - 1) * page_z)
  219. }
  220. cond := orm.NewCondition()
  221. cond1 := cond.And("T_State", 1)
  222. if T_pid > 0 {
  223. cond1 = cond1.AndCond(cond.And("T_pid", T_pid))
  224. }
  225. if len(T_name) > 0 {
  226. cond1 = cond1.AndCond(cond.Or("T_name__icontains", T_name).Or("T_user__icontains", T_name))
  227. }
  228. qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&maps)
  229. cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()
  230. for _, v := range maps {
  231. AdminList = append(AdminList, AdminToAdmin_R(v))
  232. }
  233. return AdminList, cnt
  234. }
  235. func Read_Admin_List_T_pids(T_pids string) (AdminList []Admin_) {
  236. o := orm.NewOrm()
  237. // 也可以直接使用 Model 结构体作为表名
  238. qs := o.QueryTable(new(Admin))
  239. var maps []Admin
  240. cond := orm.NewCondition()
  241. cond1 := cond.And("T_State", 1)
  242. if len(T_pids) > 0 {
  243. list := models.SplitStringIds(T_pids, "P")
  244. cond1 = cond1.And("T_pid__in", list)
  245. }
  246. qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&maps)
  247. for _, v := range maps {
  248. AdminList = append(AdminList, AdminToAdmin_(v))
  249. }
  250. return AdminList
  251. }
  252. // 获取列表
  253. func Read_Admin_List_All() (AdminList []Admin_R) {
  254. o := orm.NewOrm()
  255. // 也可以直接使用 Model 结构体作为表名
  256. qs := o.QueryTable(new(Admin))
  257. var maps []Admin
  258. qs.All(&maps)
  259. for _, v := range maps {
  260. AdminList = append(AdminList, AdminToAdmin_R(v))
  261. }
  262. return AdminList
  263. }