Blizzard
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
plantModel "sundynix-micro-go/app/plant/model"
|
|
"sundynix-micro-go/app/plant/rpc/internal/svc"
|
|
"sundynix-micro-go/app/plant/rpc/plant"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type CreatePlantLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewCreatePlantLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreatePlantLogic {
|
|
return &CreatePlantLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
// 创建植物
|
|
func (l *CreatePlantLogic) CreatePlant(in *plant.CreatePlantReq) (*plant.CommonResp, error) {
|
|
// 解析植时间
|
|
plantTime, err := time.Parse("2006-01-02", in.PlantTime)
|
|
if err != nil {
|
|
plantTime = time.Now()
|
|
}
|
|
|
|
var myPlantID string
|
|
txErr := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
|
|
myPlant := plantModel.MyPlant{
|
|
UserID: in.UserId,
|
|
Name: in.Name,
|
|
PlantTime: plantTime,
|
|
Status: 1,
|
|
Placement: in.Placement,
|
|
PotMaterial: in.PotMaterial,
|
|
PotSize: in.PotSize,
|
|
Sunlight: in.Sunlight,
|
|
PlantingMaterial: in.PlantingMaterial,
|
|
}
|
|
if err := tx.Create(&myPlant).Error; err != nil {
|
|
return err
|
|
}
|
|
myPlantID = myPlant.ID
|
|
// 处理图片关联
|
|
if len(in.ImgIds) > 0 {
|
|
var relations []map[string]interface{}
|
|
for _, id := range in.ImgIds {
|
|
relations = append(relations, map[string]interface{}{
|
|
"sundynix_my_plant_id": myPlant.ID,
|
|
"sundynix_oss_id": id,
|
|
})
|
|
}
|
|
if err := tx.Table("sundynix_plant_my_plant_oss").Create(relations).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
// 更新用户 plant_count
|
|
return tx.Model(&plantModel.UserProfile{}).Where("user_id = ?", in.UserId).
|
|
Update("plant_count", gorm.Expr("plant_count + 1")).Error
|
|
})
|
|
if txErr != nil {
|
|
return nil, txErr
|
|
}
|
|
return &plant.CommonResp{Code: 0, Msg: myPlantID}, nil
|
|
}
|