mirror of https://github.com/muety/wakapi.git
44 lines
976 B
Go
44 lines
976 B
Go
package repositories
|
|
|
|
import (
|
|
"github.com/muety/wakapi/config"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type MetricsRepository struct {
|
|
config *config.Config
|
|
db *gorm.DB
|
|
}
|
|
|
|
const sizeTplMysql = `
|
|
SELECT SUM(data_length + index_length)
|
|
FROM information_schema.tables
|
|
WHERE table_schema = ?
|
|
GROUP BY table_schema`
|
|
|
|
const sizeTplPostgres = `SELECT pg_database_size(?);`
|
|
|
|
const sizeTplSqlite = `
|
|
SELECT page_count * page_size as size
|
|
FROM pragma_page_count(), pragma_page_size();`
|
|
|
|
func NewMetricsRepository(db *gorm.DB) *MetricsRepository {
|
|
return &MetricsRepository{config: config.Get(), db: db}
|
|
}
|
|
|
|
func (srv *MetricsRepository) GetDatabaseSize() (size int64, err error) {
|
|
cfg := srv.config.Db
|
|
|
|
query := srv.db.Raw("SELECT 0")
|
|
if cfg.IsMySQL() {
|
|
query = srv.db.Raw(sizeTplMysql, cfg.Name)
|
|
} else if cfg.IsPostgres() {
|
|
query = srv.db.Raw(sizeTplPostgres, cfg.Name)
|
|
} else if cfg.IsSQLite() {
|
|
query = srv.db.Raw(sizeTplSqlite)
|
|
}
|
|
|
|
err = query.Scan(&size).Error
|
|
return size, err
|
|
}
|