【Go实战基础】gin 如何自定义和使用一个中间件

菜鸟实战

已于 2022-06-20 19:42:21 修改

阅读量746

点赞数 15

分类专栏： GO 实战基础文章标签： golang 中间件开发语言

于 2022-06-20 19:30:17 首次发布

本文链接：https://blog.csdn.net/qq_39816613/article/details/125331354

版权

GO 实战基础专栏收录该内容

23 篇文章 2 订阅

订阅专栏

本文介绍了如何在 Go 的 Gin 框架中自定义和使用中间件。通过示例展示了创建 IP 授权中间件的过程，该中间件检查客户端 IP 是否在白名单中，只有白名单内的 IP 才能通过。文章还涵盖了 gin 路由、字符串数组、数组循环、自定义中间件和 HTTP 状态码等知识点。通过运行代码，可以看到中间件成功拦截了非白名单 IP 的请求并返回了 401 状态码。

摘要由CSDN通过智能技术生成

一、简介

实战场景: gin 如何自定义和使用一个中间件

二、知识点

gin 路由

字符串数组

数组循环

自定义中间件

http 状态码

三、菜鸟实战

马上安排！

1、创建 go文件

/*
 * @Author: 菜鸟实战
 * @Description: gin 如何自定义和使用一个中间件
 */

// 知识点:
// # gin 路由
// # 字符串数组
// # 数组循环
// # 自定义中间件
// # http 状态码

package main

// 导入包
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"runtime"
)

// IP 授权中间件

func IPAuthMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 白名单
		whiteIpList := []string{
			"127.0.0.1",
		}

		flag := false

		// 获取当前访问 ip, 与白名单 ip做比较
		clientIp := c.ClientIP()
		fmt.Printf("当前 client ip 为: %s \n", clientIp)
		for _, whiteIp := range whiteIpList {
			// 验证通过
			if clientIp == whiteIp {
				flag = true
				break
			}
		}

		// 验证未通过
		if flag == false {
			c.String(401, "client ip % is not in the white ip list", clientIp)
			// 需要 abort
			c.Abort()
		}

	}

}

// 主函数
func main() {
	// 使用内置函数打印
	println("Hello", "菜鸟实战")
	println("实战场景: ", "gin 如何自定义和使用一个中间件")

	// 初始化
	r := gin.Default()

	// 使用中间件
	r.Use(IPAuthMiddleware())

	// 添加 get 路由和回调
	r.GET("/g", func(c *gin.Context) {
		// 返回的 code 和 字符串返回
		c.String(200, "这是一个使用了中间件的方法 \n")
	})

	// 使用包函数打印
	fmt.Printf("版本: %s \n", runtime.Version())

	// 启动框架程序, 默认 8080 端口
	r.Run()

}

// curl 验证方式
// bind get
// curl -X GET "http://127.0.0.1:8080/g"