package middleware_http

import (
	"errors"
	"fmt"
	"gitee.com/os-lee/easy-paas/common"
	"gitee.com/os-lee/easy-paas/common/e"
	"gitee.com/os-lee/easy-paas/common/flow"
	"github.com/gin-gonic/gin"
	"github.com/sony/gobreaker/v2"
)

// FlowReqBreaker 是一个Gin中间件，它使用gobreaker来实现熔断器。
func FlowReqBreaker() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 调用下一个处理函数，并将其包装在熔断器中
		_, err := flow.BreakerHandler.Execute(func() ([]byte, error) {
			c.Next()
			// c.AbortWithError(e.ERROR, err) 添加的错误
			if len(c.Errors) != 0 {
				// 检测到后面中间件的错误，返回错误，让断路器统计
				return nil, errors.New(fmt.Sprintf("BreakerHandler捕获错误: %v", c.Errors))
			}
			return nil, nil
		})
		// 判断错误是断路器被打开，直接返回结果，不代理服务了
		if errors.Is(err, gobreaker.ErrOpenState) {
			common.ResponseError(c, e.ErrGateway, err)
			c.Abort()
			return
		}
	}
}

// errToError 将panic的类型转换为gin.Error类型
func errToError(err interface{}) error {
	switch v := err.(type) {
	case string:
		return errors.New(v)
	case error:
		return v
	default:
		return errors.New("unknown error")
	}
}