Abort requests when session transactions cannot start

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

internal/handler/ghome/account/active.go

@@ -10,6 +10,9 @@ import (
 
 func active(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	ss.Respond(ghomeschema.ActiveResp{

internal/handler/ghome/account/initialize.go

@@ -14,6 +14,9 @@ import (
 
 func initialize(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	initData := ghomeschema.InitializeData{

internal/handler/ghome/account/login.go

@@ -34,6 +34,9 @@ const (
 
 func login(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	data, err := ctx.GetRawData()

internal/handler/ghome/account/loginauto.go

@@ -16,6 +16,9 @@ import (
 
 func loginAuto(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	data, err := ctx.GetRawData()

internal/handler/ghome/account/reportrole.go

@@ -12,6 +12,9 @@ import (
 
 func reportRole(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	randKey, err := ss.Get3DESRandKey()

internal/handler/ghome/basic/getcode.go

@@ -11,6 +11,9 @@ import (
 
 func getCode(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	codeArray := `{"codeArray":[{"btntext":"好的","code":"-10264022","msg_from":2,"text":"","title":"短信验证码被阻止","type":1},{"btntext":"","code":"-10869623","msg_from":2,"text":"","title":"网络连接失败，无法一键登录","type":2},{"btntext":"","code":"10298300","msg_from":2,"text":"","title":"","type":3},{"btntext":"","code":"10298311","msg_from":2,"text":"","title":"","type":3},{"btntext":"","code":"10298312","msg_from":2,"text":"","title":"","type":3},{"btntext":"","code":"10298313","msg_from":2,"text":"","title":"","type":1},{"btntext":"","code":"10298321","msg_from":2,"text":"","title":"","type":3},{"btntext":"","code":"10298322","msg_from":2,"text":"","title":"","type":3}],"codeVersion":"1.0.5"}`

internal/handler/ghome/basic/getproductlist.go

@@ -10,6 +10,9 @@ import (
 
 func getProductList(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	getProductListData := ghomeschema.GetProductListData{

internal/handler/ghome/basic/handshake.go

@@ -18,6 +18,9 @@ import (
 
 func handshake(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	data, err := ctx.GetRawData()

internal/handler/ghome/basic/loginarea.go

@@ -10,6 +10,9 @@ import (
 
 func loginArea(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	ss.Respond(ghomeschema.LoginAreaResp{

internal/handler/ghome/basic/publickey.go

@@ -14,6 +14,9 @@ import (
 
 func publicKey(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	publicKeyCode := 0

internal/handler/ghome/guest/status.go

@@ -10,6 +10,9 @@ import (
 
 func status(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	ss.Respond(ghomeschema.GuestStatusResp{

internal/handler/ghome/misc/agreement.go

@@ -10,6 +10,9 @@ import (
 
 func agreement(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	ss.Respond(ghomeschema.AgreementResp{

internal/handler/ghome/misc/integration.go

@@ -10,6 +10,9 @@ import (
 
 func appReport(ctx *gin.Context) {
 	ss := session.Attach(ctx)
+	if ss.Done() {
+		return
+	}
 	defer ss.FinalizeOrRollback()
 
 	ss.Respond(ghomeschema.AppReportResp{

internal/session/session.go

@@ -51,7 +51,9 @@ func New(ctx *gin.Context) *Session {
 
 	ss.MainEng = db.MainEng.NewSession()
 	ss.UserEng = db.UserEng.NewSession()
-	ss.UserEng.Begin()
+	if err := ss.UserEng.Begin(); err != nil {
+		ss.Abort(err)
+	}
 
 	return ss
 }