当前位置:首页 > 短网址资讯 > 正文内容

Go语言写的一个短网址服务

www.ft12.com7年前 (2017-06-27)短网址资讯3618

题图来自https://www.ft12.com/

“缩址,又称短址、短网址、网址缩短、缩短网址、URL缩短等,指的是一种互联网上的技术与服务。此服务可以提供一个非常短小的URL以代替原来的可能较长的URL,将长的URL地址缩短。
用户访问缩短后的URL时,通常将会重定向到原来的URL。”

– Wikipedia

虽然短网址早已不再那么受广泛关注。但是不妨拿来练手。

根据公开可以搜索到的资料,短网址一般是将一个ID转换到一串字母,生成短的网址用于传播,实际访问会重定向到原网址。如上所述。

那么使用Go来写这个有什么优势呢,优势之一当然是,Go部署简单,只需要copy执行文件即可。执行速度也快,甚至连HTTP服务器都不需要。

下边就边写边说明。

12345678910111213141516
package mainimport (	"fmt"	"strings"	"time"	"net/http"	"database/sql"	"github.com/gin-gonic/gin"	"github.com/garyburd/redigo/redis"	_ "github.com/go-sql-driver/mysql"	"github.com/speps/go-hashids")

定义hashid包需要的salt,即生成字符串的最短位数。

12345
const (	hdSalt        = "mysalt"	hdMinLength   = 5	defaultDomain = "http://localhost:8000/")

定义redis和MySQL的配置信息

123456789101112
var (	RedisClient *redis.Pool	RedisHost   = "127.0.0.1:6379"	RedisDb     = 0	RedisPwd    = ""	db      *sql.DB	DB_HOST = "tcp(127.0.0.1:3306)"	DB_NAME = "short"	DB_USER = "root"	DB_PASS = "")

main函数,首先连接redis和MySQL。定义如下路由:
  • 访问首页

  • 访问hash

  • 访问短网址信息页

  • 生成短网址接口

熟悉的朋友应该都知道,访问短网址服务的首页一般会跳转到一个固定的网址,比如渣浪微博会跳转到微博首页,Twitter则是给出“Twitter uses the t.co domain as part of a service to protect users from harmful activity”的提示。这里我们也让它跳转到一个指定的网页。

最后,以8080端口运行,实际线上会使用80端口,可以自行修改。

