UserDao.go 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. package dao
  2. import (
  3. "fmt"
  4. "go-create/configs"
  5. "go-create/domain"
  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) (domain.BaseListVo, error) {
  29. var arr []domain.User
  30. Count, err := configs.Engine.Limit(pageSize, (pageNum-1)*pageSize).Desc("id").FindAndCount(&arr, &user)
  31. if err != nil {
  32. return domain.BaseListVo{}, err
  33. }
  34. var vo domain.BaseListVo
  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.Table("user").Insert(&user)
  49. if err != nil {
  50. panic(err)
  51. return *user, err
  52. }
  53. return *user, nil
  54. }
  55. func SetUser(userMap map[string]interface{}, id int) (domain.User, bool, error) {
  56. var user domain.User
  57. _, err := configs.Engine.Table("user").Where("id = ?", id).Update(userMap)
  58. if err != nil {
  59. return user, false, err
  60. }
  61. return user, true, nil
  62. }
  63. func DeleteUser(user domain.User) bool {
  64. i, err := configs.Engine.Where("id = ?", user.Id).Delete(&user)
  65. if err != nil {
  66. return false
  67. }
  68. return i > 0
  69. }
  70. func SelectByObject(User domain.User, pageNum, pageSize int) {
  71. var res []domain.User
  72. err := configs.Engine.Limit(pageSize, (pageNum-1)*pageSize).Desc("id").Find(&res, &User)
  73. if err != nil {
  74. panic(err)
  75. return
  76. }
  77. for _, v := range res {
  78. fmt.Println(v)
  79. }
  80. }