2022-05-16 07:37:23 +08:00
|
|
|
package store
|
|
|
|
|
2022-05-22 00:59:22 +08:00
|
|
|
import (
|
2022-11-06 12:21:58 +08:00
|
|
|
"context"
|
2023-02-11 14:19:26 +08:00
|
|
|
"sync"
|
2022-06-27 22:09:06 +08:00
|
|
|
|
2023-09-17 22:55:13 +08:00
|
|
|
"github.com/usememos/memos/server/profile"
|
2022-05-22 00:59:22 +08:00
|
|
|
)
|
|
|
|
|
2022-08-24 21:53:12 +08:00
|
|
|
// Store provides database access to all raw objects.
|
2022-05-16 07:37:23 +08:00
|
|
|
type Store struct {
|
2024-04-13 10:50:25 +08:00
|
|
|
Profile *profile.Profile
|
|
|
|
driver Driver
|
|
|
|
workspaceSettingCache sync.Map // map[string]*storepb.WorkspaceSetting
|
|
|
|
userCache sync.Map // map[int]*User
|
2024-05-12 13:19:31 +08:00
|
|
|
userSettingCache sync.Map // map[string]*storepb.UserSetting
|
2024-04-17 08:56:52 +08:00
|
|
|
idpCache sync.Map // map[int]*storepb.IdentityProvider
|
2022-05-16 07:37:23 +08:00
|
|
|
}
|
|
|
|
|
2022-08-24 21:53:12 +08:00
|
|
|
// New creates a new instance of Store.
|
2023-09-27 09:27:31 +08:00
|
|
|
func New(driver Driver, profile *profile.Profile) *Store {
|
2022-05-16 07:37:23 +08:00
|
|
|
return &Store{
|
2023-09-26 17:16:58 +08:00
|
|
|
driver: driver,
|
2024-01-02 08:29:18 +08:00
|
|
|
Profile: profile,
|
2022-05-16 07:37:23 +08:00
|
|
|
}
|
|
|
|
}
|
2022-11-06 12:21:58 +08:00
|
|
|
|
2024-04-17 09:00:16 +08:00
|
|
|
func (*Store) MigrateManually(context.Context) error {
|
2024-02-21 23:43:18 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2023-09-27 09:27:31 +08:00
|
|
|
func (s *Store) Close() error {
|
|
|
|
return s.driver.Close()
|
2022-11-06 12:21:58 +08:00
|
|
|
}
|
2023-10-16 21:07:21 +08:00
|
|
|
|
|
|
|
func (s *Store) GetCurrentDBSize(ctx context.Context) (int64, error) {
|
|
|
|
return s.driver.GetCurrentDBSize(ctx)
|
|
|
|
}
|