代码拉取完成,页面将自动刷新
package api
import (
"gitee.com/h79/goutils/auth/token"
"gitee.com/h79/goutils/common/logger"
"gitee.com/h79/goutils/common/result"
"gitee.com/h79/goutils/common/stringutil"
"strings"
)
const (
KHead = "HEADER"
KToken = "TOKEN"
KUid = "UID"
)
type HeadOption func(ctx Context, header *Header)
func BindHead(ctx Context, option HeadOption) {
header := NewHeader(ReqHead{})
header.ReadFrom(ctx)
if option != nil {
option(ctx, header)
}
header.ClientIP = ctx.HttpContext().ClientIP()
ctx.CacheContext().Set(KHead, header)
}
func WithHeadOption(ctx Context, header *Header) {
header.Version = ctx.HttpContext().Query(CVersion)
header.Source = ctx.HttpContext().Query(CSource)
header.SeqId = ctx.HttpContext().Query(CSeqId)
header.ReqAt = stringutil.StringToInt64(ctx.HttpContext().Query(CTimeAt))
header.App.AppId = ctx.HttpContext().Query(CAppId)
header.App.Terminal = ctx.HttpContext().Query(CTerminal)
header.App.Version = header.Version
header.App.Source = header.Source
}
func BindClientAuth(ctx Context, authMethod, secretKey string) error {
authenticate, err := ctx.AuthContext().Client().Auth(authMethod)
if err != nil {
return err
}
flag := authenticate.EnableFlag(secretKey)
if flag == 2 {
return result.RErrNotFound
}
if flag == 0 {
return result.RErrIgnore
}
logger.Debug("Api: client authentication start")
if err := authenticateCheck(ctx, secretKey, authenticate); err != nil {
logger.Error("Api: client authentication failure, err= '%v'", err)
} else {
logger.Debug("Api: client authentication completed")
}
return nil
}
func BindServerAuth(ctx Context, authMethod, secretKey string) error {
authenticate, err := ctx.AuthContext().Server().Auth(authMethod)
if err != nil {
return err
}
flag := authenticate.EnableFlag(secretKey)
if flag == 2 {
return result.RErrNotFound
}
if flag == 0 {
return result.RErrIgnore
}
logger.Debug("Api: server authentication start")
if err := authenticateCheck(ctx, secretKey, authenticate); err != nil {
logger.Error("Api: server authentication failure, err= '%v'", err)
} else {
logger.Debug("Api: server authentication completed")
}
return nil
}
func GetTokenSource(ctx Context) token.Source {
base := GetBase(ctx)
return token.Source{AppId: base.App.AppId, Source: base.App.Source}
}
func authenticateCheck(ctx Context, secretKey string, authenticate token.Authenticate) error {
tok, err := tokenReq(ctx)
if err != nil {
authError(ctx, err)
return err
}
tk, err := authenticate.Check(secretKey, tok)
if err != nil {
authError(ctx, err)
return err
}
ctx.CacheContext().Set(KUid, tk.GetVar("uid"))
ctx.CacheContext().Set(KToken, tk)
return nil
}
func authError(ctx Context, err error) {
SendStatusNoHead(ctx, DefErrorHttpCode, result.Error(result.ErrAuth, err.Error()), nil)
ctx.HttpContext().Abort()
}
const kToken = "token"
func tokenReq(ctx Context) (string, error) {
var header = ctx.HeaderContext()
tok := header.Get(CToken)
if len(tok) > 6 {
if strings.ToUpper(tok[0:7]) == "BEARER " {
return tok[7:], nil
}
return tok, nil
}
tok = header.Get(kToken)
if len(tok) > 0 {
return tok, nil
}
tok = ctx.HttpContext().Query(kToken)
if len(tok) > 0 {
return tok, nil
}
return "", result.RErrAuth
}
func Bearer(tok string) string {
if len(tok) <= 0 {
return ""
}
if strings.ToUpper(tok[0:7]) == "BEARER " {
return tok
}
return "Bearer " + tok
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。