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的功能,也对其基本使用有了大致了解

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Java帮帮-微信公众号-技术文章全总结

【选择题】Java基础测试题一(10道)

【选择题】Java基础测试题一(10道) 1.下面哪些是合法的变量名? (DEG) A.2variable //不能以数字开头 ...

47680
来自专栏Spark生态圈

[Spark SQL] 源码解析之Analyzer

Analyzer模块将Unresolved LogicalPlan结合元数据catalog进行绑定,最终转化为Resolved LogicalPlan。跟着代码...

13520
来自专栏JMCui

MongoDB系列一(查询).

一、简述     MongoDB中使用find来进行查询。查询就是返回一个集合中文档的子集,子集合的范围从0个文档到整个集合。默认情况下,"_id"这个键总是被...

50460
来自专栏Java帮帮-微信公众号-技术文章全总结

Java类加载器(用户自定义类加载器实现)

java类加载器主要分为如下几种: jvm提供的类加载器 根类加载器:底层实现,主要加载java核心类库(如:java.lang.*) 扩展类加载器:使用jav...

36560
来自专栏JavaEdge

Netty源码阅读入门实战(八)-解码(更新 ing)

就像很多标准的架构模式都被各种专用框架所支持一样,常见的数据处理模式往往也是目标实现的很好的候选对象,它可以节省开发人员大量的时间和精力。 当然这也适应于本文...

15040
来自专栏跟着阿笨一起玩NET

C# Eval在aspx页面中的用法及作用

29920
来自专栏Android 研究

OKHttp源码解析(五)--OKIO简介及FileSystem

okio是由square公司开发的,它补充了java.io和java.nio的不足,以便能够更加方便,快速的访问、存储和处理你的数据。OKHttp底层也是用该库...

25830
来自专栏SeanCheney的专栏

《利用Python进行数据分析·第2版》第6章 数据加载、存储与文件格式6.1 读写文本格式的数据6.2 二进制数据格式6.3 Web APIs交互6.4 数据库交互6.5 总结

访问数据是使用本书所介绍的这些工具的第一步。我会着重介绍pandas的数据输入与输出,虽然别的库中也有不少以此为目的的工具。 输入输出通常可以划分为几个大类:读...

59360
来自专栏mini188

java中的锁

java中有哪些锁 这个问题在我看了一遍<java并发编程>后尽然无法回答,说明自己对于锁的概念了解的不够。于是再次翻看了一下书里的内容,突然有点打开脑门的感觉...

50690
来自专栏Android知识点总结

Java总结IO篇之File类和Properties类

打开颜色选择器 :读流I-->字符串分割-->字符串存入Map-->使用Map对象还原用户配置 修改配置时 :写流O-->创建Map对象-->字符...

19020

扫码关注云+社区

领取腾讯云代金券