diff --git a/idGenerator/store/segment.go b/idGenerator/store/segment.go index 2307182..8d673aa 100644 --- a/idGenerator/store/segment.go +++ b/idGenerator/store/segment.go @@ -1,6 +1,9 @@ package store -import "fmt" +import ( + "fmt" + "sync" +) // 号段 type Segment struct { @@ -8,22 +11,35 @@ type Segment struct { end int64 current int64 reback bool + mu sync.RWMutex } func (s *Segment) Allot() int64 { + s.mu.Lock() + defer s.mu.Unlock() + s.current++ return s.current } func (s *Segment) IsEnding() bool { + s.mu.Lock() + defer s.mu.Unlock() + return (s.current - s.start) > (s.end - s.current) } func (s *Segment) IsEmpty() bool { + s.mu.Lock() + defer s.mu.Unlock() + return s.current == s.end } func (s *Segment) Reback() bool { + s.mu.Lock() + defer s.mu.Unlock() + // println("回旋确认:", s.reback, s.current == (s.start+1)) return s.reback && s.current == (s.start+1) }