controller.go 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. package auth_token
  2. import (
  3. "github.com/astaxie/beego/logs"
  4. "nginx-ui/server/base"
  5. "nginx-ui/server/models"
  6. "nginx-ui/server/vo"
  7. )
  8. type Controller struct {
  9. base.Controller
  10. service *Service
  11. }
  12. func NewController() *Controller {
  13. return &Controller{
  14. service: NewService(),
  15. }
  16. }
  17. var AuthController = NewController()
  18. // List 获取全部用户信息
  19. func (c *Controller) List() {
  20. user := c.RequiredUser()
  21. if user == nil {
  22. return
  23. }
  24. req := vo.PageReq{}
  25. if !c.ReadBody(&req) {
  26. return
  27. }
  28. query := &ListQuery{Uid: user.Id}
  29. resp, err := c.service.List(query, &req)
  30. if err != nil {
  31. logs.Warn("Users get fail: %v", err)
  32. c.ErrorJson(err)
  33. return
  34. }
  35. c.SetData(resp).Json()
  36. }
  37. // Create 用户注册
  38. func (c *Controller) Create() {
  39. user := c.RequiredUser()
  40. if user == nil {
  41. return
  42. }
  43. req := models.AuthToken{}
  44. if !c.ReadBody(&req) {
  45. return
  46. }
  47. req.Uid = user.Id
  48. err := c.service.Creat(&req)
  49. if err != nil {
  50. logs.Warn("Users get fail: %v", err)
  51. c.ErrorJson(err)
  52. return
  53. }
  54. c.SetData(req).Json()
  55. }
  56. // Update 更新Token
  57. func (c *Controller) Update() {
  58. user := c.RequiredUser()
  59. if user == nil {
  60. return
  61. }
  62. req := models.AuthToken{}
  63. if !c.ReadBody(&req) {
  64. return
  65. }
  66. err := c.service.Update(&req)
  67. if err != nil {
  68. logs.Warn("Users get fail: %v", err)
  69. c.ErrorJson(err)
  70. return
  71. }
  72. c.SetData(req).Json()
  73. }
  74. func (c *Controller) GetDetail() {
  75. // 获取全部用户信息
  76. user := c.RequiredUser()
  77. if user == nil {
  78. return
  79. }
  80. id, err := c.GetIntQuery("id")
  81. if err != nil {
  82. logs.Warn("AuthToken get fail: %v", err)
  83. c.ErrorJson(err)
  84. return
  85. }
  86. query, err := c.service.GetById(id)
  87. if err != nil {
  88. logs.Warn("AuthToken get fail: %v", err)
  89. c.ErrorJson(err)
  90. return
  91. }
  92. c.SetData(query).Json()
  93. }
  94. func (c *Controller) DeleteById() {
  95. req := models.AuthToken{}
  96. if !c.ReadBody(&req) {
  97. return
  98. }
  99. err := c.service.DeleteById(req.Id)
  100. if err != nil {
  101. logs.Warn("AuthToken delete fail: %v", err)
  102. c.ErrorJson(err)
  103. return
  104. }
  105. c.Json()
  106. }