如何建立 Golang RESTful API，並使用中間件進行身份驗證？-Golang-PHP中文網

如何建立 Golang RESTful API，並使用中間件進行身份驗證？

WBOY

發布： 2024-06-04 14:16:56

原創

929 人瀏覽過

本文介紹如何建立 Golang RESTful API。首先，透過匯入必要的函式庫、定義資料模型和建立路由來建立 RESTful API。其次，使用 go-chi/chigot 和 go-chi/chi/middleware 建立和應用中間件來進行身份驗證。該文中進一步給出了一個實戰案例範例，展示了解決方案的實際應用。

如何构建 Golang RESTful API，并使用中间件进行身份验证？

如何建立Golang RESTful API，並使用中間件進行驗證

1. 建構RESTful API

匯入必需的函式庫：

import (
    "encoding/json"
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/mux"
)

登入後複製

定義資料模型：

type Product struct {
    ID    int    `json:"id"`
    Name  string `json:"name"`
    Price float64 `json:"price"`
}

登入後複製

建立路由：

router := mux.NewRouter()
router.HandleFunc("/products", GetProducts).Methods("GET")
router.HandleFunc("/products/{id}", GetProduct).Methods("GET")
router.HandleFunc("/products", CreateProduct).Methods("POST")
router.HandleFunc("/products/{id}", UpdateProduct).Methods("PUT")
router.HandleFunc("/products/{id}", DeleteProduct).Methods("DELETE")

登入後複製

定義控制器：

func GetProducts(w http.ResponseWriter, r *http.Request) {
    // ...
}

func GetProduct(w http.ResponseWriter, r *http.Request) {
    // ...
}

func CreateProduct(w http.ResponseWriter, r *http.Request) {
    // ...
}

func UpdateProduct(w http.ResponseWriter, r *http.Request) {
    // ...
}

func DeleteProduct(w http.ResponseWriter, r *http.Request) {
    // ...
}

登入後複製

2. 使用中間件進行驗證

#安裝中間件：

go get github.com/go-chi/chi
go get github.com/go-chi/chi/middleware

登入後複製

建立中間件：

func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // ... 检查认证信息 ...
        if !authorized {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }
        // ... 继续处理请求 ...
        next.ServeHTTP(w, r)
    })
}

登入後複製

應用中間件：

router.Use(AuthMiddleware)

登入後複製

實戰案例

範例：##

func main() {
    // 初始化数据库连接
    db, err := sql.Open("sqlite3", "mydb.db")
    if err != nil {
        log.Fatal(err)
    }

    // 创建路由器
    router := mux.NewRouter()

    // 定义路由并应用身份验证中间件
    router.HandleFunc("/products", GetProducts).Methods("GET").Use(AuthMiddleware)
    router.HandleFunc("/products/{id}", GetProduct).Methods("GET").Use(AuthMiddleware)
    router.HandleFunc("/products", CreateProduct).Methods("POST").Use(AuthMiddleware)
    router.HandleFunc("/products/{id}", UpdateProduct).Methods("PUT").Use(AuthMiddleware)
    router.HandleFunc("/products/{id}", DeleteProduct).Methods("DELETE").Use(AuthMiddleware)

    // 启动服务器
    http.Handle("/", router)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

登入後複製

以上是如何建立 Golang RESTful API，並使用中間件進行身份驗證？的詳細內容。更多資訊請關注PHP中文網其他相關文章！