refactor: add constant

This commit is contained in:
weloe
2023-05-07 02:28:24 +08:00
parent e831ad6bf6
commit 4a14a250dc
2 changed files with 15 additions and 5 deletions

View File

@@ -2,6 +2,7 @@ package model
import (
"errors"
"github.com/weloe/token-go/constant"
"github.com/weloe/token-go/util"
"sync"
)
@@ -41,11 +42,11 @@ func (g *GenerateTokenFunc) GetFunction(tokenForm string) (GenerateFunc, error)
func LoadFunctionMap() GenerateTokenFunc {
fm := &GenerateTokenFunc{fns: &sync.Map{}}
fm.AddFunc("uuid", util.GenerateUUID)
fm.AddFunc("uuid-simple", util.GenerateSimpleUUID)
fm.AddFunc("random-string32", util.GenerateRandomString32)
fm.AddFunc("random-string64", util.GenerateRandomString64)
fm.AddFunc("random-string128", util.GenerateRandomString128)
fm.AddFunc(constant.UUID, util.GenerateUUID)
fm.AddFunc(constant.SimpleUUID, util.GenerateSimpleUUID)
fm.AddFunc(constant.RandomString32, util.GenerateRandomString32)
fm.AddFunc(constant.RandomString64, util.GenerateRandomString64)
fm.AddFunc(constant.RandomString128, util.GenerateRandomString128)
return *fm
}