Gin框架,body参数只能读取一次

在使用gin框架的时候,发现请求的body数据只允许读取一次,针对这种情况有2种解决办法
第一步:base code:

package main

import (
	"fmt"
    "gopkg.in/gin-gonic/gin.v1"
	"net/http"
	"io/ioutil"
	"bytes"
	//"encoding/json"
)

type Person struct{
	Name string `json:"name"`
	Phone int64 `json:"phone"`
	Data string `json:"data"`
}

func main(){
    
    router := gin.Default()

    router.POST("/",Hello)
    router.Run(":8000")
}
func Hello(ctx *gin.Context){
	
	var info Person
	err := ctx.BindJSON(&info)
	if err != nil{
		fmt.Println(err.Error())
	}
	fmt.Printf("info: %#v\n",info)
	
    ctx.JSON(http.StatusOK, gin.H{
		"code":200,
		"msg":"success",
	})
   
}

随便一组请求数据,得到结果:

[GIN-debug] Listening and serving HTTP on :8000
info: main.Person{Name:"lisa", Phone:13700001234, Data:"thisIsTest"}

这个是一般情况下使用场景,如果需要加一个中间件【验证身份…】之类的

第二步:加一个中间件供能

package main

import (
	"fmt"
    "gopkg.in/gin-gonic/gin.v1"
	"net/http"
	"io/ioutil"
	"bytes"
	//"encoding/json"
)

type Person struct{
	Name string `json:"name"`
	Phone int64 `json:"phone"`
	Data string `json:"data"`
}

func main(){
    
    router := gin.Default()

    router.POST("/",HelloMiddleware(),Hello)
    router.Run(":8000")
}

func HelloMiddleware() gin.HandlerFunc {

	return func(ctx *gin.Context) {
  
		data,err := ctx.GetRawData()
		if err != nil{
			fmt.Println(err.Error())
		}
		fmt.Printf("data: %v\n",string(data)) 
		ctx.Next()
	}
	
}

func Hello(ctx *gin.Context){
	
	var info Person
	err := ctx.BindJSON(&info)
	if err != nil{
		fmt.Println(err.Error())
	}
	fmt.Printf("info: %#v\n",info)
	
    ctx.JSON(http.StatusOK, gin.H{
		"code":200,
		"msg":"success",
	})
   
}

此时运行结果:

[GIN-debug] Listening and serving HTTP on :8000
data: {"name":"lisa","phone":13700001234,"data":"thisIsTest"}
EOF
info: main.Person{Name:"", Phone:0, Data:""}

可以看出,BindJSON这一步报错:EOF,而先于Hello函数运行的中间件正常打印出读取的data数据,而info仅仅是默认值信息

解决这个问题的方法有2个:
第一种:利用gin自带函数Set

package main

import (
	"fmt"
    "gopkg.in/gin-gonic/gin.v1"
	"net/http"
	//"io/ioutil"
	//"bytes"
	"encoding/json"
)

type Person struct{
	Name string `json:"name"`
	Phone int64 `json:"phone"`
	Data string `json:"data"`
}

func main(){
    
    router := gin.Default()

    router.POST("/",HelloMiddleware(),Hello)
    router.Run(":8000")
}

func HelloMiddleware() gin.HandlerFunc {

	return func(ctx *gin.Context) {
		data,err := ctx.GetRawData()
		if err != nil{
			fmt.Println(err.Error())
		}
		fmt.Printf("data: %v\n",string(data)) 
		
    	var info Person
		json.Unmarshal(data,&info)

		ctx.Set("name",info.Name)
		ctx.Set("phone",info.Phone)
		ctx.Set("data",info.Data)
	
		ctx.Next()
	}
	
}
func Hello(ctx *gin.Context){
	name,_ := ctx.Get("name")
	phone,_ := ctx.Get("phone")
	data,_ := ctx.Get("data")
	info := Person{
		Name:name.(string),
		Phone:phone.(int64),
		Data:data.(string),
	}
	fmt.Printf("info: %#v\n",info)
}

运行结果【PS:这才是想要的结果】:

[GIN-debug] Listening and serving HTTP on :8000
data: {"name":"lisa","phone":13700001234,"data":"thisIsTest"}
info: main.Person{Name:"lisa", Phone:13700001234, Data:"thisIsTest"}

第二种:利用golang官方库,推荐这种方法解决这个问题

package main

import (
	"fmt"
    "gopkg.in/gin-gonic/gin.v1"
	"net/http"
	"io/ioutil"
	"bytes"
	"encoding/json"
)

type Person struct{
	Name string `json:"name"`
	Phone int64 `json:"phone"`
	Data string `json:"data"`
}

func main(){
    
    router := gin.Default()

    router.POST("/",HelloMiddleware(),Hello)
    router.Run(":8000")
}

func HelloMiddleware() gin.HandlerFunc {

	return func(ctx *gin.Context) {
		data,err := ctx.GetRawData()
		if err != nil{
			fmt.Println(err.Error())
		}
		fmt.Printf("data: %v\n",string(data)) 
		
		ctx.Request.Body = ioutil.NopCloser(bytes.NewBuffer(data)) // 关键点
		ctx.Next()
	}
	
}
func Hello(ctx *gin.Context){
	
	var info Person
	err := ctx.BindJSON(&info)
	if err != nil{
		fmt.Println(err.Error())
	}
	fmt.Printf("info: %#v\n",info)
	
    ctx.JSON(http.StatusOK, gin.H{
		"code":200,
		"msg":"success",
	})
   
}

运行结果【PS:有没有跟上面一样】:

[GIN-debug] Listening and serving HTTP on :8000
data: {"name":"lisa","phone":13700001234,"data":"thisIsTest"}
info: main.Person{Name:"lisa", Phone:13700001234, Data:"thisIsTest"}

欢迎关注微信公众号:
在这里插入图片描述

  • 4
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值