From ffee84ced273718aba211c9f42e834d63176740b Mon Sep 17 00:00:00 2001
From: aichy126 <16996097+aichy126@users.noreply.github.com>
Date: Thu, 2 Mar 2023 15:10:58 +0800
Subject: [PATCH] add en de short id

---
 pkg/uid/sid.go      | 18 ++++++++++++++++--
 pkg/uid/sid_test.go | 17 +++++++++++++----
 2 files changed, 29 insertions(+), 6 deletions(-)

diff --git a/pkg/uid/sid.go b/pkg/uid/sid.go
index a38fb007..9898f8f9 100644
--- a/pkg/uid/sid.go
+++ b/pkg/uid/sid.go
@@ -1,5 +1,7 @@
 package uid
 
+import "strconv"
+
 const salt = int64(12345678)
 
 var AlphanumericSet = []rune{
@@ -18,7 +20,7 @@ func init() {
 }
 
 // NumToString num to string
-func NumToString(id int64) string {
+func NumToShortID(id int64) string {
 	id = id + salt
 	var code []rune
 	for id > 0 {
@@ -30,7 +32,7 @@ func NumToString(id int64) string {
 }
 
 // StringToNum string to num
-func StringToNum(code string) int64 {
+func ShortIDToNum(code string) int64 {
 	var id int64
 	runes := []rune(code)
 
@@ -42,3 +44,15 @@ func StringToNum(code string) int64 {
 	id = id - salt
 	return id
 }
+
+func EnShortID(id string) string {
+	num, err := strconv.ParseInt(id, 10, 64)
+	if err != nil {
+		return ""
+	}
+	return NumToShortID(num)
+}
+func DeShortID(sid string) string {
+	num := ShortIDToNum(sid)
+	return strconv.FormatInt(num, 10)
+}
diff --git a/pkg/uid/sid_test.go b/pkg/uid/sid_test.go
index 63e7915b..1ffa1ba6 100644
--- a/pkg/uid/sid_test.go
+++ b/pkg/uid/sid_test.go
@@ -5,11 +5,20 @@ import (
 	"testing"
 )
 
-func Test_getInviteCodeById(t *testing.T) {
-	nums := []int64{0, 1, 10, 100, 1000, 10000, 100000, 10010000000001316, 10030000000001316}
+func Test_ShortID(t *testing.T) {
+	nums := []int64{0, 1, 10, 100, 1000, 10000, 100000, 10010000000001316, 10030000000001316, 999999999999999999, 1999999999999999999}
 	for _, num := range nums {
-		code := NumToString(num)
-		denum := StringToNum(code)
+		code := NumToShortID(num)
+		denum := ShortIDToNum(code)
+		fmt.Println(num, code, denum)
+	}
+}
+
+func Test_EnDeShortID(t *testing.T) {
+	nums := []string{"0", "1", "10", "100", "1000", "10000", "100000", "1234567", "10010000000001316", "10030000000001316", "99999999999999999", "999999999999999999", "1999999999999999999"}
+	for _, num := range nums {
+		code := EnShortID(num)
+		denum := DeShortID(code)
 		fmt.Println(num, code, denum)
 	}
 }