68 lines
1.4 KiB
Go
68 lines
1.4 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,join_time,update_time)
|
|
VALUES ($1,$2,$3,$4,$5,NOW(),NOW())
|
|
`
|
|
|
|
type InsertSpaceMemberParams struct {
|
|
ID int64
|
|
SpaceID int64
|
|
UserID int64
|
|
Role int16
|
|
IsDefault int16
|
|
}
|
|
|
|
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,
|
|
)
|
|
return err
|
|
}
|