首页  >  文章  >  后端开发  >  Golang 拦截器: 一览无余

Golang 拦截器: 一览无余

WBOY
WBOY原创
2024-04-07 21:24:02966浏览

综述: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 程序功能的强大工具。它们允许你在函数执行前后执行自定义代码,从而实现诸如记录、验证和授权等操作。

以上是Golang 拦截器: 一览无余的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn