diff --git a/api/account.go b/api/account.go index df48396..55764fd 100644 --- a/api/account.go +++ b/api/account.go @@ -3,6 +3,8 @@ package api import ( "database/sql" "errors" + "github.com/lib/pq" + "log" "net/http" db "github.com/cukhoaimon/SimpleBank/db/sqlc" @@ -30,6 +32,11 @@ func (server *Server) createAccount(ctx *gin.Context) { account, err := server.store.CreateAccount(ctx, arg) if err != nil { + var pqErr *pq.Error + if errors.As(err, &pqErr) { + log.Println(pqErr.Code.Name()) + } + ctx.JSON(http.StatusInternalServerError, errorResponse(err)) return } diff --git a/db/migration/000002_add_users.up.sql b/db/migration/000002_add_users.up.sql index 52c2b86..d6a2516 100644 --- a/db/migration/000002_add_users.up.sql +++ b/db/migration/000002_add_users.up.sql @@ -8,4 +8,6 @@ CREATE TABLE "users" ( ); --CREATE INDEX ON "accounts" ("owner", "currency"); -ALTER TABLE "accounts" ADD CONSTRAINT "owner_currency_key" UNIQUE ("owner", "currency"); \ No newline at end of file +ALTER TABLE "accounts" ADD CONSTRAINT "owner_currency_key" UNIQUE ("owner", "currency"); + +ALTER TABLE "accounts" ADD FOREIGN KEY ("owner") REFERENCES "users" ("username"); \ No newline at end of file diff --git a/db/sqlc/account.sql_test.go b/db/sqlc/account.sql_test.go index 225ed2b..f51b954 100644 --- a/db/sqlc/account.sql_test.go +++ b/db/sqlc/account.sql_test.go @@ -11,8 +11,10 @@ import ( ) func createRandomAccount(t *testing.T) Account { + user := createRandomUser(t) + arg := CreateAccountParams{ - Owner: utils.RandomOwner(), + Owner: user.Username, Balance: utils.RandomMoney(), Currency: utils.RandomCurrency(), }