transfer.sh/vendor/google.golang.org/grpc/balancer/xds/edsbalancer/util.go

/*
 * Copyright 2019 gRPC authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package edsbalancer

import (
	"sync"
)

type dropper struct {
	// Drop rate will be numerator/denominator.
	numerator   uint32
	denominator uint32

	mu sync.Mutex
	i  uint32
}

func newDropper(numerator, denominator uint32) *dropper {
	return &dropper{
		numerator:   numerator,
		denominator: denominator,
	}
}

func (d *dropper) drop() (ret bool) {
	d.mu.Lock()
	defer d.mu.Unlock()

	// TODO: the drop algorithm needs a design.
	// Currently, for drop rate 3/5:
	// 0 1 2 3 4
	// d d d n n
	if d.i < d.numerator {
		ret = true
	}
	d.i++
	if d.i >= d.denominator {
		d.i = 0
	}

	return
}