diff options
Diffstat (limited to 'weed/sequence/memory_sequencer.go')
| -rw-r--r-- | weed/sequence/memory_sequencer.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/weed/sequence/memory_sequencer.go b/weed/sequence/memory_sequencer.go new file mode 100644 index 000000000..d727dc723 --- /dev/null +++ b/weed/sequence/memory_sequencer.go @@ -0,0 +1,36 @@ +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 +} |
