u
This commit is contained in:
oneao committed
1 parent
8663cb2e3d
commit
83cbf18572
34 files changed
+1938
-1350
No files matched your search
@@ -11,32 +11,50 @@ import (
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
const countUsersInSpace = `-- name: CountUsersInSpace :one
|
||||
SELECT COUNT(id)
|
||||
FROM space_user
|
||||
WHERE space_id = $1
|
||||
`
|
||||
|
||||
// 获取空间下的人数
|
||||
func (q *Queries) CountUsersInSpace(ctx context.Context, spaceID int64) (int64, error) {
|
||||
row := q.db.QueryRow(ctx, countUsersInSpace, spaceID)
|
||||
var count int64
|
||||
err := row.Scan(&count)
|
||||
return count, err
|
||||
}
|
||||
|
||||
const deleteSpaceUser = `-- name: DeleteSpaceUser :exec
|
||||
DELETE FROM space_user WHERE user_id = $1
|
||||
DELETE FROM space_user
|
||||
WHERE user_id = $1 AND space_id = $2
|
||||
`
|
||||
|
||||
// 删除空间下的单个用户
|
||||
func (q *Queries) DeleteSpaceUser(ctx context.Context, userID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceUser, userID)
|
||||
type DeleteSpaceUserParams struct {
|
||||
UserID int64
|
||||
SpaceID int64
|
||||
}
|
||||
|
||||
// 修改空间用户
|
||||
func (q *Queries) DeleteSpaceUser(ctx context.Context, arg DeleteSpaceUserParams) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceUser, arg.UserID, arg.SpaceID)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteSpaceUserAll = `-- name: DeleteSpaceUserAll :exec
|
||||
DELETE FROM space_user WHERE space_id = $1
|
||||
const getSpaceUser = `-- name: GetSpaceUser :one
|
||||
SELECT id, space_id, user_id, role, join_time
|
||||
FROM space_user
|
||||
WHERE user_id = $1 AND space_id = $2
|
||||
`
|
||||
|
||||
// 删除空间下的所有用户
|
||||
func (q *Queries) DeleteSpaceUserAll(ctx context.Context, spaceID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceUserAll, spaceID)
|
||||
return err
|
||||
type GetSpaceUserParams struct {
|
||||
UserID int64
|
||||
SpaceID int64
|
||||
}
|
||||
|
||||
const getSpaceUserByUserId = `-- name: GetSpaceUserByUserId :one
|
||||
SELECT id, space_id, user_id, role, join_time FROM space_user WHERE user_id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) GetSpaceUserByUserId(ctx context.Context, userID int64) (SpaceUser, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceUserByUserId, userID)
|
||||
// 获取空间用户
|
||||
func (q *Queries) GetSpaceUser(ctx context.Context, arg GetSpaceUserParams) (SpaceUser, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceUser, arg.UserID, arg.SpaceID)
|
||||
var i SpaceUser
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
@@ -48,45 +66,29 @@ func (q *Queries) GetSpaceUserByUserId(ctx context.Context, userID int64) (Space
|
||||
return i, err
|
||||
}
|
||||
|
||||
const getUserSpaceDetail = `-- name: GetUserSpaceDetail :one
|
||||
SELECT
|
||||
s.id AS space_id,
|
||||
s.name AS space_name,
|
||||
s.invite_code AS space_invite_code,
|
||||
su.id AS space_user_id,
|
||||
su.user_id,
|
||||
su.role
|
||||
FROM space_user su
|
||||
JOIN space s ON su.space_id = s.id
|
||||
WHERE su.user_id = $1
|
||||
const hasUserInSpace = `-- name: HasUserInSpace :one
|
||||
SELECT EXISTS (
|
||||
SELECT 1
|
||||
FROM space_user
|
||||
WHERE user_id = $1 AND space_id = $2
|
||||
)
|
||||
`
|
||||
|
||||
type GetUserSpaceDetailRow struct {
|
||||
SpaceID int64
|
||||
SpaceName pgtype.Text
|
||||
SpaceInviteCode string
|
||||
SpaceUserID int64
|
||||
UserID int64
|
||||
Role int32
|
||||
type HasUserInSpaceParams struct {
|
||||
UserID int64
|
||||
SpaceID int64
|
||||
}
|
||||
|
||||
func (q *Queries) GetUserSpaceDetail(ctx context.Context, userID int64) (GetUserSpaceDetailRow, error) {
|
||||
row := q.db.QueryRow(ctx, getUserSpaceDetail, userID)
|
||||
var i GetUserSpaceDetailRow
|
||||
err := row.Scan(
|
||||
&i.SpaceID,
|
||||
&i.SpaceName,
|
||||
&i.SpaceInviteCode,
|
||||
&i.SpaceUserID,
|
||||
&i.UserID,
|
||||
&i.Role,
|
||||
)
|
||||
return i, err
|
||||
func (q *Queries) HasUserInSpace(ctx context.Context, arg HasUserInSpaceParams) (bool, error) {
|
||||
row := q.db.QueryRow(ctx, hasUserInSpace, arg.UserID, arg.SpaceID)
|
||||
var exists bool
|
||||
err := row.Scan(&exists)
|
||||
return exists, err
|
||||
}
|
||||
|
||||
const insertSpaceUser = `-- name: InsertSpaceUser :exec
|
||||
INSERT INTO space_user(id,space_id,user_id,role)
|
||||
VALUES ($1,$2,$3,$4)
|
||||
INSERT INTO space_user(id,space_id,user_id,role,join_time)
|
||||
VALUES ($1,$2,$3,$4,NOW())
|
||||
`
|
||||
|
||||
type InsertSpaceUserParams struct {
|
||||
@@ -96,7 +98,7 @@ type InsertSpaceUserParams struct {
|
||||
Role int32
|
||||
}
|
||||
|
||||
// 加入空间
|
||||
// 插入空间用户
|
||||
func (q *Queries) InsertSpaceUser(ctx context.Context, arg InsertSpaceUserParams) error {
|
||||
_, err := q.db.Exec(ctx, insertSpaceUser,
|
||||
arg.ID,
|
||||
@@ -109,31 +111,25 @@ func (q *Queries) InsertSpaceUser(ctx context.Context, arg InsertSpaceUserParams
|
||||
|
||||
const listSpaceUsers = `-- name: ListSpaceUsers :many
|
||||
SELECT
|
||||
u.id AS user_id,
|
||||
u.username,
|
||||
u.active_time,
|
||||
su.role,
|
||||
su.join_time
|
||||
u.id AS user_id,
|
||||
u.username AS user_name,
|
||||
u.avatar AS user_avatar,
|
||||
su.role
|
||||
FROM space_user su
|
||||
JOIN space s ON su.space_id = s.id
|
||||
JOIN user_info u ON su.user_id = u.id
|
||||
WHERE s.id = $1
|
||||
ORDER BY su.role ASC,
|
||||
u.active_time DESC,
|
||||
u.username ASC
|
||||
INNER JOIN "user" u ON su.user_id = u.id
|
||||
WHERE su.space_id = $1
|
||||
ORDER BY su.role DESC, u.username ASC
|
||||
`
|
||||
|
||||
type ListSpaceUsersRow struct {
|
||||
UserID int64
|
||||
Username pgtype.Text
|
||||
ActiveTime pgtype.Timestamp
|
||||
UserName string
|
||||
UserAvatar string
|
||||
Role int32
|
||||
JoinTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
// 获取空间下的所有用户信息（按 role, active_time, username 排序）
|
||||
func (q *Queries) ListSpaceUsers(ctx context.Context, id int64) ([]ListSpaceUsersRow, error) {
|
||||
rows, err := q.db.Query(ctx, listSpaceUsers, id)
|
||||
func (q *Queries) ListSpaceUsers(ctx context.Context, spaceID int64) ([]ListSpaceUsersRow, error) {
|
||||
rows, err := q.db.Query(ctx, listSpaceUsers, spaceID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -143,10 +139,48 @@ func (q *Queries) ListSpaceUsers(ctx context.Context, id int64) ([]ListSpaceUser
|
||||
var i ListSpaceUsersRow
|
||||
if err := rows.Scan(
|
||||
&i.UserID,
|
||||
&i.Username,
|
||||
&i.ActiveTime,
|
||||
&i.UserName,
|
||||
&i.UserAvatar,
|
||||
&i.Role,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const listUserSpaces = `-- name: ListUserSpaces :many
|
||||
SELECT s.id, s.name, s.invite_code, su.role
|
||||
FROM space s
|
||||
JOIN space_user su ON s.id = su.space_id
|
||||
WHERE su.user_id = $1
|
||||
`
|
||||
|
||||
type ListUserSpacesRow struct {
|
||||
ID int64
|
||||
Name pgtype.Text
|
||||
InviteCode string
|
||||
Role int32
|
||||
}
|
||||
|
||||
func (q *Queries) ListUserSpaces(ctx context.Context, userID int64) ([]ListUserSpacesRow, error) {
|
||||
rows, err := q.db.Query(ctx, listUserSpaces, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []ListUserSpacesRow
|
||||
for rows.Next() {
|
||||
var i ListUserSpacesRow
|
||||
if err := rows.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.Role,
|
||||
&i.JoinTime,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -161,8 +195,7 @@ func (q *Queries) ListSpaceUsers(ctx context.Context, id int64) ([]ListSpaceUser
|
||||
const updateSpaceUserRole = `-- name: UpdateSpaceUserRole :exec
|
||||
UPDATE space_user
|
||||
SET role = $1
|
||||
WHERE user_id = $2
|
||||
AND space_id = $3
|
||||
WHERE user_id = $2 AND space_id = $3
|
||||
`
|
||||
|
||||
type UpdateSpaceUserRoleParams struct {
|
||||
|
||||
Reference in new issue
Block a user