ALPHABET

icowan · Mar 28, 2020 · 1c21346 · 1c21346
1 parent 60651e5
commit 1c21346
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/service/service.go b/pkg/service/service.go
@@ -56,7 +56,12 @@ func (s *service) Get(ctx context.Context, code string) (redirect *Redirect, err
 func (s *service) Post(ctx context.Context, domain string) (redirect *Redirect, err error) {
 	now := time.Now()
 	now = now.In(time.Local)
-	code, _ := shortid.MustNew(0, s.alphabet, 1).Generate()
+	var code string
+	if s.alphabet != "" {
+		code, _ = shortid.MustNew(0, s.alphabet, 1).Generate()
+	} else {
+		code = shortid.MustGenerate()
+	}
 	// todo 考虑如何处理垃圾数据的问题 得复的url 不同的code
 
 	if s.maxLen > 0 {