udeServer/internal/logic/createservicelogic.go

// udeServer: CreateService 走事务 —— 同一事务内插 t_service + N 条 t_directory,
// 任意一条 INSERT 失败,全部回滚,不留脏数据。

package logic

import (
	"context"
	"fmt"
	"strings"

	"udeServer/internal/model"
	"udeServer/internal/svc"
	"udeServer/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type CreateServiceLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewCreateServiceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateServiceLogic {
	return &CreateServiceLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *CreateServiceLogic) CreateService(req *types.CreateServiceReq) (*types.ServiceResp, error) {
	var svcId int64
	err := l.svcCtx.DB.Transact(func(session sqlx.Session) error {
		// 把 model 绑到事务 session 上,后续所有写操作都走同一个连接/事务
		svcModel := l.svcCtx.ServiceModel.WithSession(session)
		dirModel := l.svcCtx.DirectoryModel.WithSession(session)

		// 1. 插服务
		svcData := &model.TService{
			Name:    nullString(req.Name),
			Ip:      nullString(req.Ip),
			Port:    nullString(req.Port),
			Iden:    nullString(req.Iden),
			KeyIden: nullString(req.KeyIden),
			KeyAlgo: nullString(req.KeyAlgo),
			Status:  nullString(req.Status),
		}
		svcRes, err := svcModel.Insert(l.ctx, svcData)
		if err != nil {
			return fmt.Errorf("insert t_service: %w", err)
		}
		svcId, err = svcRes.LastInsertId()
		if err != nil {
			return fmt.Errorf("get last insert id: %w", err)
		}

		// 2. 插保护目录(与上面的服务同事务,任一条失败整体回滚)
		for _, dir := range req.Directories {
			path := strings.TrimSpace(dir)
			if path == "" {
				continue
			}
			_, err := dirModel.Insert(l.ctx, &model.TDirectory{
				ServiceId: nullInt64(svcId),
				Directory: nullString(path),
			})
			if err != nil {
				return fmt.Errorf("insert t_directory[%s]: %w", path, err)
			}
		}
		return nil
	})
	if err != nil {
		l.Errorf("create service tx failed: %v", err)
		return nil, err
	}

	// 事务提交后再回查完整实体(CreateAt 由 DB 默认值生成)
	created, err := findService(l.ctx, l.svcCtx.ServiceModel, svcId)
	if err != nil {
		return nil, err
	}

	return toServiceResp(created), nil
}