golang gin自定义中间件的两种写法

た入场券 2023-09-28 21:45 106阅读 0赞

package main
    
    import (
    	"fmt"
    
    	"github.com/gin-gonic/gin"
    )
    
    // 第一种写法
    //func middleware1(c *gin.Context) {
        
    //
    //}
    
    // 第二种写法
    //func middleware2() gin.HandlerFunc {
        
    //	return func(c *gin.Context) {
        
    //		// *.baidu.com
    //	}
    //}
    
    func RefererMiddleware() gin.HandlerFunc {
        
    	return func(c *gin.Context) {
        
    		fmt.Println("first in....")
    		// 取到 referer
    		ref := c.GetHeader("Referer")
    		if ref == "" {
        
    			c.AbortWithStatusJSON(200, gin.H{
        
    				"msg": "非法访问",
    			})
    			return
    		}
    		c.Next()
    		fmt.Println("first done....")
    	}
    }
    
    func main() {
        
    	r := gin.Default()
    	// 第一种写法的使用
    	//r.Use(middleware1)
    	// 第二种写法的使用
    	//r.Use(middleware2())
    	r.Use(RefererMiddleware())
    	r.Use(func(c *gin.Context) {
        
    		fmt.Println("我是第二个中间件")
    		c.Next()
    		fmt.Println("second done")
    	})
    
    	r.GET("ping", func(c *gin.Context) {
        
    		c.JSON(200, gin.H{
        
    			"code": 0,
    		})
    	})
    	r.Run()
    }

发表评论取消回复

表情：

评论列表（有 0 条评论，106人围观）

还没有评论，来说两句吧...

相关阅读

相关 golang gin自定义中间件：自定义context.Request.Header

package main import ( "log" "github.com/gin-gonic/gin"

亦凉/ 2023年09月28日 21:45/ 0 赞/ 122 阅读

相关 golang gin自定义中间件的两种写法

package main import ( "fmt" "github.com/gin-gonic/gin" )

た入场券/ 2023年09月28日 21:45/ 0 赞/ 107 阅读

相关 golang gin 自定义中间件：AbortWithStatusJSON终止的使用

package main import ( "log" "github.com/gin-gonic/gin" )

忘是亡心i/ 2023年09月28日 21:44/ 0 赞/ 119 阅读

相关 golang gin内置中间件的使用

package main import "github.com/gin-gonic/gin" func main() {

雨点打透心脏的1/2处/ 2023年09月28日 21:44/ 0 赞/ 119 阅读

相关自定义中间件

我们知道在asp.net中每次请求，都要经过请求管道，依次触发管道中的一系列事件。那么我们可以这么理解，中间件是请求管道中的一个组件，可以用来拦截请求，以方便我们进行请求和响应

迈不过友情╰/ 2023年08月17日 17:14/ 0 赞/ 130 阅读

相关 node -koa 自定义中间件

koa中编写中间件，很简单：下边，我是写了一个异常捕获中间件，在app.js中使用app.use（）注册即可： // errMiddleWare .js

桃扇骨/ 2023年07月04日 12:35/ 0 赞/ 75 阅读

相关 Golang-日志中间件

restful服务中，通常需要将请求信息记录到日志中。以下例子中展示了go如何实现一个简单的日志中间件。它的基本原理是将业务函数（本例 `foo/bar`）作为参数传递给日

Bertha 。/ 2022年11月03日 15:30/ 0 赞/ 228 阅读

相关 Qt自定义控件的两种方法

在Qt开发的工程中我们往往要用到自定义一些部件，这些控件是我们需要的，所以我们做了一些特殊的处理。 Qt自定义控件的方法一是重写要自定义的控件：改进法如图我定义的工程的结

水深无声/ 2022年09月18日 08:55/ 0 赞/ 444 阅读

相关自定义django中间件实现登录

title: 自定义django中间件实现登录 date: 2018-06-06 08:11:56 tags: 中间件 > 作者:李忠林 > > G

迷南。/ 2022年05月21日 05:00/ 0 赞/ 240 阅读

相关 redux 自定义中间件

每派发一次action,就会执行设置的中间件官方记录状态和action的中间件:redux-logger; 1、在store.js中

一时失言乱红尘/ 2021年07月24日 15:17/ 0 赞/ 433 阅读