magic_stake_product.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. package app
  2. import (
  3. "go_server/base/core"
  4. model "go_server/model/biz_modules/app"
  5. "go_server/model/common/response"
  6. "go_server/service/base"
  7. "strings"
  8. "github.com/gin-gonic/gin"
  9. )
  10. type MagicStakeProductService struct {
  11. base.BizCommonService
  12. }
  13. func (s *MagicStakeProductService) Get(c *gin.Context) {
  14. s.SetDbAlias("app")
  15. base.NewBaseHandler(model.NewMagicStakeProduct()).Get(c, s.DB())
  16. }
  17. func (s *MagicStakeProductService) Find(c *gin.Context) {
  18. s.SetDbAlias("app")
  19. type request[T any] struct {
  20. base.ListRequest[T]
  21. Id *int64 `form:"id"`
  22. IsExport *bool `form:"isExport"`
  23. Fields *string `form:"fields"` // 指定返回字段 , 分割
  24. Name *string `form:"name"`
  25. BeginTime *int64 `form:"beginTime"`
  26. EndTime *int64 `form:"endTime"`
  27. }
  28. req := new(request[model.MagicStakeProduct])
  29. if err := c.BindQuery(req); err != nil {
  30. response.Resp(c, err.Error())
  31. return
  32. }
  33. db := s.DB()
  34. if req.Id != nil && *req.Id != 0 {
  35. db = db.Where("id", req.Id)
  36. }
  37. if req.Name != nil && *req.Name != "" {
  38. db = db.Where("name", req.Name)
  39. }
  40. if req.BeginTime != nil && *req.BeginTime > 0 && req.EndTime != nil && *req.EndTime > 0 {
  41. db = db.Where("created_at between ? and ?", *req.BeginTime, *req.EndTime)
  42. }
  43. // 导出处理
  44. fields := make([]string, 0)
  45. if req.Fields != nil {
  46. fields = strings.Split(*req.Fields, ",")
  47. }
  48. var url string
  49. colInfo := s.GetColumnCommentFromStruct(model.MagicStakeProduct{})
  50. var err error
  51. if req.IsExport != nil && *req.IsExport {
  52. if len(fields) == 0 {
  53. for _, col := range colInfo {
  54. fields = append(fields, col.Field)
  55. }
  56. }
  57. core.Log.Infof("导出的字段:%s", fields)
  58. url, err = base.ExportCsv[model.MagicStakeProduct](db, fields, colInfo)
  59. if err != nil {
  60. response.Resp(c, err.Error())
  61. return
  62. }
  63. }
  64. resp, err := base.NewQueryBaseHandler(model.NewMagicStakeProduct()).List(db, req)
  65. if err != nil {
  66. response.Resp(c, err.Error())
  67. return
  68. }
  69. response.Resp(c, map[string]interface{}{
  70. "url": url,
  71. "cols": colInfo,
  72. "list": resp.List,
  73. "paging": resp.Paging,
  74. })
  75. }
  76. func (s *MagicStakeProductService) Comment(c *gin.Context) {
  77. s.SetDbAlias("app")
  78. dbs, err := s.GetColumnComment("app", model.NewMagicStakeProduct().TableName())
  79. if err != nil {
  80. response.Resp(c, "获取失败")
  81. return
  82. } else {
  83. response.Resp(c, gin.H{"dbs": dbs})
  84. return
  85. }
  86. }