43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package database
|
|
|
|
import (
|
|
"github.com/gotify/server/auth"
|
|
"github.com/gotify/server/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
|
|
)
|
|
|
|
// New creates a new wrapper for the gorm database framework.
|
|
func New(dialect, connection, defaultUser, defaultPass string, strength int) (*GormDatabase, error) {
|
|
db, err := gorm.Open(dialect, connection)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// 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
|
|
if dialect == "sqlite3" {
|
|
db.DB().SetMaxOpenConns(1)
|
|
}
|
|
if !db.HasTable(new(model.User)) && !db.HasTable(new(model.Message)) &&
|
|
!db.HasTable(new(model.Client)) && !db.HasTable(new(model.Application)) {
|
|
db.AutoMigrate(new(model.User), new(model.Application), new(model.Message), new(model.Client))
|
|
db.Create(&model.User{Name: defaultUser, Pass: auth.CreatePassword(defaultPass, strength), Admin: true})
|
|
}
|
|
|
|
return &GormDatabase{DB: db}, nil
|
|
}
|
|
|
|
// 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()
|
|
}
|