Ethan před 5 měsíci
rodič
revize
7d6bdc539f

+ 16 - 0
app/dao/product_photo_dao.go

@@ -1,7 +1,9 @@
 package dao
 
 import (
+	"errors"
 	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
 	"youngee_b_api/app/entity"
 )
 
@@ -25,6 +27,20 @@ func (d ProductPhotoDAO) GetProductPhotoByProductID(productId int64) ([]entity.P
 	return productPhotos, nil
 }
 
+// 获取所属商品的主图详情
+func (d ProductPhotoDAO) GetMainProductPhotoInfoByProductID(productId int64) (*entity.ProductPhoto, error) {
+	var productPhoto entity.ProductPhoto
+	err := Db.Where("product_id = ? AND symbol = ?", productId, 1).First(&productPhoto).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		} else {
+			return nil, err
+		}
+	}
+	return &productPhoto, nil
+}
+
 // 获取所属商品的主图
 func (d ProductPhotoDAO) GetMainPhotoByProductID(productId int64) (string, error) {
 	var productPhoto entity.ProductPhoto

+ 2 - 2
app/service/selection_info_service.go

@@ -65,9 +65,9 @@ func (s SelectionInfoService) CreateSelectionInfo(selectionCreateParam *vo.Selec
 	if product == nil {
 		return nil, errors.New("未找到关联商品")
 	}
-	productPhotos, err := dao.ProductPhotoDAO{}.GetProductPhotoByProductID(selectionCreateParam.ProductId)
+	productMainPhoto, _ := dao.ProductPhotoDAO{}.GetMainProductPhotoInfoByProductID(selectionCreateParam.ProductId)
 	productInfoToJson, _ := json.Marshal(product)
-	productPhotosToJson, _ := json.Marshal(productPhotos)
+	productPhotosToJson, _ := json.Marshal(productMainPhoto)
 	// 任务截止时间
 	taskDdl, err1 := time.ParseInLocation("2006-01-02 15:04:05", selectionCreateParam.TaskDdl, time.Local)
 	if err1 != nil {