2020-01-30 14:49:39 +01:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
2020-01-31 15:22:58 +01:00
|
|
|
"net/url"
|
2020-01-30 14:49:39 +01:00
|
|
|
"os"
|
|
|
|
|
2020-04-15 16:18:51 +02:00
|
|
|
"github.com/pkg/errors"
|
2020-11-23 19:39:44 +01:00
|
|
|
"github.com/viktorstrate/photoview/api/graphql/models"
|
2020-04-15 16:18:51 +02:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
"gorm.io/driver/mysql"
|
|
|
|
"gorm.io/gorm"
|
2020-01-30 14:49:39 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
// SetupDatabase connects to the database using environment variables
|
2020-11-23 19:39:44 +01:00
|
|
|
func SetupDatabase() (*gorm.DB, error) {
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2020-01-31 15:22:58 +01:00
|
|
|
address, err := url.Parse(os.Getenv("MYSQL_URL"))
|
|
|
|
if err != nil {
|
2020-04-15 20:52:08 +02:00
|
|
|
return nil, errors.Wrap(err, "Could not parse mysql url")
|
2020-01-31 15:22:58 +01:00
|
|
|
}
|
|
|
|
|
2020-02-19 22:04:38 +01:00
|
|
|
if address.String() == "" {
|
2020-04-15 16:18:51 +02:00
|
|
|
return nil, errors.New("Environment variable MYSQL_URL missing, exiting")
|
2020-02-19 22:04:38 +01:00
|
|
|
}
|
|
|
|
|
2020-01-31 15:22:58 +01:00
|
|
|
queryValues := address.Query()
|
|
|
|
queryValues.Add("multiStatements", "true")
|
2020-02-11 14:32:35 +01:00
|
|
|
queryValues.Add("parseTime", "true")
|
2020-01-31 15:22:58 +01:00
|
|
|
|
|
|
|
address.RawQuery = queryValues.Encode()
|
|
|
|
|
2020-01-30 14:49:39 +01:00
|
|
|
log.Printf("Connecting to database: %s", address)
|
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
db, err := gorm.Open(mysql.Open(address.String()), &gorm.Config{})
|
2020-04-30 15:00:18 +02:00
|
|
|
if err != nil {
|
2020-11-23 19:39:44 +01:00
|
|
|
return nil, errors.Wrap(err, "could not connect to database")
|
2020-04-30 15:00:18 +02:00
|
|
|
}
|
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
// var db *sql.DB
|
2020-02-19 21:33:28 +01:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
// db, err = sql.Open("mysql", address.String())
|
|
|
|
// if err != nil {
|
|
|
|
// return nil, errors.New("Could not connect to database, exiting")
|
|
|
|
// }
|
2020-04-30 15:00:18 +02:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
// tryCount := 0
|
2020-02-19 21:33:28 +01:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
// for {
|
|
|
|
// ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
|
|
// defer cancel()
|
|
|
|
|
|
|
|
// if err := db.PingContext(ctx); err != nil {
|
|
|
|
// if tryCount < 4 {
|
|
|
|
// tryCount++
|
|
|
|
// log.Printf("WARN: Could not ping database: %s, Will retry after 1 second", err)
|
|
|
|
// time.Sleep(time.Second)
|
|
|
|
// continue
|
|
|
|
// } else {
|
|
|
|
// return nil, errors.Wrap(err, "Could not ping database, exiting")
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
|
|
|
|
// break
|
|
|
|
// }
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
// db.SetMaxOpenConns(80)
|
2020-02-14 14:29:41 +01:00
|
|
|
|
2020-04-15 16:18:51 +02:00
|
|
|
return db, nil
|
2020-01-30 14:49:39 +01:00
|
|
|
}
|
2020-01-31 15:22:58 +01:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
func MigrateDatabase(db *gorm.DB) error {
|
2020-01-31 15:22:58 +01:00
|
|
|
|
2020-11-23 19:39:44 +01:00
|
|
|
db.AutoMigrate(&models.User{}, &models.AccessToken{})
|
2020-01-31 15:22:58 +01:00
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|