u
This commit is contained in:
oneao committed
1 parent
0ab2796dd4
commit
f7b0c6ad2f
26 files changed
+1893
-354
No files matched your search
@@ -38,6 +38,34 @@ type MoneyUserCategory struct {
|
||||
CreateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
// 共享空间（一个空间包含多个成员）
|
||||
type Space struct {
|
||||
// 空间ID
|
||||
ID int64
|
||||
// 空间名称
|
||||
Name pgtype.Text
|
||||
// 空间邀请码
|
||||
InviteCode string
|
||||
// 空间创建时间
|
||||
CreateTime pgtype.Timestamp
|
||||
// 空间更新时间
|
||||
UpdateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
// 共享空间成员表
|
||||
type SpaceUser struct {
|
||||
// 主键ID
|
||||
ID int64
|
||||
// 所属空间ID
|
||||
SpaceID int64
|
||||
// 用户ID（唯一，每用户只加入一个空间）
|
||||
UserID int64
|
||||
// 成员角色：0创建者 1管理员 2普通
|
||||
Role int32
|
||||
// 加入空间时间
|
||||
JoinTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type UserInfo struct {
|
||||
ID int64
|
||||
SeqNo pgtype.Int4
|
||||
|
||||
@@ -0,0 +1,141 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: space.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
const deleteSpace = `-- name: DeleteSpace :exec
|
||||
DELETE FROM space WHERE id = $1
|
||||
`
|
||||
|
||||
// 删除分享空间
|
||||
func (q *Queries) DeleteSpace(ctx context.Context, id int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpace, id)
|
||||
return err
|
||||
}
|
||||
|
||||
const getSpaceById = `-- name: GetSpaceById :one
|
||||
SELECT id, name, invite_code, create_time, update_time FROM space WHERE id = $1
|
||||
`
|
||||
|
||||
// 根据id获取分享空间信息
|
||||
func (q *Queries) GetSpaceById(ctx context.Context, id int64) (Space, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceById, id)
|
||||
var i Space
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const getSpaceByInviteCode = `-- name: GetSpaceByInviteCode :one
|
||||
SELECT id, name, invite_code, create_time, update_time FROM space WHERE invite_code = $1
|
||||
`
|
||||
|
||||
// 根据验证码获取空间
|
||||
func (q *Queries) GetSpaceByInviteCode(ctx context.Context, inviteCode string) (Space, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceByInviteCode, inviteCode)
|
||||
var i Space
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const getSpaceByUserId = `-- name: GetSpaceByUserId :one
|
||||
SELECT s.id, s.name, s.invite_code, s.create_time, s.update_time
|
||||
FROM space s
|
||||
JOIN space_user m
|
||||
ON s.id = m.space_id
|
||||
WHERE m.user_id = $1
|
||||
`
|
||||
|
||||
// 查询用户所在的空间信息
|
||||
func (q *Queries) GetSpaceByUserId(ctx context.Context, userID int64) (Space, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceByUserId, userID)
|
||||
var i Space
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
)
|
||||
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)
|
||||
`
|
||||
|
||||
type InsertSpaceParams struct {
|
||||
ID int64
|
||||
InviteCode string
|
||||
}
|
||||
|
||||
// 新增分享空间
|
||||
func (q *Queries) InsertSpace(ctx context.Context, arg InsertSpaceParams) error {
|
||||
_, err := q.db.Exec(ctx, insertSpace, arg.ID, arg.InviteCode)
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,72 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: space_user.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
const deleteSpaceUser = `-- name: DeleteSpaceUser :exec
|
||||
DELETE FROM space_user WHERE user_id = $1
|
||||
`
|
||||
|
||||
// 删除空间下的单个用户
|
||||
func (q *Queries) DeleteSpaceUser(ctx context.Context, userID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceUser, userID)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteSpaceUserAll = `-- name: DeleteSpaceUserAll :exec
|
||||
DELETE FROM space_user WHERE space_id = $1
|
||||
`
|
||||
|
||||
// 删除空间下的所有用户
|
||||
func (q *Queries) DeleteSpaceUserAll(ctx context.Context, spaceID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceUserAll, spaceID)
|
||||
return err
|
||||
}
|
||||
|
||||
const insertSpaceUser = `-- name: InsertSpaceUser :exec
|
||||
INSERT INTO space_user(id,space_id,user_id,role)
|
||||
VALUES ($1,$2,$3,$4)
|
||||
`
|
||||
|
||||
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 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
|
||||
}
|
||||
Reference in new issue
Block a user