@@ -0,0 +1,65 @@
|
||||
package session
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"honoka-chan/internal/utils"
|
||||
"honoka-chan/pkg/db"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"xorm.io/xorm"
|
||||
)
|
||||
|
||||
type Session struct {
|
||||
Ctx *gin.Context
|
||||
MainEng *xorm.Session
|
||||
UserEng *xorm.Session
|
||||
}
|
||||
|
||||
func New(ctx *gin.Context) *Session {
|
||||
ss := &Session{
|
||||
Ctx: ctx,
|
||||
}
|
||||
|
||||
ss.MainEng = db.MainEng.NewSession()
|
||||
|
||||
ss.UserEng = db.UserEng.NewSession()
|
||||
ss.UserEng.Begin()
|
||||
|
||||
return ss
|
||||
}
|
||||
|
||||
func (ss *Session) Finalize() {
|
||||
ss.MainEng.Close()
|
||||
|
||||
ss.UserEng.Commit()
|
||||
ss.UserEng.Close()
|
||||
}
|
||||
|
||||
func (ss *Session) Abort(err error) {
|
||||
ss.MainEng.Close()
|
||||
|
||||
ss.UserEng.Rollback()
|
||||
ss.UserEng.Close()
|
||||
|
||||
ss.Ctx.JSON(500, gin.H{"error": err.Error()})
|
||||
ss.Ctx.Abort()
|
||||
}
|
||||
|
||||
func (ss *Session) CheckErr(err error) bool {
|
||||
if err != nil {
|
||||
ss.Abort(err)
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (ss *Session) Respond(resp any) {
|
||||
data, err := json.Marshal(resp)
|
||||
if err != nil {
|
||||
ss.Abort(err)
|
||||
return
|
||||
}
|
||||
|
||||
ss.Ctx.Header("X-Message-Sign", utils.GenXMS(data))
|
||||
ss.Ctx.String(200, string(data))
|
||||
}
|
||||
Reference in New Issue
Block a user