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 (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Category struct {
|
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;")
|
statement, err := transaction.Prepare("INSERT INTO category (category_name) VALUES ($1) RETURNING category_id;")
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
transaction.Rollback()
|
||||||
println(err.Error())
|
println(err.Error())
|
||||||
return 0, false
|
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);")
|
statement, err := transaction.Prepare("INSERT INTO subcategory(subcategory_name, category_id) VALUES($1, $2);")
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
transaction.Rollback()
|
||||||
println(err.Error())
|
println(err.Error())
|
||||||
return 0, false
|
return 0, false
|
||||||
}
|
}
|
||||||
@@ -100,7 +101,7 @@ func insertNewCategory(name string, subcategories []Subcategory) (recordID int64
|
|||||||
defer statement.Close()
|
defer statement.Close()
|
||||||
|
|
||||||
for _, subcategory := range subcategories {
|
for _, subcategory := range subcategories {
|
||||||
println(fmt.Sprintf("For cat %d, add sub '%s'.", recordID, subcategory.Name))
|
|
||||||
_, err := statement.Exec(subcategory.Name, recordID)
|
_, err := statement.Exec(subcategory.Name, recordID)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
Reference in New Issue
Block a user