base_controller.go 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200
  1. package controllers
  2. import (
  3. "github.com/go-playground/validator/v10"
  4. "github.com/kataras/iris/v12"
  5. "github.com/stretchr/objx"
  6. "io"
  7. "log"
  8. "xps/cache"
  9. "xps/viewmodels"
  10. )
  11. var (
  12. validate *validator.Validate
  13. )
  14. func init() {
  15. validate = validator.New()
  16. }
  17. func errorData(errs ...error) {
  18. var s string
  19. for _, err := range errs {
  20. if err != nil {
  21. s += err.Error() + "<br/>"
  22. }
  23. }
  24. }
  25. type Base struct {
  26. Ctx iris.Context
  27. }
  28. func (b *Base) GetCurOcId() int64 {
  29. ocId, _ := b.Ctx.Values().GetInt64("ocId")
  30. return ocId
  31. }
  32. func (b *Base) GetCurUserId() int64 {
  33. userId, _ := b.Ctx.Values().GetInt64("userId")
  34. return userId
  35. }
  36. func (b *Base) GetCurUserName() string {
  37. userId, _ := b.Ctx.Values().GetInt64("userId")
  38. userData := GetCurUser(userId)
  39. if userData != nil {
  40. return userData.UserName
  41. }
  42. return ""
  43. }
  44. func (b *Base) GetCurUserPositionId() int64 {
  45. userId, _ := b.Ctx.Values().GetInt64("userId")
  46. userData := GetCurUser(userId)
  47. if userData != nil {
  48. return userData.PositionId
  49. }
  50. return 0
  51. }
  52. func (b *Base) GetCurUserPositionName() string {
  53. userId, _ := b.Ctx.Values().GetInt64("userId")
  54. userData := GetCurUser(userId)
  55. if userData != nil {
  56. return userData.PositionName
  57. }
  58. return ""
  59. }
  60. func (b *Base) GetCurUserGroupId() int64 {
  61. userId, _ := b.Ctx.Values().GetInt64("userId")
  62. userData := GetCurUser(userId)
  63. if userData != nil {
  64. return userData.GroupId
  65. }
  66. return 0
  67. }
  68. func (b *Base) GetCurUserGroupName() string {
  69. userId, _ := b.Ctx.Values().GetInt64("userId")
  70. userData := GetCurUser(userId)
  71. if userData != nil {
  72. return userData.GroupName
  73. }
  74. return ""
  75. }
  76. func (b *Base) GetOcTypeId() int64 {
  77. ocTypeId, _ := b.Ctx.Values().GetInt64("ocTypeId")
  78. return ocTypeId
  79. }
  80. func (b *Base) GetAppKey() string {
  81. appKey := b.Ctx.Values().GetString("appKey")
  82. return appKey
  83. }
  84. func (b *Base) GetAppSecret() string {
  85. appSecret := b.Ctx.Values().GetString("appSecret")
  86. return appSecret
  87. }
  88. func (b *Base) GetJsonData() (map[string]interface{}, error) {
  89. data, err := io.ReadAll(b.Ctx.Request().Body)
  90. if err != nil {
  91. return nil, err
  92. }
  93. m, err := objx.FromJSON(string(data))
  94. if err != nil {
  95. return nil, err
  96. }
  97. return m, nil
  98. }
  99. func (b *Base) buildParams() map[string]interface{} {
  100. ocId, err := b.Ctx.Values().GetInt64("ocId")
  101. if err != nil {
  102. _ = b.Ctx.JSON(ResultErr(err.Error(), nil))
  103. b.Ctx.Next()
  104. }
  105. userId, err := b.Ctx.Values().GetInt64("userId")
  106. if err != nil {
  107. _ = b.Ctx.JSON(ResultErr(err.Error(), nil))
  108. b.Ctx.Next()
  109. }
  110. params := make(map[string]interface{})
  111. params["ocId"] = ocId
  112. params["userId"] = userId
  113. groupId := b.Ctx.URLParamInt64Default("groupId", 0)
  114. if groupId > 0 {
  115. params["groupId"] = groupId
  116. }
  117. limit := b.Ctx.URLParamIntDefault("limit", 0)
  118. if limit <= 0 {
  119. limit = 10
  120. }
  121. params["limit"] = limit
  122. page := b.Ctx.URLParamIntDefault("page", 0)
  123. if page <= 0 {
  124. page = 1
  125. }
  126. params["page"] = page
  127. keywords := b.Ctx.URLParamDefault("keywords", "")
  128. if keywords != "" {
  129. params["keywords"] = keywords
  130. }
  131. return params
  132. }
  133. // RetCode Result Code
  134. type RetCode int
  135. // Result Code List
  136. const (
  137. RetOk RetCode = 200
  138. RetErr RetCode = 500
  139. )
  140. type QueryConditions struct {
  141. OcId int64 `url:"ocId"`
  142. GroupId int64 `url:"groupId"`
  143. UserId int64 `url:"userId"`
  144. Page int `url:"page"`
  145. Limit int `url:"limit"`
  146. Keywords string `url:"keywords"`
  147. }
  148. type JsonResult struct {
  149. Code RetCode `json:"code"`
  150. Msg interface{} `json:"msg"`
  151. Data interface{} `json:"data"`
  152. }
  153. func ResultOk(msg string, objects interface{}) (r *JsonResult) {
  154. r = &JsonResult{Code: RetOk, Msg: msg, Data: objects}
  155. return
  156. }
  157. func ResultErr(msg string, objects interface{}) (r *JsonResult) {
  158. r = &JsonResult{Code: RetErr, Msg: msg, Data: objects}
  159. return
  160. }
  161. func GetCurUser(userId int64) *viewmodels.UserData {
  162. userData, err := cache.GetUserData(userId)
  163. if err != nil {
  164. log.Fatal(err.Error())
  165. }
  166. return userData
  167. }