base_controller.go 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  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. b.Ctx.URLParams()
  132. return params
  133. }
  134. // RetCode Result Code
  135. type RetCode int
  136. // Result Code List
  137. const (
  138. RetOk RetCode = 200
  139. RetErr RetCode = 500
  140. )
  141. type QueryConditions struct {
  142. OcId int64 `url:"ocId"`
  143. GroupId int64 `url:"groupId"`
  144. UserId int64 `url:"userId"`
  145. Page int `url:"page"`
  146. Limit int `url:"limit"`
  147. Keywords string `url:"keywords"`
  148. }
  149. type JsonResult struct {
  150. Code RetCode `json:"code"`
  151. Msg interface{} `json:"msg"`
  152. Data interface{} `json:"data"`
  153. }
  154. func ResultOk(msg string, objects interface{}) (r *JsonResult) {
  155. r = &JsonResult{Code: RetOk, Msg: msg, Data: objects}
  156. return
  157. }
  158. func ResultErr(msg string, objects interface{}) (r *JsonResult) {
  159. r = &JsonResult{Code: RetErr, Msg: msg, Data: objects}
  160. return
  161. }
  162. func GetCurUser(userId int64) *viewmodels.UserData {
  163. userData, err := cache.GetUserData(userId)
  164. if err != nil {
  165. log.Fatal(err.Error())
  166. }
  167. return userData
  168. }