package main

import (
	"time"

	"github.com/gin-gonic/gin"
)

func main() {
	Port := ":8080"
	route := gin.Default()

	v1 := route.Group("/api")

	v1.Use(
		timeoutFilterV3(1 * time.Second),
	)
	{
		v1.GET("/testTimeoutFilter", testTimeoutFilter)
	}

	route.Run(Port)
}

func testTimeoutFilter(ctx *gin.Context) {

	time.Sleep(3 * time.Second)
	ctx.JSON(200, "ok")
}

func timeoutFilterV3(t time.Duration) func(c *gin.Context) {
	return func(c *gin.Context) {
		finish := make(chan struct{})

		go func() {
			defer func() {
				if err := recover(); err != nil {
					c.JSON(500, "error")
					c.Abort()
					finish <- struct{}{}
				}
			}()
			c.Next()
			finish <- struct{}{}
		}()

		select {
		case <-time.After(t):
			c.JSON(504, "timeout")
			c.Abort()
		case <-finish:
		}
	}
}