public/photoview
1
0
Fork 0
mirror of https://git.vectorsigma.ru/public/photoview.git synced 2026-08-03 21:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Work on sqlite support

Browse Source
This commit is contained in:
viktorstrate
2021-01-17 12:45:23 +01:00
parent f711da4d9c
commit fcdb5b4e7b
7 changed files with 45 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
api/database/database.go
View File

@@ -18,13 +18,13 @@ import (
)
func getMysqlAddress() (*url.URL, error) {
address, err := url.Parse(os.Getenv("MYSQL_URL"))
address, err := url.Parse(os.Getenv("PHOTOVIEW_MYSQL_URL"))
if err != nil {
return nil, errors.Wrap(err, "Could not parse mysql url")
}
if address.String() == "" {
return nil, errors.New("Environment variable MYSQL_URL missing, exiting")
return nil, errors.New("Environment variable PHOTOVIEW_MYSQL_URL missing, exiting")
}
queryValues := address.Query()
@@ -35,6 +35,28 @@ func getMysqlAddress() (*url.URL, error) {
return address, nil
}
func getSqliteAddress() (*url.URL, error) {
path := os.Getenv("PHOTOVIEW_SQLITE_PATH")
if path == "" {
path = "photoview.db"
}
address, err := url.Parse(path)
if err != nil {
return nil, errors.Wrapf(err, "Could not parse sqlite url (%s)", path)
}
queryValues := address.Query()
queryValues.Add("cache", "shared")
queryValues.Add("mode", "rwc")
// queryValues.Add("_busy_timeout", "60000") // 1 minute
address.RawQuery = queryValues.Encode()
// log.Panicf("%s", address.String())
return address, nil
}
// SetupDatabase connects to the database using environment variables
func SetupDatabase() (*gorm.DB, error) {
@@ -44,7 +66,8 @@ func SetupDatabase() (*gorm.DB, error) {
config.Logger = logger.Default.LogMode(logger.Info)
var databaseDialect gorm.Dialector
switch strings.ToLower(os.Getenv("PHOTOVIEW_DATABASE_DRIVER")) {
databaseDriver := strings.ToLower(os.Getenv("PHOTOVIEW_DATABASE_DRIVER"))
switch databaseDriver {
case "mysql":
mysqlAddress, err := getMysqlAddress()
if err != nil {
@@ -54,11 +77,11 @@ func SetupDatabase() (*gorm.DB, error) {
databaseDialect = mysql.Open(mysqlAddress.String())
case "sqlite":
path := os.Getenv("PHOTOVIEW_SQLITE_PATH")
if path == "" {
path = "photoview.db"
sqliteAddress, err := getSqliteAddress()
if err != nil {
return nil, err
}
databaseDialect = sqlite.Open(path)
databaseDialect = sqlite.Open(sqliteAddress.String())
}
db, err := gorm.Open(databaseDialect, &config)