127 lines
3.0 KiB
Go
127 lines
3.0 KiB
Go
// Code generated by sqlc. DO NOT EDIT.
|
|
// versions:
|
|
// sqlc v1.30.0
|
|
// source: space_member.sql
|
|
|
|
package repo
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
const countUserSpaces = `-- name: CountUserSpaces :one
|
|
SELECT COUNT(1)
|
|
FROM space_member
|
|
WHERE user_id = $1
|
|
`
|
|
|
|
// 合计某个用户的空间数量
|
|
func (q *Queries) CountUserSpaces(ctx context.Context, userID int64) (int64, error) {
|
|
row := q.db.QueryRow(ctx, countUserSpaces, userID)
|
|
var count int64
|
|
err := row.Scan(&count)
|
|
return count, err
|
|
}
|
|
|
|
const getUserSpaceRole = `-- name: GetUserSpaceRole :one
|
|
SELECT role
|
|
FROM space_member
|
|
WHERE user_id = $1
|
|
AND space_id = $2
|
|
`
|
|
|
|
type GetUserSpaceRoleParams struct {
|
|
UserID int64
|
|
SpaceID int64
|
|
}
|
|
|
|
// 获取用户在指定空间的角色
|
|
func (q *Queries) GetUserSpaceRole(ctx context.Context, arg GetUserSpaceRoleParams) (int16, error) {
|
|
row := q.db.QueryRow(ctx, getUserSpaceRole, arg.UserID, arg.SpaceID)
|
|
var role int16
|
|
err := row.Scan(&role)
|
|
return role, err
|
|
}
|
|
|
|
const insertSpaceMember = `-- name: InsertSpaceMember :exec
|
|
INSERT INTO space_member(id,space_id,user_id,role,is_default,member_name,sort_number,join_time,update_time)
|
|
VALUES ($1,$2,$3,$4,$5,$6,$7,NOW(),NOW())
|
|
`
|
|
|
|
type InsertSpaceMemberParams struct {
|
|
ID int64
|
|
SpaceID int64
|
|
UserID int64
|
|
Role int16
|
|
IsDefault int16
|
|
MemberName string
|
|
SortNumber int32
|
|
}
|
|
|
|
// 插入空间用户
|
|
func (q *Queries) InsertSpaceMember(ctx context.Context, arg InsertSpaceMemberParams) error {
|
|
_, err := q.db.Exec(ctx, insertSpaceMember,
|
|
arg.ID,
|
|
arg.SpaceID,
|
|
arg.UserID,
|
|
arg.Role,
|
|
arg.IsDefault,
|
|
arg.MemberName,
|
|
arg.SortNumber,
|
|
)
|
|
return err
|
|
}
|
|
|
|
const updateOtherSpacesIsDefault = `-- name: UpdateOtherSpacesIsDefault :exec
|
|
UPDATE space_member
|
|
SET is_default = $1
|
|
WHERE user_id = $2
|
|
AND space_id != $3
|
|
`
|
|
|
|
type UpdateOtherSpacesIsDefaultParams struct {
|
|
IsDefault int16
|
|
UserID int64
|
|
SpaceID int64
|
|
}
|
|
|
|
// 将用户的其它空间更新为非默认
|
|
func (q *Queries) UpdateOtherSpacesIsDefault(ctx context.Context, arg UpdateOtherSpacesIsDefaultParams) error {
|
|
_, err := q.db.Exec(ctx, updateOtherSpacesIsDefault, arg.IsDefault, arg.UserID, arg.SpaceID)
|
|
return err
|
|
}
|
|
|
|
const updateSpaceIsDefault = `-- name: UpdateSpaceIsDefault :exec
|
|
UPDATE space_member SET is_default = $1
|
|
WHERE space_id = $2 AND user_id = $3
|
|
`
|
|
|
|
type UpdateSpaceIsDefaultParams struct {
|
|
IsDefault int16
|
|
SpaceID int64
|
|
UserID int64
|
|
}
|
|
|
|
// 更新空间是否为默认
|
|
func (q *Queries) UpdateSpaceIsDefault(ctx context.Context, arg UpdateSpaceIsDefaultParams) error {
|
|
_, err := q.db.Exec(ctx, updateSpaceIsDefault, arg.IsDefault, arg.SpaceID, arg.UserID)
|
|
return err
|
|
}
|
|
|
|
const updateSpaceMemberName = `-- name: UpdateSpaceMemberName :exec
|
|
UPDATE space_member SET member_name = $1
|
|
WHERE space_id = $2 AND user_id = $3
|
|
`
|
|
|
|
type UpdateSpaceMemberNameParams struct {
|
|
MemberName string
|
|
SpaceID int64
|
|
UserID int64
|
|
}
|
|
|
|
// 更新用户在空间内的名称
|
|
func (q *Queries) UpdateSpaceMemberName(ctx context.Context, arg UpdateSpaceMemberNameParams) error {
|
|
_, err := q.db.Exec(ctx, updateSpaceMemberName, arg.MemberName, arg.SpaceID, arg.UserID)
|
|
return err
|
|
}
|