kenneth/go-url-shortener
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

short url web page v1

Browse Source
This commit is contained in:
kenneth
2023-12-25 17:04:22 +08:00
parent f27023c2ac
commit c1f4731669
45 changed files with 1637 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
shortener/shortener.go
View File

@@ -15,7 +15,8 @@ func sha256Of(input string) []byte {
}
func base58Encoded(bytes []byte) (string, error) {
encoded, err := base58.BitcoinEncoding.Encode(bytes)
encoding := base58.BitcoinEncoding
encoded, err := encoding.Encode(bytes)
if err != nil {
return "", err
}
@@ -31,5 +32,5 @@ func GenerateShortLink(originUrl string, userId string) (string, error) {
return "", err
}
return result, nil
return result[:8], nil
}

6
shortener/shortener_test.go
View File

@@ -12,15 +12,15 @@ func TestGenerateShortLink(t *testing.T) {
link1 := "https://www.baidu.com/"
short1, err := GenerateShortLink(link1, userId)
assert.NoError(t, err)
assert.Equal(t, short1, "egtq236P5f3")
assert.Equal(t, short1, "egtq236P")
link2 := "https://www.163.com/"
short2, err := GenerateShortLink(link2, userId)
assert.NoError(t, err)
assert.Equal(t, short2, "DiCqg9YpV89")
assert.Equal(t, short2, "DiCqg9Yp")
link3 := "https://www.qq.com/"
short3, err := GenerateShortLink(link3, userId)
assert.NoError(t, err)
assert.Equal(t, short3, "4QhQ62cZem1")
assert.Equal(t, short3, "4QhQ62cZ")
}