Quick cleanup on a few bits of code, also added some places where a transaction rollback was missed.

This commit is contained in:
2021-09-09 15:15:05 -05:00
parent 913b7ca138
commit 5feb32a122
+3 -2
View File
@@ -2,7 +2,6 @@ package main
import (
"context"
"fmt"
)
type Category struct {
@@ -68,6 +67,7 @@ func insertNewCategory(name string, subcategories []Subcategory) (recordID int64
statement, err := transaction.Prepare("INSERT INTO category (category_name) VALUES ($1) RETURNING category_id;")
if err != nil {
transaction.Rollback()
println(err.Error())
return 0, false
}
@@ -93,6 +93,7 @@ func insertNewCategory(name string, subcategories []Subcategory) (recordID int64
statement, err := transaction.Prepare("INSERT INTO subcategory(subcategory_name, category_id) VALUES($1, $2);")
if err != nil {
transaction.Rollback()
println(err.Error())
return 0, false
}
@@ -100,7 +101,7 @@ func insertNewCategory(name string, subcategories []Subcategory) (recordID int64
defer statement.Close()
for _, subcategory := range subcategories {
println(fmt.Sprintf("For cat %d, add sub '%s'.", recordID, subcategory.Name))
_, err := statement.Exec(subcategory.Name, recordID)
if err != nil {