Golang实现短链接服务：基于map内存存储方法

用Golang的map实现短链接服务的核心是通过两个map维护长短链双向映射，配合自增ID生成Base62短串，并通过HTTP接口提供生成和跳转功能。1. 使用shortToLong和longToShort两个map实现双向映射，避免重复生成相同短链；2. 通过自增ID结合Base62编码生成唯一短标识符；3. 利用net/http库创建shorten接口接收长链接生成短链、redirect接口根据短链跳转原链接；4. 数据存储于内存适合原型开发，但需注意重启丢失、并发安全及冲突风险。

短链接服务的核心是将一个长 URL 映射成一个简短的字符串，用户访问这个短串就能跳转到原始地址。如果你只是想快速实现一个基础版本，用 Golang 的 map 做内存存储是一个简单又直接的办法。

下面我们就来看看怎么一步步用 Golang + map 实现一个最简单的短链接服务。

1. 设计数据结构和映射关系

我们使用两个 map 来保存长短链接之间的双向映射：

• 一个用于短链 → 长链（供跳转时查找）
• 一个用于长链 → 短链（避免重复生成）

var shortToLong = make(map[string]string)
var longToShort = make(map[string]string)

这样设计的好处是，当用户提交相同的长链接时，我们可以直接返回之前生成的短链接，而不是每次都生成新的。

2. 生成短链接标识符

短链接的关键在于生成一个唯一的、较短的字符串作为标识。常见做法是使用 Base62 编码（0-9a-zA-Z），也可以结合递增 ID 或随机生成。

这里我们采用一个简单的自增 ID 方式，每次生成后转换为 Base62：

func generateShortKey(id int) string {
    const base62 = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
    result := ""
    for id > 0 {
        id--
        result = string(base62[id%62]) + result
        id /= 62
    }
    return result
}

比如 ID=1 会生成 "a"，ID=10000 会生成 "27g"，足够简洁且不易冲突。

3. 处理请求逻辑

我们使用 Go 的标准库 net/http 搭建一个简单的 HTTP 服务，包含两个接口：

• /shorten：接收长链接，返回短链接
• /{key}：根据短链接 key 跳转到原链接

接口一：生成短链接

var counter = 1

func shortenHandler(w http.ResponseWriter, r *http.Request) {
    longURL := r.URL.Query().Get("url")
    if longURL == "" {
        http.Error(w, "Missing 'url' parameter", http.StatusBadRequest)
        return
    }

    if short, exists := longToShort[longURL]; exists {
        fmt.Fprintf(w, "Short URL: http://localhost:8080/%s\n", short)
        return
    }

    short := generateShortKey(counter)
    counter++

    shortToLong[short] = longURL
    longToShort[longURL] = short

    fmt.Fprintf(w, "Short URL: http://localhost:8080/%s\n", short)
}

接口二：跳转处理

func redirectHandler(w http.ResponseWriter, r *http.Request) {
    parts := strings.Split(r.URL.Path, "/")
    if len(parts) < 2 || parts[1] == "" {
        http.Error(w, "Invalid short URL", http.StatusBadRequest)
        return
    }

    key := parts[1]
    if longURL, exists := shortToLong[key]; exists {
        http.Redirect(w, r, longURL, http.StatusFound)
    } else {
        http.NotFound(w, r)
    }
}

然后注册路由并启动服务：

func main() {
    http.HandleFunc("/shorten", shortenHandler)
    http.HandleFunc("/", redirectHandler)

    fmt.Println("Starting server at :8080")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal(err)
    }
}

4. 测试一下

你可以通过浏览器或者 curl 来测试：

• 生成短链接：

  curl "http://localhost:8080/shorten?url=https://example.com/really-long-path?query=abc"

• 访问短链接跳转：

  curl -v http://localhost:8080/a

注意事项和优化空间

虽然上面的方法已经能跑起来，但还是有些需要注意的地方：

• 重启数据丢失：因为数据存在内存里，程序重启就会清空。如果要持久化，可以考虑加个文件或数据库写入。
• 并发安全问题：多个请求同时操作 map 可能会有竞态条件。可以用 sync.Mutex 或者换成 sync.Map。
• 短链冲突风险：如果使用随机生成而不是自增 ID，需要检查是否已存在该 key。
• 性能限制：内存存储适合小规模使用，如果数据量大了就需要上 Redis 这样的缓存系统。

基本上就这些。用 map 实现短链接服务虽然简单，但很适合练手或做原型开发。等你熟悉流程后，再扩展功能、换存储方式都不难。