go fiber:路由中间件

一，目录结构:

二，代码

1,中间件代码

package middleware

import (
	"fmt"
	"github.com/gofiber/fiber/v2"
	"industry/config"
)

// token校验
func CheckUser(c *fiber.Ctx) error {
	token:=c.Query("token")
	fmt.Println("token:",token)
	if token == "" {
		return c.Status(200).JSON(config.Error("参数token不存在"))
	} else {
		if token == "123" {
			var userId = 4
			c.Locals("userId", userId)
		} else {
			return c.Status(200).JSON(config.Error("参数token错误"))
		}
	}
	return c.Next()
}

2,控制器

func NewUserController() *UserController {
	return &UserController{}
}

func (uc *UserController) GetUser(c *fiber.Ctx) error {

	fmt.Println("hello","user")
	userId := c.Locals("userId")
	str2 := fmt.Sprintf("%v", userId)
	fmt.Println(str2)
	// 处理获取用户的逻辑
	return c.SendString("获取用户信息userId:"+str2)
}

func (uc *UserController) CreateUser(c *fiber.Ctx) error {
	// 处理创建用户的逻辑
	userId := c.Locals("userId")
	str2 := fmt.Sprintf("%v", userId)
	fmt.Println(str2)
	return c.SendString("创建用户:"+str2)
}

3,路由:

package routes

import (
	"github.com/gofiber/fiber/v2"
	"industry/config"
	"industry/controller"
	"industry/middleware"
)

func SetupRoutes(app *fiber.App) {

	//用户模块
	userController := controller.NewUserController()
	user := app.Group("/user")
	user.Get("/info" ,middleware.CheckUser, userController.GetUser)
	user.Get("/add", userController.CreateUser)

	//找不到路径时的处理
	app.Use(func(c *fiber.Ctx) error {
		return c.Status(fiber.StatusNotFound).JSON(config.Error("不存在的访问路径"))
	})

}

也可以把中间件添加给整组，如下:

	//用户模块
	userController := controller.NewUserController()
	user := app.Group("/user",middleware.CheckUser)
	user.Get("/info" , userController.GetUser)
	user.Get("/add", userController.CreateUser)

三，测试效果: