MaysWind
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/mayswind/lab/pkg/errs"
|
|
)
|
|
|
|
const requestIdFieldKey = "REQUEST_ID"
|
|
const tokenClaimsFieldKey = "TOKEN_CLAIMS"
|
|
const responseErrorFieldKey = "RESPONSE_ERROR"
|
|
|
|
const ClientTimezoneOffsetHeaderName = "X-Timezone-Offset"
|
|
|
|
// Context represents the request and response context
|
|
type Context struct {
|
|
*gin.Context
|
|
// DO NOT ADD ANY FIELD IN THIS CONTEXT, THIS CONTEXT IS JUST A WRAPPER
|
|
}
|
|
|
|
// SetRequestId sets the given request id to context
|
|
func (c *Context) SetRequestId(requestId string) {
|
|
c.Set(requestIdFieldKey, requestId)
|
|
}
|
|
|
|
// GetRequestId returns the current request id
|
|
func (c *Context) GetRequestId() string {
|
|
requestId, exists := c.Get(requestIdFieldKey)
|
|
|
|
if !exists {
|
|
return ""
|
|
}
|
|
|
|
return requestId.(string)
|
|
}
|
|
|
|
// SetTokenClaims sets the given user token id to context
|
|
func (c *Context) SetTokenClaims(claims *UserTokenClaims) {
|
|
c.Set(tokenClaimsFieldKey, claims)
|
|
}
|
|
|
|
// GetTokenClaims returns the current user token
|
|
func (c *Context) GetTokenClaims() *UserTokenClaims {
|
|
claims, exists := c.Get(tokenClaimsFieldKey)
|
|
|
|
if !exists {
|
|
return nil
|
|
}
|
|
|
|
return claims.(*UserTokenClaims)
|
|
}
|
|
|
|
// GetCurrentUid returns the current user uid by the current user token
|
|
func (c *Context) GetCurrentUid() int64 {
|
|
claims := c.GetTokenClaims()
|
|
|
|
if claims == nil {
|
|
return 0
|
|
}
|
|
|
|
uid, err := strconv.ParseInt(claims.Id, 10, 64)
|
|
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
|
|
return uid
|
|
}
|
|
|
|
// GetClientTimezoneOffset returns the client timezone offset
|
|
func (c *Context) GetClientTimezoneOffset() (int16, error) {
|
|
value := c.GetHeader(ClientTimezoneOffsetHeaderName)
|
|
offset, err := strconv.Atoi(value)
|
|
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return int16(offset), nil
|
|
}
|
|
|
|
// SetResponseError sets the response error
|
|
func (c *Context) SetResponseError(error *errs.Error) {
|
|
c.Set(responseErrorFieldKey, error)
|
|
}
|
|
|
|
// GetResponseError returns the response error
|
|
func (c *Context) GetResponseError() *errs.Error {
|
|
err, exists := c.Get(responseErrorFieldKey)
|
|
|
|
if !exists {
|
|
return nil
|
|
}
|
|
|
|
return err.(*errs.Error)
|
|
}
|
|
|
|
// WrapContext returns a context wrapped by this file
|
|
func WrapContext(ginCtx *gin.Context) *Context {
|
|
return &Context{
|
|
Context: ginCtx,
|
|
}
|
|
}
|