uniflake 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f80b6d4ffd25a72a9a88dcec412a922514d9c6e631ac4244c9dfb3d2e590018f
+  data.tar.gz: 5405b316f49e42ce17876f062302d44d447d165152138cc2286ce04d7e176070
+SHA512:
+  metadata.gz: 39ae675c29e7de6674043d050d0ce7b133316b481a5be6b4026dd013312afd2b49cc75717760db94b14095ea680db4d960a0576270408b8588e7306861d80c59
+  data.tar.gz: 7c9270a843f01f0600fd0212dfbdfad0d4d1a1e26b7c60c6b3f143194a9d3a34248001ce5bdf9aae22a403fe061ad1089a48756d12c9699ce178e2537ebb7023

data/lib/uniflake/generator_ids/number.rb

@@ -0,0 +1,18 @@
+module Uniflake
+
+  class Number
+    BIT_LENGTH = 10
+    MAX_ID = (1 << BIT_LENGTH) - 1
+
+    attr_reader :id
+
+    def initialize(id = 0)
+      raise OverflowError, "invalid generator id (#{id} > #{MAX_ID})" if id > MAX_ID
+      @id = id
+    end
+
+    def to_i
+      @id
+    end
+  end
+end

data/lib/uniflake/generators/discord.rb

@@ -0,0 +1,8 @@
+module Uniflake::Generators
+
+  class Discord < Uniflake
+    # https://en.wikipedia.org/wiki/Snowflake_ID
+    DEFAULT_EPOCH_START_AT = Time.new(2015,01,01)
+  end
+
+end

data/lib/uniflake/generators/snowflake.rb

@@ -0,0 +1,8 @@
+module Uniflake::Generators
+
+  class Snowflake < Uniflake
+    # https://en.wikipedia.org/wiki/Snowflake_ID
+    DEFAULT_EPOCH_START_AT = Time.at(0, 1288834974657, :millisecond)
+  end
+
+end

data/lib/uniflake/generators/uniflake.rb

@@ -0,0 +1,65 @@
+module Uniflake::Generators
+
+  class Uniflake
+    STRUCTURE = {
+      timestamp: 41,
+      generator_id: 10,
+      sequence: 12
+    }
+
+    DEFAULT_EPOCH_START_AT = Time.new(2022, 01, 01)
+
+    MAX_TIMESTAMP = (1 << STRUCTURE[:timestamp]) - 1
+    MAX_GENERATOR_ID = (1 << STRUCTURE[:generator_id]) - 1
+    MAX_SEQUENCE = (1 << STRUCTURE[:sequence]) - 1
+
+      def initialize(sequence_resolver, epoch_start_at = self::class::DEFAULT_EPOCH_START_AT, generator_id = 0)
+      @generator_id = generator_id.to_i
+      raise Error, "invalid generator_id (#{@generator_id} > #{self::class::MAX_GENERATOR_ID})" if @generator_id > self::class::MAX_GENERATOR_ID
+
+      @epoch_start_at = epoch_start_at.strftime('%s%L').to_i
+      now = Time.now.strftime('%s%L').to_i
+      raise Error, "invalid epoch (#{@epoch_start_at} > #{now})" if @epoch_start_at > now
+
+      @sequence_resolver = sequence_resolver
+    end
+
+    def generate
+      now = Time.now.strftime('%s%L').to_i
+
+      sequence = @sequence_resolver.resolve(@generator_id, now, self::class::MAX_SEQUENCE)
+      raise Error, "invalid sequence (#{sequence} > #{self::class::MAX_SEQUENCE})" if sequence > self::class::MAX_SEQUENCE
+
+      timestamp = now - @epoch_start_at
+      raise Error, "invalid timestamp (#{timestamp} > #{self::class::MAX_TIMESTAMP})" if timestamp > self::class::MAX_TIMESTAMP
+
+      self.class.compose(timestamp, @generator_id, sequence)
+    end
+
+    def self.compose(timestamp, generator_id, sequence)
+      shift = 0
+      id = 0
+      self::STRUCTURE.keys.reverse.each do |field|
+        id += binding.local_variable_get(field) << shift
+        shift += self::STRUCTURE[field]
+      end
+      id
+    end
+
+    def parse(id)
+      fields = self.class.parse(id)
+      fields[:timestamp] += @epoch_start_at if fields.has_key? :timestamp
+      fields
+    end
+
+    def self.parse(id)
+      id = id.to_s(2).rjust(self::STRUCTURE.values.sum, "0")
+      fields = {}
+      self::STRUCTURE.each do |field, length|
+        fields[field] = id[0..length - 1].to_i(2)
+        id[0..length - 1] = id[length..-1]
+      end
+      fields
+    end
+  end
+end

