Go实现短url项目的方法示例
更新时间：2018年03月16日 10:41:38 作者：python修行路
这篇文章主要介绍了Go实现短url项目的方法示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
首先说一下这种业务的应用场景：
1.把一个长url转换为一个短url网址
2.主要用于微博，二维码，等有字数限制的场景
主要实现的功能分析：
1.把长url的地址转换为短url地址
2.通过短url获取对应的原始长url地址
3.相同长url地址是否需要同样的短url地址
这里实现的是一个api服务
数据库设计
数据库的设计其实也没有非常复杂，如图所示：
这里有个设置需要主要就是关于数据库表中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=? where id=?",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?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)
}
小结
这次通过这个小代码对go也有了一个初步的认识和使用，同时也通过net/http 包实现了api的功能，也对其基本使用有了大致了解
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
相关文章
GO 函数式选项模式(Functional Options Pattern)
Option模式支持传递多个参数，并且在参数个数、类型发生变化时保持兼容性，任意顺序传递参数，下面给大家介绍GO 函数式选项模式(Functional Options Pattern)的相关知识，感兴趣的朋友一起看看吧2021-10-10
最近想学习一下golang,自己之前一直把环境全部安装在docker上,所以这次也想把golang的环境安装在docker上,下面这篇文章主要给大家介绍了关于本地使用Docker搭建go开发环境的相关资料,需要的朋友可以参考下2022-07-07
我们都知道go语言中内存管理工作都是由Go在底层完成的，这样我们可以不用过多的关注底层的内存问题。本文主要总结一下 Golang内存逃逸分析，需要的朋友可以参考以下内容，希望对大家有帮助2022-12-12
AES是一个对称密码，旨在取代DES成为广泛使用的标准，本文给大家分享Golang实现AES对称加密的过程，本文附有Golang实现AES加密ECB模式的源码，感兴趣的朋友跟随小编一起学习下吧2021-05-05
这篇文章主要为大家介绍了go通过编码缩短字符串的长度实现方法步骤,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01
在Web开发中，有时需要对网页进行截图，以便进行页面预览、测试等操作，本文为大家整理了Golang实现无头浏览器的截图的方法，感兴趣的可以了解一下2023-05-05
这篇文章主要为大家介绍了golang下grpc框架的使用编写示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步早日升职加薪2022-04-04
golang bad file descriptor问题的解决方法
这篇文章主要给大家介绍了golang bad file descriptor问题的解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-02-02
sync.Mutex可以说是sync包的核心了, sync.RWMutex, sync.WaitGroup...都依赖于他, 本章我们将带你一文读懂sync.Mutex，快跟随小编一起学习一下吧2023-04-04
这篇文章主要来和大家深入讨论一下Golang 中的字符串,并查看一些不同的场景,以避免常见错误,对大家掌握golang有一定的帮助,需要的可以了解下2023-10-10
最新评论