使用Golang构建URL缩短服务的完整指南

使用Golang构建URL缩短服务的完整指南 我最近构建了一个速度极快的URL短链接服务，名为 shortr。

我还撰写了一份易于遵循的指南，介绍如何让任何人都能搭建一个。

文章在此：如何在Go中构建一个URL短链接服务 | 4rkal的博客

希望这能帮助到一些人。

也非常希望能获得一些反馈！

2 回复

感谢分享！

更多关于使用Golang构建URL缩短服务的完整指南的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


技术实现分析

从你提供的项目来看，这个URL缩短服务采用了简洁高效的设计方案。以下是核心实现的技术要点：

1. 存储层设计

// 使用内存存储的简化示例
type URLStore struct {
    urls map[string]string
    mu   sync.RWMutex
}

func (s *URLStore) Save(longURL string) string {
    s.mu.Lock()
    defer s.mu.Unlock()
    
    shortCode := generateShortCode()
    s.urls[shortCode] = longURL
    return shortCode
}

func (s *URLStore) Get(shortCode string) (string, bool) {
    s.mu.RLock()
    defer s.mu.RUnlock()
    
    longURL, exists := s.urls[shortCode]
    return longURL, exists
}

2. 短码生成算法

func generateShortCode() string {
    // 使用Base62编码的示例
    const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
    b := make([]byte, 6)
    for i := range b {
        b[i] = charset[rand.Intn(len(charset))]
    }
    return string(b)
}

3. HTTP路由处理

func main() {
    store := NewURLStore()
    
    http.HandleFunc("/shorten", func(w http.ResponseWriter, r *http.Request) {
        if r.Method != "POST" {
            http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
            return
        }
        
        longURL := r.FormValue("url")
        shortCode := store.Save(longURL)
        
        response := map[string]string{
            "short_url": fmt.Sprintf("https://shortr.dev/%s", shortCode),
        }
        json.NewEncoder(w).Encode(response)
    })
    
    http.HandleFunc("/{shortCode}", func(w http.ResponseWriter, r *http.Request) {
        shortCode := chi.URLParam(r, "shortCode")
        if longURL, exists := store.Get(shortCode); exists {
            http.Redirect(w, r, longURL, http.StatusFound)
            return
        }
        http.NotFound(w, r)
    })
    
    http.ListenAndServe(":8080", nil)
}

4. 性能优化点

  • 使用sync.RWMutex实现并发安全的读写操作
  • 采用内存存储减少I/O延迟
  • 支持连接池配置
var httpClient = &http.Client{
    Timeout: 5 * time.Second,
    Transport: &http.Transport{
        MaxIdleConns:        100,
        MaxIdleConnsPerHost: 10,
        IdleConnTimeout:     30 * time.Second,
    },
}

5. 可扩展性考虑

如果需要持久化存储，可以抽象存储接口：

type Storage interface {
    Save(shortCode, longURL string) error
    Get(shortCode string) (string, error)
    Exists(shortCode string) bool
}

// 实现Redis存储
type RedisStorage struct {
    client *redis.Client
}

func (r *RedisStorage) Save(shortCode, longURL string) error {
    return r.client.Set(context.Background(), shortCode, longURL, 0).Err()
}

这个实现方案在保持简洁的同时，通过合理的并发控制和内存管理确保了高性能。存储层抽象设计也为后续扩展提供了灵活性。

回到顶部