go-svc-configure/internal/service/servicemanagement/create_service.go

package servicemanagement

import (
	"context"
	"fmt"

	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-base/model/response"
	"git.x2erp.com/qdy/go-base/util"
	"git.x2erp.com/qdy/go-db/factory/database"
)

// CreateServiceRequest 创建微服务请求
type CreateServiceRequest struct {
	ServiceName string `json:"service_name"`
	Description string `json:"description"`
}

// CreateService 创建微服务
func CreateService(req *CreateServiceRequest, ctx context.Context, dbFactory *database.DBFactory, reqCtx *ctx.RequestContext) *response.QueryResult[int64] {
	logger.Debug(fmt.Sprintf("CreateService-开始创建微服务: %s", req.ServiceName))

	if req.ServiceName == "" {
		logger.ErrorC(reqCtx, "微服务名称不能为空")
		return util.CreateErrorResult[int64]("微服务名称不能为空", reqCtx)
	}

	// 获取数据库连接
	db := dbFactory.GetDB()

	// 开始事务
	tx, err := db.BeginTxx(ctx, nil)
	if err != nil {
		logger.ErrorC(reqCtx, fmt.Sprintf("开始事务失败: %v", err))
		return util.CreateErrorResult[int64](fmt.Sprintf("开始事务失败: %v", err), reqCtx)
	}

	defer func() {
		if p := recover(); p != nil {
			tx.Rollback()
			panic(p)
		}
	}()

	// 检查微服务是否已存在
	var serviceExists bool
	checkQuery := `SELECT COUNT(*) FROM config_startup_svc WHERE service_name = ? AND config_name = '_service_metadata'`
	var count int
	err = tx.GetContext(ctx, &count, checkQuery, req.ServiceName)
	if err == nil && count > 0 {
		serviceExists = true
	}

	if serviceExists {
		tx.Rollback()
		logger.ErrorC(reqCtx, fmt.Sprintf("微服务 %s 已存在", req.ServiceName))
		return util.CreateErrorResult[int64](fmt.Sprintf("微服务 %s 已存在", req.ServiceName), reqCtx)
	}

	// 插入微服务元数据记录
	creator := reqCtx.UserID
	if creator == "" {
		creator = "system"
	}

	metadataID := fmt.Sprintf("config_startup_svc.%s._service_metadata.description", req.ServiceName)
	insertQuery := `
		INSERT INTO config_startup_svc 
		(id, service_name, config_name, yaml_name, yaml_value, creator, created_at)
		VALUES (?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP)
	`

	result, err := tx.ExecContext(ctx, insertQuery,
		metadataID,
		req.ServiceName,
		"_service_metadata",
		"description",
		req.Description,
		creator,
	)
	if err != nil {
		tx.Rollback()
		logger.ErrorC(reqCtx, fmt.Sprintf("创建微服务失败: %v", err))
		return util.CreateErrorResult[int64](fmt.Sprintf("创建微服务失败: %v", err), reqCtx)
	}

	rowsAffected, err := result.RowsAffected()
	if err != nil {
		tx.Rollback()
		logger.ErrorC(reqCtx, fmt.Sprintf("获取影响行数失败: %v", err))
		return util.CreateErrorResult[int64](fmt.Sprintf("获取影响行数失败: %v", err), reqCtx)
	}

	// 提交事务
	if err := tx.Commit(); err != nil {
		logger.ErrorC(reqCtx, fmt.Sprintf("提交事务失败: %v", err))
		return util.CreateErrorResult[int64](fmt.Sprintf("提交事务失败: %v", err), reqCtx)
	}

	logger.Debug(fmt.Sprintf("成功创建微服务: %s，描述: %s", req.ServiceName, req.Description))
	return util.CreateSuccessResultData[int64](rowsAffected, reqCtx)
}