Handlers, DB, Cache

internal/service/database/config/database.go

@@ -1,7 +1,8 @@
 package config
 
 type Database struct {
-	Type       string
-	Connection string
-	LogLevel   string
+	Type          string
+	Connection    string
+	LogLevel      string
+	TraceRequests bool
 }

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
+}

internal/service/database/module.go

@@ -1,16 +1,32 @@
 package database
 
 import (
-	"payouts/internal/config"
+	"context"
 
 	"go.uber.org/fx"
+
+	"payouts/internal/config"
+	"payouts/internal/service/database/orm"
 )
 
 var Module = fx.Options(
 	fx.Provide(New),
 )
 
-type DatabaseService interface {
+type params struct {
+	ctx context.Context
+}
+type Optional func(*params)
+
+func WithContext(ctx context.Context) Optional {
+	return func(p *params) {
+		p.ctx = ctx
+	}
+}
+
+type Service interface {
+	CreateUser(user orm.User, opts ...Optional) error
+	GetUser(user orm.User, opts ...Optional) (orm.User, error)
 }
 
 // Params represents the module input params
@@ -21,6 +37,6 @@ type Params struct {
 }
 
 // NewPersistence instantiates the persistence module
-func New(p Params) (DatabaseService, error) {
-	return NewDatabaseService(p.AppConfig.Database.Type, p.AppConfig.Database.Connection, p.AppConfig.Database.LogLevel)
+func New(p Params) (Service, error) {
+	return NewDatabaseService(p.AppConfig.Database)
 }

internal/service/database/orm/user.go

@@ -0,0 +1,10 @@
+package orm
+
+import "gorm.io/gorm"
+
+type User struct {
+	gorm.Model
+	TIN        string
+	Phone      string `gorm:"uniqueIndex:idx_phone"`
+	PasswdHash string
+}