u
This commit is contained in:
oneao committed
1 parent
835958886e
commit
d04d22723b
155 files changed
+543
-9793
No files matched your search
@@ -1,123 +0,0 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"allapp/conf"
|
||||
"allapp/db/repo"
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
var (
|
||||
DB *pgxpool.Pool
|
||||
Queries *repo.Queries
|
||||
)
|
||||
|
||||
type SQLCBatch interface {
|
||||
Exec(func(int, error))
|
||||
}
|
||||
|
||||
func InitDB() {
|
||||
cfg := conf.GetConf().Pgsql
|
||||
|
||||
sslMode := "disable"
|
||||
if cfg.SllMode {
|
||||
sslMode = "require"
|
||||
}
|
||||
|
||||
dsn := fmt.Sprintf(
|
||||
"postgres://%s:%s@%s:%d/%s?sslmode=%s&TimeZone=%s",
|
||||
cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.Dbname, sslMode, cfg.TimeZone,
|
||||
)
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// 使用可修改配置
|
||||
cfgPool, err := pgxpool.ParseConfig(dsn)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// 给连接池参数设置默认值
|
||||
if cfg.MaxOpenConns <= 0 {
|
||||
cfg.MaxOpenConns = 10
|
||||
}
|
||||
if cfg.MaxIdleConns <= 0 {
|
||||
cfg.MaxIdleConns = 5
|
||||
}
|
||||
if cfg.ConnMaxLifetime <= 0 {
|
||||
cfg.ConnMaxLifetime = 30 * time.Minute
|
||||
}
|
||||
|
||||
cfgPool.MaxConns = cfg.MaxOpenConns
|
||||
cfgPool.MinConns = cfg.MaxIdleConns
|
||||
cfgPool.MaxConnLifetime = cfg.ConnMaxLifetime
|
||||
|
||||
pool, err := pgxpool.NewWithConfig(ctx, cfgPool)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// 测试连接
|
||||
if err := pool.Ping(ctx); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
DB = pool
|
||||
Queries = repo.New(DB)
|
||||
}
|
||||
|
||||
// WithTx 执行事务，支持 panic 和 error 自动回滚
|
||||
// 默认使用全局 DB，调用更简洁
|
||||
func WithTx(ctx context.Context, fn func(q *repo.Queries) error) (err error) {
|
||||
return WithTxPool(ctx, DB, fn)
|
||||
}
|
||||
|
||||
// WithTxPool 支持自定义连接池
|
||||
func WithTxPool(ctx context.Context, pool *pgxpool.Pool, fn func(q *repo.Queries) error) (err error) {
|
||||
tx, err := pool.BeginTx(ctx, pgx.TxOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer func() {
|
||||
if p := recover(); p != nil {
|
||||
// panic 时回滚事务
|
||||
if rbErr := tx.Rollback(ctx); rbErr != nil {
|
||||
fmt.Printf("rollback failed during panic: %v\n", rbErr)
|
||||
}
|
||||
panic(p)
|
||||
} else if err != nil {
|
||||
// 回滚事务，并捕获 rollback 错误
|
||||
if rbErr := tx.Rollback(ctx); rbErr != nil {
|
||||
err = fmt.Errorf("rollback failed: %v, original error: %w", rbErr, err)
|
||||
}
|
||||
} else {
|
||||
// 提交事务，并捕获 commit 错误
|
||||
if commitErr := tx.Commit(ctx); commitErr != nil {
|
||||
err = fmt.Errorf("commit failed: %w", commitErr)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
q := Queries.WithTx(tx)
|
||||
err = fn(q)
|
||||
return err
|
||||
}
|
||||
|
||||
// RunBatch 通用批处理执行器（适配所有 sqlc Batch）
|
||||
func RunBatch(ctx context.Context, batch SQLCBatch) error {
|
||||
var firstErr error
|
||||
|
||||
batch.Exec(func(i int, err error) {
|
||||
if err != nil && firstErr == nil {
|
||||
firstErr = err
|
||||
}
|
||||
})
|
||||
|
||||
return firstErr
|
||||
}
|
||||
@@ -1,110 +0,0 @@
|
||||
-- name: ListMoneyRecords :many
|
||||
SELECT
|
||||
mr.*,
|
||||
u.username AS username
|
||||
FROM money_record mr
|
||||
LEFT JOIN space_member su
|
||||
ON su.user_id = mr.user_id
|
||||
LEFT JOIN "user" u
|
||||
ON u.id = mr.user_id
|
||||
WHERE
|
||||
mr.record_time BETWEEN $2 AND $3
|
||||
AND (
|
||||
-- 空间账本：属于当前空间的用户
|
||||
su.user_id IS NOT NULL
|
||||
-- 个人账本：当前用户
|
||||
OR mr.user_id = $1
|
||||
)
|
||||
ORDER BY
|
||||
mr.record_time DESC,
|
||||
mr.create_time DESC;
|
||||
|
||||
|
||||
-- name: InsertMoneyRecord :exec
|
||||
INSERT INTO money_record (
|
||||
id,
|
||||
category_id,
|
||||
user_id,
|
||||
amount,
|
||||
remark,
|
||||
record_time,
|
||||
create_time,
|
||||
update_time
|
||||
) VALUES (
|
||||
$1, $2, $3, $4, $5,$6,NOW(),NOW()
|
||||
);
|
||||
|
||||
-- name: GetMoneyRecordById :one
|
||||
SELECT * FROM money_record WHERE id = $1;
|
||||
|
||||
-- name: UpdateMoneyRecord :exec
|
||||
UPDATE money_record
|
||||
SET
|
||||
category_id = $2,
|
||||
amount = $3,
|
||||
remark = $4,
|
||||
record_time = $5,
|
||||
update_time = NOW()
|
||||
WHERE id = $1
|
||||
RETURNING *;
|
||||
|
||||
-- name: DeleteMoneyRecord :exec
|
||||
DELETE FROM money_record WHERE id = $1;
|
||||
|
||||
-- name: DeleteMoneyRecordByCategoryId :exec
|
||||
DELETE FROM money_record WHERE user_id = $1 AND category_id = $2;
|
||||
|
||||
-- 折线图：按天、月、年汇总金额
|
||||
-- name: ListMoneyRecordsByLineAnalysis :many
|
||||
SELECT
|
||||
CASE
|
||||
WHEN $1 = 0 THEN to_char(mr.record_time, 'DD') -- 日分析，转换为文本
|
||||
WHEN $1 = 1 THEN to_char(mr.record_time, 'MM') -- 月分析，转换为文本
|
||||
WHEN $1 = 2 THEN to_char(mr.record_time, 'YYYY') -- 年分析，转换为文本
|
||||
END AS period,
|
||||
SUM(mr.amount)::DECIMAL AS total_amount
|
||||
FROM
|
||||
money_record mr
|
||||
LEFT JOIN
|
||||
money_user_category muc ON mr.category_id = muc.id -- 连接 money_user_category 获取 type
|
||||
LEFT JOIN
|
||||
space_member su ON su.user_id = $2 -- 连接 space_member 表，查找该用户在空间中的信息
|
||||
LEFT JOIN
|
||||
space s ON s.id = su.space_id -- 连接 space 表，获取空间信息
|
||||
WHERE
|
||||
(
|
||||
(su.user_id IS NOT NULL) -- 用户在空间，查询该空间内所有人的数据
|
||||
OR (mr.user_id = $1) -- 用户不在空间，查询该用户个人的账本数据
|
||||
)
|
||||
AND mr.record_time BETWEEN $3 AND $4 -- 时间范围
|
||||
AND muc.type = $5 -- 收入或支出类型
|
||||
GROUP BY
|
||||
period
|
||||
ORDER BY
|
||||
period;
|
||||
|
||||
|
||||
-- 饼图：按类别汇总金额
|
||||
-- name: ListMoneyRecordsByPieAnalysis :many
|
||||
SELECT
|
||||
muc.name,
|
||||
SUM(mr.amount)::DECIMAL AS total_amount
|
||||
FROM
|
||||
money_record mr
|
||||
LEFT JOIN
|
||||
money_user_category muc ON mr.category_id = muc.id
|
||||
LEFT JOIN
|
||||
space_member su ON su.user_id = $1 -- 连接 space_member 表，查找该用户在空间中的信息
|
||||
LEFT JOIN
|
||||
space s ON s.id = su.space_id -- 连接 space 表，获取空间信息
|
||||
WHERE
|
||||
(
|
||||
(su.user_id IS NOT NULL) -- 用户在空间，查询该空间内所有人的数据
|
||||
OR (mr.user_id = $1) -- 用户不在空间，查询该用户个人的账本数据
|
||||
)
|
||||
AND mr.record_time BETWEEN $2 AND $3 -- 时间范围
|
||||
AND muc.type = $4 -- 收入或支出类型
|
||||
GROUP BY
|
||||
muc.name
|
||||
ORDER BY
|
||||
total_amount DESC;
|
||||
@@ -1,2 +0,0 @@
|
||||
-- name: ListMoneySysCategory :many
|
||||
SELECT * FROM money_sys_category;
|
||||
@@ -1,30 +0,0 @@
|
||||
-- name: BatchInsertMoneyUserCategories :copyfrom
|
||||
INSERT INTO money_user_category (id, user_id, name, icon, type, sort_number)
|
||||
VALUES ($1, $2, $3, $4, $5, $6);
|
||||
|
||||
-- name: FindUserMoneyCategoriesByUserID :many
|
||||
SELECT *
|
||||
FROM money_user_category
|
||||
WHERE user_id = $1;
|
||||
|
||||
-- name: DeleteMoneyCategoryById :exec
|
||||
DELETE FROM money_user_category
|
||||
WHERE id = $1;
|
||||
|
||||
-- name: BatchUpdateMoneyCategorySort :batchexec
|
||||
UPDATE money_user_category
|
||||
SET sort_number = $2
|
||||
WHERE id = $1;
|
||||
|
||||
-- name: DeleteMoneyCategoryByUserId :exec
|
||||
DELETE FROM money_user_category
|
||||
WHERE user_id = $1;
|
||||
|
||||
-- name: InsertMoneyUserCategory :exec
|
||||
INSERT INTO money_user_category (id, user_id, name, icon, type, sort_number,create_time,update_time)
|
||||
VALUES ($1, $2, $3, $4, $5, $6,NOW(),NOW());
|
||||
|
||||
-- name: UpdateMoneyCategory :exec
|
||||
UPDATE money_user_category
|
||||
SET icon = $2,name = $3,update_time = NOW()
|
||||
WHERE id = $1;
|
||||
@@ -1,26 +0,0 @@
|
||||
-- 新增分享空间
|
||||
-- name: InsertSpace :exec
|
||||
INSERT INTO space (
|
||||
id,name,invite_code,create_time,update_time
|
||||
) VALUES ($1,$2,$3,NOW(),NOW());
|
||||
|
||||
-- 删除分享空间
|
||||
-- name: DeleteSpace :exec
|
||||
DELETE FROM space WHERE id = $1;
|
||||
|
||||
-- 根据id获取分享空间信息
|
||||
-- name: GetSpaceById :one
|
||||
SELECT * FROM space WHERE id = $1;
|
||||
|
||||
|
||||
-- 查询用户所在的空间信息
|
||||
-- name: GetSpaceByUserId :one
|
||||
SELECT s.*, m.role
|
||||
FROM space s
|
||||
JOIN space_member m
|
||||
ON s.id = m.space_id
|
||||
WHERE m.user_id = $1;
|
||||
|
||||
-- 根据邀请码查询空间
|
||||
-- name: FindSpaceByInviteCode :one
|
||||
SELECT * FROM space WHERE invite_code = $1;
|
||||
@@ -1,44 +0,0 @@
|
||||
-- 插入空间用户
|
||||
-- name: InsertSpaceMember :exec
|
||||
INSERT INTO space_member(id,space_id,user_id,role,join_time)
|
||||
VALUES ($1,$2,$3,$4,NOW());
|
||||
|
||||
-- name: ListSpaceMembers :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_member su
|
||||
INNER JOIN "user" u ON su.user_id = u.id
|
||||
WHERE su.space_id = $1
|
||||
ORDER BY su.role DESC, u.username ASC;
|
||||
|
||||
-- 获取空间用户
|
||||
-- name: GetSpaceMember :one
|
||||
SELECT *
|
||||
FROM space_member
|
||||
WHERE user_id = $1;
|
||||
|
||||
-- 删除某个用户在所有空间的关系
|
||||
-- name: DeleteSpaceMemberByUserID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE user_id = $1;
|
||||
|
||||
-- 删除某个用户在指定空间中的关系
|
||||
-- name: DeleteSpaceMemberBySpaceIDAndUserID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE space_id = $1
|
||||
AND user_id = $2;
|
||||
|
||||
-- 删除某个空间下的所有用户
|
||||
-- name: DeleteSpaceMembersBySpaceID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE space_id = $1;
|
||||
|
||||
-- 修改角色
|
||||
-- name: UpdateSpaceMemberRole :exec
|
||||
UPDATE space_member
|
||||
SET role = $1
|
||||
WHERE user_id = $2 AND space_id = $3;
|
||||
@@ -1,32 +0,0 @@
|
||||
-- name: FindUserByOpenID :one
|
||||
SELECT
|
||||
u.id AS user_id,
|
||||
u.username,
|
||||
u.avatar,
|
||||
u.status,
|
||||
o.type AS login_type,
|
||||
o.openid
|
||||
FROM user_oauth o
|
||||
JOIN "user" u ON u.id = o.user_id
|
||||
WHERE o.openid = $1
|
||||
AND o.type = $2
|
||||
LIMIT 1;
|
||||
|
||||
-- name: InsertUser :exec
|
||||
INSERT INTO "user" (
|
||||
id,
|
||||
username,
|
||||
avatar,
|
||||
status,
|
||||
create_time,
|
||||
update_time,
|
||||
last_login_time
|
||||
) VALUES (
|
||||
$1, $2, $3,$4, NOW(), NOW(),now()
|
||||
)
|
||||
RETURNING id;
|
||||
|
||||
-- name: UpdateUserLastLoginTime :exec
|
||||
UPDATE "user"
|
||||
SET last_login_time = NOW()
|
||||
WHERE id = $1;
|
||||
@@ -1,13 +0,0 @@
|
||||
-- name: InsertUserOAuth :exec
|
||||
INSERT INTO user_oauth (
|
||||
id,
|
||||
user_id,
|
||||
type,
|
||||
openid,
|
||||
create_time,
|
||||
update_time
|
||||
) VALUES (
|
||||
$1, $2, $3, $4, NOW(), NOW()
|
||||
);
|
||||
|
||||
|
||||
@@ -1,68 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: batch.go
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrBatchAlreadyClosed = errors.New("batch already closed")
|
||||
)
|
||||
|
||||
const batchUpdateMoneyCategorySort = `-- name: BatchUpdateMoneyCategorySort :batchexec
|
||||
UPDATE money_user_category
|
||||
SET sort_number = $2
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
type BatchUpdateMoneyCategorySortBatchResults struct {
|
||||
br pgx.BatchResults
|
||||
tot int
|
||||
closed bool
|
||||
}
|
||||
|
||||
type BatchUpdateMoneyCategorySortParams struct {
|
||||
ID int64
|
||||
SortNumber int32
|
||||
}
|
||||
|
||||
func (q *Queries) BatchUpdateMoneyCategorySort(ctx context.Context, arg []BatchUpdateMoneyCategorySortParams) *BatchUpdateMoneyCategorySortBatchResults {
|
||||
batch := &pgx.Batch{}
|
||||
for _, a := range arg {
|
||||
vals := []interface{}{
|
||||
a.ID,
|
||||
a.SortNumber,
|
||||
}
|
||||
batch.Queue(batchUpdateMoneyCategorySort, vals...)
|
||||
}
|
||||
br := q.db.SendBatch(ctx, batch)
|
||||
return &BatchUpdateMoneyCategorySortBatchResults{br, len(arg), false}
|
||||
}
|
||||
|
||||
func (b *BatchUpdateMoneyCategorySortBatchResults) Exec(f func(int, error)) {
|
||||
defer b.br.Close()
|
||||
for t := 0; t < b.tot; t++ {
|
||||
if b.closed {
|
||||
if f != nil {
|
||||
f(t, ErrBatchAlreadyClosed)
|
||||
}
|
||||
continue
|
||||
}
|
||||
_, err := b.br.Exec()
|
||||
if f != nil {
|
||||
f(t, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (b *BatchUpdateMoneyCategorySortBatchResults) Close() error {
|
||||
b.closed = true
|
||||
return b.br.Close()
|
||||
}
|
||||
@@ -1,47 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: copyfrom.go
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
// iteratorForBatchInsertMoneyUserCategories implements pgx.CopyFromSource.
|
||||
type iteratorForBatchInsertMoneyUserCategories struct {
|
||||
rows []BatchInsertMoneyUserCategoriesParams
|
||||
skippedFirstNextCall bool
|
||||
}
|
||||
|
||||
func (r *iteratorForBatchInsertMoneyUserCategories) Next() bool {
|
||||
if len(r.rows) == 0 {
|
||||
return false
|
||||
}
|
||||
if !r.skippedFirstNextCall {
|
||||
r.skippedFirstNextCall = true
|
||||
return true
|
||||
}
|
||||
r.rows = r.rows[1:]
|
||||
return len(r.rows) > 0
|
||||
}
|
||||
|
||||
func (r iteratorForBatchInsertMoneyUserCategories) Values() ([]interface{}, error) {
|
||||
return []interface{}{
|
||||
r.rows[0].ID,
|
||||
r.rows[0].UserID,
|
||||
r.rows[0].Name,
|
||||
r.rows[0].Icon,
|
||||
r.rows[0].Type,
|
||||
r.rows[0].SortNumber,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (r iteratorForBatchInsertMoneyUserCategories) Err() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (q *Queries) BatchInsertMoneyUserCategories(ctx context.Context, arg []BatchInsertMoneyUserCategoriesParams) (int64, error) {
|
||||
return q.db.CopyFrom(ctx, []string{"money_user_category"}, []string{"id", "user_id", "name", "icon", "type", "sort_number"}, &iteratorForBatchInsertMoneyUserCategories{rows: arg})
|
||||
}
|
||||
@@ -1,34 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgconn"
|
||||
)
|
||||
|
||||
type DBTX interface {
|
||||
Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
|
||||
Query(context.Context, string, ...interface{}) (pgx.Rows, error)
|
||||
QueryRow(context.Context, string, ...interface{}) pgx.Row
|
||||
CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error)
|
||||
SendBatch(context.Context, *pgx.Batch) pgx.BatchResults
|
||||
}
|
||||
|
||||
func New(db DBTX) *Queries {
|
||||
return &Queries{db: db}
|
||||
}
|
||||
|
||||
type Queries struct {
|
||||
db DBTX
|
||||
}
|
||||
|
||||
func (q *Queries) WithTx(tx pgx.Tx) *Queries {
|
||||
return &Queries{
|
||||
db: tx,
|
||||
}
|
||||
}
|
||||
@@ -1,88 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
type MoneyRecord struct {
|
||||
ID int64
|
||||
CategoryID int64
|
||||
UserID int64
|
||||
Amount pgtype.Numeric
|
||||
Remark pgtype.Text
|
||||
RecordTime pgtype.Date
|
||||
CreateTime pgtype.Timestamp
|
||||
UpdateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type MoneySysCategory struct {
|
||||
ID int64
|
||||
Name string
|
||||
Icon string
|
||||
Type int32
|
||||
SortNumber int32
|
||||
CreateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type MoneyUserCategory struct {
|
||||
ID int64
|
||||
UserID int64
|
||||
Name string
|
||||
Icon string
|
||||
Type int32
|
||||
SortNumber int32
|
||||
CreateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type Space struct {
|
||||
ID int64
|
||||
Name pgtype.Text
|
||||
InviteCode string
|
||||
CreateTime pgtype.Timestamp
|
||||
UpdateTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type SpaceMember struct {
|
||||
ID int64
|
||||
SpaceID int64
|
||||
UserID int64
|
||||
Role int32
|
||||
JoinTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type User struct {
|
||||
ID int64
|
||||
Username string
|
||||
Avatar string
|
||||
Status int32
|
||||
LastLoginTime pgtype.Timestamptz
|
||||
CreateTime pgtype.Timestamptz
|
||||
UpdateTime pgtype.Timestamptz
|
||||
}
|
||||
|
||||
type UserInfo struct {
|
||||
ID int64
|
||||
SeqNo pgtype.Int4
|
||||
Account string
|
||||
Password string
|
||||
Email pgtype.Text
|
||||
Username pgtype.Text
|
||||
Gender pgtype.Int4
|
||||
CreateTime pgtype.Timestamp
|
||||
UpdateTime pgtype.Timestamp
|
||||
ActiveTime pgtype.Timestamp
|
||||
LastLoginTime pgtype.Timestamp
|
||||
}
|
||||
|
||||
type UserOauth struct {
|
||||
ID int64
|
||||
UserID int64
|
||||
Type int32
|
||||
Openid string
|
||||
CreateTime pgtype.Timestamptz
|
||||
UpdateTime pgtype.Timestamptz
|
||||
}
|
||||
@@ -1,324 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: money_record.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
const deleteMoneyRecord = `-- name: DeleteMoneyRecord :exec
|
||||
DELETE FROM money_record WHERE id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) DeleteMoneyRecord(ctx context.Context, id int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteMoneyRecord, id)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteMoneyRecordByCategoryId = `-- name: DeleteMoneyRecordByCategoryId :exec
|
||||
DELETE FROM money_record WHERE user_id = $1 AND category_id = $2
|
||||
`
|
||||
|
||||
type DeleteMoneyRecordByCategoryIdParams struct {
|
||||
UserID int64
|
||||
CategoryID int64
|
||||
}
|
||||
|
||||
func (q *Queries) DeleteMoneyRecordByCategoryId(ctx context.Context, arg DeleteMoneyRecordByCategoryIdParams) error {
|
||||
_, err := q.db.Exec(ctx, deleteMoneyRecordByCategoryId, arg.UserID, arg.CategoryID)
|
||||
return err
|
||||
}
|
||||
|
||||
const getMoneyRecordById = `-- name: GetMoneyRecordById :one
|
||||
SELECT id, category_id, user_id, amount, remark, record_time, create_time, update_time FROM money_record WHERE id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) GetMoneyRecordById(ctx context.Context, id int64) (MoneyRecord, error) {
|
||||
row := q.db.QueryRow(ctx, getMoneyRecordById, id)
|
||||
var i MoneyRecord
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.CategoryID,
|
||||
&i.UserID,
|
||||
&i.Amount,
|
||||
&i.Remark,
|
||||
&i.RecordTime,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const insertMoneyRecord = `-- name: InsertMoneyRecord :exec
|
||||
INSERT INTO money_record (
|
||||
id,
|
||||
category_id,
|
||||
user_id,
|
||||
amount,
|
||||
remark,
|
||||
record_time,
|
||||
create_time,
|
||||
update_time
|
||||
) VALUES (
|
||||
$1, $2, $3, $4, $5,$6,NOW(),NOW()
|
||||
)
|
||||
`
|
||||
|
||||
type InsertMoneyRecordParams struct {
|
||||
ID int64
|
||||
CategoryID int64
|
||||
UserID int64
|
||||
Amount pgtype.Numeric
|
||||
Remark pgtype.Text
|
||||
RecordTime pgtype.Date
|
||||
}
|
||||
|
||||
func (q *Queries) InsertMoneyRecord(ctx context.Context, arg InsertMoneyRecordParams) error {
|
||||
_, err := q.db.Exec(ctx, insertMoneyRecord,
|
||||
arg.ID,
|
||||
arg.CategoryID,
|
||||
arg.UserID,
|
||||
arg.Amount,
|
||||
arg.Remark,
|
||||
arg.RecordTime,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
const listMoneyRecords = `-- name: ListMoneyRecords :many
|
||||
SELECT
|
||||
mr.id, mr.category_id, mr.user_id, mr.amount, mr.remark, mr.record_time, mr.create_time, mr.update_time,
|
||||
u.username AS username
|
||||
FROM money_record mr
|
||||
LEFT JOIN space_member su
|
||||
ON su.user_id = mr.user_id
|
||||
LEFT JOIN "user" u
|
||||
ON u.id = mr.user_id
|
||||
WHERE
|
||||
mr.record_time BETWEEN $2 AND $3
|
||||
AND (
|
||||
-- 空间账本：属于当前空间的用户
|
||||
su.user_id IS NOT NULL
|
||||
-- 个人账本：当前用户
|
||||
OR mr.user_id = $1
|
||||
)
|
||||
ORDER BY
|
||||
mr.record_time DESC,
|
||||
mr.create_time DESC
|
||||
`
|
||||
|
||||
type ListMoneyRecordsParams struct {
|
||||
UserID int64
|
||||
RecordTime pgtype.Date
|
||||
RecordTime_2 pgtype.Date
|
||||
}
|
||||
|
||||
type ListMoneyRecordsRow struct {
|
||||
ID int64
|
||||
CategoryID int64
|
||||
UserID int64
|
||||
Amount pgtype.Numeric
|
||||
Remark pgtype.Text
|
||||
RecordTime pgtype.Date
|
||||
CreateTime pgtype.Timestamp
|
||||
UpdateTime pgtype.Timestamp
|
||||
Username pgtype.Text
|
||||
}
|
||||
|
||||
func (q *Queries) ListMoneyRecords(ctx context.Context, arg ListMoneyRecordsParams) ([]ListMoneyRecordsRow, error) {
|
||||
rows, err := q.db.Query(ctx, listMoneyRecords, arg.UserID, arg.RecordTime, arg.RecordTime_2)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []ListMoneyRecordsRow
|
||||
for rows.Next() {
|
||||
var i ListMoneyRecordsRow
|
||||
if err := rows.Scan(
|
||||
&i.ID,
|
||||
&i.CategoryID,
|
||||
&i.UserID,
|
||||
&i.Amount,
|
||||
&i.Remark,
|
||||
&i.RecordTime,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
&i.Username,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const listMoneyRecordsByLineAnalysis = `-- name: ListMoneyRecordsByLineAnalysis :many
|
||||
SELECT
|
||||
CASE
|
||||
WHEN $1 = 0 THEN to_char(mr.record_time, 'DD') -- 日分析，转换为文本
|
||||
WHEN $1 = 1 THEN to_char(mr.record_time, 'MM') -- 月分析，转换为文本
|
||||
WHEN $1 = 2 THEN to_char(mr.record_time, 'YYYY') -- 年分析，转换为文本
|
||||
END AS period,
|
||||
SUM(mr.amount)::DECIMAL AS total_amount
|
||||
FROM
|
||||
money_record mr
|
||||
LEFT JOIN
|
||||
money_user_category muc ON mr.category_id = muc.id -- 连接 money_user_category 获取 type
|
||||
LEFT JOIN
|
||||
space_member su ON su.user_id = $2 -- 连接 space_member 表，查找该用户在空间中的信息
|
||||
LEFT JOIN
|
||||
space s ON s.id = su.space_id -- 连接 space 表，获取空间信息
|
||||
WHERE
|
||||
(
|
||||
(su.user_id IS NOT NULL) -- 用户在空间，查询该空间内所有人的数据
|
||||
OR (mr.user_id = $1) -- 用户不在空间，查询该用户个人的账本数据
|
||||
)
|
||||
AND mr.record_time BETWEEN $3 AND $4 -- 时间范围
|
||||
AND muc.type = $5 -- 收入或支出类型
|
||||
GROUP BY
|
||||
period
|
||||
ORDER BY
|
||||
period
|
||||
`
|
||||
|
||||
type ListMoneyRecordsByLineAnalysisParams struct {
|
||||
Column1 interface{}
|
||||
UserID int64
|
||||
RecordTime pgtype.Date
|
||||
RecordTime_2 pgtype.Date
|
||||
Type int32
|
||||
}
|
||||
|
||||
type ListMoneyRecordsByLineAnalysisRow struct {
|
||||
Period interface{}
|
||||
TotalAmount pgtype.Numeric
|
||||
}
|
||||
|
||||
// 折线图：按天、月、年汇总金额
|
||||
func (q *Queries) ListMoneyRecordsByLineAnalysis(ctx context.Context, arg ListMoneyRecordsByLineAnalysisParams) ([]ListMoneyRecordsByLineAnalysisRow, error) {
|
||||
rows, err := q.db.Query(ctx, listMoneyRecordsByLineAnalysis,
|
||||
arg.Column1,
|
||||
arg.UserID,
|
||||
arg.RecordTime,
|
||||
arg.RecordTime_2,
|
||||
arg.Type,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []ListMoneyRecordsByLineAnalysisRow
|
||||
for rows.Next() {
|
||||
var i ListMoneyRecordsByLineAnalysisRow
|
||||
if err := rows.Scan(&i.Period, &i.TotalAmount); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const listMoneyRecordsByPieAnalysis = `-- name: ListMoneyRecordsByPieAnalysis :many
|
||||
SELECT
|
||||
muc.name,
|
||||
SUM(mr.amount)::DECIMAL AS total_amount
|
||||
FROM
|
||||
money_record mr
|
||||
LEFT JOIN
|
||||
money_user_category muc ON mr.category_id = muc.id
|
||||
LEFT JOIN
|
||||
space_member su ON su.user_id = $1 -- 连接 space_member 表，查找该用户在空间中的信息
|
||||
LEFT JOIN
|
||||
space s ON s.id = su.space_id -- 连接 space 表，获取空间信息
|
||||
WHERE
|
||||
(
|
||||
(su.user_id IS NOT NULL) -- 用户在空间，查询该空间内所有人的数据
|
||||
OR (mr.user_id = $1) -- 用户不在空间，查询该用户个人的账本数据
|
||||
)
|
||||
AND mr.record_time BETWEEN $2 AND $3 -- 时间范围
|
||||
AND muc.type = $4 -- 收入或支出类型
|
||||
GROUP BY
|
||||
muc.name
|
||||
ORDER BY
|
||||
total_amount DESC
|
||||
`
|
||||
|
||||
type ListMoneyRecordsByPieAnalysisParams struct {
|
||||
UserID int64
|
||||
RecordTime pgtype.Date
|
||||
RecordTime_2 pgtype.Date
|
||||
Type int32
|
||||
}
|
||||
|
||||
type ListMoneyRecordsByPieAnalysisRow struct {
|
||||
Name pgtype.Text
|
||||
TotalAmount pgtype.Numeric
|
||||
}
|
||||
|
||||
// 饼图：按类别汇总金额
|
||||
func (q *Queries) ListMoneyRecordsByPieAnalysis(ctx context.Context, arg ListMoneyRecordsByPieAnalysisParams) ([]ListMoneyRecordsByPieAnalysisRow, error) {
|
||||
rows, err := q.db.Query(ctx, listMoneyRecordsByPieAnalysis,
|
||||
arg.UserID,
|
||||
arg.RecordTime,
|
||||
arg.RecordTime_2,
|
||||
arg.Type,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []ListMoneyRecordsByPieAnalysisRow
|
||||
for rows.Next() {
|
||||
var i ListMoneyRecordsByPieAnalysisRow
|
||||
if err := rows.Scan(&i.Name, &i.TotalAmount); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const updateMoneyRecord = `-- name: UpdateMoneyRecord :exec
|
||||
UPDATE money_record
|
||||
SET
|
||||
category_id = $2,
|
||||
amount = $3,
|
||||
remark = $4,
|
||||
record_time = $5,
|
||||
update_time = NOW()
|
||||
WHERE id = $1
|
||||
RETURNING id, category_id, user_id, amount, remark, record_time, create_time, update_time
|
||||
`
|
||||
|
||||
type UpdateMoneyRecordParams struct {
|
||||
ID int64
|
||||
CategoryID int64
|
||||
Amount pgtype.Numeric
|
||||
Remark pgtype.Text
|
||||
RecordTime pgtype.Date
|
||||
}
|
||||
|
||||
func (q *Queries) UpdateMoneyRecord(ctx context.Context, arg UpdateMoneyRecordParams) error {
|
||||
_, err := q.db.Exec(ctx, updateMoneyRecord,
|
||||
arg.ID,
|
||||
arg.CategoryID,
|
||||
arg.Amount,
|
||||
arg.Remark,
|
||||
arg.RecordTime,
|
||||
)
|
||||
return err
|
||||
}
|
||||
@@ -1,41 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: money_sys_category.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
const listMoneySysCategory = `-- name: ListMoneySysCategory :many
|
||||
SELECT id, name, icon, type, sort_number, create_time FROM money_sys_category
|
||||
`
|
||||
|
||||
func (q *Queries) ListMoneySysCategory(ctx context.Context) ([]MoneySysCategory, error) {
|
||||
rows, err := q.db.Query(ctx, listMoneySysCategory)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []MoneySysCategory
|
||||
for rows.Next() {
|
||||
var i MoneySysCategory
|
||||
if err := rows.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.Icon,
|
||||
&i.Type,
|
||||
&i.SortNumber,
|
||||
&i.CreateTime,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
@@ -1,116 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: money_user_category.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
type BatchInsertMoneyUserCategoriesParams struct {
|
||||
ID int64
|
||||
UserID int64
|
||||
Name string
|
||||
Icon string
|
||||
Type int32
|
||||
SortNumber int32
|
||||
}
|
||||
|
||||
const deleteMoneyCategoryById = `-- name: DeleteMoneyCategoryById :exec
|
||||
DELETE FROM money_user_category
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) DeleteMoneyCategoryById(ctx context.Context, id int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteMoneyCategoryById, id)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteMoneyCategoryByUserId = `-- name: DeleteMoneyCategoryByUserId :exec
|
||||
DELETE FROM money_user_category
|
||||
WHERE user_id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) DeleteMoneyCategoryByUserId(ctx context.Context, userID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteMoneyCategoryByUserId, userID)
|
||||
return err
|
||||
}
|
||||
|
||||
const findUserMoneyCategoriesByUserID = `-- name: FindUserMoneyCategoriesByUserID :many
|
||||
SELECT id, user_id, name, icon, type, sort_number, create_time
|
||||
FROM money_user_category
|
||||
WHERE user_id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) FindUserMoneyCategoriesByUserID(ctx context.Context, userID int64) ([]MoneyUserCategory, error) {
|
||||
rows, err := q.db.Query(ctx, findUserMoneyCategoriesByUserID, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []MoneyUserCategory
|
||||
for rows.Next() {
|
||||
var i MoneyUserCategory
|
||||
if err := rows.Scan(
|
||||
&i.ID,
|
||||
&i.UserID,
|
||||
&i.Name,
|
||||
&i.Icon,
|
||||
&i.Type,
|
||||
&i.SortNumber,
|
||||
&i.CreateTime,
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items = append(items, i)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const insertMoneyUserCategory = `-- name: InsertMoneyUserCategory :exec
|
||||
INSERT INTO money_user_category (id, user_id, name, icon, type, sort_number,create_time,update_time)
|
||||
VALUES ($1, $2, $3, $4, $5, $6,NOW(),NOW())
|
||||
`
|
||||
|
||||
type InsertMoneyUserCategoryParams struct {
|
||||
ID int64
|
||||
UserID int64
|
||||
Name string
|
||||
Icon string
|
||||
Type int32
|
||||
SortNumber int32
|
||||
}
|
||||
|
||||
func (q *Queries) InsertMoneyUserCategory(ctx context.Context, arg InsertMoneyUserCategoryParams) error {
|
||||
_, err := q.db.Exec(ctx, insertMoneyUserCategory,
|
||||
arg.ID,
|
||||
arg.UserID,
|
||||
arg.Name,
|
||||
arg.Icon,
|
||||
arg.Type,
|
||||
arg.SortNumber,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
const updateMoneyCategory = `-- name: UpdateMoneyCategory :exec
|
||||
UPDATE money_user_category
|
||||
SET icon = $2,name = $3,update_time = NOW()
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
type UpdateMoneyCategoryParams struct {
|
||||
ID int64
|
||||
Icon string
|
||||
Name string
|
||||
}
|
||||
|
||||
func (q *Queries) UpdateMoneyCategory(ctx context.Context, arg UpdateMoneyCategoryParams) error {
|
||||
_, err := q.db.Exec(ctx, updateMoneyCategory, arg.ID, arg.Icon, arg.Name)
|
||||
return err
|
||||
}
|
||||
@@ -1,108 +0,0 @@
|
||||
// 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 findSpaceByInviteCode = `-- name: FindSpaceByInviteCode :one
|
||||
SELECT id, name, invite_code, create_time, update_time FROM space WHERE invite_code = $1
|
||||
`
|
||||
|
||||
// 根据邀请码查询空间
|
||||
func (q *Queries) FindSpaceByInviteCode(ctx context.Context, inviteCode string) (Space, error) {
|
||||
row := q.db.QueryRow(ctx, findSpaceByInviteCode, inviteCode)
|
||||
var i Space
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
)
|
||||
return i, 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 getSpaceByUserId = `-- name: GetSpaceByUserId :one
|
||||
SELECT s.id, s.name, s.invite_code, s.create_time, s.update_time, m.role
|
||||
FROM space s
|
||||
JOIN space_member m
|
||||
ON s.id = m.space_id
|
||||
WHERE m.user_id = $1
|
||||
`
|
||||
|
||||
type GetSpaceByUserIdRow struct {
|
||||
ID int64
|
||||
Name pgtype.Text
|
||||
InviteCode string
|
||||
CreateTime pgtype.Timestamp
|
||||
UpdateTime pgtype.Timestamp
|
||||
Role int32
|
||||
}
|
||||
|
||||
// 查询用户所在的空间信息
|
||||
func (q *Queries) GetSpaceByUserId(ctx context.Context, userID int64) (GetSpaceByUserIdRow, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceByUserId, userID)
|
||||
var i GetSpaceByUserIdRow
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.Name,
|
||||
&i.InviteCode,
|
||||
&i.CreateTime,
|
||||
&i.UpdateTime,
|
||||
&i.Role,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const insertSpace = `-- name: InsertSpace :exec
|
||||
INSERT INTO space (
|
||||
id,name,invite_code,create_time,update_time
|
||||
) VALUES ($1,$2,$3,NOW(),NOW())
|
||||
`
|
||||
|
||||
type InsertSpaceParams struct {
|
||||
ID int64
|
||||
Name pgtype.Text
|
||||
InviteCode string
|
||||
}
|
||||
|
||||
// 新增分享空间
|
||||
func (q *Queries) InsertSpace(ctx context.Context, arg InsertSpaceParams) error {
|
||||
_, err := q.db.Exec(ctx, insertSpace, arg.ID, arg.Name, arg.InviteCode)
|
||||
return err
|
||||
}
|
||||
@@ -1,159 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: space_member.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
const deleteSpaceMemberBySpaceIDAndUserID = `-- name: DeleteSpaceMemberBySpaceIDAndUserID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE space_id = $1
|
||||
AND user_id = $2
|
||||
`
|
||||
|
||||
type DeleteSpaceMemberBySpaceIDAndUserIDParams struct {
|
||||
SpaceID int64
|
||||
UserID int64
|
||||
}
|
||||
|
||||
// 删除某个用户在指定空间中的关系
|
||||
func (q *Queries) DeleteSpaceMemberBySpaceIDAndUserID(ctx context.Context, arg DeleteSpaceMemberBySpaceIDAndUserIDParams) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceMemberBySpaceIDAndUserID, arg.SpaceID, arg.UserID)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteSpaceMemberByUserID = `-- name: DeleteSpaceMemberByUserID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE user_id = $1
|
||||
`
|
||||
|
||||
// 删除某个用户在所有空间的关系
|
||||
func (q *Queries) DeleteSpaceMemberByUserID(ctx context.Context, userID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceMemberByUserID, userID)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteSpaceMembersBySpaceID = `-- name: DeleteSpaceMembersBySpaceID :exec
|
||||
DELETE FROM space_member
|
||||
WHERE space_id = $1
|
||||
`
|
||||
|
||||
// 删除某个空间下的所有用户
|
||||
func (q *Queries) DeleteSpaceMembersBySpaceID(ctx context.Context, spaceID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteSpaceMembersBySpaceID, spaceID)
|
||||
return err
|
||||
}
|
||||
|
||||
const getSpaceMember = `-- name: GetSpaceMember :one
|
||||
SELECT id, space_id, user_id, role, join_time
|
||||
FROM space_member
|
||||
WHERE user_id = $1
|
||||
`
|
||||
|
||||
// 获取空间用户
|
||||
func (q *Queries) GetSpaceMember(ctx context.Context, userID int64) (SpaceMember, error) {
|
||||
row := q.db.QueryRow(ctx, getSpaceMember, userID)
|
||||
var i SpaceMember
|
||||
err := row.Scan(
|
||||
&i.ID,
|
||||
&i.SpaceID,
|
||||
&i.UserID,
|
||||
&i.Role,
|
||||
&i.JoinTime,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const insertSpaceMember = `-- name: InsertSpaceMember :exec
|
||||
INSERT INTO space_member(id,space_id,user_id,role,join_time)
|
||||
VALUES ($1,$2,$3,$4,NOW())
|
||||
`
|
||||
|
||||
type InsertSpaceMemberParams struct {
|
||||
ID int64
|
||||
SpaceID int64
|
||||
UserID int64
|
||||
Role 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,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
const listSpaceMembers = `-- name: ListSpaceMembers :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_member 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 ListSpaceMembersRow struct {
|
||||
UserID int64
|
||||
UserName string
|
||||
UserAvatar string
|
||||
LastLoginTime pgtype.Timestamptz
|
||||
Role int32
|
||||
}
|
||||
|
||||
func (q *Queries) ListSpaceMembers(ctx context.Context, spaceID int64) ([]ListSpaceMembersRow, error) {
|
||||
rows, err := q.db.Query(ctx, listSpaceMembers, spaceID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var items []ListSpaceMembersRow
|
||||
for rows.Next() {
|
||||
var i ListSpaceMembersRow
|
||||
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 updateSpaceMemberRole = `-- name: UpdateSpaceMemberRole :exec
|
||||
UPDATE space_member
|
||||
SET role = $1
|
||||
WHERE user_id = $2 AND space_id = $3
|
||||
`
|
||||
|
||||
type UpdateSpaceMemberRoleParams struct {
|
||||
Role int32
|
||||
UserID int64
|
||||
SpaceID int64
|
||||
}
|
||||
|
||||
// 修改角色
|
||||
func (q *Queries) UpdateSpaceMemberRole(ctx context.Context, arg UpdateSpaceMemberRoleParams) error {
|
||||
_, err := q.db.Exec(ctx, updateSpaceMemberRole, arg.Role, arg.UserID, arg.SpaceID)
|
||||
return err
|
||||
}
|
||||
@@ -1,96 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: user.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
const findUserByOpenID = `-- name: FindUserByOpenID :one
|
||||
SELECT
|
||||
u.id AS user_id,
|
||||
u.username,
|
||||
u.avatar,
|
||||
u.status,
|
||||
o.type AS login_type,
|
||||
o.openid
|
||||
FROM user_oauth o
|
||||
JOIN "user" u ON u.id = o.user_id
|
||||
WHERE o.openid = $1
|
||||
AND o.type = $2
|
||||
LIMIT 1
|
||||
`
|
||||
|
||||
type FindUserByOpenIDParams struct {
|
||||
Openid string
|
||||
Type int32
|
||||
}
|
||||
|
||||
type FindUserByOpenIDRow struct {
|
||||
UserID int64
|
||||
Username string
|
||||
Avatar string
|
||||
Status int32
|
||||
LoginType int32
|
||||
Openid string
|
||||
}
|
||||
|
||||
func (q *Queries) FindUserByOpenID(ctx context.Context, arg FindUserByOpenIDParams) (FindUserByOpenIDRow, error) {
|
||||
row := q.db.QueryRow(ctx, findUserByOpenID, arg.Openid, arg.Type)
|
||||
var i FindUserByOpenIDRow
|
||||
err := row.Scan(
|
||||
&i.UserID,
|
||||
&i.Username,
|
||||
&i.Avatar,
|
||||
&i.Status,
|
||||
&i.LoginType,
|
||||
&i.Openid,
|
||||
)
|
||||
return i, err
|
||||
}
|
||||
|
||||
const insertUser = `-- name: InsertUser :exec
|
||||
INSERT INTO "user" (
|
||||
id,
|
||||
username,
|
||||
avatar,
|
||||
status,
|
||||
create_time,
|
||||
update_time,
|
||||
last_login_time
|
||||
) VALUES (
|
||||
$1, $2, $3,$4, NOW(), NOW(),now()
|
||||
)
|
||||
RETURNING id
|
||||
`
|
||||
|
||||
type InsertUserParams struct {
|
||||
ID int64
|
||||
Username string
|
||||
Avatar string
|
||||
Status int32
|
||||
}
|
||||
|
||||
func (q *Queries) InsertUser(ctx context.Context, arg InsertUserParams) error {
|
||||
_, err := q.db.Exec(ctx, insertUser,
|
||||
arg.ID,
|
||||
arg.Username,
|
||||
arg.Avatar,
|
||||
arg.Status,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
const updateUserLastLoginTime = `-- name: UpdateUserLastLoginTime :exec
|
||||
UPDATE "user"
|
||||
SET last_login_time = NOW()
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) UpdateUserLastLoginTime(ctx context.Context, id int64) error {
|
||||
_, err := q.db.Exec(ctx, updateUserLastLoginTime, id)
|
||||
return err
|
||||
}
|
||||
@@ -1,40 +0,0 @@
|
||||
// Code generated by sqlc. DO NOT EDIT.
|
||||
// versions:
|
||||
// sqlc v1.30.0
|
||||
// source: user_oauth.sql
|
||||
|
||||
package repo
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
const insertUserOAuth = `-- name: InsertUserOAuth :exec
|
||||
INSERT INTO user_oauth (
|
||||
id,
|
||||
user_id,
|
||||
type,
|
||||
openid,
|
||||
create_time,
|
||||
update_time
|
||||
) VALUES (
|
||||
$1, $2, $3, $4, NOW(), NOW()
|
||||
)
|
||||
`
|
||||
|
||||
type InsertUserOAuthParams struct {
|
||||
ID int64
|
||||
UserID int64
|
||||
Type int32
|
||||
Openid string
|
||||
}
|
||||
|
||||
func (q *Queries) InsertUserOAuth(ctx context.Context, arg InsertUserOAuthParams) error {
|
||||
_, err := q.db.Exec(ctx, insertUserOAuth,
|
||||
arg.ID,
|
||||
arg.UserID,
|
||||
arg.Type,
|
||||
arg.Openid,
|
||||
)
|
||||
return err
|
||||
}
|
||||
@@ -1,84 +0,0 @@
|
||||
CREATE TABLE user_info (
|
||||
id BIGINT PRIMARY KEY,
|
||||
seq_no SERIAL, -- 自增
|
||||
account VARCHAR(255) NOT NULL UNIQUE,
|
||||
password VARCHAR(255) NOT NULL,
|
||||
email VARCHAR(255) UNIQUE,
|
||||
username VARCHAR(50),
|
||||
gender INT DEFAULT 0,
|
||||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
active_time TIMESTAMP,
|
||||
last_login_time TIMESTAMP
|
||||
);
|
||||
|
||||
CREATE TABLE "user" (
|
||||
"id" int8 NOT NULL,
|
||||
"username" varchar(255) NOT NULL ,
|
||||
"avatar" varchar(255) NOT NULL ,
|
||||
"status" int4 NOT NULL DEFAULT 1,
|
||||
"last_login_time" timestamptz(6),
|
||||
"create_time" timestamptz(6),
|
||||
"update_time" timestamptz(6)
|
||||
);
|
||||
|
||||
CREATE TABLE "user_oauth" (
|
||||
"id" int8 NOT NULL,
|
||||
"user_id" int8 NOT NULL,
|
||||
"type" int4 NOT NULL,
|
||||
"openid" varchar(255) COLLATE "pg_catalog"."default" NOT NULL,
|
||||
"create_time" timestamptz(6) NOT NULL,
|
||||
"update_time" timestamptz(6) NOT NULL
|
||||
);
|
||||
|
||||
CREATE TABLE money_record(
|
||||
id BIGINT PRIMARY KEY,
|
||||
category_id BIGINT NOT NULL,
|
||||
user_id BIGINT NOT NULL,
|
||||
amount NUMERIC(12,2) NOT NULL,
|
||||
remark VARCHAR(255) DEFAULT '',
|
||||
record_time DATE NOT NULL,
|
||||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
CREATE TABLE money_sys_category(
|
||||
id BIGINT PRIMARY KEY ,
|
||||
name VARCHAR(255) NOT NULL ,
|
||||
icon VARCHAR(255) NOT NULL,
|
||||
type INT NOT NULL DEFAULT 0 ,
|
||||
sort_number INT NOT NULL DEFAULT 0,
|
||||
create_time timestamp DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
CREATE TABLE money_user_category(
|
||||
id BIGINT PRIMARY KEY ,
|
||||
user_id BIGINT NOT NULL,
|
||||
name VARCHAR(255) NOT NULL,
|
||||
icon VARCHAR(255) NOT NULL,
|
||||
type INT NOT NULL DEFAULT 0,
|
||||
sort_number INT NOT NULL DEFAULT 0,
|
||||
create_time timestamp DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
-- 共享空间表
|
||||
CREATE TABLE space (
|
||||
id BIGSERIAL PRIMARY KEY,
|
||||
name VARCHAR(100) DEFAULT '',
|
||||
invite_code VARCHAR(30) NOT NULL,
|
||||
create_time TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||
update_time TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
ALTER TABLE space
|
||||
ADD CONSTRAINT space_invite_code_key UNIQUE (invite_code);
|
||||
|
||||
-- 空间成员表（每个用户只能加入一个空间）
|
||||
CREATE TABLE space_member (
|
||||
id BIGSERIAL PRIMARY KEY,
|
||||
space_id BIGINT NOT NULL,
|
||||
user_id BIGINT NOT NULL,
|
||||
role INT NOT NULL DEFAULT 2,
|
||||
join_time TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
Reference in new issue
Block a user