Quick cleanup on a few bits of code, also added some places where a transaction rollback was missed.
This commit is contained in:
+3
-2
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user