基於gin的golang web開發:模型綁定

在前兩篇文章介紹路由的時候,我們了解到gin可用通過類似DefaultQuery或DefaultPostForm等方法獲取到前端提交過來的參數。參數不多的情況下也很好用,但是想想看,如果介面有很多個參數的時候再用這種方法就要調用很多次獲取參數的方法,本文將介紹一種新的接收參數的方法來解決這個問題:模型綁定。

gin中的模型綁定可以理解為:把請求的參數映射為一個具體的類型。gin支援JSON,XML,YAML和表單參數等多種參數格式,只需要在對應的欄位上聲明標籤。

綁定表單或者查詢字元串

type Person struct {
	Name    string `form:"name"`
	Address string `form:"address"`
}

func startPage(c *gin.Context) {
	var person Person
	if c.ShouldBindQuery(&person) == nil {
		log.Println(person.Name)
		log.Println(person.Address)
	}
	c.String(200, "Success")
}

在結構體Name欄位聲明form標籤,並調用ShouldBindQuery方法,gin會為我們綁定查詢字元串中的name和address兩個參數。注意雖然我們聲明了form標籤,ShouldBindQuery只綁定查詢字元串中的參數。

如果你想綁定表單中的參數的話結構體不用改變,需要把ShouldBindQuery方更改為ShouldBind方法。ShouldBind方法會區分GET和POST請求,如果是GET請求綁定查詢字元串中的參數,如果是POST請求綁定表單參數中的內容,但是不能同時綁定兩種參數。

綁定json參數

type Person struct {
	Name    string `json:"name"`
	Address string `json:"address"`
}

func startPage(c *gin.Context) {
	var person Person
	if c.ShouldBind(&person) == nil {
		log.Println(person.Name)
		log.Println(person.Address)
	}
	c.String(200, "Success")
}

json是一種常用的數據交換格式,尤其是在和web前端頁面交互的時候,似乎已經成為了一種事實標準。gin綁定json格式數據方法很簡單,只需要設置欄位的標籤為json並且調用ShouldBind方法。

其他類型參數綁定

路由參數在綁定時設置標籤為uri,並調用ShouldBindUri方法。

type Person struct {
	Id    string `uri:"id"`
}

func startPage(c *gin.Context) {
	var person Person
	if c.ShouldBindUri(&person) == nil {
		log.Println(person.Id)
	}
	c.String(200, "Success")
}

綁定在HTTP Header中的參數,欄位的標籤設置為header,調用方法為ShouldBindHeader。

還有不太常用的數組參數是欄位標籤設置為form:”colors[]”,結構體例子如下:

type myForm struct {
    Colors []string `form:"colors[]"`
}

文件上傳這種場景我很少用模型綁定的方式獲取參數,在gin中對於這種場景也提供了模型綁定支援。

type ProfileForm struct {
	Name   string                `form:"name"`
	Avatar *multipart.FileHeader `form:"avatar"`
	// Avatars []*multipart.FileHeader `form:"avatar"` 多文件上傳
}

func main() {
	router := gin.Default()
	router.POST("/profile", func(c *gin.Context) {
		var form ProfileForm
		if err := c.ShouldBind(&form); err != nil {
			c.String(http.StatusBadRequest, "bad request")
			return
		}

		err := c.SaveUploadedFile(form.Avatar, form.Avatar.Filename)
		if err != nil {
			c.String(http.StatusInternalServerError, "unknown error")
			return
		}

		c.String(http.StatusOK, "ok")
	})
	router.Run(":8080")
}

多種類型的模型綁定

如果我們有一個UpdateUser介面,PUT /user/:id,參數是{“nickname”: “nickname…”,”mobile”: “13322323232”}。程式碼如下:

type ProfileForm struct {
	Id       int    `uri:"id"`
	Nickname string `json:"nickname"` // 昵稱
	Mobile   string `json:"mobile"`   // 手機號
}

func main() {
	router := gin.Default()
	router.GET("/user/:id", func(c *gin.Context) {
		var form ProfileForm
		if err := c.ShouldBindUri(&form); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}

		if err := c.ShouldBindJSON(&form); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}

		c.String(http.StatusOK, "ok")
	})
	router.Run(":8080")
}

程式碼里調用了兩次bind方法才獲取到全部的參數。和gin社區溝通之後發現目前還不能調用一個方法同時綁定多個參數來源,當前gin版本為1.6.x,不知道未來會不會提供這種功能。

文章出處:基於gin的golang web開發:模型綁定

Tags: