Admin.go 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361
  1. package Admin
  2. import (
  3. "Cold_Api/conf"
  4. "encoding/json"
  5. "fmt"
  6. "github.com/astaxie/beego/cache"
  7. _ "github.com/astaxie/beego/cache/redis"
  8. "github.com/beego/beego/v2/client/orm"
  9. "time"
  10. )
  11. type Admin struct {
  12. Id int `orm:"column(ID);size(11);auto;pk"`
  13. Admin_uuid string `orm:"size(256);null"` //
  14. Admin_master int `orm:"size(200);null"` // 上一级
  15. Admin_power int `orm:"size(2);0"` // 权限
  16. Admin_name string `orm:"size(256);null"` //
  17. Admin_user string `orm:"size(256);null"` //
  18. Admin_pass string `orm:"size(256);null"` //
  19. Admin_telephone string `orm:"size(256);null"` // 电话
  20. Admin_mail string `orm:"size(200);null"` // 邮箱
  21. Admin_gender int `orm:"size(2);null"` // 性别 0未知 1男 2女
  22. Admin_text string `orm:"type(text);null"` // 详情
  23. Admin_panel_plan string `orm:"size(200);null"` // 平面图
  24. Admin_panel_data string `orm:"size(200);null"` // 大数据
  25. Admin_rh int `orm:"type(text);1"` // 显示湿度 0 不显示 1 显示
  26. T_State int `orm:"size(200);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. Admin_uuid string //
  32. Admin_master int // 上一级
  33. Admin_power int // 权限
  34. Admin_name string //
  35. Admin_user string //
  36. Admin_telephone string // 电话
  37. Admin_mail string // 邮箱
  38. Admin_gender int // 性别 0未知 1男 2女
  39. Admin_text string // 详情
  40. Admin_rh int // 显示湿度
  41. Admin_panel_plan string // 平面图
  42. Admin_panel_data string // 大数据
  43. }
  44. func Admin_To_Admin_R(r Admin) (v Admin_R) {
  45. v.Admin_uuid = r.Admin_uuid
  46. v.Admin_master = r.Admin_master
  47. v.Admin_power = r.Admin_power
  48. v.Admin_name = r.Admin_name
  49. v.Admin_user = r.Admin_user
  50. v.Admin_telephone = r.Admin_telephone
  51. v.Admin_mail = r.Admin_mail
  52. v.Admin_gender = r.Admin_gender
  53. v.Admin_text = r.Admin_text
  54. v.Admin_rh = r.Admin_rh
  55. v.Admin_panel_plan = r.Admin_panel_plan
  56. v.Admin_panel_data = r.Admin_panel_data
  57. return v
  58. }
  59. func (t *Admin) TableName() string {
  60. return "Admin" // 数据库名称 // ************** 替换 FormulaList **************
  61. }
  62. var redisCache_Admin cache.Cache
  63. func init() {
  64. //注册模型
  65. orm.RegisterModel(new(Admin))
  66. config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
  67. "redis_Admin", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
  68. fmt.Println(config)
  69. var err error
  70. redisCache_Admin, err = cache.NewCache("redis", config)
  71. if err != nil || redisCache_Admin == nil {
  72. errMsg := "failed to init redis"
  73. fmt.Println(errMsg, err)
  74. }
  75. }
  76. // ---------------- Redis -------------------
  77. //Redis_Set(m.T_sn,m) // Redis 更新缓存
  78. func Redis_Admin_Set(r Admin) (err error) {
  79. //json序列化
  80. str, err := json.Marshal(r)
  81. if err != nil {
  82. fmt.Print(err)
  83. return
  84. }
  85. err = redisCache_Admin.Put(r.Admin_uuid, str, 24*time.Hour)
  86. if err != nil {
  87. fmt.Println("set key:", r.Admin_uuid, ",value:", str, err)
  88. }
  89. return
  90. }
  91. //if r,is :=Redis_Get(T_sn);is{
  92. //return r,nil
  93. //}
  94. func Redis_Admin_Get(key string) (r Admin, is bool) {
  95. if redisCache_Admin.IsExist(key) {
  96. //println("找到key:",key)
  97. v := redisCache_Admin.Get(key)
  98. json.Unmarshal(v.([]byte), &r)
  99. return r, true
  100. }
  101. //println("没有 找到key:",key)
  102. return Admin{}, false
  103. }
  104. func Redis_Admin_DelK(key string) (err error) {
  105. err = redisCache_Admin.Delete(key)
  106. return
  107. }
  108. // ---------------- 特殊方法 -------------------
  109. // 添加
  110. func Add_Admin(m Admin) (id int64, err error) {
  111. o := orm.NewOrm()
  112. id, err = o.Insert(&m)
  113. if err != nil {
  114. fmt.Println(err)
  115. }
  116. m.Id = int(id)
  117. Redis_Admin_Set(m)
  118. return id, err
  119. }
  120. //// 修改
  121. //func Update_Delete_Admin_ByAdmin_uuid(Admin_uuid string) (err error) {
  122. // o := orm.NewOrm()
  123. // v := Admin{Admin_uuid: Admin_uuid}
  124. // // ascertain id exists in the database
  125. // if err = o.Read(&v, "Admin_uuid"); err == nil {
  126. // var num int64
  127. // v.T_State = 0
  128. //
  129. // if num, err = o.Update(&v, "T_State"); err == nil {
  130. // fmt.Println("Number of records updated in database:", num)
  131. // }
  132. //
  133. // Redis_Admin_Set(v)
  134. // }
  135. // return err
  136. //}
  137. // 修改
  138. func Update_Admin(r Admin, cols ...string) bool {
  139. o := orm.NewOrm()
  140. if num, err := o.Update(&r, cols...); err == nil {
  141. fmt.Println("Number of records updated in database:", num)
  142. Redis_Admin_Set(r) // Redis 更新缓存
  143. return true
  144. }
  145. return false
  146. }
  147. func Read_Admin_Admin_user(Admin_user string) (error, Admin) {
  148. o := orm.NewOrm()
  149. r := Admin{Admin_user: Admin_user}
  150. err := o.Read(&r, "Admin_user") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  151. if err != nil {
  152. fmt.Println(err)
  153. }
  154. return err, r
  155. }
  156. func ReadLogin_verification(Admin_user string, Admin_pass string) (error, Admin) {
  157. o := orm.NewOrm()
  158. r := Admin{Admin_user: Admin_user, Admin_pass: Admin_pass}
  159. err := o.Read(&r, "Admin_user", "Admin_pass") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  160. if err != nil {
  161. fmt.Println(err)
  162. }
  163. Redis_Admin_Set(r) // Redis 更新缓存
  164. return err, r
  165. }
  166. // 修改
  167. func Update_Admin_ById_Basics(m Admin) (err error) {
  168. o := orm.NewOrm()
  169. v := Admin{Admin_uuid: m.Admin_uuid}
  170. // ascertain id exists in the database
  171. if err = o.Read(&v, "Admin_uuid"); err == nil {
  172. var num int64
  173. if len(m.Admin_name) > 0 {
  174. v.Admin_name = m.Admin_name
  175. }
  176. if len(m.Admin_mail) > 0 {
  177. v.Admin_mail = m.Admin_mail
  178. }
  179. if len(m.Admin_telephone) > 0 {
  180. v.Admin_telephone = m.Admin_telephone
  181. }
  182. if len(m.Admin_text) > 0 {
  183. v.Admin_text = m.Admin_text
  184. }
  185. v.Admin_gender = m.Admin_gender
  186. if m.Admin_power > 0 {
  187. v.Admin_power = m.Admin_power
  188. }
  189. if len(m.Admin_pass) > 0 {
  190. v.Admin_pass = m.Admin_pass
  191. }
  192. println("m.Admin_pass:", m.Admin_pass)
  193. if num, err = o.Update(&v, "Admin_name", "Admin_telephone", "Admin_mail", "Admin_text", "Admin_gender", "Admin_power", "Admin_pass"); err == nil {
  194. fmt.Println("Number of records updated in database:", num)
  195. }
  196. }
  197. return err
  198. }
  199. // 获取 ById
  200. func Read_Admin_ByAdmin_uuid(Admin_uuid string) (e error, r Admin) {
  201. if r, is := Redis_Admin_Get(Admin_uuid); is {
  202. //println("Redis_Get OK")
  203. return nil, r
  204. }
  205. o := orm.NewOrm()
  206. qs := o.QueryTable(new(Admin))
  207. e = qs.Filter("Admin_uuid", Admin_uuid).One(&r)
  208. return e, r
  209. }
  210. // 获取 ById
  211. func Read_Admin_ByUuid(Admin_uuid string) (r Admin, err error) {
  212. if r, is := Redis_Admin_Get(Admin_uuid); is {
  213. //println("Redis_Get OK")
  214. return r, nil
  215. }
  216. o := orm.NewOrm()
  217. r = Admin{Admin_uuid: Admin_uuid}
  218. err = o.Read(&r, "Admin_uuid") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  219. if err != nil {
  220. fmt.Println(err)
  221. }
  222. return r, err
  223. }
  224. // 获取 ById
  225. func Read_Admin_ById(id int) (r Admin, err error) {
  226. o := orm.NewOrm()
  227. r = Admin{Id: id}
  228. err = o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
  229. if err != nil {
  230. fmt.Println(err)
  231. }
  232. return r, err
  233. }
  234. // 获取 ById
  235. func Read_Admin_ById_x(Id int) (r Admin) {
  236. o := orm.NewOrm()
  237. qs := o.QueryTable(new(Admin))
  238. qs.Filter("Id", Id).One(&r, "Id", "Admin_uuid", "Admin_master", "Admin_name", "Admin_user", "CreateTime", "UpdateTime")
  239. return r
  240. }
  241. // 获取列表
  242. func Read_Admin_List_ALL_T(page int, Admin_name string, Admin_master int) (maps []orm.Params, cnt int64) {
  243. o := orm.NewOrm()
  244. // 也可以直接使用 Model 结构体作为表名
  245. qs := o.QueryTable(new(Admin))
  246. var offset int64
  247. if page <= 1 {
  248. offset = 0
  249. } else {
  250. offset = int64((page - 1) * conf.Page_size)
  251. }
  252. qs.Limit(conf.Page_size, offset).Filter("Admin_name__icontains", Admin_name).Filter("Admin_master__gt", Admin_master).Filter("T_State", 1).OrderBy("-Id").Values(
  253. &maps,
  254. "Id", "Admin_uuid", "Admin_master", "Admin_name", "Admin_user", "CreateTime", "UpdateTime")
  255. cnt, _ = qs.Filter("Admin_name__icontains", Admin_name).Filter("Admin_master__gt", Admin_master).Filter("T_State", 1).Count()
  256. return maps, cnt
  257. }
  258. // 获取列表
  259. func Read_Admin_List_ALL_User(page int, Admin_name string, Admin_master int) (maps []Admin, cnt int64) {
  260. o := orm.NewOrm()
  261. // 也可以直接使用 Model 结构体作为表名
  262. qs := o.QueryTable(new(Admin))
  263. var offset int64
  264. if page <= 1 {
  265. offset = 0
  266. } else {
  267. offset = int64((page - 1) * conf.Page_size)
  268. }
  269. qs.Limit(conf.Page_size, offset).Filter("Admin_name__icontains", Admin_name).Exclude("Admin_master", 0).Filter("Admin_master", Admin_master).Filter("T_State", 1).OrderBy("-Id").All(&maps)
  270. cnt, _ = qs.Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).Count()
  271. return maps, cnt
  272. }
  273. // 获取列表
  274. func Read_Admin_List_ALL_Admin_master(page int, Admin_name string, Admin_master int) (maps []Admin, cnt int64) {
  275. o := orm.NewOrm()
  276. // 也可以直接使用 Model 结构体作为表名
  277. qs := o.QueryTable(new(Admin))
  278. var offset int64
  279. if page <= 1 {
  280. offset = 0
  281. } else {
  282. offset = int64((page - 1) * conf.Page_size)
  283. }
  284. qs.Limit(conf.Page_size, offset).Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).OrderBy("-Id").All(&maps)
  285. cnt, _ = qs.Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).Count()
  286. return maps, cnt
  287. }
  288. // 获取列表 - 权限相同
  289. func Read_Admin_List_ALL_AdminPower(AdminPower int) (maps []Admin) {
  290. o := orm.NewOrm()
  291. // 也可以直接使用 Model 结构体作为表名
  292. qs := o.QueryTable(new(Admin))
  293. qs.Filter("Admin_power", AdminPower).Filter("T_State", 1).OrderBy("Id").All(&maps)
  294. //cnt, _ := qs.Filter("Id", AdminPower).Count()
  295. return maps
  296. }
  297. // 获取列表 - 权限相同
  298. func Read_Admin_List_Admin_master_ALL(Admin_master int) (maps []Admin) {
  299. o := orm.NewOrm()
  300. // 也可以直接使用 Model 结构体作为表名
  301. qs := o.QueryTable(new(Admin))
  302. qs.Filter("Admin_master", Admin_master).Filter("T_State", 1).Filter("Admin_power", 6).OrderBy("Id").All(&maps)
  303. //cnt, _ := qs.Filter("Id", AdminPower).Count()
  304. return maps
  305. }