83 lines
1.5 KiB
Go
83 lines
1.5 KiB
Go
package context
|
||
|
||
import (
|
||
"admin/internal/errcode"
|
||
"admin/lib/web"
|
||
"admin/lib/xlog"
|
||
"context"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type WebHeader struct {
|
||
UserId int `json:"UserId"` // 用户id，会与下面token解析出用户id做匹配校验
|
||
Token string `json:"Token"` // jwt token，内置一些信息
|
||
Ip string `json:"IP"`
|
||
}
|
||
|
||
type WebContext struct {
|
||
context.Context
|
||
rawCtx *gin.Context
|
||
Header *WebHeader
|
||
alreadySetRsp bool
|
||
}
|
||
|
||
func NewWebContext(rawCtx *gin.Context) web.IContext {
|
||
return &WebContext{rawCtx: rawCtx}
|
||
}
|
||
|
||
func (ctx *WebContext) ExtractHeader() error {
|
||
header := &WebHeader{}
|
||
err := ctx.rawCtx.ShouldBindHeader(header)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
ctx.Header = header
|
||
return nil
|
||
}
|
||
|
||
func (ctx *WebContext) IsUserLogin() bool {
|
||
return ctx.Header != nil && ctx.Header.UserId > 0
|
||
}
|
||
|
||
func (ctx *WebContext) GinCtx() *gin.Context {
|
||
return ctx.rawCtx
|
||
}
|
||
|
||
func (ctx *WebContext) Ok(data any) {
|
||
if ctx.alreadySetRsp {
|
||
return
|
||
}
|
||
|
||
ctx.rawCtx.JSON(200, map[string]any{
|
||
"code": 200,
|
||
"msg": "",
|
||
"data": data,
|
||
})
|
||
|
||
ctx.alreadySetRsp = true
|
||
}
|
||
|
||
func (ctx *WebContext) Fail(err error) {
|
||
if ctx.alreadySetRsp {
|
||
return
|
||
}
|
||
|
||
code, stack, msg := errcode.ParseError(err)
|
||
ctx.rawCtx.JSON(200, map[string]any{
|
||
"code": code,
|
||
"stack": stack,
|
||
"msg": msg,
|
||
"data": "",
|
||
})
|
||
|
||
ctx.alreadySetRsp = true
|
||
}
|
||
|
||
func (ctx *WebContext) HandleError(path string, err error) {
|
||
xlog.Errorf("path:%v handle error:%v ", path, err)
|
||
ctx.Fail(err)
|
||
}
|
||
func (ctx *WebContext) HandleSuccess(rspData any) {
|
||
ctx.Ok(rspData)
|
||
}
|