167 lines
3.1 KiB
Go
167 lines
3.1 KiB
Go
// 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"
|
|
)
|
|
|
|
// CreateSpace .
|
|
// @router /space/create [POST]
|
|
func CreateSpace(ctx context.Context, c *app.RequestContext) {
|
|
var err error
|
|
var req space.CreateSpaceParams
|
|
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()
|
|
|
|
err = db.WithTx(ctx, func(q *repo.Queries) error {
|
|
// 创建空间
|
|
params := repo.InsertSpaceParams{
|
|
ID: spaceId,
|
|
InviteCode: inviteCode,
|
|
}
|
|
|
|
err = q.InsertSpace(ctx, params)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 将创建用户默认添加到该空间内
|
|
userParams := repo.InsertSpaceUserParams{
|
|
ID: idgen.NextId(),
|
|
SpaceID: spaceId,
|
|
UserID: userID,
|
|
Role: 0,
|
|
}
|
|
err := q.InsertSpaceUser(ctx, userParams)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
resp := &space.CreateSpaceResult{
|
|
Id: spaceId,
|
|
InviteCode: inviteCode,
|
|
}
|
|
|
|
response.Success(c).Data(resp).Send()
|
|
}
|
|
|
|
// DissolveSpace .
|
|
// @router /space/dissolve [POST]
|
|
func DissolveSpace(ctx context.Context, c *app.RequestContext) {
|
|
var err error
|
|
var req space.DissolveSpaceParams
|
|
err = c.BindAndValidate(&req)
|
|
if err != nil {
|
|
c.String(consts.StatusBadRequest, err.Error())
|
|
return
|
|
}
|
|
|
|
userID := middleware.GetUserID(ctx)
|
|
spaceByUserId, err := db.Queries.GetSpaceByUserId(ctx, userID)
|
|
|
|
if err != nil {
|
|
if errorx.IsNotFound(err) {
|
|
response.Success(c).Send()
|
|
return
|
|
} else {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
spaceId := spaceByUserId.ID
|
|
|
|
err = db.WithTx(ctx, func(q *repo.Queries) error {
|
|
err := q.DeleteSpace(ctx, spaceId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = q.DeleteSpaceUserAll(ctx, spaceId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
response.Success(c).Send()
|
|
}
|