From 4a14a250dcbcbfd01e32b80bd40eb9b8205c38f7 Mon Sep 17 00:00:00 2001
From: weloe <1345895607@qq.com>
Date: Sun, 7 May 2023 02:28:24 +0800
Subject: [PATCH] refactor: add constant

---
 constant/token_style.go |  9 +++++++++
 model/function.go       | 11 ++++++-----
 2 files changed, 15 insertions(+), 5 deletions(-)
 create mode 100644 constant/token_style.go

diff --git a/constant/token_style.go b/constant/token_style.go
new file mode 100644
index 0000000..ee93287
--- /dev/null
+++ b/constant/token_style.go
@@ -0,0 +1,9 @@
+package constant
+
+const (
+	UUID            = "uuid"
+	SimpleUUID      = "uuid-simple"
+	RandomString32  = "random-string32"
+	RandomString64  = "random-string64"
+	RandomString128 = "random-string128"
+)
diff --git a/model/function.go b/model/function.go
index 99addbc..9fb3865 100644
--- a/model/function.go
+++ b/model/function.go
@@ -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
 }