go/testing-go-code-with-postgres
1
0
Fork 0
mirror of https://github.com/xorcare/testing-go-code-with-postgres.git synced 2025-06-30 23:23:40 +02:00
Code Issues Releases Activity

Add interface do DB driver to repository

Browse Source 
This abstraction is needed to add transactional cleanup in the future.
This commit is contained in:
Vasiliy Vasilyuk
2024-02-17 16:39:34 +03:00
parent 912e9c9ab0
commit 83b2f61e41
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
user_repository.go
View File

@ -12,12 +12,17 @@ import (
"github.com/google/uuid" "github.com/google/uuid"
) )
func NewUserRepository(db *sql.DB) *UserRepository { type DB interface {
QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
}
func NewUserRepository(db DB) *UserRepository {
return &UserRepository{db: db} return &UserRepository{db: db}
} }
type UserRepository struct { type UserRepository struct {
db *sql.DB db DB
} }
func (r *UserRepository) ReadUser(ctx context.Context, userID uuid.UUID) (User, error) { func (r *UserRepository) ReadUser(ctx context.Context, userID uuid.UUID) (User, error) {