gfast/internal/app/common/service/token.go

/*
* @desc:token功能
* @company:云南奇讯科技有限公司
* @Author: yixiaohu
* @Date:   2022/3/8 15:54
 */

package service

import (
	"context"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/tiger1103/gfast-token/gftoken"
	"github.com/tiger1103/gfast/v3/internal/app/common/model"
)

type IGfToken interface {
	GenerateToken(ctx context.Context, key string, data interface{}) (keys string, err error)
	Middleware(group *ghttp.RouterGroup) error
	ParseToken(r *ghttp.Request) (*gftoken.CustomClaims, error)
}

type gfTokenImpl struct {
	*gftoken.GfToken
}

var gT = gfTokenImpl{
	GfToken: gftoken.NewGfToken(),
}

func GfToken(options *model.TokenOptions) IGfToken {
	if options.CacheModel == model.CacheModelRedis {
		gT.GfToken = gftoken.NewGfToken(
			gftoken.WithCacheKey(options.CacheKey),
			gftoken.WithTimeout(options.Timeout),
			gftoken.WithMaxRefresh(options.MaxRefresh),
			gftoken.WithMultiLogin(options.MultiLogin),
			gftoken.WithExcludePaths(options.ExcludePaths),
			gftoken.WithGRedis(),
		)
	} else {
		gT.GfToken = gftoken.NewGfToken(
			gftoken.WithCacheKey(options.CacheKey),
			gftoken.WithTimeout(options.Timeout),
			gftoken.WithMaxRefresh(options.MaxRefresh),
			gftoken.WithMultiLogin(options.MultiLogin),
			gftoken.WithExcludePaths(options.ExcludePaths),
		)
	}
	return IGfToken(&gT)
}