1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- package store
- import (
- "context"
- "database/sql"
- "sync"
- "github.com/usememos/memos/server/profile"
- )
- // Store provides database access to all raw objects.
- type Store struct {
- db *sql.DB
- profile *profile.Profile
- userCache sync.Map // map[int]*userRaw
- userSettingCache sync.Map // map[string]*userSettingRaw
- memoCache sync.Map // map[int]*memoRaw
- shortcutCache sync.Map // map[int]*shortcutRaw
- idpCache sync.Map // map[int]*identityProviderMessage
- }
- // New creates a new instance of Store.
- func New(db *sql.DB, profile *profile.Profile) *Store {
- return &Store{
- db: db,
- profile: profile,
- }
- }
- func (s *Store) Vacuum(ctx context.Context) error {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return FormatError(err)
- }
- defer tx.Rollback()
- if err := vacuum(ctx, tx); err != nil {
- return err
- }
- if err := tx.Commit(); err != nil {
- return FormatError(err)
- }
- // Vacuum sqlite database file size after deleting resource.
- if _, err := s.db.Exec("VACUUM"); err != nil {
- return err
- }
- return nil
- }
- // Exec vacuum records in a transaction.
- func vacuum(ctx context.Context, tx *sql.Tx) error {
- if err := vacuumMemo(ctx, tx); err != nil {
- return err
- }
- if err := vacuumResource(ctx, tx); err != nil {
- return err
- }
- if err := vacuumShortcut(ctx, tx); err != nil {
- return err
- }
- if err := vacuumUserSetting(ctx, tx); err != nil {
- return err
- }
- if err := vacuumMemoOrganizer(ctx, tx); err != nil {
- return err
- }
- if err := vacuumMemoResource(ctx, tx); err != nil {
- return err
- }
- if err := vacuumTag(ctx, tx); err != nil {
- // Prevent revive warning.
- return err
- }
- return nil
- }
|