从入门到精通：如何在 Go 中构建自己的短网址服务

2023-04-02 14:22:58

构建一个功能齐全的短网址服务

在当今数字世界中，短网址服务在分享长而复杂的链接方面发挥着至关重要的作用。它们不仅方便，而且可以通过跟踪链接点击次数来提供有价值的分析数据。在这篇教程中，我们将深入探讨如何使用 Go 语言构建一个全面的短网址服务，涵盖从创建缩短的 URL 到处理错误的各个方面。

1. 准备工作

首先，你需要在你的计算机上安装 Go 语言。完成安装后，创建一个名为 "shortlink" 的项目目录。接下来，初始化一个 Go 模块并导入必要的包，包括 database/sql、encoding/json、fmt、log、net/http 和 time。

2. 建立数据库连接

为了存储短网址和原始 URL，我们需要一个数据库。本教程将使用 SQLite。创建一个名为 "shortlink.db" 的文件，并使用提供的 SQL 语句创建 "urls" 表。

3. 定义数据模型

接下来，创建一个名为 "url.go" 的文件，定义一个数据模型，包括 ID、ShortURL、OriginalURL 和 CreatedAt 字段。

4. 定义路由

在 "main.go" 文件中，定义路由以处理不同的请求。包括一个用于创建短网址的 POST 路由和一个用于重定向到原始 URL 的 GET 路由。

5. 构建业务逻辑

在 main.go 文件中，定义用于插入和获取 URL 的函数。这些函数使用数据库连接来与数据库交互。

6. 生成短网址

在 main.go 文件中，定义一个函数来生成唯一的短网址。这可以使用随机字符串生成器或哈希函数来实现。

7. 错误处理

在处理业务逻辑时，错误处理至关重要。在我们的代码中，我们使用 http.Error 函数来向客户端发送错误响应，包括错误代码和消息。

代码示例

以下是 "main.go" 文件的完整代码：

package main

import (
    "context"
    "database/sql"
    "fmt"
    "log"
    "net/http"

    "github.com/go-chi/chi"
)

func main() {
    db, err := sql.Open("sqlite3", "./shortlink.db")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    r := chi.NewRouter()

    // 路由1：创建短网址
    r.Post("/shorten", func(w http.ResponseWriter, r *http.Request) {
        var url URL
        if err := json.NewDecoder(r.Body).Decode(&url); err != nil {
            http.Error(w, "Invalid JSON", http.StatusBadRequest)
            return
        }

        shortURL, err := insertURL(db, url.OriginalURL)
        if err != nil {
            http.Error(w, "Internal Server Error", http.StatusInternalServerError)
            return
        }

        json.NewEncoder(w).Encode(map[string]string{"short_url": shortURL})
    })

    // 路由2：重定向到原始 URL
    r.Get("/{shortURL}", func(w http.ResponseWriter, r *http.Request) {
        shortURL := chi.URLParam(r, "shortURL")

        originalURL, err := getOriginalURL(db, shortURL)
        if err != nil {
            http.Error(w, "Not Found", http.StatusNotFound)
            return
        }

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

    log.Println("Server listening on port 8080")
    http.ListenAndServe(":8080", r)
}

func insertURL(db *sql.DB, originalURL string) (string, error) {
    shortURL := generateShortURL()

    stmt, err := db.Prepare("INSERT INTO urls (short_url, original_url, created_at) VALUES (?, ?, ?)")
    if err != nil {
        return "", err
    }
    defer stmt.Close()

    _, err = stmt.Exec(shortURL, originalURL, time.Now())
    if err != nil {
        return "", err
    }

    return shortURL, nil
}

func getOriginalURL(db *sql.DB, shortURL string) (string, error) {
    row := db.QueryRow("SELECT original_url FROM urls WHERE short_url = ?", shortURL)
    var originalURL string
    if err := row.Scan(&originalURL); err != nil {
        return "", err
    }

    return originalURL, nil
}

func generateShortURL() string {
    return fmt.Sprintf("%d", time.Now().UnixNano())
}