go语言中接口限流案例

package main

import (
	"log"
	"net/http"

	"github.com/alibaba/sentinel-golang/api"
	"github.com/alibaba/sentinel-golang/core/base"
	"github.com/alibaba/sentinel-golang/core/flow"
	"github.com/gin-gonic/gin"
)

func initSentinel() error {
	err := api.InitDefault()
	if err != nil {
		return err
	}

	_, err = flow.LoadRules([]*flow.Rule{
		{
			Resource:               "rule1",
			TokenCalculateStrategy: flow.Direct,
			ControlBehavior:        flow.Reject,
			Threshold:              18,
			StatIntervalInMs:       1000,
		},
	})
	if err != nil {
		return err
	}

	return nil
}

func main() {
	err := initSentinel()
	if err != nil {
		log.Fatalf("初始化Sentinel失败: %v", err)
	}

	r := gin.Default()

	r.GET("/test", func(c *gin.Context) {
		e, b := api.Entry("rule1", api.WithTrafficType(base.Inbound))
		if b != nil {
			c.String(http.StatusTooManyRequests, "请求过多，请稍后再试")
		} else {
			defer e.Exit()
			c.String(http.StatusOK, "Hello, World!")
		}
	})

	r.Run(":8080")
}