uniugm/admin/internal/context/ctx_web.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)
}