nimo-full-check/checker/sample.go

package checker

import "math/rand"

const (
	SEED = 1
)

type Sample struct {
	sampleCnt int64
	totalCnt  int64
	source    *rand.Rand
}

func NewSample(sampleCnt, totalCnt int64) *Sample {
	return &Sample{
		sampleCnt: sampleCnt,
		totalCnt:  totalCnt,
		source:    rand.New(rand.NewSource(SEED)),
	}
}

func (s *Sample) Hit() bool {
	if s.sampleCnt >= s.totalCnt {
		return true
	}
	if s.sampleCnt == 0 {
		return false
	}

	return s.source.Int63n(s.totalCnt) < s.sampleCnt
}