81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"moul.io/zapgorm2"
|
|
)
|
|
|
|
type Database struct {
|
|
Host string
|
|
User string
|
|
Password string
|
|
Database string
|
|
}
|
|
|
|
const Storage = "storage/"
|
|
|
|
func init() {
|
|
os.Mkdir(Storage, os.ModePerm)
|
|
}
|
|
|
|
func (d *Database) tryDbConnection() {
|
|
i := 1
|
|
total := 20
|
|
for i <= total {
|
|
ln, err := net.DialTimeout("tcp", d.Host, 1*time.Second)
|
|
if err != nil {
|
|
if i == total {
|
|
zap.L().Fatal("Failed connecting to database", zap.Int("attempt", i))
|
|
}
|
|
zap.L().Warn("Connecting to database", zap.Int("attempt", i))
|
|
time.Sleep(2 * time.Second)
|
|
i++
|
|
} else {
|
|
_ = ln.Close()
|
|
zap.L().Info("Connected to database", zap.Int("attempt", i))
|
|
i = total + 1
|
|
}
|
|
}
|
|
}
|
|
|
|
func (d *Database) initializeMySql(conf *gorm.Config) *gorm.DB {
|
|
var err error
|
|
d.tryDbConnection()
|
|
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
d.User,
|
|
d.Password,
|
|
d.Host,
|
|
d.Database,
|
|
)
|
|
orm, err := gorm.Open(mysql.Open(dsn), conf)
|
|
if err != nil {
|
|
zap.L().Error(err.Error())
|
|
}
|
|
return orm
|
|
}
|
|
|
|
func (d *Database) initializeSqLite(conf *gorm.Config) *gorm.DB {
|
|
var err error
|
|
orm, err := gorm.Open(sqlite.Open(Storage+"db.sqlite?_pragma=foreign_keys(1)"), conf)
|
|
if err != nil {
|
|
zap.L().Error(err.Error())
|
|
}
|
|
return orm
|
|
}
|
|
|
|
func NewDatabaseConnection(d *Database) *gorm.DB {
|
|
logger := zapgorm2.New(zap.L())
|
|
conf := &gorm.Config{Logger: logger, PrepareStmt: true}
|
|
if d.Host == "" {
|
|
return d.initializeSqLite(conf)
|
|
}
|
|
return d.initializeMySql(conf)
|
|
}
|