Go语言写的一个短网址服务
“缩址，又称短址、短网址、网址缩短、缩短网址、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
