南强小屋

Go实现短url项目的方法示例
站长资源 2024/4/28 佚名
48 1538
南强小屋 Design By 杰米

首先说一下这种业务的应用场景：
1.把一个长url转换为一个短url网址
2.主要用于微博，二维码，等有字数限制的场景

主要实现的功能分析：
1.把长url的地址转换为短url地址
2.通过短url获取对应的原始长url地址
3.相同长url地址是否需要同样的短url地址

这里实现的是一个api服务

Go实现短url项目的方法示例

数据库设计

数据库的设计其实也没有非常复杂，如图所示：

Go实现短url项目的方法示例

这里有个设置需要主要就是关于数据库表中id的设计，需要设置为自增的

并且这里有个问题需要提前知道，我们的思路是根据id的值会转换为62进制关于进制转换的代码为：

// 将十进制转换为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)
}

所以这里需要设置一下数据库id的起始值，可以设置的大一点，这样转换为62进制之后不至于太短

代码逻辑

项目完整的代码git地址：https://github.com/pythonsite/go_simple_code/tree/master/short_url
当然这里的代码还有待后面继续做优化，但是这里通过golang内置的net/http 库实现了一个简单的api功能

代码的目录结构

|____logic
| |____logic.go
|____model
| |____data.go
|____api
| |____api.go
|____client
| |____client.go

logic目录为主要的处理逻辑
model是定义了request和response结构体
api目录为程序的入口程序
client 为测试请求，进行地址的转换

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) {
  // 数据库初始化
  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
    // 数据库中没有记录，重新生成一个新的短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=",shortUrl,insertId)
  if err != nil{
    fmt.Println(err)
    return
  }
  return
}

// 将十进制转换为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
}

// 用于将返回序列化数据，失败的返回
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)
}

// 用于将返回序列化数据，成功的返回
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) {
  // 这里需要说明的是发来的数据是通过post发过来一个json格式的数据
  data, err := ioutil.ReadAll(r.Body)
  if err != nil {
    fmt.Println("read all failded, ", err)
    responseError(w, 1001)
    return
  }

  var req model.Long2ShortRequest
  // 将反序列化的数据保存在结构体中
  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) {
  // 这里需要说明的是发来的数据是通过post发过来一个json格式的数据
  data, err := ioutil.ReadAll(r.Body)
  if err != nil {
    fmt.Println("read all failded, ", err)
    responseError(w, 1001)
    return
  }

  var req model.Short2LongRequest
  // 将反序列化的数据保存在结构体中
  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")
  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)
}

小结

这次通过这个小代码对go也有了一个初步的认识和使用，同时也通过net/http 包实现了api的功能，也对其基本使用有了大致了解

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。

标签：
Go实现短url
南强小屋 Design By 杰米
广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！ 如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com
南强小屋 Design By 杰米

评论“Go实现短url项目的方法示例”

暂无Go实现短url项目的方法示例的评论...
www.nqxw.com 南强小屋
8,675无损音乐
1,324高清电影
213破解软件
70,141收录教程

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

暴雪近日发布了《魔兽世界》10.2.6 更新内容，新游玩模式《强袭风暴》即将于3月21 日在亚服上线，届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。

艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕，并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时，他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中，玩家将会进入一个全新的海盗主题大逃杀式限时活动，其中包含极高的风险和史诗级的奖励。

《强袭风暴》不是普通的战场，作为一个独立于主游戏之外的活动，玩家可以用大逃杀的风格来体验《魔兽世界》，不分职业、不分装备（除了你在赛局中捡到的），光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式，玩家在加入海盗主题的预赛大厅区域前，可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹，《巨龙崛起》和《魔兽世界：巫妖王之怒 经典版》的玩家都可以获得奖励。

友情链接

杰晶网络 DDR爱好者之家 南强小屋 黑松山资源网 白云城资源网 更多链接