mirror of
https://github.com/usememos/memos.git
synced 2024-11-16 11:46:07 +08:00
122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
// Import the SQLite driver.
|
|
_ "modernc.org/sqlite"
|
|
|
|
"github.com/usememos/memos/server/profile"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
type DB struct {
|
|
db *sql.DB
|
|
profile *profile.Profile
|
|
}
|
|
|
|
// NewDB opens a database specified by its database driver name and a
|
|
// driver-specific data source name, usually consisting of at least a
|
|
// database name and connection information.
|
|
func NewDB(profile *profile.Profile) (store.Driver, error) {
|
|
// Ensure a DSN is set before attempting to open the database.
|
|
if profile.DSN == "" {
|
|
return nil, errors.New("dsn required")
|
|
}
|
|
|
|
// Connect to the database with some sane settings:
|
|
// - No shared-cache: it's obsolete; WAL journal mode is a better solution.
|
|
// - No foreign key constraints: it's currently disabled by default, but it's a
|
|
// good practice to be explicit and prevent future surprises on SQLite upgrades.
|
|
// - Journal mode set to WAL: it's the recommended journal mode for most applications
|
|
// as it prevents locking issues.
|
|
//
|
|
// Notes:
|
|
// - When using the `modernc.org/sqlite` driver, each pragma must be prefixed with `_pragma=`.
|
|
//
|
|
// References:
|
|
// - https://pkg.go.dev/modernc.org/sqlite#Driver.Open
|
|
// - https://www.sqlite.org/sharedcache.html
|
|
// - https://www.sqlite.org/pragma.html
|
|
sqliteDB, err := sql.Open("sqlite", profile.DSN+"?_pragma=foreign_keys(0)&_pragma=busy_timeout(10000)&_pragma=journal_mode(WAL)")
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to open db with dsn: %s", profile.DSN)
|
|
}
|
|
|
|
driver := DB{db: sqliteDB, profile: profile}
|
|
|
|
return &driver, nil
|
|
}
|
|
|
|
func (d *DB) GetDB() *sql.DB {
|
|
return d.db
|
|
}
|
|
|
|
func (d *DB) Vacuum(ctx context.Context) error {
|
|
tx, err := d.db.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tx.Rollback()
|
|
|
|
if err := vacuumImpl(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Vacuum sqlite database file size after deleting resource.
|
|
if _, err := d.db.Exec("VACUUM"); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func vacuumImpl(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 := vacuumUserSetting(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
if err := vacuumMemoOrganizer(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
if err := vacuumMemoRelations(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
if err := vacuumInbox(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
if err := vacuumTag(ctx, tx); err != nil {
|
|
// Prevent revive warning.
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DB) GetCurrentDBSize(context.Context) (int64, error) {
|
|
fi, err := os.Stat(d.profile.DSN)
|
|
if err != nil {
|
|
return 0, status.Errorf(codes.Internal, "failed to get file info: %v", err)
|
|
}
|
|
|
|
return fi.Size(), nil
|
|
}
|
|
|
|
func (d *DB) Close() error {
|
|
return d.db.Close()
|
|
}
|