mirror of https://github.com/usememos/memos
refactor: move pinned to memo
johnnyjoy
4 months ago
parent
2058a8ab7b
commit
d605faeffa
@ -1,70 +0,0 @@
|
|||||||
package mysql
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"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 DUPLICATE KEY UPDATE `pinned` = ?"
|
|
||||||
if _, err := d.db.ExecContext(ctx, stmt, upsert.MemoID, upsert.UserID, upsert.Pinned, upsert.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 = append(where, "`memo_id` = ?")
|
|
||||||
args = append(args, find.MemoID)
|
|
||||||
}
|
|
||||||
if find.UserID != 0 {
|
|
||||||
where = append(where, "`user_id` = ?")
|
|
||||||
args = append(args, find.UserID)
|
|
||||||
}
|
|
||||||
|
|
||||||
query := "SELECT `memo_id`, `user_id`, `pinned` FROM `memo_organizer` WHERE " + 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{}
|
|
||||||
if err := rows.Scan(
|
|
||||||
&memoOrganizer.MemoID,
|
|
||||||
&memoOrganizer.UserID,
|
|
||||||
&memoOrganizer.Pinned,
|
|
||||||
); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
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` = ?"), 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
|
|
||||||
}
|
|
||||||
@ -1,91 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
||||||
@ -1,89 +0,0 @@
|
|||||||
package sqlite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"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) ListMemoOrganizer(ctx context.Context, find *store.FindMemoOrganizer) ([]*store.MemoOrganizer, error) {
|
|
||||||
where, args := []string{"1 = 1"}, []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 "))
|
|
||||||
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{}
|
|
||||||
if err := rows.Scan(
|
|
||||||
&memoOrganizer.MemoID,
|
|
||||||
&memoOrganizer.UserID,
|
|
||||||
&memoOrganizer.Pinned,
|
|
||||||
); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
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 = ?"), 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
|
|
||||||
}
|
|
||||||
@ -1,47 +0,0 @@
|
|||||||
package store
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MemoOrganizer struct {
|
|
||||||
MemoID int32
|
|
||||||
UserID int32
|
|
||||||
Pinned bool
|
|
||||||
}
|
|
||||||
|
|
||||||
type FindMemoOrganizer struct {
|
|
||||||
MemoID int32
|
|
||||||
UserID int32
|
|
||||||
}
|
|
||||||
|
|
||||||
type DeleteMemoOrganizer struct {
|
|
||||||
MemoID *int32
|
|
||||||
UserID *int32
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) UpsertMemoOrganizer(ctx context.Context, upsert *MemoOrganizer) (*MemoOrganizer, error) {
|
|
||||||
return s.driver.UpsertMemoOrganizer(ctx, upsert)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) GetMemoOrganizer(ctx context.Context, find *FindMemoOrganizer) (*MemoOrganizer, error) {
|
|
||||||
list, err := s.ListMemoOrganizer(ctx, find)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(list) == 0 {
|
|
||||||
return nil, errors.New("not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
return list[0], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) ListMemoOrganizer(ctx context.Context, find *FindMemoOrganizer) ([]*MemoOrganizer, error) {
|
|
||||||
return s.driver.ListMemoOrganizer(ctx, find)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) DeleteMemoOrganizer(ctx context.Context, delete *DeleteMemoOrganizer) error {
|
|
||||||
return s.driver.DeleteMemoOrganizer(ctx, delete)
|
|
||||||
}
|
|
||||||
@ -1,64 +0,0 @@
|
|||||||
package teststore
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
|
|
||||||
"github.com/usememos/memos/store"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestMemoOrganizerStore(t *testing.T) {
|
|
||||||
ctx := context.Background()
|
|
||||||
ts := NewTestingStore(ctx, t)
|
|
||||||
user, err := createTestingHostUser(ctx, ts)
|
|
||||||
require.NoError(t, err)
|
|
||||||
memoCreate := &store.Memo{
|
|
||||||
UID: "main-memo",
|
|
||||||
CreatorID: user.ID,
|
|
||||||
Content: "main memo content",
|
|
||||||
Visibility: store.Public,
|
|
||||||
}
|
|
||||||
memo, err := ts.CreateMemo(ctx, memoCreate)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, memoCreate.Content, memo.Content)
|
|
||||||
|
|
||||||
memoOrganizer, err := ts.UpsertMemoOrganizer(ctx, &store.MemoOrganizer{
|
|
||||||
MemoID: memo.ID,
|
|
||||||
UserID: user.ID,
|
|
||||||
Pinned: true,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.NotNil(t, memoOrganizer)
|
|
||||||
require.Equal(t, memo.ID, memoOrganizer.MemoID)
|
|
||||||
require.Equal(t, user.ID, memoOrganizer.UserID)
|
|
||||||
require.Equal(t, true, memoOrganizer.Pinned)
|
|
||||||
|
|
||||||
memoOrganizerTemp, err := ts.GetMemoOrganizer(ctx, &store.FindMemoOrganizer{
|
|
||||||
MemoID: memo.ID,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, memoOrganizer, memoOrganizerTemp)
|
|
||||||
memoOrganizerTemp, err = ts.UpsertMemoOrganizer(ctx, &store.MemoOrganizer{
|
|
||||||
MemoID: memo.ID,
|
|
||||||
UserID: user.ID,
|
|
||||||
Pinned: false,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.NotNil(t, memoOrganizerTemp)
|
|
||||||
require.Equal(t, memo.ID, memoOrganizerTemp.MemoID)
|
|
||||||
require.Equal(t, user.ID, memoOrganizerTemp.UserID)
|
|
||||||
require.Equal(t, false, memoOrganizerTemp.Pinned)
|
|
||||||
err = ts.DeleteMemoOrganizer(ctx, &store.DeleteMemoOrganizer{
|
|
||||||
MemoID: &memo.ID,
|
|
||||||
UserID: &user.ID,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
memoOrganizers, err := ts.ListMemoOrganizer(ctx, &store.FindMemoOrganizer{
|
|
||||||
UserID: user.ID,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, 0, len(memoOrganizers))
|
|
||||||
ts.Close()
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue