Restructure project

internal/database/database.go

@@ -0,0 +1,81 @@
+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)
+}