Gin学习记录4——Controller和中间件

一. Controller

用不同的Controller可以实现业务的分类,不同类型的请求可以共用同一套中间件

1.1 单文件Controller

几乎等同于函数封装,直接将ctrl的代码写入到一个文件里然后调用:

go 复制代码
package admin

import (
	"net/http"

	"github.com/gin-gonic/gin"
)

func Index(ctx *gin.Context) {
	ctx.String(http.StatusOK, "首页")
}

调用:

go 复制代码
package main

import (
	"kanna-web/controller/admin"

	"github.com/gin-gonic/gin"
)

func main() {
	r := gin.Default()
	r.GET("/", admin.Index)
	r.Run(":80")
}

1.2 Controller类与继承

这个和类的写法也是一样的:

go 复制代码
package admin

import (
	"net/http"

	"github.com/gin-gonic/gin"
)

type UserController struct {
}

type AdminController struct {
	UserController
}

func (c AdminController) Index(ctx *gin.Context) {
	ctx.String(http.StatusOK, "首页")
}

func (c AdminController) Page(ctx *gin.Context) {
	ctx.String(http.StatusOK, "Page")
}

调用:

go 复制代码
package main

import (
	"kanna-web/controller/admin"

	"github.com/gin-gonic/gin"
)

func main() {
	r := gin.Default()
	r.GET("/", admin.AdminController{}.Index)
	r.GET("/page/", admin.AdminController{}.Page)
	r.Run(":80")
}

二. 中间件

中间件就是在RouterController之间可以执行的代码,可以给单个路由添加,也可以路由组添加,也可以全局添加。

2.1 Default or New

在我们创建路由的时候,是gin.Default(),即包含了:LoggerRecover这两个中间件。

如果不需要使用者两个中间件可以gin.New()

中间件可以拿来鉴权等各种功能。

2.2 中间件的使用

最基本的形式:

go 复制代码
func baseMid(ctx *gin.Context) {
	fmt.Println("我进入了中间件")
}

func main() {
	r := gin.Default()
	r.GET("/", baseMid, admin.AdminController{}.Index)
	r.Run(":80")
}

路由组:

go 复制代码
func main() {
	r := gin.Default()
	v1 := r.Group("/v1", baseMid)
	{
		v1.GET("/login", func(ctx *gin.Context) {
			ctx.String(http.StatusOK, "Success")
		})
	}
	r.Run(":80")
}
// or v1.Use(baseMid)

全局:

go 复制代码
func main() {
	r := gin.Default()
	r.Use(baseMid)
	v1 := r.Group("/v1")
	{
		v1.GET("/login", func(ctx *gin.Context) {
			ctx.String(http.StatusOK, "Success")
		})
	}
	r.Run(":80")
}

2.3 Next()和Abort()

Next()可以先执行Control的内容,再返回到Next()处:

go 复制代码
func baseMid(ctx *gin.Context) {
	fmt.Println("我进入了")
	ctx.Next()
	fmt.Println("我离开了")
}

func main() {
	r := gin.Default()
	r.Use(baseMid)
	r.GET("/", func(ctx *gin.Context) {
		fmt.Println("我显示了")
		ctx.String(http.StatusOK, "Success")
	})
	r.Run(":80")
}


Abort()可以结束Controller的行为:

go 复制代码
func baseMid(ctx *gin.Context) {
	fmt.Println("我进入了")
	ctx.Abort()
	fmt.Println("我离开了")
}

func main() {
	r := gin.Default()
	r.Use(baseMid)
	r.GET("/", func(ctx *gin.Context) {
		fmt.Println("我显示了")
		ctx.String(http.StatusOK, "Success")
	})
	r.Run(":80")
}


2.4 多中间件顺序

go 复制代码
func Mid1(ctx *gin.Context) {
	fmt.Println("中间件1进入")
	ctx.Next()
	fmt.Println("中间件1离开")
}

func Mid2(ctx *gin.Context) {
	fmt.Println("中间件2进入")
	ctx.Next()
	fmt.Println("中间件2离开")
}

func main() {
	r := gin.Default()
	r.Use(Mid1, Mid2)
	r.GET("/", func(ctx *gin.Context) {
		fmt.Println("我显示了")
		ctx.String(http.StatusOK, "Success")
	})
	r.Run(":80")
}
/*
中间件1进入
中间件2进入
我显示了
中间件2离开
中间件1离开
*/
go 复制代码
func Mid1(ctx *gin.Context) {
	fmt.Println("中间件1进入")
	ctx.Abort()
	fmt.Println("中间件1离开")
}

func Mid2(ctx *gin.Context) {
	fmt.Println("中间件2进入")
	ctx.Next()
	fmt.Println("中间件2离开")
}

func main() {
	r := gin.Default()
	r.Use(Mid1, Mid2)
	r.GET("/", func(ctx *gin.Context) {
		fmt.Println("我显示了")
		ctx.String(http.StatusOK, "Success")
	})
	r.Run(":80")
}
/*
中间件1进入
中间件1离开
*/

2.5 中间件与Control共享数据

可以用set*get*等函数,例如:

go 复制代码
func Mid1(ctx *gin.Context) {
	ctx.Set("uid", "kanna")
}

func main() {
	r := gin.Default()
	r.Use(Mid1)
	r.GET("/", func(ctx *gin.Context) {
		uid := ctx.GetString("uid")
		ctx.String(http.StatusOK, uid)
	})
	r.Run(":80")
}
相关推荐
架构师那点事儿5 小时前
golang 用unsafe 无所畏惧,但使用不得到会panic
架构·go·掘金技术征文
RationalDysaniaer21 小时前
Gin入门笔记
笔记·gin
于顾而言21 小时前
【笔记】Go Coding In Go Way
后端·go
qq_1728055921 小时前
GIN 反向代理功能
后端·golang·go
千年死缓21 小时前
gin中间件
中间件·gin
follycat1 天前
2024强网杯Proxy
网络·学习·网络安全·go
OT.Ter1 天前
【力扣打卡系列】单调栈
算法·leetcode·职场和发展·go·单调栈
探索云原生1 天前
GPU 环境搭建指南:如何在裸机、Docker、K8s 等环境中使用 GPU
ai·云原生·kubernetes·go·gpu
OT.Ter1 天前
【力扣打卡系列】移动零(双指针)
算法·leetcode·职场和发展·go
码财小子2 天前
k8s 集群中 Golang pprof 工具的使用
后端·kubernetes·go