Gin 极速入门 | 青训营

142 阅读13分钟

认识 Gin

框架是一系列工具的集合,能让开发变的便捷。

学习框架的目的就是为了提供项目的开发效率,使我们更加专注业务,而不是和业务无关的底层代码。

Go 流行的 Web 框架

如果学习过其他语言,可能知道Java用的比较多的是Spring框架,PHP用的比较多的是Laravel,python用的多的是Django,都在各自的语言中具有强大的统治力。

Go 从诞生之初就带有浓重的开源属性,其原生库已经很强大,即使不依赖框架,也能进行高性能开发,又因为其语言并没有一定的设计标准,所以较为灵活,也就诞生了众多的框架,各具有特色,满足不同的喜好。

Gin

地址:github.com/gin-gonic/g…

号称最快的go语言web框架,目前是go官方的推荐框架(go.dev/doc/tutoria…

iris

地址:github.com/kataras/iri…

性能比gin高一些,支持MVC,但这款框架评价不太好,使用上问题较多,近些年很少去选择使用

Beego

地址:github.com/beego/beego

国人开发,最早的go web框架之一,工具集比较完善,性能较差,据传言作者是php转行,所以框架带有浓厚的php特色,早期国内使用的多,目前少有人选择

fiber

地址:github.com/gofiber/fib…

2020年发布的框架,发展迅速,建立在fasthttp之上,性能目前最高,受Express启发,比较简洁,上手较快,和gin类似。

当然还有其他一些框架,但从star数上,以及流行程度上看,gin一骑绝尘,gin的好处在于其简洁,扩展性,稳定性以及性能都比较出色。

Go的框架其实是可以理解为库,并不是用了某一个框架就不能用别的框架,可以选择性的使用各个库中的优秀组件,进行组合

Gin 介绍

  • 快速:基于 Radix 树的路由,小内存占用。没有反射。可预测的 API 性能。
  • 支持中间件:传入的 HTTP 请求可以由一系列中间件和最终操作来处理。 例如:Logger,Authorization,GZIP,最终操作 DB。
  • Crash 处理:Gin 可以 catch 一个发生在 HTTP 请求中的 panic 并 recover 它。这样,你的服务器将始终可用。例如,你可以向 Sentry 报告这个 panic!
  • JSON 验证:Gin 可以解析并验证请求的 JSON,例如,检查所需值的存在。
  • 路由组:更好地组织路由。是否需要授权,不同的 API 版本…… 此外,这些组可以无限制地嵌套而不会降低性能。
  • 错误管理:Gin 提供了一种方便的方法来收集 HTTP 请求期间发生的所有错误。最终,中间件可以将它们写入日志文件,数据库并通过网络发送。
  • 内置渲染:Gin 为 JSON,XML 和 HTML 渲染提供了易于使用的 API。
  • 可扩展性:新建一个中间件非常简单。

快速入门

要求

  • Go 1.13及以上版本

安装

mkdir GinStudy
cd GinStudy
go work init
mkdir helloworld
cd helloworld
go mod init test.com/helloworld
go: creating new go.mod: module test.com/helloworld
cd ..
go work use ./helloworld

Goland打开:

cd helloworld
go get -u github.com/gin-gonic/gin

示例程序:

package main
​
import (
  "github.com/gin-gonic/gin"
  "log"
)
​
func main() {
​
  r := gin.Default()
  // curl http://localhost:8080/hello GET 获取到json返回值{"name":"hello world"}
  r.GET("/hello", func(ctx *gin.Context) {
    // 返回数组、map、list、结构体都可以
    ctx.JSON(200, gin.H{
      "name": "hello world",
    })
  })
  err := r.Run(":8080")
  if err != nil {
    log.Fatalln(err)
  }
}

测试:

路由

路由是URI到函数的映射。

一个URI含: http://localhost:8080/user/find?id=11

  • 协议:比如 http,https 等
  • ip端口或者域名:比如 127.0.0.1:8080 或者www.test.com
  • path:比如 /path
  • query:比如 ?query

同时访问的时候,还需要指明HTTP METHOD,比如:

  • GET:GET方法请求一个指定资源的表示形式. 使用GET的请求应该只被用于获取数据
  • POST:POST方法用于将实体提交到指定的资源,通常会导致在服务器上的状态变化
  • HEAD:HEAD方法请求一个与GET请求的响应相同的响应,但没有响应体
  • PUT:PUT方法用请求有效载荷替换目标资源的所有当前表示
  • DELETE:DELETE方法删除指定的资源
  • CONNECT:CONNECT方法建立一个到由目标资源标识的服务器的隧道
  • OPTIONS:OPTIONS方法用于描述目标资源的通信选项
  • TRACE:TRACE方法沿着到目标资源的路径执行一个消息环回测试
  • PATCH:PATCH方法用于对资源应用部分修改

使用的时候,应该尽量遵循其语义

RESTful API 规范

RESTful API 的规范建议我们使用特定的HTTP方法来对服务器上的资源进行操作。比如:

  1. GET,表示读取服务器上的资源
  2. POST,表示在服务器上创建资源
  3. PUT,表示更新或者替换服务器上的资源
  4. DELETE,表示删除服务器上的资源
  5. PATCH,表示更新/修改资源的一部分

请求方法

r.GET("/get", func(ctx *gin.Context) {
    ctx.JSON(200, "get")
})
r.POST("/post", func(ctx *gin.Context) {
    ctx.JSON(200, "post")
})
r.DELETE("/delete", func(ctx *gin.Context) {
    ctx.JSON(200, "delete")
})
r.PUT("/put", func(ctx *gin.Context) {
    ctx.JSON(200, "put")
})

如果想要支持所有:

r.Any("/any", func(ctx *gin.Context) {
    ctx.JSON(200, "any")
})

如果想要支持其中的几种:

r.GET("/hello", func(ctx *gin.Context) {
    //数组 map list 结构体
    ctx.JSON(200, gin.H{
      "name": "hello world",
    })
})
r.POST("/hello", func(ctx *gin.Context) {
    //数组 map list 结构体
    ctx.JSON(200, gin.H{
      "name": "hello world",
    })
})

URI

URI书写的时候,我们不需要关心scheme和authority这两部分,我们主要通过path和query两部分的书写来进行资源的定位。

  • 静态url,比如/hello/user/find
r.POST("/user/find", func(ctx *gin.Context) {})
  • 路径参数,比如/user/find/:id
r.POST("/user/find/:id", func(ctx *gin.Context) {
    param := ctx.Param("id")
    ctx.JSON(200, param)
})
  • 模糊匹配,比如/user/*path
r.POST("/user/*path", func(ctx *gin.Context) {
    param := ctx.Param("path")
    ctx.JSON(200, param)
})

处理函数

定义:

type HandlerFunc func(*Context)

通过上下文的参数,获取http的请求参数,响应http请求等。

分组路由

在进行开发的时候,我们往往要进行模块的划分,比如用户模块,以user开发,商品模块,以goods开头。或者进行多版本开发,不同版本之间路径是一致的,这种时候,就可以用到分组路由了。比如:

ug := r.Group("/user"){
    ug.GET("find", func(ctx *gin.Context) {
      ctx.JSON(200, "user find")
    })
    ug.POST("save", func(ctx *gin.Context) {
      ctx.JSON(200, "user save")
    })
}
​
gg := r.Group("/goods"){
    gg.GET("find", func(ctx *gin.Context) {
      ctx.JSON(200, "goods find")
    })
    gg.POST("save", func(ctx *gin.Context) {
      ctx.JSON(200, "goods save")
    })
}

请求参数

Get 请求参数

使用Get请求传参时,类似于这样 http://localhost:8080/user/save?id=11&name=zhangsan,如何获取呢?

普通参数

request url: http://localhost:8080/user/save?id=11&name=zhangsan

r.GET("/user/save", func(ctx *gin.Context) {
    id := ctx.Query("id")
    name := ctx.Query("name")
    ctx.JSON(200, gin.H{
      "id":   id,
      "name": name,
    })
})

如果参数不存在,就给一个默认值:

r.GET("/user/save", func(ctx *gin.Context) {
    id := ctx.Query("id")
    name := ctx.Query("name")
    address := ctx.DefaultQuery("address", "北京")
    ctx.JSON(200, gin.H{
      "id":      id,
      "name":    name,
      "address": address,
    })
})

判断参数是否存在:

r.GET("/user/save", func(ctx *gin.Context) {
    id, ok := ctx.GetQuery("id")
    address, aok := ctx.GetQuery("address")
    ctx.JSON(200, gin.H{
      "id":      id,
      "idok":    ok,
      "address": address,
      "aok":     aok,
    })
})

id是数值类型,上述获取的都是string类型,根据类型获取:(结构体中要加tag噢)

type User struct {
  Id   int64  `form:"id"`
  Name string `form:"name"`
}
r.GET("/user/save", func(ctx *gin.Context) {
    var user User
    err := ctx.BindQuery(&user)
    if err != nil {
      log.Println(err)
    }
    ctx.JSON(200, user)
})

也可以:

r.GET("/user/save", func(ctx *gin.Context) {
    var user User
    err := ctx.ShouldBindQuery(&user)
    if err != nil {
      log.Println(err)
    }
    ctx.JSON(200, user)
})

区别:

type User struct {
  Id      int64  `form:"id"`
  Name    string `form:"name"`
  Address string `form:"address" binding:"required"`
}
  • 当bind是必须的时候,ShouldBindQuery会报错,开发者自行处理,状态码不变。
  • BindQuery则报错的同时,会将状态码改为400。所以一般建议是使用Should开头的bind

数组参数

请求url:http://localhost:8080/user/save?address=Beijing&address=shanghai

r.GET("/user/save", func(ctx *gin.Context) {
    address := ctx.QueryArray("address")
    ctx.JSON(200, address)
})
r.GET("/user/save", func(ctx *gin.Context) {
    address, ok := ctx.GetQueryArray("address")
    fmt.Println(ok)
    ctx.JSON(200, address)
})
r.GET("/user/save", func(ctx *gin.Context) {
    var user User
    err := ctx.ShouldBindQuery(&user)
    fmt.Println(err)
    ctx.JSON(200, user)
})

map 参数

请求url:http://localhost:8080/user/saveaddressMap[home]=Beijing&addressMap[company]=shanghai

r.GET("/user/save", func(ctx *gin.Context) {
    addressMap := ctx.QueryMap("addressMap")
    ctx.JSON(200, addressMap)
})
r.GET("/user/save", func(ctx *gin.Context) {
    addressMap, _ := ctx.GetQueryMap("addressMap")
    ctx.JSON(200, addressMap)
})

map参数,bind并没有支持

Post请求参数

post请求一般是表单参数和json参数。

表单参数

r.POST("/user/save", func(ctx *gin.Context) {
    id := ctx.PostForm("id")
    name := ctx.PostForm("name")
    address := ctx.PostFormArray("address")
    addressMap := ctx.PostFormMap("addressMap")
    ctx.JSON(200, gin.H{
      "id":         id,
      "name":       name,
      "address":    address,
      "addressMap": addressMap,
    })
})
r.POST("/user/save", func(ctx *gin.Context) {
    var user User
    err := ctx.ShouldBind(&user)
    addressMap, _ := ctx.GetPostFormMap("addressMap")
    user.AddressMap = addressMap
    fmt.Println(err)
    ctx.JSON(200, user)
})

不支持map绑定

json 参数

{
    "id":1111,
    "name":"zhangsan",
    "address": [
        "beijing",
        "shanghai"
    ],
    "addressMap":{
        "home":"beijing"
    }
}
r.POST("/user/save", func(ctx *gin.Context) {
    var user User
    err := ctx.ShouldBindJSON(&user)
    fmt.Println(err)
    ctx.JSON(200, user)
})

其他类型参数注入xml,yaml等和json道理一样

路径参数

请求url:http://localhost:8080/user/save/111

r.POST("/user/save/:id", func(ctx *gin.Context) {
    ctx.JSON(200, ctx.Param("id"))
})

如果需要使用ctx.ShouldBindUri,则需要绑定tag

文件参数

r.POST("/user/save", func(ctx *gin.Context) {
    form, err := ctx.MultipartForm()
    if err != nil {
      log.Println(err)
    }
    files := form.File
    for _, fileArray := range files {
      for _, v := range fileArray {
        ctx.SaveUploadedFile(v, "./"+v.Filename)
      }
    }
    ctx.JSON(200, form.Value)
})

响应

字符串方式

r.GET("/user/save", func(ctx *gin.Context) {
    ctx.String(http.StatusOK, "this is a %s", "ms string response")
})

JSON 方式

r.GET("/user/save", func(ctx *gin.Context) {
    ctx.JSON(http.StatusOK, gin.H{
      "success": true,
    })
})

XML 方式

type XmlUser struct {
  Id   int64  `xml:"id"`
  Name string `xml:"name"`
}
r.GET("/user/save", func(ctx *gin.Context) {
    u := XmlUser{
      Id:   11,
      Name: "zhangsan",
    }
    ctx.XML(http.StatusOK, u)
})

文件格式

r.GET("/user/save", func(ctx *gin.Context) {
    //ctx.File("./1.png")
    ctx.FileAttachment("./1.png", "2.png")  // 自定义名字
})

设置 http 响应头

r.GET("/user/save", func(ctx *gin.Context) {
    ctx.Header("test", "headertest")
})

重定向

r.GET("/user/save", func(ctx *gin.Context) {
    ctx.Redirect(http.StatusMovedPermanently, "http://www.baidu.com")
})

YAML 方式

r.GET("/user/save", func(ctx *gin.Context) {
    ctx.YAML(200, gin.H{"name": "ms", "age": 19})
})

模版渲染

模板是 golang 语言的一个标准库,使用场景很多,gin框架同样支持模板。

基本使用

定义一个存放模板文件的templates文件夹:

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <title>gin_templates</title>
</head>
<body>
{{.title}}
</body>
</html>

后端代码:

package main
​
import (
  "github.com/gin-gonic/gin"
  "net/http"
)
​
func main() {
  r := gin.Default()
  // 模板解析
  r.LoadHTMLFiles("templates/index.tmpl")
​
  r.GET("/index", func(c *gin.Context) {
    // HTML请求
    // 模板的渲染
    c.HTML(http.StatusOK, "index.tmpl", gin.H{
      "title": "hello 模板",
    })
  })
  r.Run(":9090") // 启动server
}

多个模板渲染

如果有多个模板,可以统一进行渲染:

// 模板解析,解析templates目录下的所有模板文件
r.LoadHTMLGlob("templates/**")

如果目录为templates/post/index.tmpltemplates/user/index.tmpl这种,可以:

// **/* 代表所有子目录下的所有文件
r.LoadHTMLGlob("templates/**/*")

自定义模板函数

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <title>gin_templates</title>
</head>
<body>
{{.title | safe}}
</body>
</html>
// gin框架给模板添加自定义函数
r.SetFuncMap(template.FuncMap{
    "safe": func(str string) template.HTML {
      return template.HTML(str)
    },
})
​
// 模板解析,解析templates目录下的所有模板文件
r.LoadHTMLGlob("templates/**")
​
r.GET("/index", func(c *gin.Context) {
    // HTML请求
    // 模板的渲染
    c.HTML(http.StatusOK, "index.tmpl", gin.H{
      "title": "<a href='http://baidu.com'>跳转到其他地方</a>",
    })
})

静态文件处理

如果在模板中引入静态文件,比如样式文件 index.css

body{
    background-color: aqua;
}
<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <title>gin_templates</title>
    <link rel="stylesheet" href="/css/index.css">
</head>
<body>
{{.title}}
</body>
</html>
// 加载静态文件
r.Static("/css", "./static/css")

会话

会话控制涉及到cookie和session的使用。

cookie

  1. HTTP是无状态协议,服务器不能记录浏览器的访问状态,也就是说服务器不能区分两次请求是否由同一个客户端发出
  2. Cookie就是解决HTTP协议无状态的方案之一
  3. Cookie实际上就是服务器保存在浏览器上的一段信息。浏览器有了Cookie之后,每次向服务器发送请求时都会同时将该信息发送服务器,服务器收到请求后,就可以根据该信息处理请求
  4. Cookie由服务器创建,并发送给浏览器,最终由浏览器保存

设置 cookie

func (c *Context) SetCookie(name, value string, maxAge int, path, domain string, secure, httpOnly bool)

参数说明:

参数名类型说明
namestringcookie名字
valuestringcookie值
maxAgeint有效时间,单位是秒,MaxAge=0 忽略MaxAge属性,MaxAge<0 相当于删除cookie, 通常可以设置-1代表删除,MaxAge>0 多少秒后cookie失效
pathstringcookie路径
domainstringcookie作用域
secureboolSecure=true,那么这个cookie只能用https协议发送给服务器
httpOnlybool设置HttpOnly=true的cookie不能被js获取到
r.GET("/cookie", func(c *gin.Context) {
    // 设置cookie
    c.SetCookie("site_cookie", "cookievalue", 3600, "/", "localhost", false, true)
})

读取 cookie

r.GET("/read", func(c *gin.Context) {
    // 根据cookie名字读取cookie值
    data, err := c.Cookie("site_cookie")
    if err != nil {
      c.String(200,"not found!")
      return
    }
    // 直接返回cookie值
    c.String(200,data)
})

删除 cookie

通过将cookie的MaxAge设置为-1, 达到删除cookie的目的。

r.GET("/del", func(c *gin.Context) {
    // 设置cookie  MaxAge设置为-1,表示删除cookie
    c.SetCookie("site_cookie", "cookievalue", -1, "/", "localhost", false, true)
    c.String(200,"删除cookie")
})

Session

在Gin框架中,我们可以依赖gin-contrib/sessions中间件处理session。

安装session包

go get github.com/gin-contrib/sessions
package main
​
import (
  "fmt"
  "github.com/gin-contrib/sessions"
  "github.com/gin-contrib/sessions/cookie"
  "github.com/gin-gonic/gin"
)
​
func main() {
  r := gin.Default()
  // 创建基于cookie的存储引擎,secret 参数是用于加密的密钥
  store := cookie.NewStore([]byte("secret"))
  // 设置session中间件,参数mysession,指的是session的名字,也是cookie的名字
  // store是前面创建的存储引擎,我们可以替换成其他存储引擎
  r.Use(sessions.Sessions("mysession", store))
​
  r.GET("/hello", func(c *gin.Context) {
    // 初始化session对象
    session := sessions.Default(c)
    // 通过session.Get读取session值
    // session是键值对格式数据,因此需要通过key查询数据
    if session.Get("hello") != "world" {
      fmt.Println("没读到")
      // 设置session数据
      session.Set("hello", "world")
      session.Save()
    }
    c.JSON(200, gin.H{"hello": session.Get("hello")})
  })
  r.Run(":8080")
}

多 session

package main
​
import (
  "github.com/gin-contrib/sessions"
  "github.com/gin-contrib/sessions/cookie"
  "github.com/gin-gonic/gin"
)
​
func main() {
  r := gin.Default()
  store := cookie.NewStore([]byte("secret"))
  sessionNames := []string{"a", "b"}
  r.Use(sessions.SessionsMany(sessionNames, store))
​
  r.GET("/hello", func(c *gin.Context) {
    sessionA := sessions.DefaultMany(c, "a")
    sessionB := sessions.DefaultMany(c, "b")
​
    if sessionA.Get("hello") != "world!" {
      sessionA.Set("hello", "world!")
      sessionA.Save()
    }
​
    if sessionB.Get("hello") != "world?" {
      sessionB.Set("hello", "world?")
      sessionB.Save()
    }
​
    c.JSON(200, gin.H{
      "a": sessionA.Get("hello"),
      "b": sessionB.Get("hello"),
    })
  })
  r.Run(":8080")
}

基于 redis 存储引擎的 session

如果我们想将 session 数据保存到 redis 中,只要将 session 的存储引擎改成 redis 即可。

使用 redis 作为存储引擎的例子:

首先安装 redis 存储引擎的包:

go get github.com/gin-contrib/sessions/redis
package main
​
import (
  "github.com/gin-contrib/sessions"
  "github.com/gin-contrib/sessions/redis"
  "github.com/gin-gonic/gin"
)
​
func main() {
  r := gin.Default()
  // 初始化基于redis的存储引擎
  // 参数说明:
  //    第1个参数 - redis最大的空闲连接数
  //    第2个参数 - 数通信协议tcp或者udp
  //    第3个参数 - redis地址, 格式,host:port
  //    第4个参数 - redis密码
  //    第5个参数 - session加密密钥
  store, _ := redis.NewStore(10, "tcp", "localhost:6379", "", []byte("secret"))
  r.Use(sessions.Sessions("mysession", store))
​
  r.GET("/incr", func(c *gin.Context) {
    session := sessions.Default(c)
    var count int
    v := session.Get("count")
    if v == nil {
      count = 0
    } else {
      count = v.(int)
      count++
    }
    session.Set("count", count)
    session.Save()
    c.JSON(200, gin.H{"count": count})
  })
  r.Run(":8080")
}

中间件

在Gin框架中,中间件(Middleware)指的是可以拦截http请求-响应生命周期的特殊函数,在请求-响应生命周期中可以注册多个中间件,每个中间件执行不同的功能,一个中间执行完再轮到下一个中间件执行。

中间件的常见应用场景如下:

  • 请求限速
  • api接口签名处理
  • 权限校验
  • 统一错误处理

Gin支持设置全局中间件和针对路由分组设置中间件,设置全局中间件意思就是会拦截所有请求,针对分组路由设置中间件,意思就是仅对这个分组下的路由起作用。

中间件使用

r := gin.New()
  // 通过use设置全局中间件
  // 设置日志中间件,主要用于打印请求日志
  r.Use(gin.Logger())
  // 设置Recovery中间件,主要用于拦截panic错误,不至于导致进程崩掉
  r.Use(gin.Recovery())
  r.GET("/test", func(ctx *gin.Context) {
    panic(errors.New("test error"))
  })
  r.Run(":8080")

自定义中间件

使用 Use 可以使用 Gin 自带的中间件或者其他第三方中间件,也可以自己开发中间件。

package main
// 导入gin包
import (
  "github.com/gin-gonic/gin"
  "log"
  "time"
)
​
// 自定义个日志中间件
func Logger() gin.HandlerFunc {
  return func(c *gin.Context) {
    t := time.Now()
​
    // 可以通过上下文对象,设置一些依附在上下文对象里面的键/值数据
    c.Set("example", "12345")
​
    // 在这里处理请求到达控制器函数之前的逻辑
     
    // 调用下一个中间件,或者控制器处理函数,具体得看注册了多少个中间件。
    c.Next()
​
    // 在这里可以处理请求返回给用户之前的逻辑
    latency := time.Since(t)
    log.Print(latency)
​
    // 例如,查询请求状态吗
    status := c.Writer.Status()
    log.Println(status)
  }
}
​
func main() {
  r := gin.New()
  // 注册上面自定义的日志中间件
  r.Use(Logger())
​
  r.GET("/test", func(c *gin.Context) {
    // 查询我们之前在日志中间件,注入的键值数据
    example := c.MustGet("example").(string)
    // it would print: "12345"
    log.Println(example)
  })
​
  // Listen and serve on 0.0.0.0:8080
  r.Run(":8080")
}