101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package pgdb
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
|
|
dberrors "git-molva.ru/Molva/molva-backend/services/api_gateway/internal/database/errors"
|
|
dbtypes "git-molva.ru/Molva/molva-backend/services/api_gateway/internal/database/types"
|
|
"github.com/Masterminds/squirrel"
|
|
)
|
|
|
|
func (c *client) GetBalance(
|
|
ctx context.Context,
|
|
request *dbtypes.BalanceGetRequest,
|
|
) (*dbtypes.BalanceGetResponse, error) {
|
|
if request == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
var (
|
|
psql = squirrel.StatementBuilder.PlaceholderFormat(squirrel.Dollar)
|
|
|
|
balancesTable = fmt.Sprintf("%s.%s", c.config.Schema, BalancesTableName)
|
|
)
|
|
|
|
getBalance := psql.Select(
|
|
"raw_balance", "clean_balance",
|
|
).From(balancesTable).
|
|
Where(squirrel.Eq{"owner_id": request.OwnerId})
|
|
|
|
query, args, err := getBalance.ToSql()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%w: error building get balance query: %v", dberrors.ErrInternal, err)
|
|
}
|
|
|
|
row := c.db.QueryRowContext(ctx, query, args...)
|
|
|
|
var (
|
|
balance dbtypes.Balance
|
|
)
|
|
|
|
if err := row.Scan(&balance.RawBalance, &balance.CleanBalance); err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, dberrors.ErrNotFound
|
|
}
|
|
|
|
return nil, fmt.Errorf("%w: error scanning row: %v", dberrors.ErrInternal, err)
|
|
}
|
|
|
|
return &dbtypes.BalanceGetResponse{
|
|
Balance: &balance,
|
|
}, nil
|
|
}
|
|
|
|
func (c *client) createBalance(
|
|
ctx context.Context,
|
|
driver Driver,
|
|
request *dbtypes.BalanceCreateRequest,
|
|
) error {
|
|
if request == nil {
|
|
return fmt.Errorf("%w: request is nil", dberrors.ErrBadRequest)
|
|
}
|
|
|
|
var (
|
|
psql = squirrel.StatementBuilder.PlaceholderFormat(squirrel.Dollar)
|
|
|
|
balancesTable = fmt.Sprintf("%s.%s", c.config.Schema, BalancesTableName)
|
|
)
|
|
|
|
createBalance := psql.Insert(balancesTable).
|
|
Columns(
|
|
"id", "owner_id", "raw_balance", "clean_balance",
|
|
).
|
|
Values(
|
|
request.Id, request.OwnerId, request.RawBalance, request.CleanBalance,
|
|
)
|
|
|
|
query, args, err := createBalance.ToSql()
|
|
if err != nil {
|
|
return fmt.Errorf("%w: error building 'create balance' query: %v", dberrors.ErrInternal, err)
|
|
}
|
|
|
|
res, err := driver.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
return fmt.Errorf("%w: error executing 'create balance' query: %v", dberrors.ErrInternal, err)
|
|
}
|
|
|
|
rowsAffected, err := res.RowsAffected()
|
|
if err != nil {
|
|
return fmt.Errorf("%w: error getting rows affected for 'create balance' query: %v", dberrors.ErrInternal, err)
|
|
}
|
|
|
|
if rowsAffected == 0 {
|
|
return dberrors.ErrInternal
|
|
}
|
|
|
|
return nil
|
|
}
|