fix: add foreign key for account table

cukhoaimon · Feb 1, 2024 · 964b54b · 964b54b
1 parent 9268412
commit 964b54b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git 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
@@ -8,4 +8,6 @@ CREATE TABLE "users" (
 );
 
 --CREATE INDEX ON "accounts" ("owner", "currency");
-ALTER TABLE "accounts" ADD CONSTRAINT "owner_currency_key" UNIQUE ("owner", "currency");
+ALTER TABLE "accounts" ADD CONSTRAINT "owner_currency_key" UNIQUE ("owner", "currency");
+
+ALTER TABLE "accounts" ADD FOREIGN KEY ("owner") REFERENCES "users" ("username");
diff --git 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(),
 	}