diff --git a/dao/README.md b/dao/README.md index a9a43cf..f3f7500 100644 --- a/dao/README.md +++ b/dao/README.md @@ -10,6 +10,7 @@ - Метод `Delete` удаляет пользователя из таблицы. - Метод `Lookup` возвращает `Name` пользователя по `ID`. - Метод `List` возвращает список всех пользователей в таблице. +- Метод `Close` закрывает коннекты к базе данных. ## Запуск тестов на linux diff --git a/dao/dao_test.go b/dao/dao_test.go index deae678..a891b22 100644 --- a/dao/dao_test.go +++ b/dao/dao_test.go @@ -17,6 +17,7 @@ func TestDao(t *testing.T) { dao, err := CreateDao(ctx, dsn) require.NoError(t, err) + defer func() { _ = dao.Close() }() _, err = dao.Lookup(ctx, 42) require.ErrorIs(t, err, sql.ErrNoRows) diff --git a/dao/model.go b/dao/model.go index d2cebbe..5cdaf59 100644 --- a/dao/model.go +++ b/dao/model.go @@ -18,5 +18,6 @@ type ( Delete(ctx context.Context, id UserID) error Lookup(ctx context.Context, id UserID) (User, error) List(ctx context.Context) ([]User, error) + Close() error } ) diff --git a/ledger/README.md b/ledger/README.md index bf402b8..873753b 100644 --- a/ledger/README.md +++ b/ledger/README.md @@ -10,6 +10,7 @@ Если на счету недостаточно денег, метод должен возвращать ошибку `ledger.ErrNoMoney`. - Метод `Transfer` должен переводить деньги со счёта `from` на счёт `to`. Если на счету `from` недостаточно денег, метод должен возвращать ошибку `ledger.ErrNoMoney`. +- Метод `Close` закрывает коннекты к базе данных. Все операции должны быть атомарными. Для реализации некоторых методов вам потребуется использовать транзакции и row-level локи. Ваша реализация не должна создавать дедлоки на уровне базы данных. diff --git a/ledger/ledger_test.go b/ledger/ledger_test.go index cfc909f..81d7491 100644 --- a/ledger/ledger_test.go +++ b/ledger/ledger_test.go @@ -20,6 +20,7 @@ func TestLedger(t *testing.T) { l0, err := ledger.New(ctx, dsn) require.NoError(t, err) + defer func() { _ = l0.Close() }() t.Run("SimpleCommands", func(t *testing.T) { checkBalance := func(account ledger.ID, amount ledger.Money) { diff --git a/ledger/model.go b/ledger/model.go index 3bafcc0..727ed62 100644 --- a/ledger/model.go +++ b/ledger/model.go @@ -18,4 +18,5 @@ type Ledger interface { Deposit(ctx context.Context, id ID, amount Money) error Withdraw(ctx context.Context, id ID, amount Money) error Transfer(ctx context.Context, from, to ID, amount Money) error + Close() error }