Rollback live sessions on panic

Signed-off-by: Sean Du <do4suki@gmail.com>

internal/handler/live/gameover.go

@@ -12,12 +12,11 @@ import (
 
 func gameOver(ctx *gin.Context) {
 	ss := session.Get(ctx)
-	defer func() {
+	defer ss.FinalizeOrRollbackAfter(func() {
 		if ss.UserEng != nil {
 			ss.ClearLiveInProgress()
 		}
-		ss.Finalize()
-	}()
+	})
 
 	ss.Respond(BuildGameOverResp())
 }

internal/handler/live/reward.go

@@ -17,12 +17,11 @@ import (
 
 func reward(ctx *gin.Context) {
 	ss := session.Get(ctx)
-	defer func() {
+	defer ss.FinalizeOrRollbackAfter(func() {
 		if ss.UserEng != nil {
 			ss.ClearLiveInProgress()
 		}
-		ss.Finalize()
-	}()
+	})
 
 	playRewardReq := liveschema.RewardReq{}
 	err := honokautils.ParseRequestData(ctx, &playRewardReq)

internal/handler/rlive/gameover.go

@@ -14,13 +14,12 @@ import (
 func gameOver(ctx *gin.Context) {
 	ss := session.Get(ctx)
 	token := ""
-	defer func() {
+	defer ss.FinalizeOrRollbackAfter(func() {
 		if ss.UserEng != nil {
 			deleteRandomLiveByToken(ss, token)
 			ss.ClearLiveInProgress()
 		}
-		ss.Finalize()
-	}()
+	})
 
 	req := rliveschema.TokenReq{}
 	err := honokautils.ParseRequestData(ctx, &req)

internal/handler/rlive/reward.go

@@ -15,13 +15,12 @@ import (
 func reward(ctx *gin.Context) {
 	ss := session.Get(ctx)
 	token := ""
-	defer func() {
+	defer ss.FinalizeOrRollbackAfter(func() {
 		if ss.UserEng != nil {
 			deleteRandomLiveByToken(ss, token)
 			ss.ClearLiveInProgress()
 		}
-		ss.Finalize()
-	}()
+	})
 
 	req := rliveschema.RewardReq{}
 	err := honokautils.ParseRequestData(ctx, &req)