From 1c21346faed9b4bb24e75fb9a790d62bbe444932 Mon Sep 17 00:00:00 2001
From: dudulu <wack1001@163.com>
Date: Sat, 28 Mar 2020 22:06:37 +0800
Subject: [PATCH] ALPHABET

---
 pkg/service/service.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pkg/service/service.go b/pkg/service/service.go
index e98826c..43a2a9f 100644
--- 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 {