YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-03-18 03:30:07 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: split sql to driver (#2279)

Browse Source 
* Add new database interface for SQL operations

* Move SQL code of Activity into Database

* Rename `Database` into `Driver`

* Move SQL code of SystemSetting into Driver

* Fix store.New in text code

* Change database into driver in the variables

* Change sqlite3.New into sqlite3.NewDriver
This commit is contained in:
Athurg Gooth 2023-09-26 17:16:58 +08:00 committed by GitHub
parent 85ed0202d8
commit 41eba71f0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
12 changed files with 145 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/memos.go
View File

@ -17,6 +17,7 @@ import (
_profile "github.com/usememos/memos/server/profile"
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
"github.com/usememos/memos/store/sqlite3"
)
const (
@ -54,7 +55,9 @@ var (
return
}
store := store.New(db.DBInstance, profile)
driver := sqlite3.NewDriver(db.DBInstance)
store := store.New(db.DBInstance, driver, profile)
s, err := server.NewServer(ctx, profile, store)
if err != nil {
cancel()

5
cmd/mvrss.go
View File

@ -10,6 +10,7 @@ import (
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
"github.com/usememos/memos/store/sqlite3"
)
var (
@ -49,7 +50,9 @@ var (
return
}
s := store.New(db.DBInstance, profile)
driver := sqlite3.NewDriver(db.DBInstance)
s := store.New(db.DBInstance, driver, profile)
resources, err := s.ListResources(ctx, &store.FindResource{})
if err != nil {
fmt.Printf("failed to list resources, error: %+v\n", err)

5
cmd/setup.go
View File

@ -12,6 +12,7 @@ import (
"github.com/usememos/memos/common/util"
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
"github.com/usememos/memos/store/sqlite3"
)
var (
@ -46,7 +47,9 @@ var (
return
}
store := store.New(db.DBInstance, profile)
driver := sqlite3.NewDriver(db.DBInstance)
store := store.New(db.DBInstance, driver, profile)
if err := ExecuteSetup(ctx, store, hostUsername, hostPassword); err != nil {
fmt.Printf("failed to setup, error: %+v\n", err)
return

20
store/activity.go
View File

@ -18,23 +18,5 @@ type Activity struct {
}
func (s *Store) CreateActivity(ctx context.Context, create *Activity) (*Activity, error) {
stmt := `
INSERT INTO activity (
creator_id,
type,
level,
payload
)
VALUES (?, ?, ?, ?)
RETURNING id, created_ts
`
if err := s.db.QueryRowContext(ctx, stmt, create.CreatorID, create.Type, create.Level, create.Payload).Scan(
&create.ID,
&create.CreatedTs,
); err != nil {
return nil, err
}
activity := create
return activity, nil
return s.driver.CreateActivity(ctx, create)
}

10
store/driver.go Normal file
View File

@ -0,0 +1,10 @@
package store
import "context"
type Driver interface {
CreateActivity(ctx context.Context, create *Activity) (*Activity, error)
UpsertSystemSetting(ctx context.Context, upsert *SystemSetting) (*SystemSetting, error)
ListSystemSettings(ctx context.Context, find *FindSystemSetting) ([]*SystemSetting, error)
}

28
store/sqlite3/activity.go Normal file
View File

@ -0,0 +1,28 @@
package sqlite3
import (
"context"
"github.com/usememos/memos/store"
)
func (d *Driver) CreateActivity(ctx context.Context, create *store.Activity) (*store.Activity, error) {
stmt := `
INSERT INTO activity (
creator_id,
type,
level,
payload
)
VALUES (?, ?, ?, ?)
RETURNING id, created_ts
`
if err := d.db.QueryRowContext(ctx, stmt, create.CreatorID, create.Type, create.Level, create.Payload).Scan(
&create.ID,
&create.CreatedTs,
); err != nil {
return nil, err
}
return create, nil
}

15
store/sqlite3/driver.go Normal file
View File

@ -0,0 +1,15 @@
package sqlite3
import (
"database/sql"
"github.com/usememos/memos/store"
)
type Driver struct {
db *sql.DB
}
func NewDriver(db *sql.DB) store.Driver {
return &Driver{db: db}
}

66
store/sqlite3/system_setting.go Normal file
View File

@ -0,0 +1,66 @@
package sqlite3
import (
"context"
"strings"
"github.com/usememos/memos/store"
)
func (d *Driver) UpsertSystemSetting(ctx context.Context, upsert *store.SystemSetting) (*store.SystemSetting, error) {
stmt := `
INSERT INTO system_setting (
name, value, description
)
VALUES (?, ?, ?)
ON CONFLICT(name) DO UPDATE
SET
value = EXCLUDED.value,
description = EXCLUDED.description
`
if _, err := d.db.ExecContext(ctx, stmt, upsert.Name, upsert.Value, upsert.Description); err != nil {
return nil, err
}
return upsert, nil
}
func (d *Driver) ListSystemSettings(ctx context.Context, find *store.FindSystemSetting) ([]*store.SystemSetting, error) {
where, args := []string{"1 = 1"}, []any{}
if find.Name != "" {
where, args = append(where, "name = ?"), append(args, find.Name)
}
query := `
SELECT
name,
value,
description
FROM system_setting
WHERE ` + strings.Join(where, " AND ")
rows, err := d.db.QueryContext(ctx, query, args...)
if err != nil {
return nil, err
}
defer rows.Close()
list := []*store.SystemSetting{}
for rows.Next() {
systemSettingMessage := &store.SystemSetting{}
if err := rows.Scan(
&systemSettingMessage.Name,
&systemSettingMessage.Value,
&systemSettingMessage.Description,
); err != nil {
return nil, err
}
list = append(list, systemSettingMessage)
}
if err := rows.Err(); err != nil {
return nil, err
}
return list, nil
}

4
store/store.go
View File

@ -16,6 +16,7 @@ import (
type Store struct {
Profile *profile.Profile
db *sql.DB
driver Driver
systemSettingCache sync.Map // map[string]*SystemSetting
userCache sync.Map // map[int]*User
userSettingCache sync.Map // map[string]*UserSetting
@ -23,10 +24,11 @@ type Store struct {
}
// New creates a new instance of Store.
func New(db *sql.DB, profile *profile.Profile) *Store {
func New(db *sql.DB, driver Driver, profile *profile.Profile) *Store {
return &Store{
Profile: profile,
db: db,
driver: driver,
}
}

51
store/system_setting.go
View File

@ -2,7 +2,6 @@ package store
import (
"context"
"strings"
)
type SystemSetting struct {
@ -16,60 +15,14 @@ type FindSystemSetting struct {
}
func (s *Store) UpsertSystemSetting(ctx context.Context, upsert *SystemSetting) (*SystemSetting, error) {
stmt := `
INSERT INTO system_setting (
name, value, description
)
VALUES (?, ?, ?)
ON CONFLICT(name) DO UPDATE
SET
value = EXCLUDED.value,
description = EXCLUDED.description
`
if _, err := s.db.ExecContext(ctx, stmt, upsert.Name, upsert.Value, upsert.Description); err != nil {
return nil, err
}
systemSetting := upsert
return systemSetting, nil
return s.driver.UpsertSystemSetting(ctx, upsert)
}
func (s *Store) ListSystemSettings(ctx context.Context, find *FindSystemSetting) ([]*SystemSetting, error) {
where, args := []string{"1 = 1"}, []any{}
if find.Name != "" {
where, args = append(where, "name = ?"), append(args, find.Name)
}
query := `
SELECT
name,
value,
description
FROM system_setting
WHERE ` + strings.Join(where, " AND ")
rows, err := s.db.QueryContext(ctx, query, args...)
list, err := s.driver.ListSystemSettings(ctx, find)
if err != nil {
return nil, err
}
defer rows.Close()
list := []*SystemSetting{}
for rows.Next() {
systemSettingMessage := &SystemSetting{}
if err := rows.Scan(
&systemSettingMessage.Name,
&systemSettingMessage.Value,
&systemSettingMessage.Description,
); err != nil {
return nil, err
}
list = append(list, systemSettingMessage)
}
if err := rows.Err(); err != nil {
return nil, err
}
for _, systemSettingMessage := range list {
s.systemSettingCache.Store(systemSettingMessage.Name, systemSettingMessage)

5
test/server/server.go
View File

@ -19,6 +19,7 @@ import (
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
"github.com/usememos/memos/store/sqlite3"
"github.com/usememos/memos/test"
)
@ -39,7 +40,9 @@ func NewTestingServer(ctx context.Context, t *testing.T) (*TestingServer, error)
return nil, errors.Wrap(err, "failed to migrate db")
}
store := store.New(db.DBInstance, profile)
driver := sqlite3.NewDriver(db.DBInstance)
store := store.New(db.DBInstance, driver, profile)
server, err := server.NewServer(ctx, profile, store)
if err != nil {
return nil, errors.Wrap(err, "failed to create server")

5
test/store/store.go
View File

@ -7,6 +7,7 @@ import (
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
"github.com/usememos/memos/store/sqlite3"
"github.com/usememos/memos/test"
// sqlite driver.
@ -23,6 +24,8 @@ func NewTestingStore(ctx context.Context, t *testing.T) *store.Store {
fmt.Printf("failed to migrate db, error: %+v\n", err)
}
store := store.New(db.DBInstance, profile)
driver := sqlite3.NewDriver(db.DBInstance)
store := store.New(db.DBInstance, driver, profile)
return store
}