This commit is contained in:
oneao committed 2025-12-02 22:44:12 +08:00
1 parent f7b0c6ad2f
commit e81f605999
23 files changed
+1286 -413

No files matched your search

@@ -18,19 +18,6 @@ FROM space s
ON s.id = m.space_id
WHERE m.user_id = $1;
-- 获取空间下的所有用户信息
-- name: GetSpaceUsers :many
SELECT
u.id as user_id,
u.username,
su.role,
su.join_time
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.join_time ASC;
-- 根据验证码获取空间
-- name: GetSpaceByInviteCode :one
SELECT * FROM space WHERE invite_code = $1;
@@ -17,3 +17,20 @@ UPDATE space_user
SET role = $1
WHERE user_id = $2
AND space_id = $3;
-- 获取空间下的所有用户信息
-- name: ListSpaceUsers :many
SELECT
u.id as user_id,
u.username,
su.role,
su.join_time
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.join_time ASC;
-- name: GetSpaceUserByUserId :one
SELECT * FROM space_user WHERE user_id = $1;
@@ -7,8 +7,6 @@ package repo
import (
"context"
"github.com/jackc/pgx/v5/pgtype"
)
const deleteSpace = `-- name: DeleteSpace :exec
@@ -79,52 +77,6 @@ func (q *Queries) GetSpaceByUserId(ctx context.Context, userID int64) (Space, er
return i, err
}
const getSpaceUsers = `-- name: GetSpaceUsers :many
SELECT
u.id as user_id,
u.username,
su.role,
su.join_time
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.join_time ASC
`
type GetSpaceUsersRow struct {
UserID int64
Username pgtype.Text
Role int32
JoinTime pgtype.Timestamp
}
// 获取空间下的所有用户信息
func (q *Queries) GetSpaceUsers(ctx context.Context, id int64) ([]GetSpaceUsersRow, error) {
rows, err := q.db.Query(ctx, getSpaceUsers, id)
if err != nil {
return nil, err
}
defer rows.Close()
var items []GetSpaceUsersRow
for rows.Next() {
var i GetSpaceUsersRow
if err := rows.Scan(
&i.UserID,
&i.Username,
&i.Role,
&i.JoinTime,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const insertSpace = `-- name: InsertSpace :exec
INSERT INTO space (id,invite_code) VALUES ($1,$2)
`
@@ -7,6 +7,8 @@ package repo
import (
"context"
"github.com/jackc/pgx/v5/pgtype"
)
const deleteSpaceUser = `-- name: DeleteSpaceUser :exec
@@ -29,6 +31,23 @@ func (q *Queries) DeleteSpaceUserAll(ctx context.Context, spaceID int64) error {
return err
}
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)
var i SpaceUser
err := row.Scan(
&i.ID,
&i.SpaceID,
&i.UserID,
&i.Role,
&i.JoinTime,
)
return i, err
}
const insertSpaceUser = `-- name: InsertSpaceUser :exec
INSERT INTO space_user(id,space_id,user_id,role)
VALUES ($1,$2,$3,$4)
@@ -52,6 +71,52 @@ func (q *Queries) InsertSpaceUser(ctx context.Context, arg InsertSpaceUserParams
return err
}
const listSpaceUsers = `-- name: ListSpaceUsers :many
SELECT
u.id as user_id,
u.username,
su.role,
su.join_time
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.join_time ASC
`
type ListSpaceUsersRow struct {
UserID int64
Username pgtype.Text
Role int32
JoinTime pgtype.Timestamp
}
// 获取空间下的所有用户信息
func (q *Queries) ListSpaceUsers(ctx context.Context, id int64) ([]ListSpaceUsersRow, error) {
rows, err := q.db.Query(ctx, listSpaceUsers, id)
if err != nil {
return nil, err
}
defer rows.Close()
var items []ListSpaceUsersRow
for rows.Next() {
var i ListSpaceUsersRow
if err := rows.Scan(
&i.UserID,
&i.Username,
&i.Role,
&i.JoinTime,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const updateSpaceUserRole = `-- name: UpdateSpaceUserRole :exec
UPDATE space_user
SET role = $1
@@ -58,6 +58,9 @@ COMMENT ON COLUMN space.invite_code IS '空间邀请码';
COMMENT ON COLUMN space.create_time IS '空间创建时间';
COMMENT ON COLUMN space.update_time IS '空间更新时间';
ALTER TABLE space
ADD CONSTRAINT space_invite_code_key UNIQUE (invite_code);
-- 空间成员表(每个用户只能加入一个空间)
CREATE TABLE space_user (