go实现一个极简的短链接服务

首先放一下Github-OEZ

短链接原理

原理其实很简单，前端给后端传一个字符串，然后根据这个字符串找到原网址，最后返回一个301即可。所以短网址算法主要就是研究如何根据原网址算出这个字符串以及如何根据字符串找到这个网址。

短链接算法

常用的短链接算法主要有两种：自增ID和哈希

哈希

顾名思义，哈希就是将原网址计算出一个哈希字符串，但是这种算法碰到哈希碰撞时，会存在重复的情况，这个时候就有可能需要再哈希或者进行其他的处理。

自增ID

所以这里我们选择使用自增ID的方法，我们先将原网址存入数据，然后使用数据库返回的id，将id换算成六十二进制的字符串（随机的大小写字母+数字），数据库也不需要存储这个字符串，直接返回给前端即可。当前端请求这个字符串时，再转成十进制的id即可找到原网址。go进制转换与随机62进制字符串代码：

import (
	"io"
	"log"
	"math"
	"math/rand"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

var CHARS = "InsV3Sf0obzp2i4gj1yYGqQv6wUtmBxlMAP7KHd8uTXFk9aRJWNC5EOhZDcLer"

const (
	// 0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ
	SCALE = 62
	REGEX = "^[0-9a-zA-Z]+$"
	NUM = 6
)

func RandomStr(str string) string {
	chars := []rune(str)
	for i := len(chars) - 1; i > 0; i-- {
		num := rand.Intn(i + 1)
		chars[i], chars[num] = chars[num], chars[i]
	}
	return string(chars)
}

func Encode10To62(val uint) string {
	if val < 0 {
		panic("val cannot be negative.")
	}
	str := ""
	var remainder int
	for math.Abs(float64(val)) > SCALE-1 {
		remainder = int(val % SCALE)
		str = string(CHARS[remainder]) + str
		val = val / SCALE
	}
	str = string(CHARS[val]) + str
	//for i := len(str); i < NUM; i++ {
	//	str = string(CHARS[0]) + str
	//}
	return str
}

func Decode62To10(val string) uint {
	if match, _ := regexp.MatchString(REGEX, val); !match {
		panic("input illegal.")
	}
	var result uint = 0
	index, length := 0, len(val)
	for i := 0; i < length; i++ {
		index = strings.Index(CHARS, string(val[i]))
		result += uint(index * int(math.Pow(float64(SCALE), float64(length-i-1))))
	}
	return result
}

Demo

Oh,easy!
支持API调用与添加书签栏快捷缩短。