gin框架提高篇（二）

自定义中间件

package main

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
	"strconv"
)

func main() {
	//r := gin.New() //没有任何中间件的路由引擎
	r := gin.Default() //默认路由引擎：Logger and Recovery middleware
	r.Use(Middleware()) //先执行中间件，相当于过滤，减少不必要的服务器性能损耗
	r.GET("/middleware", func(context *gin.Context) {
		fmt.Println("服务端开始执行...")
		name := context.Query("name")
		ageStr := context.Query("age")
		age, _ := strconv.Atoi(ageStr)
		res := struct {
			Name string `json:"name"` //添加标签
			Age  int    `json:"age"`
		}{name, age}
		context.JSON(http.StatusOK, res)
	})
	r.Run(":9090")
}

func Middleware() gin.HandlerFunc {
	return func(context *gin.Context) {
		fmt.Println("中间件开始执行...")
		name := context.Query("name")
		ageStr := context.Query("age")
		age, err := strconv.Atoi(ageStr)
		//失败拦截
		if err != nil {
			context.AbortWithStatusJSON(http.StatusBadRequest, "输入格式错误，年龄不是整数")
		}
		if len(name) < 6 || len(name) > 12 {
			context.AbortWithStatusJSON(http.StatusBadRequest, "用户名只能是6-12位")
		}
		if age < 0 || age > 100 {
			context.AbortWithStatusJSON(http.StatusBadRequest, "年龄非法")
		}
		//成功进入服务端
		context.Next()
	}
}

登录中间件

package main

import (
	"github.com/gin-gonic/gin"
	"net/http"
)

func main() {
	r := gin.Default()
	r.Use(AuthMiddleware())
	r.GET("/login", func(context *gin.Context) {
		user := context.MustGet(gin.AuthUserKey).(string)
		context.JSON(http.StatusOK, "登录成功"+"欢迎您："+user)
	})
	r.Run(":9090")
}

func AuthMiddleware() gin.HandlerFunc {
	// 静态添加
	account := gin.Accounts{
		"admin":  "adminpw",
		"system": "systempw",
	}
	// 动态添加
	account["go"] = "go123"
	account["gin"] = "gin123"
	return gin.BasicAuth(account)
}

同步异步

package main

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
	"strconv"
	"time"
)

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

	r.GET("/sync", func(context *gin.Context) {
		sync(context)
		context.JSON(http.StatusOK, ">>>主程序同步已经执行<<<")
	})

	r.GET("/async", func(context *gin.Context) {
		for i := 0; i < 6; i++ {
			cCp := context.Copy()
			go async(cCp, i) // 异步通过 go程 实现
		}
		context.JSON(http.StatusOK, ">>>主程序同步已经执行<<<")
	})

	r.Run(":9090")
}

func async(cp *gin.Context, i int) {
	fmt.Println("第" + strconv.Itoa(i) + "个 go 程序开始执行：" + cp.Request.URL.Path)
	time.Sleep(time.Second * 3)
	println("第" + strconv.Itoa(i) + "个同步任务执行完成")
}

func sync(context *gin.Context) {
	println("开始执行同步任务：" + context.Request.URL.Path)
	time.Sleep(time.Second * 3)
	println("同步任务执行完成")
}