Go-web(三)Gin参数校验&cookie&session

￡神魔★判官ぃ 2023-01-19 07:19 5阅读 0赞

### 文章目录 ###

*  Go-web(三)Gin参数校验&cookie&session
 *   *  标签校验
     *  自定义校验
     *  cookie
     *  session

# Go-web(三)Gin参数校验&cookie&session #

## 标签校验 ##

gin中对于参数校验提供了非常方便的标签校验，即在定义接收结构体时即可通过添加标签进行校验

// 参数校验通过绑定tag实现
    type ValiadStruct struct {
        
    	// required：非空校验
    	Name string `form:"name" json:"name" binding:"required"`
    	// min，max最大最小值校验
    	Age int `form:"age" json:"age" binding:"required,min=0,max=100"`
    }

在进行接受绑定时，gin会进行校验，不符合则直接报错：

// gin参数校验
    	r.POST("/vaild", func(context *gin.Context) {
        
    		var valid ValiadStruct
    		if err := context.Bind(&valid);err != nil{
        
    			context.JSON(http.StatusBadRequest, gin.H{
        "error": err.Error()})
    			return
    		}
    		fmt.Println("name: ",valid.Name," age: ",valid.Age)
    		context.JSON(200,gin.H{
        
    			"vaild":valid,
    		})
    	})

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTkyMjI4OQ_size_16_color_FFFFFF_t_70]

## 自定义校验 ##

go get gopkg.in/go-playground/validator.v8

*  自定义校验方法
 *  注册校验方法，key就是在结构体中使用的标签

import (
        "net/http"
        "reflect"
        "github.com/gin-gonic/gin"
        "github.com/gin-gonic/gin/binding"
        "gopkg.in/go-playground/validator.v8"
    )
    
    /*
        对绑定解析到结构体上的参数，自定义验证功能
        比如我们要对 name 字段做校验，要不能为空，并且不等于 admin ，类似这种需求，就无法 binding 现成的方法
        需要我们自己验证方法才能实现 官网示例（https://godoc.org/gopkg.in/go-playground/validator.v8#hdr-Custom_Functions）
        这里需要下载引入下 gopkg.in/go-playground/validator.v8
    */
    type Person struct {
        
        Age int `form:"age" binding:"required,gt=10"`
        // 2、在参数 binding 上使用自定义的校验方法函数注册时候的名称
        Name    string `form:"name" binding:"NotNullAndAdmin"`
        Address string `form:"address" binding:"required"`
    }
    
    // 1、自定义的校验方法
    func nameNotNullAndAdmin(v *validator.Validate, topStruct reflect.Value, currentStructOrField reflect.Value, field reflect.Value, fieldType reflect.Type, fieldKind reflect.Kind, param string) bool {
        
    
        if value, ok := field.Interface().(string); ok {
        
            // 字段不能为空，并且不等于  admin
            return value != "" && !("5lmh" == value)
        }
    
        return true
    }
    
    func main() {
        
        r := gin.Default()
    
        // 3、将我们自定义的校验方法注册到 validator中
        if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
        
            // 这里的 key 和 fn 可以不一样最终在 struct 使用的是 key
            v.RegisterValidation("NotNullAndAdmin", nameNotNullAndAdmin)
        }
    
        /*
            curl -X GET "http://127.0.0.1:8080/testing?name=&age=12&address=beijing"
            curl -X GET "http://127.0.0.1:8080/testing?name=lmh&age=12&address=beijing"
            curl -X GET "http://127.0.0.1:8080/testing?name=adz&age=12&address=beijing"
        */
        r.GET("/5lmh", func(c *gin.Context) {
        
            var person Person
            if e := c.ShouldBind(&person); e == nil {
        
                c.String(http.StatusOK, "%v", person)
            } else {
        
                c.String(http.StatusOK, "person bind err:%v", e.Error())
            }
        })
        r.Run()
    }

## cookie ##

// 服务端接收客户端cookie，下发cookie
    	r.GET("/test_cookie", func(context *gin.Context) {
        
    		cookie,err := context.Cookie("cookie_key")
    		if err != nil{
        
    			cookie = "cookie_val"
    			// 给客户端设置cookie
    			// maxAge int, 单位为秒
    			// path,cookie所在目录
    			// domain string,域名
    			// secure 是否智能通过https访问
    			// httpOnly bool  是否允许别人通过js获取自己的cookie
    			context.SetCookie("cookie_key",cookie,60,"/","localhost",false,false)
    		}
    		context.JSON(200,cookie)
    	})

## session ##

// github.com/gorilla/sessions，session管理神奇
    var store = sessions.NewCookieStore([]byte("something-very-secret"))
    
    func saveSession(w http.ResponseWriter,r *http.Request)  {
        
    	// 获取一个session
    	session,err := store.Get(r,"session_name")
    	if err != nil{
        
    		return
    	}
    	// set session
    	session.Values["session_name"] = "session_val"
    	// set session life time
    	session.Options.MaxAge = 10
    	session.Save(r,w)
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTkyMjI4OQ_size_16_color_FFFFFF_t_70]: /images/20221021/709809101a5d4058b26c9b96ae6f64d4.png