Handlers, DB, Cache

internal/service/database/db_service.go

@@ -1,12 +1,17 @@
 package database
 
 import (
+	"context"
 	"errors"
+	"log/slog"
 
 	slogGorm "github.com/orandin/slog-gorm"
 	"gorm.io/driver/postgres"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
+
+	"payouts/internal/service/database/config"
+	"payouts/internal/service/database/orm"
 )
 
 type dbService struct {
@@ -14,31 +19,67 @@ type dbService struct {
 	db     *gorm.DB
 }
 
-func NewDatabaseService(dbType, connection, logLevel string) (DatabaseService, error) {
+func NewDatabaseService(conf config.Database) (Service, error) {
 
 	var dialector gorm.Dialector
-	switch dbType {
+	switch conf.Type {
 	case "sqlite":
-		dialector = sqlite.Open(connection)
+		dialector = sqlite.Open(conf.Connection)
 	case "postgres":
-		dialector = postgres.Open(connection)
+		dialector = postgres.Open(conf.Connection)
 	default:
 		return nil, errors.New("unknown dbType")
 	}
 
+	level := slog.LevelInfo
+	level.UnmarshalText([]byte(conf.LogLevel))
+
+	slogGormOpts := []slogGorm.Option{
+		slogGorm.SetLogLevel(slogGorm.DefaultLogType, level),
+	}
+	if conf.TraceRequests {
+		slogGormOpts = append(slogGormOpts, slogGorm.WithTraceAll())
+	}
+
+	slogGorm := slogGorm.New(slogGormOpts...)
+
 	db, err := gorm.Open(dialector, &gorm.Config{
-		Logger: slogGorm.New(),
+		Logger: slogGorm,
 	})
 
 	if err == nil {
 		db.DB()
-		db.AutoMigrate()
-		// db.LogMode(true)
+		db.AutoMigrate(&orm.User{})
 	}
 	result := &dbService{}
-	result.dbType = dbType
+	result.dbType = conf.Type
 	result.db = db
 
 	return result, err
 
 }
+
+func getParams(options ...Optional) *params {
+	params := &params{
+		ctx: context.Background(),
+	}
+	for _, opt := range options {
+		opt(params)
+	}
+	return params
+}
+
+// AddUser implements [Service].
+func (d *dbService) CreateUser(userModel orm.User, opts ...Optional) error {
+	p := getParams(opts...)
+
+	return gorm.G[orm.User](d.db).Create(p.ctx, &userModel)
+}
+
+// GetUser implements [Service].
+func (d *dbService) GetUser(userModel orm.User, opts ...Optional) (orm.User, error) {
+	p := getParams(opts...)
+
+	userResp, err := gorm.G[orm.User](d.db).Where(&userModel).First(p.ctx)
+	return userResp, err
+}