Golang中有类似Nestjs装饰器的功能吗

Golang中有类似Nestjs装饰器的功能吗 你好，我想实现类似这样的功能：https://docs.nestjs.com/openapi/decorators 我希望能够像 NestJS 那样使用装饰器来处理身份验证等功能。 请问有相关的包吗？

2 回复

在 Go 语言中，等效的模式是编写一个中间件函数。

请参阅 https://www.alexedwards.net/blog/making-and-using-middleware

更多关于Golang中有类似Nestjs装饰器的功能吗的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在Go语言中，虽然没有像TypeScript装饰器那样的原生语法支持，但可以通过反射和代码生成来实现类似的功能。以下是一个使用github.com/gin-gonic/gin和自定义中间件模拟装饰器行为的示例：

package main

import (
    "net/http"
    "github.com/gin-gonic/gin"
)

// 定义装饰器函数类型
type HandlerFunc func(*gin.Context)

// 认证装饰器
func AuthDecorator(handler HandlerFunc) HandlerFunc {
    return func(c *gin.Context) {
        token := c.GetHeader("Authorization")
        if token != "valid-token" {
            c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
            c.Abort()
            return
        }
        handler(c)
    }
}

// 日志装饰器
func LogDecorator(handler HandlerFunc) HandlerFunc {
    return func(c *gin.Context) {
        c.Next()
        fmt.Printf("Request: %s %s\n", c.Request.Method, c.Request.URL.Path)
    }
}

func main() {
    r := gin.Default()

    // 使用装饰器包装处理函数
    r.GET("/protected", AuthDecorator(func(c *gin.Context) {
        c.JSON(http.StatusOK, gin.H{"message": "Access granted"})
    }))

    r.GET("/public", LogDecorator(func(c *gin.Context) {
        c.JSON(http.StatusOK, gin.H{"message": "Public endpoint"})
    }))

    r.Run(":8080")
}

对于更接近NestJS风格的实现，可以考虑以下库：

  1. go-annotation：通过代码生成实现注解功能
// 使用go:generate指令生成代码
//go:generate go-annotation -t route

// @Route(path="/users", method="GET")
func GetUsers(c *gin.Context) {
    // 处理逻辑
}
  1. go-decorator：提供装饰器模式的实现
import "github.com/dengsgo/go-decorator/decor"

// 使用装饰器注解
//go:decor-lint use:AuthRequired
func handler(c *gin.Context) {
    // 业务逻辑
}
  1. 自定义结构体标签：利用反射处理结构体标签
type UserController struct {
    Path string `route:"/users"`
}

func (uc *UserController) RegisterRoutes(r *gin.Engine) {
    v := reflect.ValueOf(uc)
    t := v.Type()
    
    for i := 0; i < t.NumMethod(); i++ {
        method := t.Method(i)
        if route, ok := method.Tag.Lookup("route"); ok {
            r.GET(route, v.Method(i).Interface().(func(*gin.Context)))
        }
    }
}

这些方法都能在Go中实现类似装饰器的功能，但需要根据具体需求选择合适的方案。

回到顶部