feat: support add token generate function

constant/token_style.go

@@ -1,5 +1,6 @@
 package constant
 
+// tokenStyle constant
 const (
 	UUID            = "uuid"
 	SimpleUUID      = "uuid-simple"

enforcer.go

@@ -445,6 +445,12 @@ func (e *Enforcer) GetRequestToken(ctx ctx.Context) string {
 	return tokenValue
 }
 
+// AddTokenGenerateFun add token generate strategy
+func (e *Enforcer) AddTokenGenerateFun(tokenStyle string, f model.GenerateFunc) error {
+	e.generateFunc.AddFunc(tokenStyle, f)
+	return nil
+}
+
 func (e *Enforcer) GetSession(id string) *model.Session {
 	if v := e.adapter.Get(e.spliceSessionKey(id)); v != nil {
 		session := v.(*model.Session)

enforcer_interface.go

@@ -23,6 +23,7 @@ type IEnforcer interface {
 	Kickout(id string, device string) error
 
 	GetRequestToken(ctx ctx.Context) string
+	AddTokenGenerateFun(tokenStyle string, f model.GenerateFunc) error
 
 	CheckLogin(ctx ctx.Context) error