Go實現(xiàn)短url項目的方法示例
更新時間：2018年03月16日 10:41:38 作者：python修行路
這篇文章主要介紹了Go實現(xiàn)短url項目的方法示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
首先說一下這種業(yè)務的應用場景：
1.把一個長url轉(zhuǎn)換為一個短url網(wǎng)址
2.主要用于微博，二維碼，等有字數(shù)限制的場景
主要實現(xiàn)的功能分析：
1.把長url的地址轉(zhuǎn)換為短url地址
2.通過短url獲取對應的原始長url地址
3.相同長url地址是否需要同樣的短url地址
這里實現(xiàn)的是一個api服務
數(shù)據(jù)庫設計
數(shù)據(jù)庫的設計其實也沒有非常復雜，如圖所示：
這里有個設置需要主要就是關于數(shù)據(jù)庫表中id的設計，需要設置為自增的
并且這里有個問題需要提前知道，我們的思路是根據(jù)id的值會轉(zhuǎn)換為62進制關于進制轉(zhuǎn)換的代碼為：
// 將十進制轉(zhuǎn)換為62進制 0-9a-zA-Z 六十二進制
func transTo62(id int64)string{
// 1 -- > 1
// 10-- > a
// 61-- > Z
charset := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
var shortUrl []byte
for{
var result byte
number := id % 62
result = charset[number]
var tmp []byte
tmp = append(tmp,result)
shortUrl = append(tmp,shortUrl...)
id = id / 62
if id == 0{
break
}
}
fmt.Println(string(shortUrl))
return string(shortUrl)
}
所以這里需要設置一下數(shù)據(jù)庫id的起始值，可以設置的大一點，這樣轉(zhuǎn)換為62進制之后不至于太短
代碼邏輯
項目完整的代碼git地址：https://github.com/pythonsite/go_simple_code/tree/master/short_url
當然這里的代碼還有待后面繼續(xù)做優(yōu)化，但是這里通過golang內(nèi)置的net/http 庫實現(xiàn)了一個簡單的api功能
代碼的目錄結(jié)構(gòu)
|____logic | |____logic.go |____model | |____data.go |____api | |____api.go |____client | |____client.go
logic目錄為主要的處理邏輯
model是定義了request和response結(jié)構(gòu)體
api目錄為程序的入口程序
client 為測試請求，進行地址的轉(zhuǎn)換
model 代碼為：
package model
type Long2ShortRequest struct {
OriginUrl string `json:"origin_url"`
}
type ResponseHeader struct {
Code int `json:"code"`
Message string `json:"message"`
}
type Long2ShortResponse struct {
ResponseHeader
ShortUrl string `json:"short_url"`
}
type Short2LongRequest struct {
ShortUrl string `json:"short_url"`
}
type Short2LongResponse struct {
ResponseHeader
OriginUrl string `json:"origin_url"`
}
logic的代碼為：
package logic
import(
"go_dev/11/short_url/model"
"github.com/jmoiron/sqlx"
"fmt"
"crypto/md5"
"database/sql"
)
var (
Db *sqlx.DB
)
type ShortUrl struct {
Id int64 `db:"id"`
ShortUrl string `db:"short_url"`
OriginUrl string `db:"origin_url"`
HashCode string `db:"hash_code"`
}
func InitDb(dsn string)(err error) {
// 數(shù)據(jù)庫初始化
Db, err = sqlx.Open("mysql",dsn)
if err != nil{
fmt.Println("connect to mysql failed:",err)
return
}
return
}
func Long2Short(req *model.Long2ShortRequest) (response *model.Long2ShortResponse, err error) {
response = &model.Long2ShortResponse{}
urlMd5 := fmt.Sprintf("%x",md5.Sum([]byte(req.OriginUrl)))
var short ShortUrl
err = Db.Get(&short,"select id,short_url,origin_url,hash_code from short_url where hash_code=?",urlMd5)
if err == sql.ErrNoRows{
err = nil
// 數(shù)據(jù)庫中沒有記錄，重新生成一個新的短url
shortUrl,errRet := generateShortUrl(req,urlMd5)
if errRet != nil{
err = errRet
return
}
response.ShortUrl = shortUrl
return
}
if err != nil{
return
}
response.ShortUrl = short.ShortUrl
return
}
func generateShortUrl(req *model.Long2ShortRequest,hashcode string)(shortUrl string,err error){
result,err := Db.Exec("insert INTO short_url(origin_url,hash_code)VALUES (?,?)",req.OriginUrl,hashcode)
if err != nil{
return
}
// 0-9a-zA-Z 六十二進制
insertId,_:= result.LastInsertId()
shortUrl = transTo62(insertId)
_,err = Db.Exec("update short_url set short_url=? where id=?",shortUrl,insertId)
if err != nil{
fmt.Println(err)
return
}
return
}
// 將十進制轉(zhuǎn)換為62進制 0-9a-zA-Z 六十二進制
func transTo62(id int64)string{
// 1 -- > 1
// 10-- > a
// 61-- > Z
charset := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
var shortUrl []byte
for{
var result byte
number := id % 62
result = charset[number]
var tmp []byte
tmp = append(tmp,result)
shortUrl = append(tmp,shortUrl...)
id = id / 62
if id == 0{
break
}
}
fmt.Println(string(shortUrl))
return string(shortUrl)
}
func Short2Long(req *model.Short2LongRequest) (response *model.Short2LongResponse, err error) {
response = &model.Short2LongResponse{}
var short ShortUrl
err = Db.Get(&short,"select id,short_url,origin_url,hash_code from short_url where short_url=?",req.ShortUrl)
if err == sql.ErrNoRows{
response.Code = 404
return
}
if err != nil{
response.Code = 500
return
}
response.OriginUrl = short.OriginUrl
return
}
api的代碼為：
package main
import (
"io/ioutil"
"net/http"
"fmt"
"encoding/json"
"go_dev/11/short_url/logic"
"go_dev/11/short_url/model"
_ "github.com/go-sql-driver/mysql"
)
const (
ErrSuccess = 0
ErrInvalidParameter = 1001
ErrServerBusy = 1002
)
func getMessage(code int) (msg string){
switch code {
case ErrSuccess:
msg = "success"
case ErrInvalidParameter:
msg = "invalid parameter"
case ErrServerBusy:
msg = "server busy"
default:
msg = "unknown error"
}
return
}
// 用于將返回序列化數(shù)據(jù)，失敗的返回
func responseError(w http.ResponseWriter, code int) {
var response model.ResponseHeader
response.Code = code
response.Message = getMessage(code)
data, err := json.Marshal(response)
if err != nil {
w.Write([]byte("{\"code\":500, \"message\": \"server busy\"}"))
return
}
w.Write(data)
}
// 用于將返回序列化數(shù)據(jù)，成功的返回
func responseSuccess(w http.ResponseWriter, data interface{}) {
dataByte, err := json.Marshal(data)
if err != nil {
w.Write([]byte("{\"code\":500, \"message\": \"server busy\"}"))
return
}
w.Write(dataByte)
}
// 長地址到短地址
func Long2Short(w http.ResponseWriter, r *http.Request) {
// 這里需要說明的是發(fā)來的數(shù)據(jù)是通過post發(fā)過來一個json格式的數(shù)據(jù)
data, err := ioutil.ReadAll(r.Body)
if err != nil {
fmt.Println("read all failded, ", err)
responseError(w, 1001)
return
}
var req model.Long2ShortRequest
// 將反序列化的數(shù)據(jù)保存在結(jié)構(gòu)體中
err = json.Unmarshal(data, &req)
if err != nil {
fmt.Println("Unmarshal failded, ", err)
responseError(w, 1002)
return
}
resp, err := logic.Long2Short(&req)
if err != nil {
fmt.Println("Long2Short failded, ", err)
responseError(w, 1003)
return
}
responseSuccess(w, resp)
}
// 短地址到長地址
func Short2Long(w http.ResponseWriter, r *http.Request) {
// 這里需要說明的是發(fā)來的數(shù)據(jù)是通過post發(fā)過來一個json格式的數(shù)據(jù)
data, err := ioutil.ReadAll(r.Body)
if err != nil {
fmt.Println("read all failded, ", err)
responseError(w, 1001)
return
}
var req model.Short2LongRequest
// 將反序列化的數(shù)據(jù)保存在結(jié)構(gòu)體中
err = json.Unmarshal(data, &req)
if err != nil {
fmt.Println("Unmarshal failded, ", err)
responseError(w, 1002)
return
}
resp, err := logic.Short2Long(&req)
if err != nil {
fmt.Println("Long2Short failded, ", err)
responseError(w, 1003)
return
}
responseSuccess(w, resp)
}
func main(){
err := logic.InitDb("root:123456@tcp(192.168.50.145:3306)/short_url?parseTime=true")
if err != nil{
fmt.Printf("init db failed,err:%v\n",err)
return
}
http.HandleFunc("/trans/long2short", Long2Short)
http.HandleFunc("/trans/short2long", Short2Long)
http.ListenAndServe(":18888", nil)
}
小結(jié)
這次通過這個小代碼對go也有了一個初步的認識和使用，同時也通過net/http 包實現(xiàn)了api的功能，也對其基本使用有了大致了解
以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關文章
在 Golang 中,交叉編譯指的是在同一臺機器上生成針對不同操作系統(tǒng)或硬件架構(gòu)的二進制文件,這在開發(fā)跨平臺應用或構(gòu)建特定平臺的發(fā)布版本時非常有用,本文就詳細的給大家介紹一下Golang中的交叉編譯,需要的朋友可以參考下2023-08-08
解析GOROOT、GOPATH、Go-Modules-三者的關系
這篇文章主要介紹了解析GOROOT、GOPATH、Go-Modules-三者的關系,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-10-10
今天小編就為大家分享一篇關于go語言匿名函數(shù)的使用，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-04-04
logrus是一個非常強大的日志框架,具有靈活的功能和易于使用的API,適合處理各種類型的日志需求,這篇文章主要介紹了golang日志框架之logrus的安裝使用,需要的朋友可以參考下2023-08-08
Golang異常處理之defer,panic,recover的使用詳解
這篇文章主要為大家介紹了Go語言異常處理機制中defer、panic和recover三者的使用方法，文中示例代碼講解詳細,需要的朋友可以參考下2022-05-05
Go語言實現(xiàn)關閉http請求的方式總結(jié)
面試的時候問到如何關閉http請求，一般人脫口而出的是關閉response.body，這是錯誤的。本文為大家整理了三個正確關閉http請求的方法，希望對大家有所幫助2023-02-02
在本篇內(nèi)容里小編給大家整理的是一篇關于Go語言空白表示符_的實例用法，有興趣的朋友們可以學習參考下。2021-07-07
今天小編就為大家分享一篇關于在Go中復制文件最流行的3種方法，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2018-10-10
這篇文章主要介紹了Go語言HTTP請求流式寫入body,本文通過示例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-06-06
下面小編就為大家?guī)硪黄猤olang之tcp自動重連實現(xiàn)方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-06-06