init

repo/pg/client.go

@@ -0,0 +1,54 @@
+package pg
+
+import (
+	"gerald/internal/adapters/config"
+	"gerald/internal/models"
+	"github.com/rs/zerolog"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	gormlogger "gorm.io/gorm/logger"
+)
+
+type DB struct {
+	Ticket *Feedback
+	logger zerolog.Logger
+}
+
+func NewDB(
+	db *gorm.DB,
+	logger zerolog.Logger,
+) *DB {
+	return &DB{
+		Ticket: NewFeedback(db),
+		logger: logger,
+	}
+}
+
+func NewClient(cfg *config.Config, logger zerolog.Logger) *gorm.DB {
+	db, err := gorm.Open(postgres.Open(cfg.PostgresDSN), &gorm.Config{
+		Logger: gormlogger.Default.LogMode(gormlogger.Silent),
+	})
+	if err != nil {
+		logger.Fatal().Err(err).Msg("Cannot connect to database")
+	}
+
+	return db
+}
+
+func Close(db *gorm.DB) error {
+	dbConn, _ := db.DB()
+
+	return dbConn.Close()
+}
+
+func CreateSchema(db *gorm.DB) error {
+	err := db.AutoMigrate(
+		&models.Feedback{},
+	)
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}