123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413 |
- package store
- import (
- "context"
- "database/sql"
- "encoding/json"
- "fmt"
- "strings"
- "github.com/usememos/memos/api"
- "github.com/usememos/memos/common"
- )
- // memoRaw is the store model for an Memo.
- // Fields have exactly the same meanings as Memo.
- type memoRaw struct {
- ID int
- // Standard fields
- RowStatus api.RowStatus
- CreatorID int
- CreatedTs int64
- UpdatedTs int64
- // Domain specific fields
- Content string
- Visibility api.Visibility
- }
- // toMemo creates an instance of Memo based on the memoRaw.
- // This is intended to be called when we need to compose an Memo relationship.
- func (raw *memoRaw) toMemo() *api.Memo {
- return &api.Memo{
- ID: raw.ID,
- // Standard fields
- RowStatus: raw.RowStatus,
- CreatorID: raw.CreatorID,
- CreatedTs: raw.CreatedTs,
- UpdatedTs: raw.UpdatedTs,
- // Domain specific fields
- Content: raw.Content,
- Visibility: raw.Visibility,
- DisplayTs: raw.CreatedTs,
- }
- }
- func (s *Store) ComposeMemo(ctx context.Context, memo *api.Memo) (*api.Memo, error) {
- memoOrganizer, err := s.FindMemoOrganizer(ctx, &api.MemoOrganizerFind{
- MemoID: memo.ID,
- UserID: memo.CreatorID,
- })
- if err != nil && common.ErrorCode(err) != common.NotFound {
- return nil, err
- } else if memoOrganizer != nil {
- memo.Pinned = memoOrganizer.Pinned
- }
- if err = s.ComposeMemoCreator(ctx, memo); err != nil {
- return nil, err
- }
- if err = s.ComposeMemoResourceList(ctx, memo); err != nil {
- return nil, err
- }
- memoDisplayTsOptionKey := api.UserSettingMemoDisplayTsOptionKey
- memoDisplayTsOptionSetting, err := s.FindUserSetting(ctx, &api.UserSettingFind{
- UserID: memo.CreatorID,
- Key: &memoDisplayTsOptionKey,
- })
- if err != nil {
- return nil, err
- }
- memoDisplayTsOptionValue := "created_ts"
- if memoDisplayTsOptionSetting != nil {
- err = json.Unmarshal([]byte(memoDisplayTsOptionSetting.Value), &memoDisplayTsOptionValue)
- if err != nil {
- return nil, fmt.Errorf("failed to unmarshal user setting memo display ts option value")
- }
- }
- if memoDisplayTsOptionValue == "updated_ts" {
- memo.DisplayTs = memo.UpdatedTs
- }
- return memo, nil
- }
- func (s *Store) CreateMemo(ctx context.Context, create *api.MemoCreate) (*api.Memo, error) {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, FormatError(err)
- }
- defer tx.Rollback()
- memoRaw, err := createMemoRaw(ctx, tx, create)
- if err != nil {
- return nil, err
- }
- if err := tx.Commit(); err != nil {
- return nil, FormatError(err)
- }
- if err := s.cache.UpsertCache(api.MemoCache, memoRaw.ID, memoRaw); err != nil {
- return nil, err
- }
- memo, err := s.ComposeMemo(ctx, memoRaw.toMemo())
- if err != nil {
- return nil, err
- }
- return memo, nil
- }
- func (s *Store) PatchMemo(ctx context.Context, patch *api.MemoPatch) (*api.Memo, error) {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, FormatError(err)
- }
- defer tx.Rollback()
- memoRaw, err := patchMemoRaw(ctx, tx, patch)
- if err != nil {
- return nil, err
- }
- if err := tx.Commit(); err != nil {
- return nil, FormatError(err)
- }
- if err := s.cache.UpsertCache(api.MemoCache, memoRaw.ID, memoRaw); err != nil {
- return nil, err
- }
- memo, err := s.ComposeMemo(ctx, memoRaw.toMemo())
- if err != nil {
- return nil, err
- }
- return memo, nil
- }
- func (s *Store) FindMemoList(ctx context.Context, find *api.MemoFind) ([]*api.Memo, error) {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, FormatError(err)
- }
- defer tx.Rollback()
- memoRawList, err := findMemoRawList(ctx, tx, find)
- if err != nil {
- return nil, err
- }
- list := []*api.Memo{}
- for _, raw := range memoRawList {
- memo, err := s.ComposeMemo(ctx, raw.toMemo())
- if err != nil {
- return nil, err
- }
- list = append(list, memo)
- }
- return list, nil
- }
- func (s *Store) FindMemo(ctx context.Context, find *api.MemoFind) (*api.Memo, error) {
- if find.ID != nil {
- memoRaw := &memoRaw{}
- has, err := s.cache.FindCache(api.MemoCache, *find.ID, memoRaw)
- if err != nil {
- return nil, err
- }
- if has {
- memo, err := s.ComposeMemo(ctx, memoRaw.toMemo())
- if err != nil {
- return nil, err
- }
- return memo, nil
- }
- }
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, FormatError(err)
- }
- defer tx.Rollback()
- list, err := findMemoRawList(ctx, tx, find)
- if err != nil {
- return nil, err
- }
- if len(list) == 0 {
- return nil, &common.Error{Code: common.NotFound, Err: fmt.Errorf("not found")}
- }
- memoRaw := list[0]
- if err := s.cache.UpsertCache(api.MemoCache, memoRaw.ID, memoRaw); err != nil {
- return nil, err
- }
- memo, err := s.ComposeMemo(ctx, memoRaw.toMemo())
- if err != nil {
- return nil, err
- }
- return memo, nil
- }
- func (s *Store) DeleteMemo(ctx context.Context, delete *api.MemoDelete) error {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return FormatError(err)
- }
- defer tx.Rollback()
- if err := deleteMemo(ctx, tx, delete); err != nil {
- return FormatError(err)
- }
- if err := vacuum(ctx, tx); err != nil {
- return err
- }
- if err := tx.Commit(); err != nil {
- return FormatError(err)
- }
- s.cache.DeleteCache(api.MemoCache, delete.ID)
- return nil
- }
- func createMemoRaw(ctx context.Context, tx *sql.Tx, create *api.MemoCreate) (*memoRaw, error) {
- set := []string{"creator_id", "content", "visibility"}
- args := []interface{}{create.CreatorID, create.Content, create.Visibility}
- placeholder := []string{"?", "?", "?"}
- query := `
- INSERT INTO memo (
- ` + strings.Join(set, ", ") + `
- )
- VALUES (` + strings.Join(placeholder, ",") + `)
- RETURNING id, creator_id, created_ts, updated_ts, row_status, content, visibility
- `
- var memoRaw memoRaw
- if err := tx.QueryRowContext(ctx, query, args...).Scan(
- &memoRaw.ID,
- &memoRaw.CreatorID,
- &memoRaw.CreatedTs,
- &memoRaw.UpdatedTs,
- &memoRaw.RowStatus,
- &memoRaw.Content,
- &memoRaw.Visibility,
- ); err != nil {
- return nil, FormatError(err)
- }
- return &memoRaw, nil
- }
- func patchMemoRaw(ctx context.Context, tx *sql.Tx, patch *api.MemoPatch) (*memoRaw, error) {
- set, args := []string{}, []interface{}{}
- if v := patch.CreatedTs; v != nil {
- set, args = append(set, "created_ts = ?"), append(args, *v)
- }
- if v := patch.UpdatedTs; v != nil {
- set, args = append(set, "updated_ts = ?"), append(args, *v)
- }
- if v := patch.RowStatus; v != nil {
- set, args = append(set, "row_status = ?"), append(args, *v)
- }
- if v := patch.Content; v != nil {
- set, args = append(set, "content = ?"), append(args, *v)
- }
- if v := patch.Visibility; v != nil {
- set, args = append(set, "visibility = ?"), append(args, *v)
- }
- args = append(args, patch.ID)
- query := `
- UPDATE memo
- SET ` + strings.Join(set, ", ") + `
- WHERE id = ?
- RETURNING id, creator_id, created_ts, updated_ts, row_status, content, visibility
- `
- var memoRaw memoRaw
- if err := tx.QueryRowContext(ctx, query, args...).Scan(
- &memoRaw.ID,
- &memoRaw.CreatorID,
- &memoRaw.CreatedTs,
- &memoRaw.UpdatedTs,
- &memoRaw.RowStatus,
- &memoRaw.Content,
- &memoRaw.Visibility,
- ); err != nil {
- return nil, FormatError(err)
- }
- return &memoRaw, nil
- }
- func findMemoRawList(ctx context.Context, tx *sql.Tx, find *api.MemoFind) ([]*memoRaw, error) {
- where, args := []string{"1 = 1"}, []interface{}{}
- if v := find.ID; v != nil {
- where, args = append(where, "id = ?"), append(args, *v)
- }
- if v := find.CreatorID; v != nil {
- where, args = append(where, "creator_id = ?"), append(args, *v)
- }
- if v := find.RowStatus; v != nil {
- where, args = append(where, "row_status = ?"), append(args, *v)
- }
- if v := find.Pinned; v != nil {
- where = append(where, "id IN (SELECT memo_id FROM memo_organizer WHERE pinned = 1 AND user_id = memo.creator_id)")
- }
- if v := find.ContentSearch; v != nil {
- where, args = append(where, "content LIKE ?"), append(args, "%"+*v+"%")
- }
- if v := find.VisibilityList; len(v) != 0 {
- list := []string{}
- for _, visibility := range v {
- list = append(list, fmt.Sprintf("$%d", len(args)+1))
- args = append(args, visibility)
- }
- where = append(where, fmt.Sprintf("visibility in (%s)", strings.Join(list, ",")))
- }
- query := `
- SELECT
- id,
- creator_id,
- created_ts,
- updated_ts,
- row_status,
- content,
- visibility
- FROM memo
- WHERE ` + strings.Join(where, " AND ") + `
- ORDER BY created_ts DESC
- `
- rows, err := tx.QueryContext(ctx, query, args...)
- if err != nil {
- return nil, FormatError(err)
- }
- defer rows.Close()
- memoRawList := make([]*memoRaw, 0)
- for rows.Next() {
- var memoRaw memoRaw
- if err := rows.Scan(
- &memoRaw.ID,
- &memoRaw.CreatorID,
- &memoRaw.CreatedTs,
- &memoRaw.UpdatedTs,
- &memoRaw.RowStatus,
- &memoRaw.Content,
- &memoRaw.Visibility,
- ); err != nil {
- return nil, FormatError(err)
- }
- memoRawList = append(memoRawList, &memoRaw)
- }
- if err := rows.Err(); err != nil {
- return nil, FormatError(err)
- }
- return memoRawList, nil
- }
- func deleteMemo(ctx context.Context, tx *sql.Tx, delete *api.MemoDelete) error {
- where, args := []string{"id = ?"}, []interface{}{delete.ID}
- stmt := `DELETE FROM memo WHERE ` + strings.Join(where, " AND ")
- result, err := tx.ExecContext(ctx, stmt, args...)
- if err != nil {
- return FormatError(err)
- }
- rows, _ := result.RowsAffected()
- if rows == 0 {
- return &common.Error{Code: common.NotFound, Err: fmt.Errorf("memo not found")}
- }
- return nil
- }
- func vacuumMemo(ctx context.Context, tx *sql.Tx) error {
- stmt := `
- DELETE FROM
- memo
- WHERE
- creator_id NOT IN (
- SELECT
- id
- FROM
- user
- )`
- _, err := tx.ExecContext(ctx, stmt)
- if err != nil {
- return FormatError(err)
- }
- return nil
- }
|