micro限流实现及原理

限流原理

代码实现

服务端

/*
限流相关的包：
			jujuratelimit "github.com/juju/ratelimit"
			ratelimit "github.com/micro/go-plugins/wrapper/ratelimiter/ratelimit"
*/

func startRPCService() {
	reg := etcdv3.NewRegistry(func(options *registry.Options) {
		options.Addrs = []string{
			conf.DefaultConfig.EtcdAddr,
		}
		etcdv3.Auth(conf.DefaultConfig.EtcdUser,conf.DefaultConfig.EtcdPwd)(options)
	})
  // 限制每秒访问次数(令牌桶容量)
	limit := 2
	// 生成令牌桶，limit=1每秒钟放入一个令牌,limit=2每0.5秒放入一个令牌以此类推
	//（fillInterval := time.Duration(1e9 * float64(quantum) / rate)）
	rate := jujuratelimit.NewBucketWithRate(float64(limit), int64(limit))
  
	service := micro.NewService(
		micro.Registry(reg),
		micro.Name("go.micro.service.user"),
		micro.RegisterTTL(time.Second*10),
		micro.RegisterInterval(time.Second*5),
		micro.Flags(common.CustomFlags...),
		// 配置micro false为不等待抛出500，true为等待
		micro.WrapHandler(ratelimit.NewHandlerWrapper(rate,true)),
		)
  
	service.Init()
	dbproxy.Init(service)
	userProto.RegisterUserServiceHandler(service.Server(), new(handler.User))
	err := service.Run()
	if err != nil {
		log.Println(err)
	}
}

客户端

var (
	userCli userProto.UserService
)

func init(){
	conf.InitConfig("./conf/config.json")
	reg := etcdv3.NewRegistry(func(op *registry.Options) {
		op.Addrs= []string{conf.DefaultConfig.EtcdAddr}
		etcdv3.Auth(conf.DefaultConfig.EtcdUser,conf.DefaultConfig.EtcdPwd)(op)
	})
	service := micro.NewService(
		micro.Registry(reg),
		micro.Name("go.micro.service.apigw"),
		micro.Flags(common.CustomFlags...),
	)
	service.Init()
	cli := service.Client()
	userCli = userProto.NewUserService("go.micro.service.user",cli)
}