Gin项目实战07-middleware（接口设置超时时间）及原理剖析

原创

已于 2024-03-08 17:56:33 修改 · 1.3k 阅读

11 ·

CC 4.0 BY-SA版权

文章标签：

#gin #go #golang

于 2024-03-05 16:06:00 首次发布

本篇博客从接口超时middleware 的案例入手深入剖析其原理

1、接口超时中间件

在项目根目录创建middleware 目录，然后在middleware目录创建time_out.go，代码如下

package middleware

import (
	"github.com/gin-gonic/gin"
	"golang.org/x/net/context"
	"net/http"
	"time"
)

// TimeoutMiddleware 是一个用于设置超时时间的中间件
func TimeoutMiddleware(timeout time.Duration) gin.HandlerFunc {
   
   
	return func(c *gin.Context) {
   
   
		ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
		defer cancel()

		c.Request = c.Request.WithContext(ctx)

		response := make(chan bool, 1)

		go func() {
   
   
			c.Next