UserDao.go 1.6 KB

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