深度解析：Golang攔截器的實際應用效果-Golang-PHP中文網

深度解析：Golang攔截器的實際應用效果

WBOY

發布： 2024-03-21 09:18:04

原創

732 人瀏覽過

深度解析：Golang攔截器的實際應用效果

Golang攔截器（interceptor）是一種強大的設計模式，在實際應用中可以實現諸多功能，如日誌記錄、錯誤處理、權限控制等。本文將深度解析Golang攔截器的實際應用效果，並透過具體的程式碼範例來展示其用法和效果。

1. 什麼是Golang攔截器

Golang攔截器是一種面向切面程式設計（AOP）的設計模式，透過在函數呼叫前後添加一層代理，可以實現對函數的攔截和擴展。攔截器可以在函數執行之前、之後或發生錯誤時對函數進行幹預，以實現更靈活的控制和操作。

2. 實際應用效果

2.1 日誌記錄

攔截器常用於記錄函數的輸入輸出參數、執行時間等信息，方便追蹤調試和效能最佳化。以下是一個簡單的日誌記錄攔截器範例：

package interceptor

import (
    "log"
    "time"
)

func Logger(next http.HandlerFunc) http.HandlerFunc {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        log.Printf("Start: %s %s", r.Method, r.URL.Path)

        next(w, r)

        elapsed := time.Since(start)
        log.Printf("End: %s %s took %s", r.Method, r.URL.Path, elapsed)
    })
}

登入後複製

2.2 錯誤處理

攔截器可以捕捉函數執行過程中的錯誤，進行統一的處理和返回，提高程式碼的健全性。以下是一個簡單的錯誤處理攔截器範例：

package interceptor

import (
    "log"
    "net/http"
)

func ErrorHandler(next http.HandlerFunc) http.HandlerFunc {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                log.Printf("Panic: %v", err)
                http.Error(w, "Internal Server Error", http.StatusInternalServerError)
            }
        }()

        next(w, r)
    })
}

登入後複製

2.3 權限控制

攔截器也可以用來實現權限控制，判斷使用者的身分和權限，並決定是否允許存取某個函數。以下是一個簡單的權限控制攔截器範例：

package interceptor

import (
    "net/http"
)

func AuthMiddleware(next http.HandlerFunc) http.HandlerFunc {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if checkPermission(r) {
            next(w, r)
        } else {
            http.Error(w, "Permission Denied", http.StatusForbidden)
        }
    })
}

func checkPermission(r *http.Request) bool {
    // 根據請求判斷使用者權限
    return true
}

登入後複製

3. 程式碼範例

下面透過一個簡單的HTTP伺服器範例來示範如何使用攔截器實作日誌記錄、錯誤處理和權限控制：

package main

import (
    "net/http"

    "github.com/gorilla/mux"
    "github.com/yourusername/interceptor" // 導入攔截器包

func main() {
    r := mux.NewRouter()

    r.HandleFunc("/hello", interceptor.Logger(interceptor.ErrorHandler(handleHello)))
    r.HandleFunc("/admin", interceptor.Logger(interceptor.ErrorHandler(interceptor.AuthMiddleware(handleAdmin))))

    http.Handle("/", r)
    http.ListenAndServe(":8080", nil)
}

func handleHello(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello, World!"))
}

func handleAdmin(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Admin Panel"))
}

登入後複製

上述範例中，透過使用interceptor.Logger、interceptor.ErrorHandler和interceptor.AuthMiddleware三個攔截器，實作了對/hello和/admin兩個路由的日誌記錄、錯誤處理和權限控制。