1234567891011121314151617
func main() {	initRedis()	initMysql()	gin.SetMode(gin.DebugMode)	r := gin.Default()	r.GET("/", func(c *gin.Context) {		//http code can be StatusFound or StatusMovedPermanently 		c.Redirect(http.StatusFound, defaultDomain)	})	r.GET("/:hash", expandUrl)	r.GET("/:hash/info", expandUrlApi)	r.POST("/short", shortUrl)	r.Run(":8000")}

连接redis和MySQL

12345678910111213141516171819202122232425262728
func initRedis() {	// 建立连接池	RedisClient = &redis.Pool{		MaxIdle:     1,		MaxActive:   10,		IdleTimeout: 180 * time.Second,		Dial: func() (redis.Conn, error) {			c, err := redis.Dial("tcp", RedisHost)			if err != nil {				return nil, err			}			if _, err := c.Do("AUTH", RedisPwd); err != nil {				c.Close()				return nil, err			}			c.Do("SELECT", RedisDb)			return c, nil		},	}}func initMysql() {	dsn := DB_USER + ":" + DB_PASS + "@" + DB_HOST + "/" + DB_NAME + "?charset=utf8"	db, _ = sql.Open("mysql", dsn)	db.SetMaxOpenConns(5)	db.SetMaxIdleConns(20)	db.Ping()}

生成短网址的接口函数。

根据传入的URL参数,进行简单的验证后,写入数据库。根据写入后生成的ID,再生成一个字符串,输出短链接,然后返回给调用方。

1234567891011121314151617181920212223
func shortUrl(c *gin.Context) {	longUrl := c.PostForm("url")	if longUrl == "" {		c.JSON(200, gin.H{			"status":  500,			"message": "请传入网址",		})		return	}	if !strings.HasPrefix(longUrl, "http") {		longUrl = "http://" + longUrl	}	if hash, ok := insert(longUrl); ok {		c.JSON(200, gin.H{			"status":  200,			"message": "ok",			"short":   defaultDomain + hash,		})	}}

根据HASH解析并跳转到对应的长URL,不存在则跳转到默认地址

1234567891011
func expandUrl(c *gin.Context) {	hash := c.Param("hash")	if url, ok := findByHash(hash); ok {		c.Redirect(http.StatusMovedPermanently, url)	}	// 注意:	// 	实际中,此应用的运行域名可能与默认域名不同,如a.com运行此程序,默认域名为b.com	// 	当访问一个不存在的HASH或a.com时,可以跳转到任意域名,即defaultDomain	c.Redirect(http.StatusMovedPermanently, defaultDomain)}

根据HASH在redis中查找并返回结果,不存在则返回404状态

123456789101112131415161718
func expandUrlApi(c *gin.Context) {	hash := c.Param("hash")	if url, ok := findByHash(hash); ok {		c.JSON(200, gin.H{			"status":  200,			"message": "ok",			"data":    url,		})		return	}	// 此处可以尝试在MySQL中再次查询	c.JSON(200, gin.H{		"status":  404,		"message": "url of hash is not exist",	})}

将ID转换成对应的HASH值,hdSalt与hdMinLength 会影响生成结果,确定后不要改动

12345678910
func shortenURL(id int) string {	hd := hashids.NewData()	hd.Salt = hdSalt	hd.MinLength = hdMinLength	h := hashids.NewWithData(hd)	e, _ := h.Encode([]int{id})	return e}

根据HASH解析出对应的ID值, hdSalt与hdMinLength 会影响生成结果,确定后不要改动

12345678910
func expand(hash string) int {	hd := hashids.NewData()	hd.Salt = hdSalt	hd.MinLength = hdMinLength	h := hashids.NewWithData(hd)	d, _ := h.DecodeWithError(hash)	return d[0]}

数据库中根据ID查找

123456789
func find(id int) (string, bool) {	var url string	err := db.QueryRow("SELECT url FROM url WHERE id = ?", id).Scan(&url)	if err == nil {		return url, true	} else {		return "", false	}}

在redis中根据HASH查找

1234567891011121314151617
func findByHash(h string) (string, bool) {	rc := RedisClient.Get()	defer rc.Close()	url, _ := redis.String(rc.Do("GET", "URL:"+h))	if url != "" {		return url, true	}	id := expand(h)	if urldb, ok := find(id); ok {		return urldb, true	}	return "", false}

将长网址插入到数据库中,并把返回的ID生成HASH和长网址存入redis

123456789101112131415
func insert(url string) (string, bool) {	stmt, _ := db.Prepare(`INSERT INTO url (url) values (?)`)	res, err := stmt.Exec(url)	checkErr(err)	id, _ := res.LastInsertId()	rc := RedisClient.Get()	defer rc.Close()	hash := shortenURL(int(id))	rc.Do("SET", "URL:"+hash, url)	return hash, true}

打印方法,和检查错误的方法

123456789
func Log(v ...interface{}) {	fmt.Println(v...)}func checkErr(err error) {	if err != nil {		panic(err)	}}

有些地方还需修改,就算是抛砖引玉吧。

感谢hashids

Github地址 : shortme


扫描二维码推送至手机访问。

版权声明:本文由短链接发布,如需转载请注明出处。

本文链接:https://www.ft12.com/article_244.html

分享给朋友:

相关文章

个性与风骨何去何从

个性与风骨何去何从

夜深,耳机里放着蒋勋说红楼,渐入睡境,却忽然听见”嵇康”二字,便又坐起,细细听之。这个名字似乎伴随了我的中学时代,在那些要交周记却又缺乏写作话题的日子里,嵇康便成为了我周记本里的常客。我总是装作很懂他,很崇拜他的样子,在文字里替他抒抒情,抱...

阿里巴巴马云:快递公司注定将成为技术公司,保证所有小企业通货权

阿里巴巴马云:快递公司注定将成为技术公司,保证所有小企业通货权

[ 短网址资讯导读 ] 5月22日,杭州云栖小镇,在2017全球才智物流峰会上,阿里巴巴董事局主席马云说,物流公司应当出资人才、技能,而且要联合作战,方能应对天天10亿包裹的业务体量。“我通知我们,一天十亿只包裹,不会超过八年,估...

干货:实战告诉大家如何脚踏实地赚钱

干货:实战告诉大家如何脚踏实地赚钱

每一个混站长圈的朋友,几乎都听说过卢松松博客。我入行较晚,是2012年来到外地读大学,才开始自己摸索着进入站长圈,然后结识松哥博客的。最早,在松哥博客上学习各种建站和优化技术、运营技巧。踏踏实实专研了两年技术以后,开始做自己的个人工作室,然...

FT12短网址:绘制随机不规则三角彩条——小谈FT12短网址主页的实现

FT12短网址:绘制随机不规则三角彩条——小谈FT12短网址主页的实现

引言8月6号早读文章由@FT12短网址分享。正文从这开始~缘起最近感觉自己搭的一个项目资料汇总的项目,主页真的是一点美感都没有,非常简单粗暴。就想来点改版,让主页显得高大上一点。改版前的效果是这样的:寻思着怎么改版,刚开始想就简单加个背景图...

哪些设置会导致短网址的 MySQL 慢查询

哪些设置会导致短网址的 MySQL 慢查询

作者:短链接团队:短网址服务提供商前言无论是开发同学还是DBA,想必大家都遇到慢查询(select,update,insert,delete 语句慢),影响短网址服务的稳定性。这里说的慢,有两个含义一是比正常的慢,有可能正常执行时间是10m...

短网址网站对每条短链接安全性的检测方法

短网址站,顾名思义就是把长网址缩短成很短的短链接并提供跳转服务。当用户访问短链接时,短网址站会检索数据库中此条短链接对应的真实网址,并快速跳转到真实的长网址。世界本来就是一片祥和,但是总有一些不法分子利用短链接来隐藏真实网址,然后群发恶意网...

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。