YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix postgres stmts

Browse Source
This commit is contained in:
Steven
2024-01-05 21:27:16 +08:00
parent ee13927607
commit 501f8898f6
16 changed files with 589 additions and 672 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
store/db/postgres/memo_organizer.go
View File

@ -4,8 +4,7 @@ import (
"context"
"database/sql"
"fmt"
"github.com/Masterminds/squirrel"
"strings"
"github.com/usememos/memos/store"
)
@ -15,99 +14,94 @@ func (d *DB) UpsertMemoOrganizer(ctx context.Context, upsert *store.MemoOrganize
if upsert.Pinned {
pinned = 1
}
stmt := "INSERT INTO memo_organizer (memo_id, user_id, pinned) VALUES ($1, $2, $3) ON CONFLICT (memo_id, user_id) DO UPDATE SET pinned = $4"
if _, err := d.db.ExecContext(ctx, stmt, upsert.MemoID, upsert.UserID, pinned, pinned); err != nil {
stmt := `
INSERT INTO memo_organizer (
memo_id,
user_id,
pinned
)
VALUES (` + placeholders(3) + `)
ON CONFLICT(memo_id, user_id) DO UPDATE
SET pinned = EXCLUDED.pinned`
if _, err := d.db.ExecContext(ctx, stmt, upsert.MemoID, upsert.UserID, pinned); err != nil {
return nil, err
}
return upsert, nil
}
func (d *DB) ListMemoOrganizer(ctx context.Context, find *store.FindMemoOrganizer) ([]*store.MemoOrganizer, error) {
qb := squirrel.Select("memo_id", "user_id", "pinned").
From("memo_organizer").
Where("1 = 1").
PlaceholderFormat(squirrel.Dollar)
where, args := []string{"1 = 1"}, []any{}
if find.MemoID != 0 {
qb = qb.Where(squirrel.Eq{"memo_id": find.MemoID})
where, args = append(where, "memo_id = "+placeholder(len(args)+1)), append(args, find.MemoID)
}
if find.UserID != 0 {
qb = qb.Where(squirrel.Eq{"user_id": find.UserID})
}
query, args, err := qb.ToSql()
if err != nil {
return nil, err
where, args = append(where, "user_id = "+placeholder(len(args)+1)), append(args, find.UserID)
}
query := fmt.Sprintf(`
SELECT
memo_id,
user_id,
pinned
FROM memo_organizer
WHERE %s
`, strings.Join(where, " AND "))
rows, err := d.db.QueryContext(ctx, query, args...)
if err != nil {
return nil, err
}
defer rows.Close()
var list []*store.MemoOrganizer
list := []*store.MemoOrganizer{}
for rows.Next() {
memoOrganizer := &store.MemoOrganizer{}
if err := rows.Scan(&memoOrganizer.MemoID, &memoOrganizer.UserID, &memoOrganizer.Pinned); err != nil {
pinned := 0
if err := rows.Scan(
&memoOrganizer.MemoID,
&memoOrganizer.UserID,
&pinned,
); err != nil {
return nil, err
}
memoOrganizer.Pinned = pinned == 1
list = append(list, memoOrganizer)
}
return list, rows.Err()
if err := rows.Err(); err != nil {
return nil, err
}
return list, nil
}
func (d *DB) DeleteMemoOrganizer(ctx context.Context, delete *store.DeleteMemoOrganizer) error {
qb := squirrel.Delete("memo_organizer").
PlaceholderFormat(squirrel.Dollar)
where, args := []string{}, []any{}
if v := delete.MemoID; v != nil {
qb = qb.Where(squirrel.Eq{"memo_id": *v})
where, args = append(where, "memo_id = "+placeholder(len(args)+1)), append(args, *v)
}
if v := delete.UserID; v != nil {
qb = qb.Where(squirrel.Eq{"user_id": *v})
where, args = append(where, "user_id = "+placeholder(len(args)+1)), append(args, *v)
}
stmt, args, err := qb.ToSql()
if err != nil {
stmt := `DELETE FROM memo_organizer WHERE ` + strings.Join(where, " AND ")
if _, err := d.db.ExecContext(ctx, stmt, args...); err != nil {
return err
}
return nil
}
if _, err = d.db.ExecContext(ctx, stmt, args...); err != nil {
func vacuumMemoOrganizer(ctx context.Context, tx *sql.Tx) error {
stmt := `
DELETE FROM
memo_organizer
WHERE
memo_id NOT IN (SELECT id FROM memo)
OR user_id NOT IN (SELECT id FROM "user")`
_, err := tx.ExecContext(ctx, stmt)
if err != nil {
return err
}
return nil
}
func vacuumMemoOrganizer(ctx context.Context, tx *sql.Tx) error {
// First, build the subquery for memo_id
subQueryMemo, subArgsMemo, err := squirrel.Select("id").From("memo").PlaceholderFormat(squirrel.Dollar).ToSql()
if err != nil {
return err
}
// Build the subquery for user_id
subQueryUser, subArgsUser, err := squirrel.Select("id").From(`"user"`).PlaceholderFormat(squirrel.Dollar).ToSql()
if err != nil {
return err
}
// Now, build the main delete query using the subqueries
query, args, err := squirrel.Delete("memo_organizer").
Where(fmt.Sprintf("memo_id NOT IN (%s)", subQueryMemo), subArgsMemo...).
Where(fmt.Sprintf("user_id NOT IN (%s)", subQueryUser), subArgsUser...).
PlaceholderFormat(squirrel.Dollar).
ToSql()
if err != nil {
return err
}
// Combine the arguments from both subqueries
args = append(args, subArgsUser...)
// Execute the query
_, err = tx.ExecContext(ctx, query, args...)
return err
}