143 lines
2.8 KiB
Go
143 lines
2.8 KiB
Go
package config
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
|
"github.com/photoprism/photoprism/internal/entity"
|
|
"github.com/photoprism/photoprism/internal/tidb"
|
|
)
|
|
|
|
// DatabaseDriver returns the database driver name.
|
|
func (c *Config) DatabaseDriver() string {
|
|
if c.config.DatabaseDriver == "" {
|
|
return DbTiDB
|
|
}
|
|
|
|
return c.config.DatabaseDriver
|
|
}
|
|
|
|
// DatabaseDsn returns the database data source name (DSN).
|
|
func (c *Config) DatabaseDsn() string {
|
|
if c.config.DatabaseDsn == "" {
|
|
return "root:photoprism@tcp(localhost:4000)/photoprism?parseTime=true"
|
|
}
|
|
|
|
return c.config.DatabaseDsn
|
|
}
|
|
|
|
// Db returns the db connection.
|
|
func (c *Config) Db() *gorm.DB {
|
|
if c.db == nil {
|
|
log.Fatal("database not initialised.")
|
|
}
|
|
|
|
return c.db
|
|
}
|
|
|
|
// CloseDb closes the db connection (if any).
|
|
func (c *Config) CloseDb() error {
|
|
if c.db != nil {
|
|
if err := c.db.Close(); err == nil {
|
|
c.db = nil
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// MigrateDb will start a migration process.
|
|
func (c *Config) MigrateDb() {
|
|
db := c.Db()
|
|
|
|
// db.LogMode(true)
|
|
|
|
db.AutoMigrate(
|
|
&entity.File{},
|
|
&entity.Photo{},
|
|
&entity.Event{},
|
|
&entity.Place{},
|
|
&entity.Location{},
|
|
&entity.Camera{},
|
|
&entity.Lens{},
|
|
&entity.Country{},
|
|
&entity.Share{},
|
|
|
|
&entity.Album{},
|
|
&entity.PhotoAlbum{},
|
|
&entity.Label{},
|
|
&entity.Category{},
|
|
&entity.PhotoLabel{},
|
|
&entity.Keyword{},
|
|
&entity.PhotoKeyword{},
|
|
)
|
|
|
|
entity.CreateUnknownPlace(db)
|
|
entity.CreateUnknownCountry(db)
|
|
}
|
|
|
|
// connectToDatabase establishes a database connection.
|
|
// When used with the internal driver, it may create a new database server instance.
|
|
// It tries to do this 12 times with a 5 second sleep interval in between.
|
|
func (c *Config) connectToDatabase(ctx context.Context) error {
|
|
dbDriver := c.DatabaseDriver()
|
|
dbDsn := c.DatabaseDsn()
|
|
|
|
if dbDriver == "" {
|
|
return errors.New("can't connect: database driver not specified")
|
|
}
|
|
|
|
if dbDsn == "" {
|
|
return errors.New("can't connect: database DSN not specified")
|
|
}
|
|
|
|
isTiDB := false
|
|
initSuccess := false
|
|
|
|
if dbDriver == DbTiDB {
|
|
isTiDB = true
|
|
dbDriver = DbMySQL
|
|
}
|
|
|
|
db, err := gorm.Open(dbDriver, dbDsn)
|
|
if err != nil || db == nil {
|
|
if isTiDB {
|
|
log.Infof("starting database server at %s:%d\n", c.SqlServerHost(), c.SqlServerPort())
|
|
|
|
go tidb.Start(ctx, c.SqlServerPath(), c.SqlServerPort(), c.SqlServerHost(), c.Debug())
|
|
}
|
|
|
|
for i := 1; i <= 12; i++ {
|
|
time.Sleep(5 * time.Second)
|
|
|
|
db, err = gorm.Open(dbDriver, dbDsn)
|
|
|
|
if db != nil && err == nil {
|
|
break
|
|
}
|
|
|
|
if isTiDB && !initSuccess {
|
|
err = tidb.InitDatabase(c.SqlServerPort(), c.SqlServerPassword())
|
|
|
|
if err != nil {
|
|
log.Debug(err)
|
|
} else {
|
|
initSuccess = true
|
|
}
|
|
}
|
|
}
|
|
|
|
if err != nil || db == nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
c.db = db
|
|
return err
|
|
}
|