owner.go 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. package owner
  2. import (
  3. "errors"
  4. "fmt"
  5. "server/dao"
  6. "server/global"
  7. "server/model/owner/request"
  8. )
  9. type ownerService struct{}
  10. func NewOwnerService() *ownerService {
  11. return &ownerService{}
  12. }
  13. // CreateOwner 创建车主信息
  14. func (s *ownerService) CreateOwner(req request.OwnerCreate) error {
  15. // 检查手机号是否已存在
  16. var existingOwner dao.Owner
  17. result := global.GVA_DB.Where("owner_phone = ?", req.OwnerPhone).First(&existingOwner)
  18. if result.RowsAffected > 0 {
  19. return errors.New("手机号已存在")
  20. }
  21. owner := dao.Owner{
  22. OwnerName: req.OwnerName,
  23. OwnerSurname: req.OwnerSurname,
  24. OwnerPhone: req.OwnerPhone,
  25. IsVip: req.IsVip,
  26. VipExpireTime: req.VipExpireTime,
  27. }
  28. return global.GVA_DB.Create(&owner).Error
  29. }
  30. // UpdateOwner 更新车主信息
  31. func (s *ownerService) UpdateOwner(req request.OwnerUpdate) error {
  32. var owner dao.Owner
  33. result := global.GVA_DB.First(&owner, req.ID)
  34. if result.RowsAffected == 0 {
  35. return errors.New("车主不存在")
  36. }
  37. // 检查手机号是否被其他车主使用
  38. var existingOwner dao.Owner
  39. result = global.GVA_DB.Where("owner_phone = ? AND id != ?", req.OwnerPhone, req.ID).First(&existingOwner)
  40. if result.RowsAffected > 0 {
  41. return errors.New("手机号已被其他车主使用")
  42. }
  43. owner.OwnerName = req.OwnerName
  44. owner.OwnerSurname = req.OwnerSurname
  45. owner.OwnerPhone = req.OwnerPhone
  46. owner.IsVip = req.IsVip
  47. owner.VipExpireTime = req.VipExpireTime
  48. return global.GVA_DB.Save(&owner).Error
  49. }
  50. // GetOwnerByID 根据ID获取车主信息
  51. func (s *ownerService) GetOwnerByID(id uint) (dao.Owner, error) {
  52. var owner dao.Owner
  53. result := global.GVA_DB.First(&owner, id)
  54. if result.RowsAffected == 0 {
  55. return dao.Owner{}, errors.New("车主不存在")
  56. }
  57. return owner, nil
  58. }
  59. // GetOwnerByPhone 根据手机号获取车主信息
  60. func (s *ownerService) GetOwnerByPhone(phone string) ([]dao.Owner, error) {
  61. var ownerList []dao.Owner
  62. result := global.GVA_DB.Where("owner_phone LIKE ?", "%"+phone+"%").Find(&ownerList)
  63. fmt.Println(result)
  64. //if result.RowsAffected == 0 {
  65. // return dao.Owner{}, errors.New("车主不存在")
  66. //}
  67. return ownerList, nil
  68. }
  69. // ListOwners 获取车主列表
  70. func (s *ownerService) ListOwners(req request.OwnerQuery) (int64, []dao.Owner, error) {
  71. var owners []dao.Owner
  72. var total int64
  73. query := global.GVA_DB.Model(&dao.Owner{})
  74. if req.OwnerName != "" {
  75. query = query.Where("owner_name LIKE ?", "%"+req.OwnerName+"%")
  76. }
  77. if req.OwnerPhone != "" {
  78. query = query.Where("owner_phone LIKE ?", "%"+req.OwnerPhone+"%")
  79. }
  80. if req.IsVip != nil {
  81. query = query.Where("is_vip = ?", *req.IsVip)
  82. }
  83. // 计算总数
  84. query.Count(&total)
  85. // 分页查询
  86. offset := (req.Page - 1) * req.PageSize
  87. result := query.Offset(offset).Limit(req.PageSize).Find(&owners)
  88. if result.Error != nil {
  89. return 0, nil, result.Error
  90. }
  91. return total, owners, nil
  92. }
  93. // DeleteOwner 删除车主
  94. func (s *ownerService) DeleteOwner(id uint) error {
  95. // 检查车主是否存在
  96. var owner dao.Owner
  97. result := global.GVA_DB.First(&owner, id)
  98. if result.RowsAffected == 0 {
  99. return errors.New("车主不存在")
  100. }
  101. // 检查车主是否有车辆
  102. var vehicleCount int64
  103. global.GVA_DB.Model(&dao.Vehicle{}).Where("owner_id = ?", id).Count(&vehicleCount)
  104. if vehicleCount > 0 {
  105. return errors.New("车主有车辆,无法删除")
  106. }
  107. return global.GVA_DB.Delete(&owner, id).Error
  108. }