fix: gorm hooks

ElaBosak233 · Jun 2, 2024 · 47561c3 · 47561c3
1 parent bb2c400
commit 47561c3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/internal/model/challenge.go b/internal/model/challenge.go
@@ -71,16 +71,16 @@ func (c *Challenge) BeforeUpdate(db *gorm.DB) (err error) {
 }
 
 func (c *Challenge) BeforeDelete(db *gorm.DB) (err error) {
+	var pods []Pod
+	db.Table("pods").Where("challenge_id = ?", c.ID).Find(&pods)
+	for _, pod := range pods {
+		db.Table("pods").Delete(&pod)
+	}
+
 	db.Table("flags").Where("challenge_id = ?", c.ID).Delete(&Flag{})
 	db.Table("ports").Where("challenge_id = ?", c.ID).Delete(&Port{})
 	db.Table("envs").Where("challenge_id = ?", c.ID).Delete(&Env{})
 	db.Table("submissions").Where("challenge_id = ?", c.ID).Delete(&Submission{})
 	db.Table("game_challenges").Where("challenge_id = ?", c.ID).Delete(&GameChallenge{})
-
-	var pods []Pod
-	db.Table("pods").Where("challenge_id = ?", c.ID).Find(&pods)
-	for _, pod := range pods {
-		db.Table("pods").Where("id = ?", pod.ID).Delete(&Pod{})
-	}
 	return nil
 }
diff --git a/internal/model/pod.go b/internal/model/pod.go
@@ -22,7 +22,7 @@ func (p *Pod) Simplify() {
 	}
 }
 
-func (p *Pod) BeforeDelete(db *gorm.DB) error {
+func (p *Pod) BeforeDelete(db *gorm.DB) (err error) {
 	db.Table("nats").Where("pod_id = ?", p.ID).Delete(&Nat{})
 	db.Table("flag_gens").Where("pod_id = ?", p.ID).Delete(&FlagGen{})
 	return nil