From c3e181c7a9686b88fa4cd6444b6a3e70e75d0ae5 Mon Sep 17 00:00:00 2001
From: Jonathan Barrow <jonbarrow1998@gmail.com>
Date: Sun, 26 Sep 2021 19:44:26 -0400
Subject: [PATCH] Added DataStoreIDGenerator to make IDs

---
 datastore_id_generator.go | 48 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 datastore_id_generator.go

diff --git a/datastore_id_generator.go b/datastore_id_generator.go
new file mode 100644
index 0000000..6be5b39
--- /dev/null
+++ b/datastore_id_generator.go
@@ -0,0 +1,48 @@
+package main
+
+import (
+	"encoding/binary"
+)
+
+// DataStoreIDGenerator represents a safe, structured, ID for unique NEX DataStore objects
+type DataStoreIDGenerator struct {
+	inUse    bool
+	regionID uint8
+	nodeID   uint8
+	Value    uint32
+}
+
+// InUse is used to check if a DataStoreIDGenerator Node is currently generating a number
+func (dataStoreIDGenerator *DataStoreIDGenerator) InUse() bool {
+	return dataStoreIDGenerator.inUse
+}
+
+// SetInUse is used to check if a DataStoreIDGenerator Node is currently generating a number
+func (dataStoreIDGenerator *DataStoreIDGenerator) SetInUse(inUse bool) {
+	dataStoreIDGenerator.inUse = inUse
+}
+
+// Next is used to get the next value from the DataStoreIDGenerator Node
+func (dataStoreIDGenerator *DataStoreIDGenerator) Next() uint64 {
+	if dataStoreIDGenerator.inUse {
+		panic("Cannot call DataStoreIDGenerator.Next while in use")
+	}
+
+	dataStoreIDGenerator.Value += 1
+
+	data := make([]byte, 8)
+
+	data[2] = dataStoreIDGenerator.regionID
+	data[3] = dataStoreIDGenerator.nodeID
+	binary.BigEndian.PutUint32(data[4:], dataStoreIDGenerator.Value)
+
+	return binary.BigEndian.Uint64(data)
+}
+
+func NewDataStoreIDGenerator(regionID, nodeID uint8, value uint32) *DataStoreIDGenerator {
+	return &DataStoreIDGenerator{
+		regionID: regionID,
+		nodeID:   nodeID,
+		Value:    value,
+	}
+}