aboutsummaryrefslogtreecommitdiff
path: root/weed/sequence/snowflake_sequencer.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/sequence/snowflake_sequencer.go')
-rw-r--r--weed/sequence/snowflake_sequencer.go43
1 files changed, 43 insertions, 0 deletions
diff --git a/weed/sequence/snowflake_sequencer.go b/weed/sequence/snowflake_sequencer.go
new file mode 100644
index 000000000..d6479fc68
--- /dev/null
+++ b/weed/sequence/snowflake_sequencer.go
@@ -0,0 +1,43 @@
+package sequence
+
+import (
+ "fmt"
+ "hash/fnv"
+
+ "github.com/bwmarrin/snowflake"
+)
+
+// a simple snowflake Sequencer
+type SnowflakeSequencer struct {
+ node *snowflake.Node
+}
+
+func NewSnowflakeSequencer(nodeid string) (*SnowflakeSequencer, error) {
+ node, err := snowflake.NewNode(int64(hash(nodeid) & 0x3ff))
+ if err != nil {
+ fmt.Println(err)
+ return nil, err
+ }
+
+ sequencer := &SnowflakeSequencer{node: node}
+ return sequencer, nil
+}
+
+func hash(s string) uint32 {
+ h := fnv.New32a()
+ h.Write([]byte(s))
+ return h.Sum32()
+}
+
+func (m *SnowflakeSequencer) NextFileId(count uint64) uint64 {
+ return uint64(m.node.Generate().Int64())
+}
+
+// ignore setmax as we are snowflake
+func (m *SnowflakeSequencer) SetMax(seenValue uint64) {
+}
+
+// return a new id as no Peek is stored
+func (m *SnowflakeSequencer) Peek() uint64 {
+ return uint64(m.node.Generate().Int64())
+}