test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-01 11:29:48 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

[skip ci]Fix golint warnings in uniquegenerator (#9098)

Browse Source 
Signed-off-by: Xiangyu Wang <xiangyu.wang@zilliz.com>
This commit is contained in:
Xiangyu Wang 2021-10-03 00:46:22 +08:00 committed by GitHub
parent 1659a9ac03
commit f8fd416b4d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/util/uniquegenerator/unique_int_generator.go
View File

@ -14,16 +14,19 @@ package uniquegenerator
import "sync"
type (
// UniqueIntGenerator provide methods to generate unique integers
UniqueIntGenerator interface {
GetInt() int
GetInts(count int) (int, int)
}
// NaiveUniqueIntGenerator is a native implement of UniqueIntGenerator
NaiveUniqueIntGenerator struct {
now int
mtx sync.Mutex
}
)
// GetInts returns the begin and end of a batch of unique integers
func (generator *NaiveUniqueIntGenerator) GetInts(count int) (int, int) {
generator.mtx.Lock()
defer func() {
@ -33,11 +36,13 @@ func (generator *NaiveUniqueIntGenerator) GetInts(count int) (int, int) {
return generator.now, generator.now + count
}
// GetInt returns a unique integer
func (generator *NaiveUniqueIntGenerator) GetInt() int {
begin, _ := generator.GetInts(1)
return begin
}
// NewNaiveUniqueIntGenerator returns a new NaiveUniqueIntGenerator object
func NewNaiveUniqueIntGenerator() *NaiveUniqueIntGenerator {
return &NaiveUniqueIntGenerator{
now: 0,
@ -47,6 +52,7 @@ func NewNaiveUniqueIntGenerator() *NaiveUniqueIntGenerator {
var uniqueIntGeneratorIns UniqueIntGenerator
var getUniqueIntGeneratorInsOnce sync.Once
// GetUniqueIntGeneratorIns returns a singleton of UniqueIntGenerator
func GetUniqueIntGeneratorIns() UniqueIntGenerator {
getUniqueIntGeneratorInsOnce.Do(func() {
uniqueIntGeneratorIns = NewNaiveUniqueIntGenerator()