mirror of
https://github.com/gotify/server.git
synced 2024-01-28 15:20:56 +03:00
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package database
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/gotify/server/v2/auth/password"
|
|
"github.com/gotify/server/v2/model"
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/mysql" // enable the mysql dialect.
|
|
_ "github.com/jinzhu/gorm/dialects/postgres" // enable the postgres dialect.
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite" // enable the sqlite3 dialect.
|
|
)
|
|
|
|
var mkdirAll = os.MkdirAll
|
|
|
|
// New creates a new wrapper for the gorm database framework.
|
|
func New(dialect, connection, defaultUser, defaultPass string, strength int, createDefaultUserIfNotExist bool) (*GormDatabase, error) {
|
|
createDirectoryIfSqlite(dialect, connection)
|
|
|
|
db, err := gorm.Open(dialect, connection)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// We normally don't need that much connections, so we limit them. F.ex. mysql complains about
|
|
// "too many connections", while load testing Gotify.
|
|
db.DB().SetMaxOpenConns(10)
|
|
|
|
if dialect == "sqlite3" {
|
|
// We use the database connection inside the handlers from the http
|
|
// framework, therefore concurrent access occurs. Sqlite cannot handle
|
|
// concurrent writes, so we limit sqlite to one connection.
|
|
// see https://github.com/mattn/go-sqlite3/issues/274
|
|
db.DB().SetMaxOpenConns(1)
|
|
}
|
|
|
|
if dialect == "mysql" {
|
|
// Mysql has a setting called wait_timeout, which defines the duration
|
|
// after which a connection may not be used anymore.
|
|
// The default for this setting on mariadb is 10 minutes.
|
|
// See https://github.com/docker-library/mariadb/issues/113
|
|
db.DB().SetConnMaxLifetime(9 * time.Minute)
|
|
}
|
|
|
|
if err := db.AutoMigrate(new(model.User), new(model.Application), new(model.Message), new(model.Client), new(model.PluginConf)).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := prepareBlobColumn(dialect, db); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
userCount := 0
|
|
db.Find(new(model.User)).Count(&userCount)
|
|
if createDefaultUserIfNotExist && userCount == 0 {
|
|
db.Create(&model.User{Name: defaultUser, Pass: password.CreatePassword(defaultPass, strength), Admin: true})
|
|
}
|
|
|
|
return &GormDatabase{DB: db}, nil
|
|
}
|
|
|
|
func prepareBlobColumn(dialect string, db *gorm.DB) error {
|
|
blobType := ""
|
|
switch dialect {
|
|
case "mysql":
|
|
blobType = "longblob"
|
|
case "postgres":
|
|
blobType = "bytea"
|
|
}
|
|
if blobType != "" {
|
|
for _, target := range []struct {
|
|
Table interface{}
|
|
Column string
|
|
}{
|
|
{model.Message{}, "extras"},
|
|
{model.PluginConf{}, "config"},
|
|
{model.PluginConf{}, "storage"},
|
|
} {
|
|
if err := db.Model(target.Table).ModifyColumn(target.Column, blobType).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func createDirectoryIfSqlite(dialect, connection string) {
|
|
if dialect == "sqlite3" {
|
|
if _, err := os.Stat(filepath.Dir(connection)); os.IsNotExist(err) {
|
|
if err := mkdirAll(filepath.Dir(connection), 0o777); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// GormDatabase is a wrapper for the gorm framework.
|
|
type GormDatabase struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
// Close closes the gorm database connection.
|
|
func (d *GormDatabase) Close() {
|
|
d.DB.Close()
|
|
}
|