feat: 迁移plant

app/plant/rpc/internal/logic/createWikiLogic.go

@@ -0,0 +1,56 @@
+package logic
+
+import (
+	"context"
+	"errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"gorm.io/gorm"
+	plantModel "sundynix-micro-go/app/plant/model"
+	"sundynix-micro-go/app/plant/rpc/internal/svc"
+	"sundynix-micro-go/app/plant/rpc/plant"
+)
+
+type CreateWikiLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewCreateWikiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateWikiLogic {
+	return &CreateWikiLogic{ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx)}
+}
+
+func (l *CreateWikiLogic) CreateWiki(in *plant.CreateWikiReq) (*plant.CommonResp, error) {
+	if !errors.Is(l.svcCtx.DB.Where("name LIKE ?", "%"+in.Name+"%").First(&plantModel.Wiki{}).Error, gorm.ErrRecordNotFound) {
+		return nil, errors.New("植物已经存在")
+	}
+	classID := in.ClassId
+	if classID == "" && len(in.ClassIds) > 0 {
+		classID = in.ClassIds[0]
+	}
+	w := plantModel.Wiki{
+		Name: in.Name, LatinName: in.LatinName, Aliases: in.Aliases,
+		Genus: in.Genus, Difficulty: int(in.Difficulty), IsHot: int(in.IsHot),
+		GrowthHabit: in.GrowthHabit, LightIntensity: in.LightIntensity,
+		OptimalTempPeriod: in.OptimalTempPeriod, ClassID: classID,
+		DistributionArea: in.DistributionArea, LifeCycle: in.LifeCycle,
+		ReproductionMethod: in.ReproductionMethod, PestsDiseases: in.PestsDiseases,
+		LightType: in.LightType, Stem: in.Stem, Fruit: in.Fruit,
+		FoliageType: in.FoliageType, FoliageColor: in.FoliageColor,
+		FoliageShape: in.FoliageShape, Height: int(in.Height),
+		FloweringPeriod: in.FloweringPeriod, FloweringColor: in.FloweringColor,
+		FloweringShape: in.FloweringShape, FlowerDiameter: int(in.FloweringDiameter),
+	}
+	if err := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Create(&w).Error; err != nil {
+			return err
+		}
+		return replaceWikiRelations(tx, w.ID, in.ClassIds, in.OssIds, in.RelatedWikiIds)
+	}); err != nil {
+		return nil, err
+	}
+	go func() {
+		_, _ = NewSyncWikiVectorLogic(context.Background(), l.svcCtx).SyncWikiVector(&plant.SyncWikiVectorReq{WikiId: w.ID})
+	}()
+	return &plant.CommonResp{Code: 0, Msg: w.ID}, nil
+}