跟着教程用Golang实现一个简单的URL缩短服务-Golang-PHP中文网

跟着教程用Golang实现一个简单的URL缩短服务

P粉602998670

发布： 2025-09-07 09:57:02

原创

462人浏览过

使用Golang可快速搭建短链接服务，通过内存map存储短码与长URL映射，生成6位随机短码，支持POST创建短链和浏览器重定向访问。

跟着教程用golang实现一个简单的url缩短服务

想快速搭建一个属于自己的短链接服务？用 Golang 可以很轻松地实现。下面带你一步步写一个简单的 URL 缩短服务，包含基本的生成短码、重定向和存储功能，不依赖外部数据库，适合学习和本地测试。

1. 基本功能设计

短链接服务的核心逻辑是：

接收一个长 URL，生成一个唯一的短码（比如 6 位字符串）
将短码和原始 URL 存储映射关系
通过短码访问时，重定向到原始 URL

我们使用内存 map 作为存储，用随机生成的 6 位字符串作为短码，HTTP 路由使用标准库 net/http。

2. 项目结构和依赖

新建一个项目目录，比如 url-shortener，创建 main.go。不需要额外依赖，只用 Golang 标准库。

立即学习“go语言免费学习笔记（深入）”；

3. 实现代码

package main

import (
    "encoding/json"
    "fmt"
    "math/rand"
    "net/http"
    "strings"
    "time"
)

// 存储映射：短码 -> 原始 URL
var urlStore = make(map[string]string)

// 字符集用于生成短码
const shortChars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
const shortLength = 6

// 生成唯一的短码
func generateShortCode() string {
    rand.New(rand.NewSource(time.Now().UnixNano()))
    b := make([]byte, shortLength)
    for i := range b {
        b[i] = shortChars[rand.Intn(len(shortChars))]
    }
    return string(b)
}

// 创建短链接
func shortenHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method != "POST" {
        http.Error(w, "只支持 POST", http.StatusMethodNotAllowed)
        return
    }

    type request struct {
        URL string `json:"url"`
    }

    var req request
    err := json.NewDecoder(r.Body).Decode(&req)
    if err != nil || req.URL == "" {
        http.Error(w, "无效的请求体", http.StatusBadRequest)
        return
    }

    // 确保 URL 以 http 开头
    if !strings.HasPrefix(req.URL, "http://") && !strings.HasPrefix(req.URL, "https://") {
        req.URL = "http://" + req.URL
    }

    // 生成短码（简单起见，不检查重复，实际应用需循环检查）
    var shortCode string
    for {
        shortCode = generateShortCode()
        if _, exists := urlStore[shortCode]; !exists {
            break
        }
    }

    urlStore[shortCode] = req.URL

    // 返回短链接
    result := map[string]string{
        "short_url": fmt.Sprintf("http://localhost:8080/%s", shortCode),
    }
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(result)
}

// 重定向到原始 URL
func redirectHandler(w http.ResponseWriter, r *http.Request) {
    path := strings.TrimPrefix(r.URL.Path, "/")
    if path == "" {
        http.Error(w, "请输入短码", http.StatusBadRequest)
        return
    }

    originalURL, exists := urlStore[path]
    if !exists {
        http.Error(w, "短码不存在", http.StatusNotFound)
        return
    }

    http.Redirect(w, r, originalURL, http.StatusFound)
}

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

    fmt.Println("服务启动在 http://localhost:8080")
    http.ListenAndServe(":8080", nil)
}

登录后复制