快捷導(dǎo)航

使用Go語言開發(fā)短鏈接服務(wù)的方法

更新時間：2024年03月27日 09:00:41 作者：ALXPS

短鏈接一般是通過映射關(guān)系，將長長的一串網(wǎng)址，映射到幾個字符的短鏈接上，建立好這種映射關(guān)系之后保存到數(shù)據(jù)庫里，用戶每次訪問短鏈接的時候，這篇文章主要介紹了使用Go語言開發(fā)一個短鏈接服務(wù),需要的朋友可以參考下

應(yīng)用場景

　　假如我們正在運營一個在線課程網(wǎng)站，運營人員策劃雙11促銷大降價！于是乎，產(chǎn)品經(jīng)理鞭策研發(fā)對應(yīng)的活動頁面。我們的開發(fā)非常給力，經(jīng)過數(shù)周007力度的勞作，解決無數(shù)bug后，活動頁面不負(fù)眾望開發(fā)完成。我們產(chǎn)品經(jīng)理開心地把促銷活動頁面鏈接交給運營人員。運營人員早已準(zhǔn)備好活動的短信、微博和微信文案模板，就等活動鏈接下鍋了！但是當(dāng)看到鏈接URL時，運營人員陷入了沉思……

https://www.mywebsite.com/courses/promotional-activities/date-1111/ac60ffe3-8ef0-4efa-81d1-edc626569ff0

　　鏈接URL比預(yù)想的長幾倍，加入URL后文案模板給運營人員描繪活動信息的文字空間所剩無幾。接下來，產(chǎn)品運營、產(chǎn)品經(jīng)理、技術(shù)研發(fā)三方，展開長達(dá)兩年半年的激烈掰扯。不出意料，出了問題當(dāng)然還是研發(fā)來解決。聰明的研發(fā)小伙小明提出了解決方案，在Nginx配置一個短的URL跳轉(zhuǎn)到活動鏈接URL，問題解決、下班！

　　運營人員拿到“修改后的”活動鏈接，得到此次活動的短信大致長下面這樣。

　　當(dāng)然上面Nginx配置跳轉(zhuǎn)只是硬編碼方式的一種臨時解決方案，將來有更多類似的活動，維護(hù)起來就像是“千層漿糊”，根本無從滿足運營全鏈路深度營銷、矩陣式打法。

　　魯迅曾經(jīng)說過：“當(dāng)軟件設(shè)計上遇到問題時，解決方案就是，加一層。” 所以研發(fā)組決定開發(fā)一個短鏈接服務(wù)，用來維護(hù)短URL映射跳轉(zhuǎn)到長URL。

原理

　　短鏈接一般是通過映射關(guān)系，將長長的一串網(wǎng)址，映射到幾個字符的短鏈接上，建立好這種映射關(guān)系之后保存到數(shù)據(jù)庫里，用戶每次訪問短鏈接的時候，需要到數(shù)據(jù)庫里查詢這個短鏈接對應(yīng)的源網(wǎng)址，然后給用戶跳轉(zhuǎn)到目標(biāo)長鏈接。

　　短鏈接從生成到使用分為以下幾步：

申請者，請求短鏈接服務(wù)，申請將長鏈接B生成對應(yīng)的短鏈接
短鏈接服務(wù)器生成對應(yīng)的短鏈接A，并保存短鏈接和長鏈接的映射關(guān)系到數(shù)據(jù)庫，并返回短鏈接A給申請者
把短鏈接A拼接到短信等的內(nèi)容上發(fā)送。
用戶點擊短鏈接A，瀏覽器用301/302進(jìn)行重定向，訪問到對應(yīng)的長鏈接B。
展示對應(yīng)的內(nèi)容。

　　這里注意http重定向狀態(tài)碼301和302的區(qū)別：301 永久重定向，302 是臨時重定向。瀏覽器接收到301重定向后會先請求短鏈接服務(wù)，由短鏈接服務(wù)再定向到目標(biāo)長鏈接地址，后續(xù)瀏覽器再次訪問短鏈接URL后，便不再經(jīng)短鏈接服務(wù)跳轉(zhuǎn)，而是直接訪問目標(biāo)長鏈接服務(wù)，302的話則每次要經(jīng)過短鏈接服務(wù)重定向跳轉(zhuǎn)。（HTTP 中的 301、302、303、307、308 響應(yīng)狀態(tài)碼) 因此，如果要統(tǒng)計訪問量，可以使用302；如果要減少短鏈接服務(wù)器壓力，可以使用301。

代碼實踐

　　魯迅又說: "Talk is cheap, show me the code."

　　接下來我們用Gin框架實現(xiàn)一個簡單的短鏈接示例

package main
import (
	"fmt"
	"log"
	"net/http"
	"github.com/gin-gonic/gin"
)
// shortLong 短鏈接ID和目標(biāo)長鏈接映射關(guān)系，模擬數(shù)據(jù)庫存儲
var shortLong = map[string]string{
	"bd": "https://baike.baidu.com/item/%E7%9F%AD%E9%93%BE%E6%8E%A5/7224556?fr=ge_ala",
	"sg": "https://baike.sogou.com/v72514301.htm?fromTitle=%E7%9F%AD%E9%93%BE%E6%8E%A5",
}
// redirectHandler 查找鏈接映射，跳轉(zhuǎn)到目標(biāo)長鏈接
func redirectHandler(c *gin.Context) {
	shortCode := c.Param("code")
	longUrl, ok := shortLong[shortCode]
	if !ok {
		c.IndentedJSON(http.StatusNotFound, gin.H{
			"detail": fmt.Sprintf("短鏈接(%s)無對應(yīng)的長鏈接地址", shortCode),
		})
		return
	}
	c.Redirect(http.StatusMovedPermanently, longUrl)
}
func main() {
	engine := gin.Default()
	engine.GET("/:code", redirectHandler)
	if err := engine.Run(":9999"); err != nil {
		log.Fatalf("啟動gin server失?。?v", err)
	}
}

代碼邏輯比較簡單，就不一一解釋了??。XX，啟動！

[GIN-debug] [WARNING] Creating an Engine instance with the Logger and Recovery middleware already attached. [GIN-debug] [WARNING] Running in "debug" mode. Switch to "release" mode in production. - using env: export GIN_MODE=release - using code: gin.SetMode(gin.ReleaseMode) [GIN-debug] GET /:short --> main.redirectHandler (3 handlers) [GIN-debug] [WARNING] You trusted all proxies, this is NOT safe. We recommend you to set a value. Please check https://pkg.go.dev/github.com/gin-gonic/gin#readme-don-t-trust-all-proxies for details. [GIN-debug] Listening and serving HTTP on :9999

在瀏覽器輸入短鏈接URL：http://127.0.0.1:9999/bd。我們就能訪問到“長鏈接”（https://baike.baidu.com/item/%E7%9F%AD%E9%93%BE%E6%8E%A5/7224556?fr=ge_ala）了。

Gin日志在terminal輸入如下