feat: 客户关联产品并支持按产品圈选通知群体

新增客户-产品多对多关联，列表可按产品筛选并复制邮箱；发证/下单自动回填历史关联。 Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-12 21:58:35 +08:00
1 parent d0e1ca3729
commit 4c46088f4f
@@ -18,15 +18,19 @@ func NewCustomerService(db *gorm.DB) *CustomerService {
 	return &CustomerService{db: db}
 }

-// List 分页查询客户 // Logan.Li
-func (s *CustomerService) List(page, pageSize int, keyword string) ([]dto.CustomerResponse, int64, error) {
+// List 分页查询客户（可按产品筛选通知群体） // Logan.Li
+func (s *CustomerService) List(page, pageSize int, keyword string, productID uint) ([]dto.CustomerResponse, int64, error) {
 	var customers []model.Customer
 	var total int64

-	query := s.db.Model(&model.Customer{})
+	query := s.db.Model(&model.Customer{}).Preload("Products")
 	if keyword != "" {
 		query = query.Where("name LIKE ? OR email LIKE ? OR company LIKE ?", "%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
 	}
+	if productID > 0 {
+		sub := s.db.Table("customer_products").Select("customer_id").Where("product_id = ?", productID)
+		query = query.Where("id IN (?)", sub)
+	}

 	query.Count(&total)
 	if err := query.Order("created_at DESC").
@@ -42,10 +46,15 @@ func (s *CustomerService) List(page, pageSize int, keyword string) ([]dto.Custom
 	return results, total, nil
 }

-// GetAll 获取全部客户（下拉选项用） // Logan.Li
-func (s *CustomerService) GetAll() ([]dto.CustomerResponse, error) {
+// GetAll 获取全部启用客户（下拉；可按产品过滤） // Logan.Li
+func (s *CustomerService) GetAll(productID uint) ([]dto.CustomerResponse, error) {
 	var customers []model.Customer
-	if err := s.db.Where("status = ?", 1).Order("created_at DESC").Find(&customers).Error; err != nil {
+	query := s.db.Where("status = ?", 1).Preload("Products")
+	if productID > 0 {
+		sub := s.db.Table("customer_products").Select("customer_id").Where("product_id = ?", productID)
+		query = query.Where("id IN (?)", sub)
+	}
+	if err := query.Order("created_at DESC").Find(&customers).Error; err != nil {
 		return nil, err
 	}
 	results := make([]dto.CustomerResponse, 0, len(customers))
@@ -58,14 +67,14 @@ func (s *CustomerService) GetAll() ([]dto.CustomerResponse, error) {
 // GetByID 根据 ID 获取客户 // Logan.Li
 func (s *CustomerService) GetByID(id uint) (*dto.CustomerResponse, error) {
 	var customer model.Customer
-	if err := s.db.First(&customer, id).Error; err != nil {
+	if err := s.db.Preload("Products").First(&customer, id).Error; err != nil {
 		return nil, errors.New("客户不存在")
 	}
 	resp := toCustomerResponse(customer)
 	return &resp, nil
 }

-// Create 创建客户 // Logan.Li
+// Create 创建客户并关联产品 // Logan.Li
 func (s *CustomerService) Create(req dto.CustomerCreate) (*dto.CustomerResponse, error) {
 	status := 1
 	if req.Status != nil {
@@ -83,11 +92,13 @@ func (s *CustomerService) Create(req dto.CustomerCreate) (*dto.CustomerResponse,
 	if err := s.db.Create(&customer).Error; err != nil {
 		return nil, errors.New("创建客户失败")
 	}
-	resp := toCustomerResponse(customer)
-	return &resp, nil
+	if err := s.replaceProducts(customer.ID, req.ProductIDs); err != nil {
+		return nil, err
+	}
+	return s.GetByID(customer.ID)
 }

-// Update 更新客户 // Logan.Li
+// Update 更新客户及产品关联 // Logan.Li
 func (s *CustomerService) Update(id uint, req dto.CustomerUpdate) (*dto.CustomerResponse, error) {
 	var customer model.Customer
 	if err := s.db.First(&customer, id).Error; err != nil {
@@ -109,9 +120,13 @@ func (s *CustomerService) Update(id uint, req dto.CustomerUpdate) (*dto.Customer
 	if err := s.db.Model(&customer).Updates(updates).Error; err != nil {
 		return nil, errors.New("更新客户失败")
 	}
-	s.db.First(&customer, id)
-	resp := toCustomerResponse(customer)
-	return &resp, nil
+	// ProductIDs 为 nil 时不改关联；非 nil（含空切片）则全量替换 // Logan.Li
+	if req.ProductIDs != nil {
+		if err := s.replaceProducts(id, *req.ProductIDs); err != nil {
+			return nil, err
+		}
+	}
+	return s.GetByID(id)
 }

 // Delete 删除客户（软删除） // Logan.Li
@@ -121,6 +136,7 @@ func (s *CustomerService) Delete(id uint) error {
 	if count > 0 {
 		return errors.New("该客户下存在 License，无法删除")
 	}
+	_ = s.db.Where("customer_id = ?", id).Delete(&model.CustomerProduct{}).Error
 	return s.db.Delete(&model.Customer{}, id).Error
 }

@@ -129,18 +145,66 @@ func (s *CustomerService) UpdateStatus(id uint, status int) error {
 	return s.db.Model(&model.Customer{}).Where("id = ?", id).Update("status", status).Error
 }

+// AttachProduct 为客户追加产品关联（发证/下单时调用，幂等） // Logan.Li
+func AttachProduct(db *gorm.DB, customerID, productID uint) error {
+	if db == nil || customerID == 0 || productID == 0 {
+		return nil
+	}
+	var n int64
+	db.Model(&model.CustomerProduct{}).
+		Where("customer_id = ? AND product_id = ?", customerID, productID).
+		Count(&n)
+	if n > 0 {
+		return nil
+	}
+	return db.Create(&model.CustomerProduct{CustomerID: customerID, ProductID: productID}).Error
+}
+
+// replaceProducts 全量替换客户产品关联 // Logan.Li
+func (s *CustomerService) replaceProducts(customerID uint, productIDs []uint) error {
+	if err := s.db.Where("customer_id = ?", customerID).Delete(&model.CustomerProduct{}).Error; err != nil {
+		return errors.New("更新产品关联失败")
+	}
+	seen := map[uint]struct{}{}
+	for _, pid := range productIDs {
+		if pid == 0 {
+			continue
+		}
+		if _, ok := seen[pid]; ok {
+			continue
+		}
+		seen[pid] = struct{}{}
+		var product model.Product
+		if err := s.db.First(&product, pid).Error; err != nil {
+			return errors.New("关联产品不存在")
+		}
+		if err := s.db.Create(&model.CustomerProduct{CustomerID: customerID, ProductID: pid}).Error; err != nil {
+			return errors.New("更新产品关联失败")
+		}
+	}
+	return nil
+}
+
 // toCustomerResponse 模型转响应 DTO // Logan.Li
 func toCustomerResponse(c model.Customer) dto.CustomerResponse {
+	products := make([]dto.CustomerProductBrief, 0, len(c.Products))
+	ids := make([]uint, 0, len(c.Products))
+	for _, p := range c.Products {
+		products = append(products, dto.CustomerProductBrief{ID: p.ID, Name: p.Name, Code: p.Code})
+		ids = append(ids, p.ID)
+	}
 	return dto.CustomerResponse{
-		ID:        c.ID,
-		Name:      c.Name,
-		Company:   c.Company,
-		Email:     c.Email,
-		Phone:     c.Phone,
-		Address:   c.Address,
-		Notes:     c.Notes,
-		Status:    c.Status,
-		CreatedAt: c.CreatedAt.Format("2006-01-02 15:04:05"),
-		UpdatedAt: c.UpdatedAt.Format("2006-01-02 15:04:05"),
+		ID:         c.ID,
+		Name:       c.Name,
+		Company:    c.Company,
+		Email:      c.Email,
+		Phone:      c.Phone,
+		Address:    c.Address,
+		Notes:      c.Notes,
+		Status:     c.Status,
+		ProductIDs: ids,
+		Products:   products,
+		CreatedAt:  c.CreatedAt.Format("2006-01-02 15:04:05"),
+		UpdatedAt:  c.UpdatedAt.Format("2006-01-02 15:04:05"),
 	}
 }