chore: update package name

2025-06-05 22:09:59 +02:00 · 2023-10-05 23:11:29 +08:00
parent 99d9bd2d75
commit a6ee61e96d
79 changed files with 172 additions and 314 deletions
--- a/store/db/sqlite/memo_organizer.go
+++ b/store/db/sqlite/memo_organizer.go
@ -0,0 +1,108 @@
+package sqlite
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"strings"
+
+	"github.com/usememos/memos/store"
+)
+
+func (d *DB) UpsertMemoOrganizer(ctx context.Context, upsert *store.MemoOrganizer) (*store.MemoOrganizer, error) {
+	stmt := `
+		INSERT INTO memo_organizer (
+			memo_id,
+			user_id,
+			pinned
+		)
+		VALUES (?, ?, ?)
+		ON CONFLICT(memo_id, user_id) DO UPDATE 
+		SET
+			pinned = EXCLUDED.pinned
+	`
+	if _, err := d.db.ExecContext(ctx, stmt, upsert.MemoID, upsert.UserID, upsert.Pinned); err != nil {
+		return nil, err
+	}
+
+	return upsert, nil
+}
+
+func (d *DB) GetMemoOrganizer(ctx context.Context, find *store.FindMemoOrganizer) (*store.MemoOrganizer, error) {
+	where, args := []string{}, []any{}
+	if find.MemoID != 0 {
+		where = append(where, "memo_id = ?")
+		args = append(args, find.MemoID)
+	}
+	if find.UserID != 0 {
+		where = append(where, "user_id = ?")
+		args = append(args, find.UserID)
+	}
+
+	query := fmt.Sprintf(`
+		SELECT
+			memo_id,
+			user_id,
+			pinned
+		FROM memo_organizer
+		WHERE %s
+	`, strings.Join(where, " AND "))
+	row := d.db.QueryRowContext(ctx, query, args...)
+	if err := row.Err(); err != nil {
+		return nil, err
+	}
+	if row == nil {
+		return nil, nil
+	}
+
+	memoOrganizer := &store.MemoOrganizer{}
+	if err := row.Scan(
+		&memoOrganizer.MemoID,
+		&memoOrganizer.UserID,
+		&memoOrganizer.Pinned,
+	); err != nil {
+		return nil, err
+	}
+
+	return memoOrganizer, nil
+}
+
+func (d *DB) DeleteMemoOrganizer(ctx context.Context, delete *store.DeleteMemoOrganizer) error {
+	where, args := []string{}, []any{}
+	if v := delete.MemoID; v != nil {
+		where, args = append(where, "memo_id = ?"), append(args, *v)
+	}
+	if v := delete.UserID; v != nil {
+		where, args = append(where, "user_id = ?"), append(args, *v)
+	}
+	stmt := `DELETE FROM memo_organizer WHERE ` + strings.Join(where, " AND ")
+	if _, err := d.db.ExecContext(ctx, stmt, args...); err != nil {
+		return err
+	}
+	return 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
+}