卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章45530本站已运行3113

Golang 拦截器: 一览无余

综述:go 拦截器是通过中介软件实现的一种处理函数前后逻辑的机制。原理:拦截器作为参数传递给函数,在函数执行前后的逻辑。实战:通过创建记录请求时间和校验请求参数的拦截器,并将其链接到 http 处理程序,展示了拦截器的应用。

Golang 拦截器: 一览无余

Go 拦截器:一览无余

简介

拦截器是一种在函数执行前后做一些处理的机制。在 Golang 中,可以使用中介软件(middleware)来实现拦截器。

原理

拦截器通过将自己作为参数传递给要拦截的函数来工作。当函数被调用时,拦截器将在函数执行前后执行其逻辑。

实战案例

我们创建一个简单的 HTTP 服务器来演示如何使用拦截器:

package main

import (
    "log"
    "net/http"
)

func main() {
    // 创建一个记录请求时间的拦截器
    timeInterceptor := func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            start := time.Now()
            defer func() { log.Printf("请求处理耗时:%fs", time.Since(start).Seconds()) }()
            next.ServeHTTP(w, r)
        })
    }

    // 创建一个校验请求参数的拦截器
    paramInterceptor := func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            if r.URL.Query().Get("id") == "" {
                http.Error(w, "缺少参数:id", http.StatusBadRequest)
                return
            }
            next.ServeHTTP(w, r)
        })
    }

    // 将拦截器与 HTTP 处理程序链接
    handler := paramInterceptor(timeInterceptor(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello, World!"))
    })))

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

在这个示例中,timeInterceptor 拦截器记录请求处理时间,而 paramInterceptor 拦截器校验请求参数。通过将拦截器链接起来,我们可以在处理请求之前和之后执行这些逻辑。

总结

拦截器是增强 Golang 程序功能的强大工具。它们允许你在函数执行前后执行自定义代码,从而实现诸如记录、验证和授权等操作。

卓越飞翔博客
上一篇: 揭秘go语言反射的奥秘:三定律深入浅出
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