mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-13 01:19:07 +08:00
37 lines
655 B
Go
37 lines
655 B
Go
package sequence
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// just for testing
|
|
type MemorySequencer struct {
|
|
counter uint64
|
|
sequenceLock sync.Mutex
|
|
}
|
|
|
|
func NewMemorySequencer() (m *MemorySequencer) {
|
|
m = &MemorySequencer{counter: 1}
|
|
return
|
|
}
|
|
|
|
func (m *MemorySequencer) NextFileId(count uint64) (uint64, uint64) {
|
|
m.sequenceLock.Lock()
|
|
defer m.sequenceLock.Unlock()
|
|
ret := m.counter
|
|
m.counter += uint64(count)
|
|
return ret, count
|
|
}
|
|
|
|
func (m *MemorySequencer) SetMax(seenValue uint64) {
|
|
m.sequenceLock.Lock()
|
|
defer m.sequenceLock.Unlock()
|
|
if m.counter <= seenValue {
|
|
m.counter = seenValue + 1
|
|
}
|
|
}
|
|
|
|
func (m *MemorySequencer) Peek() uint64 {
|
|
return m.counter
|
|
}
|