UserWalletDao.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. package manage
  2. import (
  3. "demo/configs"
  4. "demo/data/domain"
  5. "demo/data/domain/vo"
  6. )
  7. func GetByIdUserWallet(id int) (domain.UserWallet, error) {
  8. var userWallet domain.UserWallet
  9. _, err := configs.Engine.Where("id = ?", id).Get(&userWallet)
  10. if err != nil {
  11. return userWallet, err
  12. }
  13. return userWallet, nil
  14. }
  15. func GetInIdListUserWallet(ids []int) ([]domain.UserWallet, error) {
  16. rows, err := configs.Engine.In("id", ids).Rows(&domain.UserWallet{})
  17. if err != nil {
  18. return make([]domain.UserWallet, 0), err
  19. }
  20. defer rows.Close()
  21. var list []domain.UserWallet
  22. for rows.Next() {
  23. var u domain.UserWallet
  24. list = append(list, u)
  25. }
  26. return list, nil
  27. }
  28. func GetListUserWallet(userWallet domain.UserWallet, pageNum, pageSize int) (vo.BaseListVo, error) {
  29. rows, err := configs.Engine.Limit(pageSize, (pageNum-1)*pageSize).Desc("id").Rows(&userWallet)
  30. var vo vo.BaseListVo
  31. if err != nil {
  32. return vo, err
  33. }
  34. vo.PageNum = pageNum
  35. vo.PageSize = pageSize
  36. vo.List = make([]any, 0)
  37. for rows.Next() {
  38. var u domain.UserWallet
  39. rows.Scan(&u)
  40. vo.List = append(vo.List, u)
  41. }
  42. //查询数量
  43. count, err := configs.Engine.Count(&userWallet)
  44. vo.Total = int(count)
  45. return vo, nil
  46. }
  47. func SaveUserWallet(userWallet *domain.UserWallet) (domain.UserWallet, error) {
  48. _, err := configs.Engine.Insert(&userWallet)
  49. if err != nil {
  50. return *userWallet, err
  51. }
  52. return *userWallet, nil
  53. }
  54. func SetUserWallet(userWalletMap map[string]interface{}, id int) (domain.UserWallet, bool, error) {
  55. var userWallet domain.UserWallet
  56. _, err := configs.Engine.Table("user_wallet").Where("id = ?", id).Update(userWalletMap)
  57. if err != nil {
  58. return userWallet, false, err
  59. }
  60. return userWallet, true, nil
  61. }
  62. func DeleteUserWallet(userWallet domain.UserWallet) bool {
  63. i, err := configs.Engine.Where("id = ?", userWallet.Id).Delete(&userWallet)
  64. if err != nil {
  65. return false
  66. }
  67. return i > 0
  68. }