56 lines
917 B
Go
56 lines
917 B
Go
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 {
|
|
Feedback *Feedback
|
|
logger zerolog.Logger
|
|
}
|
|
|
|
func NewDB(
|
|
db *gorm.DB,
|
|
logger zerolog.Logger,
|
|
) *DB {
|
|
return &DB{
|
|
Feedback: NewFeedback(db, logger),
|
|
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{},
|
|
&models.Upvote{},
|
|
)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|