From 5603d79b29086b680e215a2ff1140ded087f1e2c Mon Sep 17 00:00:00 2001 From: Andre Medeiros Date: Sat, 24 Jul 2021 16:17:54 -0400 Subject: [PATCH] add account repository --- .../repository/postgres/postgres_account.go | 145 ++++++++++++++++++ internal/domain/account.go | 6 +- internal/domain/device.go | 24 +++ internal/domain/errors.go | 10 ++ 4 files changed, 182 insertions(+), 3 deletions(-) create mode 100644 internal/account/repository/postgres/postgres_account.go create mode 100644 internal/domain/device.go create mode 100644 internal/domain/errors.go diff --git a/internal/account/repository/postgres/postgres_account.go b/internal/account/repository/postgres/postgres_account.go new file mode 100644 index 0000000..c6709c4 --- /dev/null +++ b/internal/account/repository/postgres/postgres_account.go @@ -0,0 +1,145 @@ +package postgres + +import ( + "context" + "fmt" + + "github.com/christianselig/apollo-backend/internal/domain" + "github.com/jackc/pgx/v4/pgxpool" +) + +type postgresAccountRepository struct { + pool *pgxpool.Pool +} + +func NewPostgresAccountRepository(pool *pgxpool.Pool) domain.AccountRepository { + return &postgresAccountRepository{pool: pool} +} + +func (p *postgresAccountRepository) fetch(ctx context.Context, query string, args ...interface{}) ([]domain.Account, error) { + rows, err := p.pool.Query(ctx, query, args...) + if err != nil { + return nil, err + } + defer rows.Close() + + var accs []domain.Account + for rows.Next() { + var acc domain.Account + if err := rows.Scan( + &acc.ID, + &acc.Username, + &acc.AccountID, + &acc.AccessToken, + &acc.RefreshToken, + &acc.ExpiresAt, + &acc.LastMessageID, + &acc.LastCheckedAt, + ); err != nil { + return nil, err + } + accs = append(accs, acc) + } + + return accs, nil +} + +func (p *postgresAccountRepository) GetByID(ctx context.Context, id int64) (domain.Account, error) { + query := ` + SELECT id, username, account_id, access_token, refresh_token, expires_at, last_message_id, last_checked_at + FROM accounts + WHERE id = $1` + + accs, err := p.fetch(ctx, query, id) + if err != nil { + return domain.Account{}, err + } + + if len(accs) == 0 { + return domain.Account{}, domain.ErrNotFound + } + + return accs[0], nil +} + +func (p *postgresAccountRepository) GetByRedditID(ctx context.Context, id string) (domain.Account, error) { + query := ` + SELECT id, username, account_id, access_token, refresh_token, expires_at, last_message_id, last_checked_at + FROM accounts + WHERE account_id = $1` + + accs, err := p.fetch(ctx, query, id) + if err != nil { + return domain.Account{}, err + } + + if len(accs) == 0 { + return domain.Account{}, domain.ErrNotFound + } + + return accs[0], nil +} + +func (p *postgresAccountRepository) Create(ctx context.Context, acc *domain.Account) error { + query := ` + INSERT INTO accounts + (username, account_id, access_token, refresh_token, expires_at, last_message_id, last_checked_at) + VALUES ($1, $2, $3, $4, $5, $6, $7) + RETURNING id` + + res, err := p.pool.Query( + ctx, + query, + acc.Username, + acc.AccountID, + acc.AccessToken, + acc.RefreshToken, + acc.ExpiresAt, + acc.LastMessageID, + acc.LastCheckedAt, + ) + if err != nil { + return err + } + + return res.Scan(&acc.ID) +} + +func (p *postgresAccountRepository) Update(ctx context.Context, acc *domain.Account) error { + query := ` + UPDATE accounts + SET username = $2, + account_id = $3, + access_token = $4, + refresh_token = $5, + expires_at = $6, + last_message_id = $7, + last_checked_at = $8 + WHERE id = $1` + + res, err := p.pool.Exec( + ctx, + query, + acc.AccountID, + acc.AccessToken, + acc.RefreshToken, + acc.ExpiresAt, + acc.LastMessageID, + acc.LastCheckedAt, + ) + + if res.RowsAffected() != 1 { + return fmt.Errorf("weird behaviour, total rows affected: %d", res.RowsAffected()) + } + return err +} + +func (p *postgresAccountRepository) Delete(ctx context.Context, id int64) error { + query := `DELETE FROM accounts WHERE id = $1` + res, err := p.pool.Exec(ctx, query, id) + + if res.RowsAffected() != 1 { + return fmt.Errorf("weird behaviour, total rows affected: %d", res.RowsAffected()) + } + return err +} diff --git a/internal/domain/account.go b/internal/domain/account.go index 82ab2a6..2c98c5c 100644 --- a/internal/domain/account.go +++ b/internal/domain/account.go @@ -23,8 +23,8 @@ type AccountRepository interface { GetByID(ctx context.Context, id int64) (Account, error) GetByRedditID(ctx context.Context, id string) (Account, error) - Update(ctx context.Context, ac *Account) error - Create(ctx context.Context, ac *Account) error + Update(ctx context.Context, acc *Account) error + Create(ctx context.Context, acc *Account) error Delete(ctx context.Context, id int64) error } @@ -32,6 +32,6 @@ type AccountRepository interface { type AccountUsecase interface { GetByID(ctx context.Context, id int64) (Account, error) GetByRedditID(ctx context.Context, id string) (Account, error) - CreateOrUpdate(ctx context.Context, ac *Account) error + CreateOrUpdate(ctx context.Context, acc *Account) error Delete(ctx context.Context, id int64) error } diff --git a/internal/domain/device.go b/internal/domain/device.go new file mode 100644 index 0000000..70f4cad --- /dev/null +++ b/internal/domain/device.go @@ -0,0 +1,24 @@ +package domain + +import "context" + +type Device struct { + ID int64 + APNSToken string + Sandbox bool + LastPingedAt int64 +} + +type DeviceRepository interface { + GetByAPNSToken(ctx context.Context, token string) (Device, error) + + Update(ctx context.Context, dev *Device) error + Create(ctx context.Context, dev *Device) error + Delete(ctx context.Context, token string) error +} + +type DeviceUsecase interface { + GetByAPNSToken(ctx context.Context, token string) (Device, error) + CreateOrUpdate(ctx context.Context, dev *Device) error + Delete(ctx context.Context, token string) error +} diff --git a/internal/domain/errors.go b/internal/domain/errors.go new file mode 100644 index 0000000..991600f --- /dev/null +++ b/internal/domain/errors.go @@ -0,0 +1,10 @@ +package domain + +import "errors" + +var ( + // ErrNotFound will be returned if the requested item is not found + ErrNotFound = errors.New("requested item was not found") + // ErrConflict will be returned if the item being persisted already exists + ErrConflict = errors.New("item already exists") +)