fusenapi/goctl_template/api/context.tpl

package svc

import (
	{{.configImport}}
	"fusenapi/initalize"
	"gorm.io/gorm"
)

type ServiceContext struct {
	Config {{.config}}
	{{.middleware}}
	MysqlConn *gorm.DB
}

func NewServiceContext(c {{.config}}) *ServiceContext {

	return &ServiceContext{
		Config: c,
		MysqlConn: initalize.InitMysql(c.SourceMysql),
		{{.middlewareAssignment}}
	}
}


func (svcCxt *ServiceContext) ParseJwtToken(r *http.Request) (jwt.MapClaims, error) {
	AuthKey := r.Header.Get("Authorization")
	if len(AuthKey) <= 50 {
		return nil, errors.New(fmt.Sprint("Error parsing token, len:", len(AuthKey)))
	}

	token, err := jwt.Parse(AuthKey, func(token *jwt.Token) (interface{}, error) {
		// 检查签名方法是否为 HS256
		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
			return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
		}
		// 返回用于验证签名的密钥
		return svcCxt.Config.Auth.AccessSecret, nil
	})
	if err != nil {
		return nil, errors.New(fmt.Sprint("Error parsing token:", err))
	}

	// 验证成功返回
	if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
		return claims, nil
	}

	return nil, errors.New(fmt.Sprint("Invalid token", err))
}