YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: storage setting changed don't take effect (#2385)

Browse Source 
* fix: Storage setting changed don't take effect

* fix: Storage setting changed don't take effect

* fix: Storage setting changed don't take effect
This commit is contained in:
guopeng
2023-10-16 21:07:21 +08:00
committed by GitHub
parent c999d71455
commit 73f59eaf09
6 changed files with 67 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
store/db/mysql/mysql.go
View File

@ -5,8 +5,10 @@ import (
"database/sql"
"fmt"
"github.com/go-sql-driver/mysql"
"github.com/pkg/errors"
"github.com/usememos/memos/common/log"
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/store"
)
@ -14,18 +16,28 @@ import (
type DB struct {
db *sql.DB
profile *profile.Profile
config *mysql.Config
}
func NewDB(profile *profile.Profile) (store.Driver, error) {
// Open MySQL connection with parameter.
// multiStatements=true is required for migration.
// See more in: https://github.com/go-sql-driver/mysql#multistatements
db, err := sql.Open("mysql", fmt.Sprintf("%s?multiStatements=true", profile.DSN))
dsn := fmt.Sprintf("%s?multiStatements=true", profile.DSN)
var err error
driver := DB{profile: profile}
driver.config, err = mysql.ParseDSN(dsn)
if err != nil {
log.Error(fmt.Sprintf("DSN parse error: %s", dsn))
return nil, errors.New("Parse DSN eroor")
}
driver.db, err = sql.Open("mysql", dsn)
if err != nil {
return nil, errors.Wrapf(err, "failed to open db: %s", profile.DSN)
}
driver := DB{db: db, profile: profile}
return &driver, nil
}
@ -67,6 +79,32 @@ func (*DB) BackupTo(context.Context, string) error {
return errors.New("Please use mysqldump to backup")
}
func (d *DB) GetCurrentDBSize(ctx context.Context) (int64, error) {
query := "SELECT SUM(`data_length` + `index_length`) AS `size` " +
" FROM information_schema.TABLES" +
" WHERE `table_schema` = ?" +
" GROUP BY `table_schema`"
rows, err := d.db.QueryContext(ctx, query, d.config.DBName)
if err != nil {
log.Error("Query db size error, make sure you have enough privilege")
return 0, err
}
defer rows.Close()
var size int64
for rows.Next() {
if err := rows.Scan(&size); err != nil {
return 0, err
}
}
if rows.Err() != nil {
return 0, rows.Err()
}
return size, nil
}
func (d *DB) Close() error {
return d.db.Close()
}