12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091 |
- package postgres
- import (
- "context"
- "fmt"
- "strings"
- "github.com/usememos/memos/store"
- )
- func (d *DB) UpsertMemoOrganizer(ctx context.Context, upsert *store.MemoOrganizer) (*store.MemoOrganizer, error) {
- pinned := 0
- if upsert.Pinned {
- pinned = 1
- }
- 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) {
- where, args := []string{"1 = 1"}, []any{}
- if find.MemoID != 0 {
- where, args = append(where, "memo_id = "+placeholder(len(args)+1)), append(args, find.MemoID)
- }
- if find.UserID != 0 {
- 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()
- list := []*store.MemoOrganizer{}
- for rows.Next() {
- memoOrganizer := &store.MemoOrganizer{}
- pinned := 0
- if err := rows.Scan(
- &memoOrganizer.MemoID,
- &memoOrganizer.UserID,
- &pinned,
- ); err != nil {
- return nil, err
- }
- memoOrganizer.Pinned = pinned == 1
- list = append(list, memoOrganizer)
- }
- if err := rows.Err(); err != nil {
- return nil, err
- }
- return list, 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 = "+placeholder(len(args)+1)), append(args, *v)
- }
- if v := delete.UserID; v != nil {
- where, args = append(where, "user_id = "+placeholder(len(args)+1)), 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
- }
|