Implement session module

Signed-off-by: Sean Du <do4suki@gmail.com>
This commit is contained in:
2025-08-15 19:24:36 +08:00
parent 985895a780
commit 80a686849f
29 changed files with 837 additions and 694 deletions
+65
View File
@@ -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))
}