160 lines
3.6 KiB
Go
160 lines
3.6 KiB
Go
// Code generated by sqlc. DO NOT EDIT.
|
|
// versions:
|
|
// sqlc v1.30.0
|
|
// source: space_user.sql
|
|
|
|
package repo
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
const deleteSpaceUserBySpaceIDAndUserID = `-- name: DeleteSpaceUserBySpaceIDAndUserID :exec
|
|
DELETE FROM space_user
|
|
WHERE space_id = $1
|
|
AND user_id = $2
|
|
`
|
|
|
|
type DeleteSpaceUserBySpaceIDAndUserIDParams struct {
|
|
SpaceID int64
|
|
UserID int64
|
|
}
|
|
|
|
// 删除某个用户在指定空间中的关系
|
|
func (q *Queries) DeleteSpaceUserBySpaceIDAndUserID(ctx context.Context, arg DeleteSpaceUserBySpaceIDAndUserIDParams) error {
|
|
_, err := q.db.Exec(ctx, deleteSpaceUserBySpaceIDAndUserID, arg.SpaceID, arg.UserID)
|
|
return err
|
|
}
|
|
|
|
const deleteSpaceUserByUserID = `-- name: DeleteSpaceUserByUserID :exec
|
|
DELETE FROM space_user
|
|
WHERE user_id = $1
|
|
`
|
|
|
|
// 删除某个用户在所有空间的关系
|
|
func (q *Queries) DeleteSpaceUserByUserID(ctx context.Context, userID int64) error {
|
|
_, err := q.db.Exec(ctx, deleteSpaceUserByUserID, userID)
|
|
return err
|
|
}
|
|
|
|
const deleteSpaceUsersBySpaceID = `-- name: DeleteSpaceUsersBySpaceID :exec
|
|
DELETE FROM space_user
|
|
WHERE space_id = $1
|
|
`
|
|
|
|
// 删除某个空间下的所有用户
|
|
func (q *Queries) DeleteSpaceUsersBySpaceID(ctx context.Context, spaceID int64) error {
|
|
_, err := q.db.Exec(ctx, deleteSpaceUsersBySpaceID, spaceID)
|
|
return err
|
|
}
|
|
|
|
const getSpaceUser = `-- name: GetSpaceUser :one
|
|
SELECT id, space_id, user_id, role, join_time
|
|
FROM space_user
|
|
WHERE user_id = $1
|
|
`
|
|
|
|
// 获取空间用户
|
|
func (q *Queries) GetSpaceUser(ctx context.Context, userID int64) (SpaceUser, error) {
|
|
row := q.db.QueryRow(ctx, getSpaceUser, 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,join_time)
|
|
VALUES ($1,$2,$3,$4,NOW())
|
|
`
|
|
|
|
type InsertSpaceUserParams struct {
|
|
ID int64
|
|
SpaceID int64
|
|
UserID int64
|
|
Role int32
|
|
}
|
|
|
|
// 插入空间用户
|
|
func (q *Queries) InsertSpaceUser(ctx context.Context, arg InsertSpaceUserParams) error {
|
|
_, err := q.db.Exec(ctx, insertSpaceUser,
|
|
arg.ID,
|
|
arg.SpaceID,
|
|
arg.UserID,
|
|
arg.Role,
|
|
)
|
|
return err
|
|
}
|
|
|
|
const listSpaceUsers = `-- name: ListSpaceUsers :many
|
|
SELECT
|
|
u.id AS user_id,
|
|
u.username AS user_name,
|
|
u.avatar AS user_avatar,
|
|
u.last_login_time AS last_login_time,
|
|
su.role
|
|
FROM space_user su
|
|
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 string
|
|
UserAvatar string
|
|
LastLoginTime pgtype.Timestamptz
|
|
Role int32
|
|
}
|
|
|
|
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
|
|
}
|
|
defer rows.Close()
|
|
var items []ListSpaceUsersRow
|
|
for rows.Next() {
|
|
var i ListSpaceUsersRow
|
|
if err := rows.Scan(
|
|
&i.UserID,
|
|
&i.UserName,
|
|
&i.UserAvatar,
|
|
&i.LastLoginTime,
|
|
&i.Role,
|
|
); 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
|
|
WHERE user_id = $2 AND space_id = $3
|
|
`
|
|
|
|
type UpdateSpaceUserRoleParams struct {
|
|
Role int32
|
|
UserID int64
|
|
SpaceID int64
|
|
}
|
|
|
|
// 修改角色
|
|
func (q *Queries) UpdateSpaceUserRole(ctx context.Context, arg UpdateSpaceUserRoleParams) error {
|
|
_, err := q.db.Exec(ctx, updateSpaceUserRole, arg.Role, arg.UserID, arg.SpaceID)
|
|
return err
|
|
}
|