data/lib/uniflake/sequence_resolvers/memory.rb

@@ -0,0 +1,18 @@
+module Uniflake::SequenceResolvers
+
+  class Memory
+
+    def initialize
+      @sequences = {}
+      @last_time = {}
+    end
+
+    def resolve(int_generator_id, time, max_value = 0)
+      raise "invalid time" if time < @last_time[int_generator_id].to_i
+
+      @sequences[int_generator_id] = time == @last_time[int_generator_id].to_i ? @sequences[int_generator_id].to_i + 1 : 0
+      @last_time[int_generator_id] = time
+      @sequences[int_generator_id]
+    end
+  end
+end

data/lib/uniflake/sequence_resolvers/random.rb

@@ -0,0 +1,15 @@
+module Uniflake::SequenceResolvers
+
+  class Random
+
+    def initialize
+      @prng = Random.new
+    end
+
+    def resolve(int_generator_id, time, max_value = 0)
+      raise "invalid time" if time < @last_time.to_i
+      @last_time = time
+      @prng.rand max_value
+    end
+  end
+end

data/lib/uniflake/sequence_resolvers/redis.rb

@@ -0,0 +1,37 @@
+module Uniflake::SequenceResolvers
+
+  class Redis
+
+    def initialize(client, key_prefix = '')
+      @client = client
+      @prefix = key_prefix
+    end
+
+    def resolve(int_generator_id, time, max_value = 0)
+      last_time = connection.get(@prefix + int_generator_id.to_s + "_last_time").to_i
+      raise "invalid time" if time < last_time
+
+      connection.set @prefix + int_generator_id.to_s + "_last_time", time
+
+      connection.eval lua, [@prefix + int_generator_id.to_s + time.to_s], ['0', '10']
+    end
+
+    def connection
+      if @client.class.method_defined? 'with'
+        @client.with { |connection| return connection }
+      end
+
+      @client
+    end
+
+    def lua
+      <<~LUA
+        if redis.call('set', KEYS[1], ARGV[1], "EX", ARGV[2], "NX") then
+          return 0
+        else
+          return redis.call('incr', KEYS[1])
+        end
+      LUA
+    end
+  end
+end

data/lib/uniflake.rb

@@ -0,0 +1,11 @@
+module Uniflake
+  require_relative 'uniflake/generators/uniflake.rb'
+  require_relative 'uniflake/generators/snowflake.rb'
+  require_relative 'uniflake/generators/discord.rb'
+
+  require_relative 'uniflake/sequence_resolvers/memory.rb'
+  require_relative 'uniflake/sequence_resolvers/random.rb'
+  require_relative 'uniflake/sequence_resolvers/redis.rb'
+
+  require_relative 'uniflake/generator_ids/number.rb'
+end

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: uniflake
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- nebo7
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-02-19 00:00:00.000000000 Z
+dependencies: []
+description: ''
+email: ''
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/uniflake.rb
+- lib/uniflake/generator_ids/number.rb
+- lib/uniflake/generators/discord.rb
+- lib/uniflake/generators/snowflake.rb
+- lib/uniflake/generators/uniflake.rb
+- lib/uniflake/sequence_resolvers/memory.rb
+- lib/uniflake/sequence_resolvers/random.rb
+- lib/uniflake/sequence_resolvers/redis.rb
+homepage: https://github.com/nebo7/uniflake
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.22
+signing_key:
+specification_version: 4
+summary: ''
+test_files: []