u

code/base-project/base-go-v3/biz/handler/space/space_service.go

@@ -0,0 +1,95 @@
+// Code generated by hertz generator.
+
+package space
+
+import (
+	"cn/oneao/base-go/biz/router/middleware"
+	"cn/oneao/base-go/db"
+	"cn/oneao/base-go/db/repo"
+	"cn/oneao/base-go/utils"
+	"cn/oneao/base-go/utils/errorx"
+	"cn/oneao/base-go/utils/idgen"
+	"cn/oneao/base-go/utils/response"
+	"context"
+
+	space "cn/oneao/base-go/biz/model/space"
+	"github.com/cloudwego/hertz/pkg/app"
+	"github.com/cloudwego/hertz/pkg/protocol/consts"
+)
+
+// InsertSpace .
+// @router /space/insert [POST]
+func InsertSpace(ctx context.Context, c *app.RequestContext) {
+	var err error
+	var req space.InsertSpaceParams
+	err = c.BindAndValidate(&req)
+	if err != nil {
+		c.String(consts.StatusBadRequest, err.Error())
+		return
+	}
+
+	userID := middleware.GetUserID(ctx)
+
+	// 检查该用户是否已经加入空间了
+	_, err = db.Queries.GetSpaceByUserId(ctx, userID)
+
+	if err != nil {
+		if errorx.IsNotFound(err) {
+			// 忽略
+		} else {
+			errorx.AddError(c, err)
+			return
+		}
+	} else {
+		response.Fail(c).Message("请先退出已加入的空间").Send()
+		return
+	}
+
+	// 生成邀请码
+	var inviteCode string
+
+	for {
+		// 生成邀请码
+		code, err := utils.GenerateCode(6)
+		if err != nil {
+			errorx.AddError(c, err)
+			return
+		}
+
+		// 检查数据库是否存在
+		_, err = db.Queries.GetSpaceByInviteCode(ctx, code)
+		if err != nil {
+			if errorx.IsNotFound(err) {
+				// 不存在，唯一，直接使用
+				inviteCode = code
+				break
+			} else {
+				// 系统错误
+				errorx.AddError(c, err)
+				return
+			}
+		}
+
+		// 如果存在，继续循环重新生成
+	}
+
+	// 创建空间
+	spaceId := idgen.NextId()
+	params := repo.InsertSpaceParams{
+		ID:         spaceId,
+		InviteCode: inviteCode,
+	}
+
+	err = db.Queries.InsertSpace(ctx, params)
+
+	if err != nil {
+		errorx.AddError(c, err)
+		return
+	}
+
+	var resp space.InsertSpaceResult
+	resp.Id = spaceId
+	resp.InviteCode = inviteCode
+
+	response.Success(c).Data(resp).Send()
+}