f

app/main/model/productFeatureModel.go

@@ -1,6 +1,12 @@
 package model
 
 import (
+	"context"
+	"fmt"
+	"strings"
+
+	"qnc-server/common/globalkey"
+
 	"github.com/zeromicro/go-zero/core/stores/cache"
 	"github.com/zeromicro/go-zero/core/stores/sqlx"
 )
@@ -12,6 +18,10 @@ type (
 	// and implement the added methods in customProductFeatureModel.
 	ProductFeatureModel interface {
 		productFeatureModel
+		// SumFeatureCostPriceByProductId 汇总该产品下已关联且未删除的模块成本（feature.cost_price 之和）
+		SumFeatureCostPriceByProductId(ctx context.Context, productId string) (float64, error)
+		// SumFeatureCostPriceByProductIds 批量汇总，未出现在结果中的产品视为 0
+		SumFeatureCostPriceByProductIds(ctx context.Context, productIds []string) (map[string]float64, error)
 	}
 
 	customProductFeatureModel struct {
@@ -25,3 +35,48 @@ func NewProductFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) ProductFeature
 		defaultProductFeatureModel: newProductFeatureModel(conn, c),
 	}
 }
+
+func (m *customProductFeatureModel) SumFeatureCostPriceByProductId(ctx context.Context, productId string) (float64, error) {
+	query := `
+SELECT COALESCE(SUM(f.cost_price), 0)
+FROM product_feature pf
+INNER JOIN feature f ON f.id = pf.feature_id AND f.del_state = ?
+WHERE pf.product_id = ? AND pf.del_state = ?`
+	var sum float64
+	err := m.QueryRowNoCacheCtx(ctx, &sum, query, globalkey.DelStateNo, productId, globalkey.DelStateNo)
+	if err != nil {
+		return 0, err
+	}
+	return sum, nil
+}
+
+func (m *customProductFeatureModel) SumFeatureCostPriceByProductIds(ctx context.Context, productIds []string) (map[string]float64, error) {
+	out := make(map[string]float64)
+	if len(productIds) == 0 {
+		return out, nil
+	}
+	placeholders := strings.TrimSuffix(strings.Repeat("?,", len(productIds)), ",")
+	query := fmt.Sprintf(`
+SELECT pf.product_id, COALESCE(SUM(f.cost_price), 0) AS total
+FROM product_feature pf
+INNER JOIN feature f ON f.id = pf.feature_id AND f.del_state = ?
+WHERE pf.del_state = ? AND pf.product_id IN (%s)
+GROUP BY pf.product_id`, placeholders)
+	args := make([]interface{}, 0, 2+len(productIds))
+	args = append(args, globalkey.DelStateNo, globalkey.DelStateNo)
+	for _, id := range productIds {
+		args = append(args, id)
+	}
+	var rows []struct {
+		ProductId string  `db:"product_id"`
+		Total     float64 `db:"total"`
+	}
+	err := m.QueryRowsNoCacheCtx(ctx, &rows, query, args...)
+	if err != nil {
+		return nil, err
+	}
+	for i := range rows {
+		out[rows[i].ProductId] = rows[i].Total
+	}
+	return out, nil
+}